From 8d50561d186f0dc927ba1184f360ccbde877b3e0 Mon Sep 17 00:00:00 2001 From: Pratap Vardhan Date: Mon, 15 Dec 2014 13:21:15 +0530 Subject: [PATCH] TST: Test for ellipse structuring element --- skimage/morphology/tests/test_selem.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/skimage/morphology/tests/test_selem.py b/skimage/morphology/tests/test_selem.py index 35559fe4..545cb0d4 100644 --- a/skimage/morphology/tests/test_selem.py +++ b/skimage/morphology/tests/test_selem.py @@ -102,6 +102,25 @@ class TestSElem(): assert_equal(expected_mask1, actual_mask1) assert_equal(expected_mask2, actual_mask2) + def test_selem_ellipse(self): + """Test ellipse structuring elements""" + expected_mask1 = np.array([[0, 0, 1, 1, 1, 1, 1, 1, 1, 0, 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, 0, 1, 1, 1, 1, 1, 1, 1, 0, 0]], dtype=uint8) + actual_mask1 = selem.ellipse(5, 3) + expected_mask2 = np.array([[1, 1, 1], + [1, 1, 1], + [1, 1, 1]], dtype=np.uint8) + actual_mask2 = selem.ellipse(1, 1) + assert_equal(expected_mask1, actual_mask1) + assert_equal(expected_mask2, actual_mask2) + assert_equal(expected_mask1, selem.ellipse(3, 5).T) + assert_equal(expected_mask2, selem.ellipse(1, 1).T) + def test_selem_star(self): """Test star structuring elements""" expected_mask1 = np.array([[0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0],