wassname/scikit-image
1
0
Fork 0
mirror of https://github.com/wassname/scikit-image.git synced 2026-07-02 10:46:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

test case for in place

Browse Source
This commit is contained in:
Vighnesh Birodkar
2014-08-09 21:35:07 +05:30
parent 3973b30300
commit ded138cbcb
1 changed files with 15 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/graph/tests/test_rag.py
+15 -5
View File
@@ -64,11 +64,15 @@ def test_threshold_cut():
labels[50:, 50:] = 3
rag = graph.rag_mean_color(img, labels)
new_labels = graph.cut_threshold(labels, rag, 10)
new_labels = graph.cut_threshold(labels, rag, 10, in_place=False)
# Two labels
assert new_labels.max() == 1
new_labels = graph.cut_threshold(labels, rag, 10)
# Two labels
assert new_labels.max() == 1
@skipif(not is_installed('networkx'))
def test_cut_normalized():
@@ -85,14 +89,20 @@ def test_cut_normalized():
labels[50:, 50:] = 3
rag = graph.rag_mean_color(img, labels, mode='similarity')
new_labels = graph.cut_normalized(labels, rag)
new_labels, _, _ = segmentation.relabel_sequential(new_labels)
new_labels = graph.cut_normalized(labels, rag, in_place=False)
new_labels, _, _ = segmentation.relabel_sequential(new_labels)
# Two labels
assert new_labels.max() == 1
new_labels = graph.cut_normalized(labels, rag)
new_labels, _, _ = segmentation.relabel_sequential(new_labels)
assert new_labels.max() == 1
@skipif(not is_installed('networkx'))
def test_rag_error():
img = np.zeros((10, 10, 3), dtype='uint8')
labels = np.zeros((10, 10), dtype='uint8')
testing.assert_raises(ValueError, graph.rag_mean_color, img, labels, 2, 'non existant mode')
testing.assert_raises(ValueError, graph.rag_mean_color, img, labels,
2, 'non existant mode')