Fix even more tests still importing skimage.graph

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')

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)

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