mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-09-21 19:38:57 +03:00
33dabf986f
The unittest in helper.py has not offered an interesting abstraction
since dbe9f85c7d so import from the more
typical stdlib location.
46 lines
1.1 KiB
Python
46 lines
1.1 KiB
Python
import sys
|
|
import unittest
|
|
|
|
from PIL import Image
|
|
|
|
from .helper import 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).
|
|
|
|
|
|
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()
|