30 lines
1.0 KiB
Python
30 lines
1.0 KiB
Python
import numpy as np
|
|
|
|
|
|
def test_mask_morphology():
|
|
from backend.nodes.mask_morphology import MaskMorphology
|
|
node = MaskMorphology()
|
|
|
|
mask = np.zeros((64, 64), dtype=np.uint8)
|
|
mask[28:36, 28:36] = 255
|
|
orig_count = np.count_nonzero(mask)
|
|
|
|
dilated, = node.process(mask, operation="dilate", radius=1, shape="square")
|
|
assert dilated.dtype == np.uint8
|
|
assert np.count_nonzero(dilated) > orig_count
|
|
|
|
eroded, = node.process(mask, operation="erode", radius=1, shape="square")
|
|
assert np.count_nonzero(eroded) < orig_count
|
|
|
|
opened, = node.process(mask, operation="open", radius=1, shape="square")
|
|
assert np.count_nonzero(opened) <= orig_count
|
|
|
|
mask_hole = mask.copy()
|
|
mask_hole[32, 32] = 0
|
|
assert np.count_nonzero(mask_hole) == orig_count - 1
|
|
closed, = node.process(mask_hole, operation="close", radius=1, shape="square")
|
|
assert closed[32, 32] == 255
|
|
|
|
dilated_disk, = node.process(mask, operation="dilate", radius=2, shape="disk")
|
|
assert np.count_nonzero(dilated_disk) > orig_count
|