Merge pull request #1 from radarhere/truncated

Test error is raised without LOAD_TRUNCATED_IMAGES
lajiyuan · Jan 16, 2024 · 6c55dff · 6c55dff
2 parents b2711c3 + 62e6d62
commit 6c55dff
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/Tests/images/end_trunc_file.png → Tests/images/truncated_end_chunk.png b/Tests/images/end_trunc_file.png → Tests/images/truncated_end_chunk.png
diff --git a/Tests/test_file_png.py b/Tests/test_file_png.py
@@ -778,11 +778,15 @@ class MyStdOut:
         with Image.open(mystdout) as reloaded:
             assert_image_equal_tofile(reloaded, TEST_PNG_FILE)
 
-    def test_end_truncated_file(self):
+    def test_truncated_end_chunk(self):
+        with Image.open("Tests/images/truncated_end_chunk.png") as im:
+            with pytest.raises(OSError):
+                im.load()
+
         ImageFile.LOAD_TRUNCATED_IMAGES = True
         try:
-            with Image.open("Tests/images/end_trunc_file.png") as im:
-                assert_image_equal_tofile(im, "Tests/images/end_trunc_file.png")
+            with Image.open("Tests/images/truncated_end_chunk.png") as im:
+                assert_image_equal_tofile(im, "Tests/images/hopper.png")
         finally:
             ImageFile.LOAD_TRUNCATED_IMAGES = False