def test_ndim_no_shift(self):
correct = (
np.array([[-2], [-1], [0]]),
np.array([[-2], [-1], [0], [1]]),
np.array([
[-1, -1],
[-1, 0],
[0, -1],
[0, 0],
]),
np.array([[-1, -1, -1], [-1, -1, 0], [-1, 0, -1], [-1, 0, 0]]),
)
for index, shape in enumerate(self.shapes):
with self.subTest():
old_pos = [np.arange(-val // 2, val // 2) for val in shape]
new_pos = reg.calc_new_positions(old_pos,
shift=(0, ) * len(shape))
self.assertTrue(np.allclose(new_pos, correct[index]))
def test_ndim_with_shift(self):
correct = (
np.array([[-3], [-2], [-1]]),
np.array([[-1.8], [-0.8], [0.2], [1.2]]),
np.array([
[-3.3, -0.9],
[-3.3, 0.1],
[-2.3, -0.9],
[-2.3, 0.1],
]),
np.array([[-2.1, -1.3, -1], [-2.1, -1.3, 0], [-2.1, -0.3, -1],
[-2.1, -0.3, 0]]),
)
for index, shape in enumerate(self.shapes):
with self.subTest():
old_pos = [np.arange(-val // 2, val // 2) for val in shape]
new_pos = reg.calc_new_positions(old_pos,
shift=self.shifts[index])
self.assertTrue(np.allclose(new_pos, correct[index]))
def test_wrong_shift_none(self):
with self.assertRaises(ValueError):
reg.calc_new_positions([np.arange(-2, 1)], shift=None)
def test_empty_positions(self):
with self.assertRaises(ValueError):
reg.calc_new_positions([], shift=[])