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

Convert large_memory_numpy_test.py to use unittest (and change second XDIM on line 36 to YDIM)

Browse Source
This commit is contained in:
hugovk 2014-06-25 12:43:01 +03:00
parent c6386a294d
commit cf07aa60a1
2 changed files with 27 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
Tests/large_memory_numpy_test.py
View File

@ -1,4 +1,4 @@
from tester import * from helper import *
# This test is not run automatically. # This test is not run automatically.
# #
@ -6,32 +6,37 @@ from tester import *
# second test. Running this automatically would amount to a denial of # second test. Running this automatically would amount to a denial of
# service on our testing infrastructure. I expect this test to fail # service on our testing infrastructure. I expect this test to fail
# on any 32 bit machine, as well as any smallish things (like # on any 32 bit machine, as well as any smallish things (like
# raspberrypis). # Raspberry Pis).
from PIL import Image from PIL import Image
try: try:
import numpy as np import numpy as np
except: except:
skip() sys.exit("Skipping: Numpy not installed")
ydim = 32769 YDIM = 32769
xdim = 48000 XDIM = 48000
f = tempfile('temp.png')
def _write_png(xdim,ydim):
class LargeMemoryNumpyTest(PillowTestCase):
def _write_png(self, XDIM, YDIM):
dtype = np.uint8 dtype = np.uint8
a = np.zeros((xdim, ydim), dtype=dtype) a = np.zeros((XDIM, YDIM), dtype=dtype)
f = self.tempfile('temp.png')
im = Image.fromarray(a, 'L') im = Image.fromarray(a, 'L')
im.save(f) im.save(f)
success()
def test_large(): def test_large(self):
""" succeeded prepatch""" """ succeeded prepatch"""
_write_png(xdim,ydim) self._write_png(XDIM, YDIM)
def test_2gpx():
def test_2gpx(self):
"""failed prepatch""" """failed prepatch"""
_write_png(xdim,xdim) self._write_png(XDIM, YDIM)
if __name__ == '__main__':
unittest.main()
# End of file

4
Tests/large_memory_test.py
View File

@ -14,7 +14,7 @@ YDIM = 32769
XDIM = 48000 XDIM = 48000
class TestImage(PillowTestCase): class LargeMemoryTest(PillowTestCase):
def _write_png(self, XDIM, YDIM): def _write_png(self, XDIM, YDIM):
f = self.tempfile('temp.png') f = self.tempfile('temp.png')
@ -27,7 +27,7 @@ class TestImage(PillowTestCase):
def test_2gpx(self): def test_2gpx(self):
"""failed prepatch""" """failed prepatch"""
self._write_png(XDIM, XDIM) self._write_png(XDIM, YDIM)
if __name__ == '__main__': if __name__ == '__main__':