test for OpenCV bug #89

Signed-off-by: Alexander Piskun <[email protected]>

tests/opencv_bug.py

@@ -0,0 +1,11 @@
+# https://github.com/bigcat88/pillow_heif/issues/89
+
+from io import BytesIO
+
+import cv2  # noqa
+from PIL import Image
+
+from pillow_heif import from_pillow, libheif_info
+
+print(libheif_info())
+from_pillow(Image.linear_gradient(mode="L")).save(BytesIO(), format="HEIF")

tests/opencv_test.py

@@ -1,8 +1,10 @@
 # With OpenCV, we test BGR, BGRA, BGR;16 and BGRA;16 modes
 
 import os
+import sys
 from io import BytesIO
 from pathlib import Path
+from subprocess import CalledProcessError, run
 
 import pytest
 from helpers import compare_hashes, hevc_enc
@@ -145,3 +147,13 @@ def test_read_8_10_12_bit(img):
     path_to_png = path_to_png.replace("_10_", "_16_")
     path_to_png = path_to_png.replace("_12_", "_16_")
     compare_hashes([BytesIO(img_encode), path_to_png], hash_size=16)
+
+
+def test_opencv_crash():
+    # https://github.com/bigcat88/pillow_heif/issues/89
+    path_to_test_file = os.path.join(os.path.dirname(os.path.abspath(__file__)), "opencv_bug.py")
+    if sys.platform.lower() == "darwin":
+        with pytest.raises(CalledProcessError):
+            run([sys.executable, path_to_test_file], check=True)
+    else:
+        run([sys.executable, path_to_test_file], check=True)