From 67165363bfd3f8aaeed67a609137cd4bae91486e Mon Sep 17 00:00:00 2001 From: Ankit Agrawal Date: Tue, 6 Aug 2013 03:38:52 +0530 Subject: [PATCH] Test for Octagon --- skimage/morphology/tests/test_selem.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/skimage/morphology/tests/test_selem.py b/skimage/morphology/tests/test_selem.py index 9a8ca895..945a9ede 100644 --- a/skimage/morphology/tests/test_selem.py +++ b/skimage/morphology/tests/test_selem.py @@ -66,6 +66,21 @@ class TestSElem(): def test_selem_octahedron(self): self.strel_worker_3d("diamond-matlab-output.npz", selem.octahedron) + def test_selem_octagon(self): + expected_mask = array([[0, 0, 0, 1, 1, 1, 1, 1, 0, 0, 0], + [0, 0, 1, 1, 1, 1, 1, 1, 1, 0, 0], + [0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0], + [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1], + [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1], + [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1], + [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1], + [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1], + [0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0], + [0, 0, 1, 1, 1, 1, 1, 1, 1, 0, 0], + [0, 0, 0, 1, 1, 1, 1, 1, 0, 0, 0]], dtype=uint8) + actual_mask = selem.octagon(5, 3) + assert_equal(expected_mask, actual_mask) + if __name__ == '__main__': np.testing.run_module_suite()