mirror of
https://github.com/python-pillow/Pillow.git
synced 2025-07-12 09:12:22 +03:00
Inwcreased max memory limit, and switched save test to use BytesIO instead of StringIO
This commit is contained in:
parent
e791aa0325
commit
11e34d695a
|
@ -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()
|
||||||
|
|
Loading…
Reference in New Issue
Block a user