Added context managers

python-pillow · Aug 24, 2021 · 1b39775 · 1b39775
1 parent 0f11d22
commit 1b39775
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/Tests/test_file_jpeg.py b/Tests/test_file_jpeg.py
@@ -895,8 +895,8 @@ class MyStdOut:
 
         if buffer:
             mystdout = mystdout.buffer
-        reloaded = Image.open(mystdout)
-        assert_image_equal(reloaded, im_roundtrip)
+        with Image.open(mystdout) as reloaded:
+            assert_image_equal(reloaded, im_roundtrip)
 
 
 @pytest.mark.skipif(not is_win32(), reason="Windows only")

diff --git a/Tests/test_file_ppm.py b/Tests/test_file_ppm.py
@@ -108,5 +108,5 @@ class MyStdOut:
 
     if buffer:
         mystdout = mystdout.buffer
-    reloaded = Image.open(mystdout)
-    assert_image_equal_tofile(reloaded, TEST_FILE)
+    with Image.open(mystdout) as reloaded:
+        assert_image_equal_tofile(reloaded, TEST_FILE)