diff options
Diffstat (limited to 'networkx/generators/tests/test_nonisomorphic_trees.py')
-rw-r--r-- | networkx/generators/tests/test_nonisomorphic_trees.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/networkx/generators/tests/test_nonisomorphic_trees.py b/networkx/generators/tests/test_nonisomorphic_trees.py index cb6eea86..ced1cbe1 100644 --- a/networkx/generators/tests/test_nonisomorphic_trees.py +++ b/networkx/generators/tests/test_nonisomorphic_trees.py @@ -6,7 +6,7 @@ Generators - Non Isomorphic Trees Unit tests for WROM algorithm generator in generators/nonisomorphic_trees.py """ import networkx as nx -from networkx.testing import assert_edges_equal +from networkx.utils import edges_equal class TestGeneratorNonIsomorphicTrees: @@ -48,9 +48,9 @@ class TestGeneratorNonIsomorphicTrees: def f(x): return list(nx.nonisomorphic_trees(x)) - assert_edges_equal(f(3)[0].edges(), [(0, 1), (0, 2)]) - assert_edges_equal(f(4)[0].edges(), [(0, 1), (0, 3), (1, 2)]) - assert_edges_equal(f(4)[1].edges(), [(0, 1), (0, 2), (0, 3)]) + assert edges_equal(f(3)[0].edges(), [(0, 1), (0, 2)]) + assert edges_equal(f(4)[0].edges(), [(0, 1), (0, 3), (1, 2)]) + assert edges_equal(f(4)[1].edges(), [(0, 1), (0, 2), (0, 3)]) def test_nonisomorphic_trees_matrix(self): trees_2 = [[[0, 1], [1, 0]]] |