mirror of
https://github.com/wassname/scikit-image.git
synced 2026-07-14 11:18:06 +08:00
31 lines
1.2 KiB
Python
31 lines
1.2 KiB
Python
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()
|