✅ Fix transforms tests

ENSTA-U2IS-AI · Dec 20, 2023 · 268b283 · 268b283
1 parent 6cc8e34
commit 268b283
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/tests/transforms/test_transforms.py → tests/transforms/test_image.py b/tests/transforms/test_transforms.py → tests/transforms/test_image.py
@@ -12,8 +12,8 @@
     MIMOBatchFormat,
     Posterize,
     RepeatTarget,
-    Rotation,
-    Sharpness,
+    Rotate,
+    Sharpen,
     Shear,
     Solarize,
     Translate,
@@ -78,11 +78,11 @@ def test_failures(self, img_input):
             _ = aug(img_input, -1)
 
 
-class TestRotation:
-    """Testing the Rotation transform."""
+class TestRotate:
+    """Testing the Rotate transform."""
 
     def test_pil(self, img_input):
-        aug = Rotation(random_direction=True)
+        aug = Rotate(random_direction=True)
         _ = aug(img_input, 10)
 
 
@@ -136,15 +136,15 @@ def test_failures(self, img_input):
             _ = aug(img_input, -1)
 
 
-class TestSharpness:
-    """Testing the Sharpness transform."""
+class TestSharpen:
+    """Testing the Sharpen transform."""
 
     def test_pil(self, img_input):
-        aug = Sharpness()
+        aug = Sharpen()
         _ = aug(img_input, 2)
 
     def test_failures(self, img_input):
-        aug = Sharpness()
+        aug = Sharpen()
         with pytest.raises(ValueError):
             _ = aug(img_input, -1)