Update Tests/check_avif_leaks.py

fdintino · radarhere · web-flow · commit 8b8bbba0f77c · 2024-10-18T09:33:55.000-04:00
Co-authored-by: Andrew Murray &lt;3112309+radarhere@users.noreply.github.com&gt;
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()
