Added context managers

python-pillow · Mar 21, 2021 · b9c841c · b9c841c
1 parent b6f6fba
commit b9c841c
Showing 1 changed file with 12 additions and 6 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):
@@ -101,7 +105,8 @@ def test_truncated_file(tmp_path):
         f.write("P6")
 
     with pytest.raises(ValueError):
-        Image.open(path)
+        with Image.open(path):
+            pass
 
 
 def test_neg_ppm():
@@ -111,7 +116,8 @@ def test_neg_ppm():
     # sizes.
 
     with pytest.raises(OSError):
-        Image.open("Tests/images/negative_size.ppm")
+        with Image.open("Tests/images/negative_size.ppm"):
+            pass
 
 
 def test_mimetypes(tmp_path):