Optionally test image support with PIL

votchallenge · Jun 15, 2023 · 3147619 · 3147619
1 parent d0942d4
commit 3147619
Showing 1 changed file with 15 additions and 7 deletions.
diff --git a/support/tests/python/test_images.py b/support/tests/python/test_images.py
@@ -2,7 +2,6 @@
 import io
 import numpy as np
 from trax import MemoryImage, BufferImage
-from PIL import Image
 
 image1 = (np.random.random((100, 100, 3)) * 255).astype(np.uint8)
 
@@ -16,12 +15,21 @@
 
 assert(np.array_equal(image2, timage2.array()))
 
+try:
+    # Optional test if PIL is installed
+    from PIL import Image
+
+    image3 = Image.fromarray(image1)
+    bimage = io.BytesIO()
+    image3.save(bimage, format='JPEG')
+    simage = bimage.getvalue()
+
+    timage3 = BufferImage.create(simage)
+
+    assert(simage == timage3.buffer())
+
+except ImportError:
+    print("PIL not installed, skipping file buffer test")
 
-image3 = Image.fromarray(image1)
-bimage = io.BytesIO()
-image3.save(bimage, format='JPEG')
-simage = bimage.getvalue()
 
-timage3 = BufferImage.create(simage)
 
-assert(simage == timage3.buffer())