Pillow/Tests/test_webp_leaks.py
2019-06-13 18:54:46 +03:00

23 lines
544 B
Python

from .helper import unittest, PillowLeakTestCase
from PIL import Image, features
from io import BytesIO
test_file = "Tests/images/hopper.webp"
@unittest.skipUnless(features.check("webp"), "WebP is not installed")
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)