Pillow/Tests/test_webp_leaks.py
Hugo van Kemenade 53c3cd9f8e isort Tests
2024-01-20 13:23:03 +02:00

26 lines
541 B
Python

from __future__ import annotations
from io import BytesIO
from PIL import Image
from .helper import PillowLeakTestCase, skip_unless_feature
test_file = "Tests/images/hopper.webp"
@skip_unless_feature("webp")
class TestWebPLeaks(PillowLeakTestCase):
mem_limit = 3 * 1024 # kb
iterations = 100
def test_leak_load(self):
with open(test_file, "rb") as f:
im_data = f.read()
def core():
with Image.open(BytesIO(im_data)) as im:
im.load()
self._test_leak(core)