mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-12-25 01:16:16 +03:00
4f185329f4
This adds a new test decorator: skip_unless_feature(). The argument is the same as passed to features.check(). If the feature is not supported, the test will be skipped. This removes several kinds of boilerplate copied and pasted around tests so test feature checking is handled and displayed more consistently. Refs #4193
25 lines
506 B
Python
25 lines
506 B
Python
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)
|