From 5222fb5ff7eef0115603e99d8f266eda884fb141 Mon Sep 17 00:00:00 2001 From: Juan Nunez-Iglesias Date: Sun, 3 Mar 2013 14:50:20 +1100 Subject: [PATCH] Add tests for remove_small_connected_components --- skimage/morphology/tests/test_misc.py | 30 +++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) create mode 100644 skimage/morphology/tests/test_misc.py diff --git a/skimage/morphology/tests/test_misc.py b/skimage/morphology/tests/test_misc.py new file mode 100644 index 00000000..77bb20c3 --- /dev/null +++ b/skimage/morphology/tests/test_misc.py @@ -0,0 +1,30 @@ +import numpy as np +from numpy.testing import assert_array_equal, assert_equal +from skimage.morphology import remove_small_connected_components + +test_image = np.array([[0, 0, 0, 1, 0], + [1, 1, 1, 0, 0], + [1, 1, 1, 0, 1]], bool) +def test_one_connectivity(): + expected = np.array([[0, 0, 0, 0, 0], + [1, 1, 1, 0, 0], + [1, 1, 1, 0, 0]], bool) + observed = remove_small_connected_components(test_image, min_size=6) + assert_array_equal(observed, expected) + +def test_two_connectivity(): + expected = np.array([[0, 0, 0, 1, 0], + [1, 1, 1, 0, 0], + [1, 1, 1, 0, 0]], bool) + observed = remove_small_connected_components(test_image, min_size=7, + connectivity=2) + assert_array_equal(observed, expected) + +def test_in_place(): + observed = remove_small_connected_components(test_image, min_size=6, + in_place=True) + assert_equal(observed is test_image, True, + "remove_small_connected_components in_place argument failed.") + +if __name__ == "__main__": + np.testing.run_module_suite()