diff options
Diffstat (limited to 'networkx/drawing/tests/test_agraph.py')
-rw-r--r-- | networkx/drawing/tests/test_agraph.py | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/networkx/drawing/tests/test_agraph.py b/networkx/drawing/tests/test_agraph.py index bf91f335..33e77c36 100644 --- a/networkx/drawing/tests/test_agraph.py +++ b/networkx/drawing/tests/test_agraph.py @@ -6,7 +6,7 @@ import pytest pygraphviz = pytest.importorskip("pygraphviz") -from networkx.testing import assert_edges_equal, assert_nodes_equal, assert_graphs_equal +from networkx.utils import nodes_equal, edges_equal, graphs_equal import networkx as nx @@ -20,8 +20,8 @@ class TestAGraph: return G def assert_equal(self, G1, G2): - assert_nodes_equal(G1.nodes(), G2.nodes()) - assert_edges_equal(G1.edges(), G2.edges()) + assert nodes_equal(G1.nodes(), G2.nodes()) + assert edges_equal(G1.edges(), G2.edges()) assert G1.graph["metal"] == G2.graph["metal"] def agraph_checks(self, G): @@ -188,21 +188,21 @@ class TestAGraph: G = nx.Graph() A = nx.nx_agraph.to_agraph(G) H = nx.nx_agraph.from_agraph(A) - # assert_graphs_equal(G, H) + # assert graphs_equal(G, H) AA = nx.nx_agraph.to_agraph(H) HH = nx.nx_agraph.from_agraph(AA) - assert_graphs_equal(H, HH) + assert graphs_equal(H, HH) G.graph["graph"] = {} G.graph["node"] = {} G.graph["edge"] = {} - assert_graphs_equal(G, HH) + assert graphs_equal(G, HH) @pytest.mark.xfail(reason="integer->string node conversion in round trip") def test_round_trip_integer_nodes(self): G = nx.complete_graph(3) A = nx.nx_agraph.to_agraph(G) H = nx.nx_agraph.from_agraph(A) - assert_graphs_equal(G, H) + assert graphs_equal(G, H) def test_graphviz_alias(self): G = self.build_graph(nx.Graph()) |