diff --git a/skimage/morphology/tests/test_selem.py b/skimage/morphology/tests/test_selem.py index ccaf43a0..9a8ca895 100644 --- a/skimage/morphology/tests/test_selem.py +++ b/skimage/morphology/tests/test_selem.py @@ -8,7 +8,7 @@ from numpy.testing import * from skimage import data_dir from skimage.io import * from skimage import data_dir -from skimage.morphology import * +from skimage.morphology import selem class TestSElem(): @@ -36,7 +36,7 @@ class TestSElem(): expected_mask = expected_mask[:, np.newaxis] assert_equal(expected_mask, actual_mask) k = k + 1 - + def strel_worker_3d(self, fn, func): matlab_masks = np.load(os.path.join(data_dir, fn)) k = 0 @@ -48,20 +48,24 @@ class TestSElem(): # Test center slice for each dimension. This gives a good # indication of validity without the need for a 3D reference # mask. - c = int(expected_mask.shape[0]/2) + c = int(expected_mask.shape[0]/2) assert_equal(expected_mask, actual_mask[c,:,:]) assert_equal(expected_mask, actual_mask[:,c,:]) assert_equal(expected_mask, actual_mask[:,:,c]) k = k + 1 - + def test_selem_disk(self): self.strel_worker("disk-matlab-output.npz", selem.disk) def test_selem_diamond(self): self.strel_worker("diamond-matlab-output.npz", selem.diamond) - + def test_selem_ball(self): self.strel_worker_3d("disk-matlab-output.npz", selem.ball) def test_selem_octahedron(self): self.strel_worker_3d("diamond-matlab-output.npz", selem.octahedron) + + +if __name__ == '__main__': + np.testing.run_module_suite()