Fix test func.

python-pillow · Jan 12, 2024 · 8326ec5 · 8326ec5
1 parent d7120dd
commit 8326ec5
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/Tests/test_file_png.py b/Tests/test_file_png.py
@@ -781,9 +781,8 @@ class MyStdOut:
     def test_end_truncated_filed(self):
         ImageFile.LOAD_TRUNCATED_IMAGES = True
         try:
-            img =  Image.open("Tests/images/end_trunc_file.png")
-            img.load()
-            assert img is not None
+            with Image.open("Tests/images/end_trunc_file.png") as im:
+                assert_image_equal_tofile(im, "Tests/images/end_trunc_file.png")
         finally:
             ImageFile.LOAD_TRUNCATED_IMAGES = False