Update Tests/check_avif_leaks.py

Co-authored-by: Andrew Murray <[email protected]>
python-pillow · Oct 18, 2024 · 8b8bbba · 8b8bbba
1 parent e5494a2
commit 8b8bbba
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/Tests/check_avif_leaks.py b/Tests/check_avif_leaks.py
@@ -36,9 +36,8 @@ def test_leak_save() -> None:
     setrlimit(RLIMIT_STACK, (stack_size, stack_size))
     setrlimit(RLIMIT_AS, (mem_limit, mem_limit))
     for _ in range(iterations):
+        test_output = BytesIO()
         with Image.open(test_file) as im:
-            im.load()
-            test_output = BytesIO()
             im.save(test_output, "AVIF")
-            test_output.seek(0)
-            test_output.read()
+        test_output.seek(0)
+        test_output.read()