Pillow/Tests/check_imaging_leaks.py

46 lines
1.4 KiB
Python
Raw Normal View History

#!/usr/bin/env python
from __future__ import division
from helper import unittest, PillowTestCase
import sys
2015-07-03 08:03:25 +03:00
from PIL import Image
min_iterations = 100
max_iterations = 10000
2015-04-24 02:26:52 +03:00
2018-03-19 11:36:07 +03:00
@unittest.skipIf(sys.platform.startswith('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
self.assertLessEqual(mem, mem_limit,
msg='memory usage limit exceeded after %d iterations'
% (i + 1))
def test_leak_putdata(self):
im = Image.new('RGB', (25, 25))
2015-07-03 09:22:56 +03:00
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)))
2018-03-03 12:54:00 +03:00
if __name__ == '__main__':
unittest.main()