diff options
| -rw-r--r-- | miasm2/ir/ir.py | 3 | ||||
| -rw-r--r-- | test/ir/ir.py | 1 |
2 files changed, 4 insertions, 0 deletions
diff --git a/miasm2/ir/ir.py b/miasm2/ir/ir.py index bc23d471..e5f0c8db 100644 --- a/miasm2/ir/ir.py +++ b/miasm2/ir/ir.py @@ -139,6 +139,9 @@ class AssignBlock(object): for dst, src in self._assigns.iteritems(): yield dst, src + def items(self): + return [(dst, src) for dst, src in self.iteritems()] + def itervalues(self): for src in self._assigns.itervalues(): yield src diff --git a/test/ir/ir.py b/test/ir/ir.py index 5c428a94..05936d75 100644 --- a/test/ir/ir.py +++ b/test/ir/ir.py @@ -37,6 +37,7 @@ assert assignblk1.get_rw() == {id_a: set([id_b])} assert assignblk1.keys() == [id_a] assert dict(assignblk1) == {id_a: id_b} assert assignblk1[id_a] == id_b +assert list(assignblk1.iteritems()) == assignblk1.items() ## Simplify assignblk3 = AssignBlock({id_a: id_b - id_b}) |