mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-12-29 19:36:19 +03:00
44 lines
1.1 KiB
Python
44 lines
1.1 KiB
Python
import sys
|
|
|
|
from helper import unittest, PillowTestCase
|
|
|
|
# This test is not run automatically.
|
|
#
|
|
# It requires > 2gb memory for the >2 gigapixel image generated in the
|
|
# second test. Running this automatically would amount to a denial of
|
|
# service on our testing infrastructure. I expect this test to fail
|
|
# on any 32-bit machine, as well as any smallish things (like
|
|
# Raspberry Pis).
|
|
|
|
from PIL import Image
|
|
try:
|
|
import numpy as np
|
|
except ImportError:
|
|
raise unittest.SkipTest("numpy not installed")
|
|
|
|
YDIM = 32769
|
|
XDIM = 48000
|
|
|
|
|
|
@unittest.skipIf(sys.maxsize <= 2**32, "requires 64-bit system")
|
|
class LargeMemoryNumpyTest(PillowTestCase):
|
|
|
|
def _write_png(self, xdim, ydim):
|
|
dtype = np.uint8
|
|
a = np.zeros((xdim, ydim), dtype=dtype)
|
|
f = self.tempfile('temp.png')
|
|
im = Image.fromarray(a, 'L')
|
|
im.save(f)
|
|
|
|
def test_large(self):
|
|
""" succeeded prepatch"""
|
|
self._write_png(XDIM, YDIM)
|
|
|
|
def test_2gpx(self):
|
|
"""failed prepatch"""
|
|
self._write_png(XDIM, XDIM)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|