mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-11-10 19:56:47 +03:00
33dabf986f
The unittest in helper.py has not offered an interesting abstraction
since dbe9f85c7d
so import from the more
typical stdlib location.
26 lines
552 B
Python
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)
|