def test_tramsforms_cupy_batch(self, params, input, expected):
input = {"image": cp.asarray(input)[cp.newaxis, ...]}
expected = cp.asarray(expected)[cp.newaxis, ...]
output = RandCuCIMd(keys="image", **params)(input)["image"]
self.assertTrue(output.dtype == expected.dtype)
self.assertTrue(isinstance(output, cp.ndarray))
cp.testing.assert_allclose(output, expected)
output = RandCuCIMd(keys="image", apply_prob=0.0,
**params)(input)["image"]
self.assertTrue(output.dtype == input["image"].dtype)
self.assertTrue(isinstance(output, cp.ndarray))
cp.testing.assert_allclose(output, input["image"])
def test_tramsforms_numpy_single(self, params, input, expected):
input = {"image": np.copy(input)}
output = RandCuCIMd(keys="image", apply_prob=1.0,
**params)(input)["image"]
self.assertTrue(output.dtype == expected.dtype)
self.assertTrue(isinstance(output, np.ndarray))
cp.testing.assert_allclose(output, expected)
output = RandCuCIMd(keys="image", apply_prob=0.0,
**params)(input)["image"]
self.assertTrue(output.dtype == input["image"].dtype)
self.assertTrue(isinstance(output, np.ndarray))
cp.testing.assert_allclose(output, input["image"])