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))