Fix even more tests still importing skimage.graph

scikit-image · Feb 3, 2015 · e5fe82d · e5fe82d
1 parent 9cbeeca
commit e5fe82d
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/skimage/future/graph/graph_cut.py b/skimage/future/graph/graph_cut.py
@@ -37,7 +37,8 @@ def cut_threshold(labels, rag, thresh, in_place=True):
 
     Examples
     --------
-    >>> from skimage import data, graph, segmentation
+    >>> from skimage import data, segmentation
+    >>> from skimage.future import graph
     >>> img = data.astronaut()
     >>> labels = segmentation.slic(img)
     >>> rag = graph.rag_mean_color(img, labels)
@@ -107,7 +108,8 @@ def cut_normalized(labels, rag, thresh=0.001, num_cuts=10, in_place=True,
 
     Examples
     --------
-    >>> from skimage import data, graph, segmentation
+    >>> from skimage import data, segmentation
+    >>> from skimage.future import graph
     >>> img = data.astronaut()
     >>> labels = segmentation.slic(img, compactness=30, n_segments=400)
     >>> rag = graph.rag_mean_color(img, labels, mode='similarity')

diff --git a/skimage/future/graph/rag.py b/skimage/future/graph/rag.py
@@ -365,7 +365,8 @@ def draw_rag(labels, rag, img, border_color=None, node_color='#ffff00',
 
     Examples
     --------
-    >>> from skimage import data, graph, segmentation
+    >>> from skimage import data, segmentation
+    >>> from skimage.future import graph
     >>> img = data.coffee()
     >>> labels = segmentation.slic(img)
     >>> g =  graph.rag_mean_color(img, labels)

diff --git a/skimage/future/graph/tests/test_rag.py b/skimage/future/graph/tests/test_rag.py
@@ -1,5 +1,5 @@
 import numpy as np
-from skimage import graph
+from skimage.future import graph
 from skimage._shared.version_requirements import is_installed
 from numpy.testing.decorators import skipif
 from skimage import segmentation