def test_base_2d_volume(self):
mask = np.zeros((7, 7), dtype=np.uint8)
mask[1:-1, 1:-1] = 1
result_mask = MaskDistanceSplit.split(mask, 1, True, (1, 1))
assert np.all(mask == result_mask)
mask2 = np.copy(mask)
mask2[2:-2, 2:-2] = 2
result_mask = MaskDistanceSplit.split(mask, 2, True, (1, 1))
assert np.all(mask2 == result_mask)
def test_base_3d_thick(self):
mask = np.zeros((6, 6, 6), dtype=np.uint8)
mask[1:-1, 1:-1, 1:-1] = 1
result_mask = MaskDistanceSplit.split(mask, 1, False, (1, 1, 1))
assert np.all(mask == result_mask)
mask2 = np.copy(mask)
mask2[2:-2, 2:-2, 2:-2] = 2
result_mask = MaskDistanceSplit.split(mask, 2, False, (1, 1, 1))
assert np.all(mask2 == result_mask)
def test_scaling_3d_thick(self):
mask = np.zeros((6, 12, 12), dtype=np.uint8)
mask[1:-1, 2:-2, 2:-2] = 1
result_mask = MaskDistanceSplit.split(mask, 1, False, (2, 1, 1))
assert np.all(mask == result_mask)
mask2 = np.copy(mask)
mask2[2:-2, 4:-4, 4:-4] = 2
result_mask = MaskDistanceSplit.split(mask, 2, False, (2, 1, 1))
assert np.all(mask2 == result_mask)
def test_scaling_3d_volume(self):
mask = np.zeros((7, 14, 14), dtype=np.uint8)
mask[1:-1, 2:-2, 2:-2] = 1
result_mask = MaskDistanceSplit.split(mask, 1, True, (2, 1, 1))
assert np.all(mask == result_mask)
mask2 = np.copy(mask)
mask2[2:-2, 4:-4, 4:-4] = 2
result_mask = MaskDistanceSplit.split(mask, 2, True, (2, 1, 1))
assert np.all(mask2 == result_mask)
