Added tests

python-pillow · Aug 1, 2021 · 8045ecc · 8045ecc
1 parent 8828080
commit 8045ecc
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 1 deletion.
diff --git a/Tests/images/expected_to_read.jp2 b/Tests/images/expected_to_read.jp2
diff --git a/Tests/images/invalid_header_length.jp2 b/Tests/images/invalid_header_length.jp2
diff --git a/Tests/images/not_enough_data.jp2 b/Tests/images/not_enough_data.jp2
diff --git a/Tests/test_file_jpeg2k.py b/Tests/test_file_jpeg2k.py
@@ -4,7 +4,7 @@
 
 import pytest
 
-from PIL import Image, ImageFile, Jpeg2KImagePlugin, features
+from PIL import Image, ImageFile, Jpeg2KImagePlugin, UnidentifiedImageError, features
 
 from .helper import (
     assert_image_equal,
@@ -159,6 +159,20 @@ def test_load_dpi():
         assert "dpi" not in im.info
 
 
+def test_header_errors():
+    for path in (
+        "Tests/images/invalid_header_length.jp2",
+        "Tests/images/not_enough_data.jp2",
+    ):
+        with pytest.raises(UnidentifiedImageError):
+            with Image.open(path):
+                pass
+
+    with pytest.raises(OSError):
+        with Image.open("Tests/images/expected_to_read.jp2"):
+            pass
+
+
 def test_layers_type(tmp_path):
     outfile = str(tmp_path / "temp_layers.jp2")
     for quality_layers in [[100, 50, 10], (100, 50, 10), None]: