Use monkeypatch

radarhere · May 16, 2024 · 7554e2c · 7554e2c
1 parent 2a703a2
commit 7554e2c
Showing 1 changed file with 7 additions and 9 deletions.
diff --git a/Tests/test_image.py b/Tests/test_image.py
@@ -116,16 +116,14 @@ def test_open_formats(self) -> None:
                 assert im.mode == "RGB"
                 assert im.size == (128, 128)
 
-    def test_open_verbose_failure(self) -> None:
+    def test_open_verbose_failure(self, monkeypatch: pytest.MonkeyPatch) -> None:
+        monkeypatch.setattr(Image, "WARN_POSSIBLE_FORMATS", True)
+
         im = io.BytesIO(b"")
-        Image.WARN_POSSIBLE_FORMATS = True
-        try:
-            with pytest.warns(UserWarning):
-                with pytest.raises(UnidentifiedImageError):
-                    with Image.open(im):
-                        pass
-        finally:
-            Image.WARN_POSSIBLE_FORMATS = False
+        with pytest.warns(UserWarning):
+            with pytest.raises(UnidentifiedImageError):
+                with Image.open(im):
+                    pass
 
     def test_width_height(self) -> None:
         im = Image.new("RGB", (1, 2))