Fix imports

networkx · Nov 14, 2020 · bbdf8fe · bbdf8fe
1 parent d8f167c
commit bbdf8fe
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/networkx/generators/random_graphs.py b/networkx/generators/random_graphs.py
@@ -32,6 +32,7 @@
     "random_powerlaw_tree",
     "random_powerlaw_tree_sequence",
     "random_kernel_graph",
+    "random_ordered_tree",
 ]
 
 
@@ -1318,12 +1319,12 @@ def random_ordered_tree(n, seed=None, directed=False):
 
     Example
     -------
-    >>> assert len(random_ordered_tree(n=1, seed=0).nodes) == 1
-    >>> assert len(random_ordered_tree(n=2, seed=0).nodes) == 2
-    >>> assert len(random_ordered_tree(n=3, seed=0).nodes) == 3
-    >>> from networkx.readwrite.text import forest_str
-    >>> otree = random_ordered_tree(n=5, seed=3, directed=True)
-    >>> print(forest_str(otree))
+    >>> import networkx as nx
+    >>> assert len(nx.random_ordered_tree(n=1, seed=0).nodes) == 1
+    >>> assert len(nx.random_ordered_tree(n=2, seed=0).nodes) == 2
+    >>> assert len(nx.random_ordered_tree(n=3, seed=0).nodes) == 3
+    >>> otree = nx.random_ordered_tree(n=5, seed=3, directed=True)
+    >>> print(nx.forest_str(otree))
     ╙── 1
         ├─╼ 0
         └─╼ 4