def test_construct_global_shift_array_static(self):
shift_axis_len = 5
shift_sigmas = (0.0, 1.0)
prob_slip = 0
prob_shift = 0
lcm_voxel_size = Coordinate((1, 1))
shift_array = np.zeros(shape=(shift_axis_len, len(shift_sigmas)),
dtype=int)
result = ShiftAugment.construct_global_shift_array(
shift_axis_len, shift_sigmas, prob_shift, prob_slip,
lcm_voxel_size)
self.assertTrue(np.array_equal(result, shift_array))
def test_construct_global_shift_array2(self):
shift_axis_len = 5
shift_sigmas = (0.0, 1.0)
prob_slip = 0
prob_shift = 1
lcm_voxel_size = Coordinate((1, 1))
shift_array = np.array([[0, 0], [0, -1], [0, 0], [0, 0], [0, 1]],
dtype=int)
result = ShiftAugment.construct_global_shift_array(
shift_axis_len, shift_sigmas, prob_slip, prob_shift,
lcm_voxel_size)
self.assertTrue(len(result) == shift_axis_len)
for position_shift in result:
self.assertTrue(position_shift[0] == 0)
self.assertTrue(np.array_equal(shift_array, result))