python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-07-23 06:29:48 +03:00
Code Issues Packages Projects Releases Wiki Activity

Inwcreased max memory limit, and switched save test to use BytesIO instead of StringIO

Browse Source
This commit is contained in:
Josh Ware 2014-08-27 16:03:07 +10:00 committed by wiredfool
parent e791aa0325
commit 11e34d695a
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
Tests/test_jp2k_leaks.py
View File

@ -1,9 +1,10 @@
from helper import unittest, PillowTestCase, tearDownModule from helper import unittest, PillowTestCase, tearDownModule
import sys import sys
from PIL import Image from PIL import Image
from io import BytesIO
# Limits for testing the leak # Limits for testing the leak
mem_limit = 768*1048576 mem_limit = 1024*1048576
stack_size = 8*1048576 stack_size = 8*1048576
iterations = int((mem_limit/stack_size)*2) iterations = int((mem_limit/stack_size)*2)
codecs = dir(Image.core) codecs = dir(Image.core)
@ -26,16 +27,12 @@ class TestJpegLeaks(PillowTestCase):
def test_leak_save(self): def test_leak_save(self):
from resource import setrlimit, RLIMIT_AS, RLIMIT_STACK from resource import setrlimit, RLIMIT_AS, RLIMIT_STACK
try:
from cStringIO import StringIO
except ImportError:
from io import StringIO
setrlimit(RLIMIT_STACK, (stack_size, stack_size)) setrlimit(RLIMIT_STACK, (stack_size, stack_size))
setrlimit(RLIMIT_AS, (mem_limit, mem_limit)) setrlimit(RLIMIT_AS, (mem_limit, mem_limit))
for count in range(iterations): for count in range(iterations):
with Image.open(test_file) as im: with Image.open(test_file) as im:
im.load() im.load()
test_output = StringIO() test_output = BytesIO()
im.save(test_output, "JPEG2000") im.save(test_output, "JPEG2000")
test_output.seek(0) test_output.seek(0)
output = test_output.read() output = test_output.read()