Pillow/Tests/test_webp_leaks.py
2024-01-27 15:19:43 +11:00

26 lines
557 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) -> None:
with open(test_file, "rb") as f:
im_data = f.read()
def core() -> None:
with Image.open(BytesIO(im_data)) as im:
im.load()
self._test_leak(core)