mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-11-14 05:36:48 +03:00
33dabf986f
The unittest in helper.py has not offered an interesting abstraction
since dbe9f85c7d so import from the more
typical stdlib location.
47 lines
1.3 KiB
Python
Executable File
47 lines
1.3 KiB
Python
Executable File
#!/usr/bin/env python
|
|
import unittest
|
|
|
|
from PIL import Image
|
|
|
|
from .helper import PillowTestCase, is_win32
|
|
|
|
min_iterations = 100
|
|
max_iterations = 10000
|
|
|
|
|
|
@unittest.skipIf(is_win32(), "requires Unix or macOS")
|
|
class TestImagingLeaks(PillowTestCase):
|
|
def _get_mem_usage(self):
|
|
from resource import getpagesize, getrusage, RUSAGE_SELF
|
|
|
|
mem = getrusage(RUSAGE_SELF).ru_maxrss
|
|
return mem * getpagesize() / 1024 / 1024
|
|
|
|
def _test_leak(self, min_iterations, max_iterations, fn, *args, **kwargs):
|
|
mem_limit = None
|
|
for i in range(max_iterations):
|
|
fn(*args, **kwargs)
|
|
mem = self._get_mem_usage()
|
|
if i < min_iterations:
|
|
mem_limit = mem + 1
|
|
continue
|
|
msg = "memory usage limit exceeded after %d iterations" % (i + 1)
|
|
self.assertLessEqual(mem, mem_limit, msg)
|
|
|
|
def test_leak_putdata(self):
|
|
im = Image.new("RGB", (25, 25))
|
|
self._test_leak(min_iterations, max_iterations, im.putdata, im.getdata())
|
|
|
|
def test_leak_getlist(self):
|
|
im = Image.new("P", (25, 25))
|
|
self._test_leak(
|
|
min_iterations,
|
|
max_iterations,
|
|
# Pass a new list at each iteration.
|
|
lambda: im.point(range(256)),
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|