mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-12-27 10:26:19 +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
43 lines
1.2 KiB
Python
Executable File
43 lines
1.2 KiB
Python
Executable File
import unittest
|
|
from io import BytesIO
|
|
|
|
from PIL import Image
|
|
|
|
from .helper import PillowTestCase, is_win32, skip_unless_feature
|
|
|
|
# Limits for testing the leak
|
|
mem_limit = 1024 * 1048576
|
|
stack_size = 8 * 1048576
|
|
iterations = int((mem_limit / stack_size) * 2)
|
|
test_file = "Tests/images/rgb_trns_ycbc.jp2"
|
|
|
|
|
|
@unittest.skipIf(is_win32(), "requires Unix or macOS")
|
|
@skip_unless_feature("jpg_2000")
|
|
class TestJpegLeaks(PillowTestCase):
|
|
def test_leak_load(self):
|
|
from resource import setrlimit, RLIMIT_AS, RLIMIT_STACK
|
|
|
|
setrlimit(RLIMIT_STACK, (stack_size, stack_size))
|
|
setrlimit(RLIMIT_AS, (mem_limit, mem_limit))
|
|
for _ in range(iterations):
|
|
with Image.open(test_file) as im:
|
|
im.load()
|
|
|
|
def test_leak_save(self):
|
|
from resource import setrlimit, RLIMIT_AS, RLIMIT_STACK
|
|
|
|
setrlimit(RLIMIT_STACK, (stack_size, stack_size))
|
|
setrlimit(RLIMIT_AS, (mem_limit, mem_limit))
|
|
for _ in range(iterations):
|
|
with Image.open(test_file) as im:
|
|
im.load()
|
|
test_output = BytesIO()
|
|
im.save(test_output, "JPEG2000")
|
|
test_output.seek(0)
|
|
test_output.read()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|