add transform noop test

pytorch · Jun 28, 2023 · 95c80d2 · 95c80d2
1 parent 5dd4e53
commit 95c80d2
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/test/test_transforms_v2_refactored.py b/test/test_transforms_v2_refactored.py
@@ -804,7 +804,7 @@ def test_dispatcher_signature(self, kernel, input_type):
     def test_transform(self, input_type, device):
         input = self._make_input(input_type, device=device)
 
-        check_transform(transforms.RandomHorizontalFlip, input)
+        check_transform(transforms.RandomHorizontalFlip, input, p=1)
 
     @pytest.mark.parametrize(
         "fn", [F.horizontal_flip, transform_cls_to_functional(transforms.RandomHorizontalFlip, p=1)]
@@ -846,3 +846,17 @@ def test_bounding_box_correctness(self, format, fn):
         expected = self._reference_horizontal_flip_bounding_box(bounding_box)
 
         torch.testing.assert_close(actual, expected)
+
+    @pytest.mark.parametrize(
+        "input_type",
+        [torch.Tensor, PIL.Image.Image, datapoints.Image, datapoints.BoundingBox, datapoints.Mask, datapoints.Video],
+    )
+    @pytest.mark.parametrize("device", cpu_and_cuda())
+    def test_transform_noop(self, input_type, device):
+        input = self._make_input(input_type, device=device)
+
+        transform = transforms.RandomHorizontalFlip(p=0)
+
+        output = transform(input)
+
+        assert_equal(output, input)