More robust check in tests for 16 bits images (#5652)

90dace8d · Nicolas Hug · GitHub · fc63f828 · 90dace8d
Unverified Commit 90dace8d authored Mar 21, 2022 by Nicolas Hug Committed by GitHub Mar 21, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

test/test_image.py test/test_image.py +1 -1

No files found.
--- a/test/test_image.py
+++ b/test/test_image.py
@@ -158,7 +158,7 @@ def test_decode_png(img_path, pil_mode, mode):

    img_pil = normalize_dimensions(img_pil)

-    if "16" in img_path:
+    if img_path.endswith("16.png"):
        # 16 bits image decoding is supported, but only as a private API
        # FIXME: see https://github.com/pytorch/vision/issues/4731 for potential solutions to making it public
        with pytest.raises(RuntimeError, match="At most 8-bit PNG images are supported"):