def test_register(self):
"""
Test register.
"""
dtype = "int"
def add1(t, x):
return x + 1
def flip_sub_width(t, x):
return x - t.width
T.Transform.register_type(dtype, add1)
T.HFlipTransform.register_type(dtype, flip_sub_width)
transforms = T.TransformList([
T.ScaleTransform(0, 0, 0, 0, 0),
T.CropTransform(0, 0, 0, 0),
T.HFlipTransform(3),
])
self.assertEqual(transforms.apply_int(3), 2)
# Testing __add__, __iadd__, __radd__, __len__.
transforms = transforms + transforms
transforms += transforms
transforms = T.NoOpTransform() + transforms
self.assertEqual(len(transforms), 13)
with self.assertRaises(AssertionError):
T.HFlipTransform.register_type(dtype, lambda x: 1)
with self.assertRaises(AttributeError):
transforms.no_existing
def test_print_transform(self):
t0 = T.HFlipTransform(width=100)
self.assertEqual(str(t0), "HFlipTransform(width=100)")
t = T.TransformList([T.NoOpTransform(), t0])
self.assertEqual(str(t), f"TransformList[NoOpTransform(), {t0}]")
t = T.BlendTransform(np.zeros((100, 100, 100)), 1.0, 1.0)
self.assertEqual(
str(t),
"BlendTransform(src_image=..., src_weight=1.0, dst_weight=1.0)")
def test_noop_transform_no_register(self):
"""
NoOpTransform does not need register - it's by default no-op.
"""
t = T.NoOpTransform()
self.assertEqual(t.apply_anything(1), 1)
