Pillow/Tests/test_webp_leaks.py
Jon Dufresne 33dabf986f Import unittest from stdlib rather than helper.py
The unittest in helper.py has not offered an interesting abstraction
since dbe9f85c7d so import from the more
typical stdlib location.
2019-11-20 18:42:52 -08:00

26 lines
552 B
Python

import unittest
from io import BytesIO
from PIL import Image, features
from .helper import PillowLeakTestCase
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)