Add SSLEYTransform test

lightly/transforms/__init__.py

@@ -31,6 +31,7 @@
 from lightly.transforms.simsiam_transform import SimSiamTransform, SimSiamViewTransform
 from lightly.transforms.smog_transform import SMoGTransform, SmoGViewTransform
 from lightly.transforms.solarize import RandomSolarization
+from lightly.transforms.ssley_transform import SSLEYTransform
 from lightly.transforms.swav_transform import SwaVTransform, SwaVViewTransform
 from lightly.transforms.vicreg_transform import VICRegTransform, VICRegViewTransform
 from lightly.transforms.vicregl_transform import VICRegLTransform, VICRegLViewTransform

tests/transforms/test_ssley_transform.py

@@ -0,0 +1,13 @@
+from PIL import Image
+
+from lightly.transforms import SSLEYTransform
+
+
+class TestSSLEYTransform:
+    def test__call__(self) -> None:
+        transform = SSLEYTransform(input_size=32)
+
+        sample = Image.new("RGB", (100, 100))
+        output = transform(sample)
+        assert len(output) == 2
+        assert all(out.shape == (3, 32, 32) for out in output)