def test_lambda_slicing(self):
img = self.imt
def slice_func(x):
return x[:, :, :6, ::-2]
lambd = Lambda(func=slice_func)
self.assertTrue(np.allclose(slice_func(img), lambd(img)))
def test_lambda_identity(self):
img = self.imt
def identity_func(x):
return x
lambd = Lambda(func=identity_func)
self.assertTrue(np.allclose(identity_func(img), lambd(img)))
def test_lambda_slicing(self):
for p in TEST_NDARRAYS:
img = p(self.imt)
def slice_func(x):
return x[:, :, :6, ::2]
lambd = Lambda(func=slice_func)
assert_allclose(slice_func(img), lambd(img))
def test_lambda_identity(self):
for p in TEST_NDARRAYS:
img = p(self.imt)
def identity_func(x):
return x
lambd = Lambda(func=identity_func)
assert_allclose(identity_func(img), lambd(img))
def __init__(
self,
keys: KeysCollection,
func: Union[Sequence[Callable], Callable],
overwrite: Union[Sequence[bool], bool] = True,
) -> None:
super().__init__(keys)
self.func = ensure_tuple_rep(func, len(self.keys))
self.overwrite = ensure_tuple_rep(overwrite, len(self.keys))
self._lambd = Lambda()
def test_lambda_slicing(self):
for p in TEST_NDARRAYS:
img = p(self.imt)
def slice_func(x):
return x[:, :, :6, ::2]
lambd = Lambda(func=slice_func)
out = lambd(img)
assert_allclose(slice_func(img), out, type_test=False)
self.assertIsInstance(out, MetaTensor)
self.assertEqual(len(out.applied_operations), 1)
out = lambd.inverse(out)
self.assertEqual(len(out.applied_operations), 0)
def __init__(self, keys: KeysCollection, func: Union[Sequence[Callable],
Callable]) -> None:
super().__init__(keys)
self.func = ensure_tuple_rep(func, len(self.keys))
self.lambd = Lambda()
