From a3f55b1ce478abdaf2025bdd323fbe888aa1a07c Mon Sep 17 00:00:00 2001 From: Philip Meier Date: Mon, 3 Jul 2023 12:53:36 +0200 Subject: [PATCH] add expand=True parameterization --- test/test_transforms_v2_refactored.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/test/test_transforms_v2_refactored.py b/test/test_transforms_v2_refactored.py index 42d182bdc3b..8dfcde8264a 100644 --- 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):