Fix some tests

Author: NicolasHug

test/smoke_test.py

@@ -34,16 +34,21 @@ def smoke_test_torchvision_read_decode() -> None:
         raise RuntimeError(f"Unexpected shape of img_webp: {img_webp.shape}")
 
     if sys.platform == "linux":
-        pass
-        # img_avif = decode_avif(read_file(str(SCRIPT_DIR / "assets/fakedata/logos/rgb_pytorch.avif")))
-        # if img_avif.shape != (3, 100, 100):
-        #     raise RuntimeError(f"Unexpected shape of img_avif: {img_avif.shape}")
-
-        # img_heic = decode_heic(
-        #     read_file(str(SCRIPT_DIR / "assets/fakedata/logos/rgb_pytorch_incorrectly_encoded_but_who_cares.heic"))
-        # )
-        # if img_heic.shape != (3, 100, 100):
-        #     raise RuntimeError(f"Unexpected shape of img_heic: {img_heic.shape}")
+        if torch.cuda.is_available():
+            # TODO: For whatever reason this only passes on the runners that
+            # support CUDA.
+            # Strangely, on the CPU runners where this fails, the AVIF/HEIC
+            # tests (ran with pytest) are passing. This is likely related to a
+            # libcxx symbol thing.
+            img_avif = decode_avif(read_file(str(SCRIPT_DIR / "assets/fakedata/logos/rgb_pytorch.avif")))
+            if img_avif.shape != (3, 100, 100):
+                raise RuntimeError(f"Unexpected shape of img_avif: {img_avif.shape}")
+
+            img_heic = decode_heic(
+                read_file(str(SCRIPT_DIR / "assets/fakedata/logos/rgb_pytorch_incorrectly_encoded_but_who_cares.heic"))
+            )
+            if img_heic.shape != (3, 100, 100):
+                raise RuntimeError(f"Unexpected shape of img_heic: {img_heic.shape}")
     else:
         try:
             decode_avif(str(SCRIPT_DIR / "assets/fakedata/logos/rgb_pytorch.avif"))

test/test_image.py

@@ -974,6 +974,7 @@ def test_decode_avif_heic_against_pil(decode_fun, mode, pil_mode, filename):
                 "no 'ispe' property",
                 "'iref' has double references",
                 "Invalid image grid",
+                "decode_heif failed: Invalid input: No 'meta' box",
             )
         ):
             pytest.skip(reason="Expected failure, that's OK")
@@ -990,7 +991,7 @@ def test_decode_avif_heic_against_pil(decode_fun, mode, pil_mode, filename):
     try:
         from_pil = F.pil_to_tensor(Image.open(filename).convert(pil_mode))
     except RuntimeError as e:
-        if "Invalid image grid" in str(e):
+        if any(s in str(e) for s in ("Invalid image grid", "Failed to decode image: Not implemented")):
             pytest.skip(reason="PIL failure")
         else:
             raise e