Fix unclosed test files

matplotlib · Feb 14, 2024 · a9713dc · a9713dc
1 parent 3bd3703
commit a9713dc
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/tests/test_default_tolerance.py b/tests/test_default_tolerance.py
@@ -9,11 +9,11 @@
 @pytest.fixture(scope="module")
 def baseline_image(tmp_path_factory):
     path = Path(__file__).parent / "baseline" / "2.0.x" / f"{TEST_NAME}.png"
-    image = Image.open(path)
-    draw = ImageDraw.Draw(image)
-    draw.rectangle(((0, 0), (100, 100)), fill="red")
-    output = tmp_path_factory.mktemp("data") / f"{TEST_NAME}.png"
-    image.save(output)
+    with Image.open(path) as image:
+        draw = ImageDraw.Draw(image)
+        draw.rectangle(((0, 0), (100, 100)), fill="red")
+        output = tmp_path_factory.mktemp("data") / f"{TEST_NAME}.png"
+        image.save(output)
     return output
 
 

diff --git a/tests/test_deterministic.py b/tests/test_deterministic.py
@@ -113,7 +113,8 @@ def test_mpl():
                 assert key_in_file
 
     else:  # "eps" or "png"
-        actual_metadata = Image.open(str(baseline_image)).info
+        with Image.open(str(baseline_image)) as image:
+            actual_metadata = image.info
         for k, expected in deterministic_metadata.items():
             actual = actual_metadata.get(k, None)
             if success_expected:  # metadata keys should not be in the file