updating tests to pass with libheif==1.13 #38

Author: bigcat88

tests/basic_test.py

@@ -1,9 +1,9 @@
 import builtins
 import os
-import sys
 from pathlib import Path
 
 import dataset
+import helpers
 import pytest
 
 import pillow_heif
@@ -13,13 +13,15 @@
 
 def test_libheif_info():
     info = pillow_heif.libheif_info()
-    assert info["version"]["libheif"] == "1.12.0"
+    assert info["version"]["libheif"] in ("1.12.0", "1.13.0")
     assert info["decoders"]["HEVC"]
-    if os.getenv("PH_LIGHT", "0") != "0" or sys.maxsize <= 2**32:
-        return
-    assert info["decoders"]["AV1"]
-    assert info["encoders"]["AV1"]
-    assert info["encoders"]["HEVC"]
+
+
+@pytest.mark.skipif(helpers.aom_enc() and helpers.aom_dec(), reason="Only when AOM missing.")
+@pytest.mark.skipif(pillow_heif.libheif_info()["version"]["aom"] == "Rav1e encoder", reason="Rav1e not supported")
+def test_pillow_register_avif_plugin():
+    with pytest.warns(UserWarning):
+        pillow_heif.register_avif_opener()
 
 
 @pytest.mark.parametrize("img_path", dataset.FULL_DATASET)
@@ -63,3 +65,21 @@ def test_heif_str():
     assert str(heif_file.thumbnails[0]) == f"{str_thumb_nl} Original:{str_img_l_1}"
     heif_file.thumbnails[0].load()  # Should not change anything, thumbnails are cloned without data.
     assert str(heif_file.thumbnails[0]) == f"{str_thumb_nl} Original:{str_img_l_1}"
+
+
+@pytest.mark.skipif(not helpers.RELEASE_FULL_FLAG, reason="Only when building full release")
+def test_full_build():
+    info = pillow_heif.libheif_info()
+    assert info["decoders"]["AV1"]
+    assert info["encoders"]["AV1"]
+    assert info["encoders"]["HEVC"]
+    assert info["version"]["libheif"] == os.getenv("EXP_PH_LIBHEIF_VERSION", "1.13.0")
+
+
+@pytest.mark.skipif(not helpers.RELEASE_LIGHT_FLAG, reason="Only when building light release")
+def test_light_build():
+    info = pillow_heif.libheif_info()
+    assert not info["decoders"]["AV1"]
+    assert not info["encoders"]["AV1"]
+    assert not info["encoders"]["HEVC"]
+    assert info["version"]["libheif"] == os.getenv("EXP_PH_LIBHEIF_VERSION", "1.13.0")

tests/helpers.py

@@ -22,6 +22,10 @@
     np = None
 
 
+RELEASE_FULL_FLAG = getenv("PH_FULL_ACTION", "0") == "1"
+RELEASE_LIGHT_FLAG = getenv("PH_LIGHT_ACTION", "0") == "1"
+
+
 def assert_image_equal(a, b):
     assert a.mode == b.mode
     assert a.size == b.size