diff --git a/cyaron/tests/graph_test.py b/cyaron/tests/graph_test.py index 1e3f1db..5c9987c 100644 --- a/cyaron/tests/graph_test.py +++ b/cyaron/tests/graph_test.py @@ -91,7 +91,7 @@ def test_repeated_edges(self): for _ in range(10): graph = Graph.graph(graph_size, int(graph_size*2), repeated_edges=False) - edges = list(graph.iterate_edges()) + edges = [(e.start, e.end) for e in graph.iterate_edges()] self.assertEqual(len(edges), len(set(edges))) def test_tree_connected(self):