Pillow/Tests/test_map.py

38 lines
1.2 KiB
Python
Raw Normal View History

import sys
2016-09-29 17:05:00 +03:00
from PIL import Image
2019-09-25 12:46:54 +03:00
from .helper import PillowTestCase, is_win32, unittest
try:
import numpy
except ImportError:
numpy = None
2019-09-25 12:46:54 +03:00
@unittest.skipIf(is_win32(), "Win32 does not call map_buffer")
2016-09-29 17:05:00 +03:00
class TestMap(PillowTestCase):
def test_overflow(self):
# There is the potential to overflow comparisons in map.c
# if there are > SIZE_MAX bytes in the image or if
# the file encodes an offset that makes
# (offset + size(bytes)) > SIZE_MAX
# Note that this image triggers the decompression bomb warning:
max_pixels = Image.MAX_IMAGE_PIXELS
Image.MAX_IMAGE_PIXELS = None
# This image hits the offset test.
2019-06-13 18:54:46 +03:00
im = Image.open("Tests/images/l2rgb_read.bmp")
2016-10-03 17:33:29 +03:00
with self.assertRaises((ValueError, MemoryError, IOError)):
2016-09-29 17:05:00 +03:00
im.load()
Image.MAX_IMAGE_PIXELS = max_pixels
@unittest.skipIf(sys.maxsize <= 2 ** 32, "requires 64-bit system")
@unittest.skipIf(numpy is None, "Numpy is not installed")
def test_ysize(self):
# Should not raise 'Integer overflow in ysize'
arr = numpy.zeros((46341, 46341), dtype=numpy.uint8)
Image.fromarray(arr)