minor changes

networkx · Nov 12, 2020 · 3057a64 · 3057a64
1 parent 384b416
commit 3057a64
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/networkx/readwrite/tests/test_text.py b/networkx/readwrite/tests/test_text.py
@@ -1,3 +1,4 @@
+import pytest
 import networkx as nx
 from textwrap import dedent
 
@@ -191,14 +192,12 @@ def test_undirected_tree_str():
 
 
 def test_forest_str_errors():
-    import pytest
-
-    ugraph = nx.generators.complete_graph(3, create_using=nx.Graph)
+    ugraph = nx.complete_graph(3, create_using=nx.Graph)
 
     with pytest.raises(nx.NetworkXNotImplemented):
         nx.forest_str(ugraph)
 
-    dgraph = nx.generators.complete_graph(3, create_using=nx.DiGraph)
+    dgraph = nx.complete_graph(3, create_using=nx.DiGraph)
 
     with pytest.raises(nx.NetworkXNotImplemented):
         nx.forest_str(dgraph)