def test_erode_disk(self):
"""Test 4."""
I = np.zeros((8, 8))
I[1:7, 1:7] = 1
Ie = im_utils.erode(I, n=1, strel='disk')
# make assertion
assert np.all(Ie == np.zeros_like(Ie))
def test_erode_square(self):
"""Test 2."""
I = np.zeros((8, 8))
I[1:7, 1:7] = 1
Ie = im_utils.erode(I, n=1, strel='square')
# make assertion
assert np.all(Ie == np.array([
[0., 0., 0., 0., 0., 0., 0., 0.], [0., 0., 0., 0., 0., 0., 0., 0.],
[0., 0., 1., 1., 1., 1., 0., 0.], [0., 0., 1., 1., 1., 1., 0., 0.],
[0., 0., 1., 1., 1., 1., 0., 0.], [0., 0., 1., 1., 1., 1., 0., 0.],
[0., 0., 0., 0., 0., 0., 0., 0.], [0., 0., 0., 0., 0., 0., 0., 0.]
]))
def test_erode_plus(self):
"""Test 3."""
I = np.zeros((8, 8))
I[0:7, 1:3] = 1
I[2:5, 1:7] = 1
Ie = im_utils.erode(I, n=1, strel='plus')
# make assertion
assert np.all(Ie == np.array([
[0., 0., 0., 0., 0., 0., 0., 0.], [0., 0., 0., 0., 0., 0., 0., 0.],
[0., 0., 1., 0., 0., 0., 0., 0.], [0., 0., 1., 1., 1., 1., 0., 0.],
[0., 0., 1., 0., 0., 0., 0., 0.], [0., 0., 0., 0., 0., 0., 0., 0.],
[0., 0., 0., 0., 0., 0., 0., 0.], [0., 0., 0., 0., 0., 0., 0., 0.]
]))
def test_erode_no_n(self):
"""Test 1."""
I = np.zeros((3, 3))
Ie = im_utils.erode(I, n=0, strel='square')
# make assertion
assert np.all(I == Ie)
