def test_transformation_interval(self, x, params):
transformation = RotationZ()
actual = transformation.transform(x, params)
self.assertSound(actual, params, partial(transformation.transform, x))
def test_transformation_float(self, x, alpha):
transformation = RotationZ()
expected = rotate_z(x, alpha.item())
actual = transformation.transform(x, alpha)
self.assertAlmostEqualNumpy(expected, actual)