Added context managers

python-pillow · Mar 21, 2021 · 41a439d · 41a439d
1 parent b6f6fba
commit 41a439d
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/Tests/test_file_ppm.py b/Tests/test_file_ppm.py
@@ -56,7 +56,8 @@ def test_not_ppm(tmp_path):
         f.write(b"PyInvalid")
 
     with pytest.raises(UnidentifiedImageError):
-        Image.open(path)
+        with Image.open(path):
+            pass
 
 
 def test_header_with_comments(tmp_path):
@@ -74,7 +75,8 @@ def test_nondecimal_header(tmp_path):
         f.write(b"P6\n128\x00")
 
     with pytest.raises(ValueError):
-        Image.open(path)
+        with Image.open(path):
+            pass
 
 
 def test_token_too_long(tmp_path):
@@ -83,7 +85,8 @@ def test_token_too_long(tmp_path):
         f.write(b"P6\n 0123456789")
 
     with pytest.raises(ValueError):
-        Image.open(path)
+        with Image.open(path):
+            pass
 
 
 def test_too_many_colors(tmp_path):
@@ -92,7 +95,8 @@ def test_too_many_colors(tmp_path):
         f.write(b"P6\n1 1\n1000\n")
 
     with pytest.raises(ValueError):
-        Image.open(path)
+        with Image.open(path):
+            pass
 
 
 def test_truncated_file(tmp_path):