From 682d0535cd31b5af65237adacf6d9fb5a12885fd Mon Sep 17 00:00:00 2001 From: JDWarner Date: Mon, 27 Aug 2012 13:41:41 -0500 Subject: [PATCH] Added multispectral random walker test. Since the multispectral path is equivalent except for gradient calcs, only one test case is needed. This test is modeled on the 3-D non-multispectral version. If deemed necessary, adding a 2-D case would be simple. --- skimage/segmentation/tests/test_random_walker.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/skimage/segmentation/tests/test_random_walker.py b/skimage/segmentation/tests/test_random_walker.py index 4e5a74c3..e0415498 100644 --- a/skimage/segmentation/tests/test_random_walker.py +++ b/skimage/segmentation/tests/test_random_walker.py @@ -138,6 +138,18 @@ def test_3d_inactive(): assert (labels.reshape(data.shape)[13:17, 13:17, 13:17] == 2).all() return data, labels, old_labels, after_labels + +def test_multispectral(): + n = 30 + lx, ly, lz = n, n, n + data, labels = make_3d_syntheticdata( lx, ly, lz ) + data = [data, data] # Result should be identical + multi_labels = random_walker(data, labels, mode='cg') + single_labels = random_walker(data[0], labels, mode='cg') + assert (multi_labels.reshape(data[0].shape)[13:17, 13:17, 13:17] == 2).all() + assert (single_labels.reshape(data[0].shape)[13:17, 13:17, 13:17] == 2).all() + return data, multi_labels, single_labels, labels + if __name__ == '__main__': from numpy import testing testing.run_module_suite()