diff options
Diffstat (limited to 'networkx/generators/tests/test_line.py')
-rw-r--r-- | networkx/generators/tests/test_line.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/networkx/generators/tests/test_line.py b/networkx/generators/tests/test_line.py index fec8893c..e072728d 100644 --- a/networkx/generators/tests/test_line.py +++ b/networkx/generators/tests/test_line.py @@ -2,7 +2,7 @@ import networkx as nx import pytest import networkx.generators.line as line -from networkx.testing.utils import assert_edges_equal +from networkx.utils import edges_equal def test_node_func(): @@ -29,7 +29,7 @@ def test_edge_func(): G.add_edge(2, 3) ef = line._edge_func(G) expected = [(1, 2), (2, 3)] - assert_edges_equal(ef(), expected) + assert edges_equal(ef(), expected) # digraph G = nx.MultiDiGraph() @@ -74,19 +74,19 @@ class TestGeneratorLine: G = nx.DiGraph() G.add_edges_from([(0, 1), (1, 2), (2, 3)]) L = nx.line_graph(G) - assert_edges_equal(L.edges(), [((0, 1), (1, 2)), ((1, 2), (2, 3))]) + assert edges_equal(L.edges(), [((0, 1), (1, 2)), ((1, 2), (2, 3))]) def test_create1(self): G = nx.DiGraph() G.add_edges_from([(0, 1), (1, 2), (2, 3)]) L = nx.line_graph(G, create_using=nx.Graph()) - assert_edges_equal(L.edges(), [((0, 1), (1, 2)), ((1, 2), (2, 3))]) + assert edges_equal(L.edges(), [((0, 1), (1, 2)), ((1, 2), (2, 3))]) def test_create2(self): G = nx.Graph() G.add_edges_from([(0, 1), (1, 2), (2, 3)]) L = nx.line_graph(G, create_using=nx.DiGraph()) - assert_edges_equal(L.edges(), [((0, 1), (1, 2)), ((1, 2), (2, 3))]) + assert edges_equal(L.edges(), [((0, 1), (1, 2)), ((1, 2), (2, 3))]) class TestGeneratorInverseLine: |