From c70fa17964eb34e00c41119ebf816e83b7dcb95f Mon Sep 17 00:00:00 2001 From: weilycoder Date: Thu, 10 Oct 2024 16:34:27 +0800 Subject: [PATCH] Modify the judgment logic --- cyaron/tests/graph_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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):