add expand=True parameterization

pytorch · Jul 3, 2023 · a3f55b1 · a3f55b1
1 parent ad3b2a9
commit a3f55b1
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/test/test_transforms_v2_refactored.py b/test/test_transforms_v2_refactored.py
@@ -1512,8 +1512,7 @@ def test_transform(self, input_type, device):
     @pytest.mark.parametrize(
         "interpolation", [transforms.InterpolationMode.NEAREST, transforms.InterpolationMode.BILINEAR]
     )
-    # TODO: investigate why expand=True leads to different shapes between PIL and tensor
-    @pytest.mark.parametrize("expand", [False])
+    @pytest.mark.parametrize("expand", [False, True])
     @pytest.mark.parametrize("fill", CORRECTNESS_FILLS)
     def test_functional_image_correctness(self, angle, center, interpolation, expand, fill):
         image = make_input(torch.Tensor, dtype=torch.uint8, device="cpu")
@@ -1534,8 +1533,7 @@ def test_functional_image_correctness(self, angle, center, interpolation, expand
     @pytest.mark.parametrize(
         "interpolation", [transforms.InterpolationMode.NEAREST, transforms.InterpolationMode.BILINEAR]
     )
-    # TODO: investigate why expand=True leads to different shapes between PIL and tensor
-    @pytest.mark.parametrize("expand", [False])
+    @pytest.mark.parametrize("expand", [False, True])
     @pytest.mark.parametrize("fill", CORRECTNESS_FILLS)
     @pytest.mark.parametrize("seed", list(range(5)))
     def test_transform_image_correctness(self, center, interpolation, expand, fill, seed):