Use utils.testing to handle testing edge and node equality

networkx · Jan 28, 2016 · 5665c71 · 5665c71
1 parent c0dd9a1
commit 5665c71
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/networkx/drawing/tests/test_agraph.py b/networkx/drawing/tests/test_agraph.py
@@ -5,6 +5,7 @@
 
 from nose import SkipTest
 from nose.tools import assert_true,assert_equal
+from networkx.testing import assert_edges_equal, assert_nodes_equal
 
 import networkx as nx
 
@@ -27,8 +28,8 @@ def build_graph(self, G):
         return G
 
     def assert_equal(self, G1, G2):
-        assert_true( sorted(G1.nodes())==sorted(G2.nodes()) )
-        assert_true( sorted(G1.edges())==sorted(G2.edges()) )
+        assert_nodes_equal(G1.nodes(),G2.nodes())
+        assert_edges_equal(G1.edges(),G2.edges())
 
 
     def agraph_checks(self, G):