diff options
| -rw-r--r-- | miasm2/core/graph.py | 2 | ||||
| -rw-r--r-- | test/core/graph.py | 4 |
2 files changed, 3 insertions, 3 deletions
diff --git a/miasm2/core/graph.py b/miasm2/core/graph.py index c64c7b72..b74717da 100644 --- a/miasm2/core/graph.py +++ b/miasm2/core/graph.py @@ -523,7 +523,7 @@ class DiGraph(object): """ for a, b in self.compute_back_edges(head): body = self._compute_natural_loop_body(b, a) - yield ((b, a), body) + yield ((a, b), body) def compute_back_edges(self, head): """ diff --git a/test/core/graph.py b/test/core/graph.py index e148d70f..9f8afcae 100644 --- a/test/core/graph.py +++ b/test/core/graph.py @@ -184,8 +184,8 @@ g3.add_edge(7, 8) g3.add_edge(8, 7) loops = set([(backedge, frozenset(body)) for backedge, body in g3.compute_natural_loops(1)]) -assert(loops == {((1, 9), frozenset({1, 2, 4, 5, 9})), - ((2, 9), frozenset({2, 4, 5, 9}))}) +assert(loops == {((9, 1), frozenset({1, 2, 4, 5, 9})), + ((9, 2), frozenset({2, 4, 5, 9}))}) sccs = set([frozenset(scc) for scc in g3.compute_strongly_connected_components()]) assert(sccs == {frozenset({6}), |