def test_shift(self, dtype):
a = cupy.ones((32,), dtype=dtype)
shift = cupyx.scipy.ndimage.shift
output = self._prep_output(a)
if _util._is_integer_output(output, a) and self.mode == 'constant':
with pytest.raises(NotImplementedError):
shift(a, 1, output=output, order=self.order, mode=self.mode,
cval=self.cval)
else:
shift(a, 1, output=output, order=self.order, mode=self.mode,
cval=self.cval)
def test_map_coordinates(self, dtype):
a = cupy.ones((32,), dtype=dtype)
coords = cupy.arange(32)[cupy.newaxis, :] + 2.5
map_coordinates = cupyx.scipy.ndimage.map_coordinates
output = self._prep_output(a)
if _util._is_integer_output(output, a) and self.mode == 'constant':
with pytest.raises(NotImplementedError):
map_coordinates(a, coords, output=output, order=self.order,
mode=self.mode, cval=self.cval)
else:
map_coordinates(a, coords, output=output, order=self.order,
mode=self.mode, cval=self.cval)
def test_affine(self, dtype):
a = cupy.ones((16, 16), dtype=dtype)
affine = cupy.eye(2)
affine_transform = cupyx.scipy.ndimage.affine_transform
output = self._prep_output(a)
if _util._is_integer_output(output, a) and self.mode == 'constant':
with pytest.raises(NotImplementedError):
affine_transform(a, affine, output=output, order=self.order,
mode=self.mode, cval=self.cval)
else:
affine_transform(a, affine, output=output, order=self.order,
mode=self.mode, cval=self.cval)
def test_rotate(self, dtype):
a = cupy.ones((16, 16), dtype=dtype)
rotate = cupyx.scipy.ndimage.rotate
output = self._prep_output(a)
if _util._is_integer_output(output, a) and self.mode == 'constant':
with pytest.raises(NotImplementedError):
# rotate by 0 to keep same shape
rotate(a, 0, output=output, order=self.order, mode=self.mode,
cval=self.cval)
else:
rotate(a, 0, output=output, order=self.order, mode=self.mode,
cval=self.cval)
def test_zoom(self, dtype):
a = cupy.ones((32,), dtype=dtype)
zoom = cupyx.scipy.ndimage.zoom
output = self._prep_output(a)
if _util._is_integer_output(output, a) and self.mode == 'constant':
with pytest.raises(NotImplementedError):
# zoom of 1.0 to keep same shape
zoom(a, 1, output=output, order=self.order, mode=self.mode,
cval=self.cval)
else:
zoom(a, 1, output=output, order=self.order, mode=self.mode,
cval=self.cval)
