def test_zero_origin(self):
kwargs = dict(shape=5, resolution=1, origin="zero")
correct_output = [
np.arange(dim_size) * dim_res for dim_size, dim_res in zip((5,), (1,))
]
for dim, coord in enumerate(_compute_axes(**kwargs)):
assert np.array_equal(coord, correct_output[dim])
def test_anisotropic_resolution(self):
kwargs = dict(shape=(2, 3, 4), resolution=[1, 1.5, 2], origin="center")
correct_output = [
np.arange(dim_size) * dim_res - np.mean(np.arange(dim_size) * dim_res)
for dim_size, dim_res in zip((2, 3, 4), (1, 1.5, 2))
]
for dim, coord in enumerate(_compute_axes(**kwargs)):
assert np.array_equal(coord, correct_output[dim])
def test_zero_and_one_in_shape(self):
kwargs = dict(shape=(0, 1, 2), resolution=1, origin="center")
correct_output = [
np.arange(dim_size) * dim_res - np.mean(np.arange(dim_size) * dim_res)
for dim_size, dim_res in zip((0, 1, 2), (1, 1, 1))
]
for dim, coord in enumerate(_compute_axes(**kwargs)):
assert np.array_equal(coord, correct_output[dim])