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)