Pillow/Tests/test_map.py

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

47 lines
1.4 KiB
Python
Raw Normal View History

from __future__ import annotations
2023-06-27 07:43:58 +03:00
import sys
2020-02-12 19:29:19 +03:00
import pytest
2016-09-29 17:05:00 +03:00
from PIL import Image
2020-02-12 19:29:19 +03:00
def test_overflow():
# 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
2016-09-29 17:05:00 +03:00
2020-02-12 19:29:19 +03:00
# Note that this image triggers the decompression bomb warning:
max_pixels = Image.MAX_IMAGE_PIXELS
Image.MAX_IMAGE_PIXELS = None
2016-09-29 17:05:00 +03:00
2020-02-12 19:29:19 +03:00
# This image hits the offset test.
with Image.open("Tests/images/l2rgb_read.bmp") as im:
with pytest.raises((ValueError, MemoryError, OSError)):
2020-02-12 19:29:19 +03:00
im.load()
2020-02-12 19:29:19 +03:00
Image.MAX_IMAGE_PIXELS = max_pixels
2021-01-30 05:01:42 +03:00
def test_tobytes():
# Note that this image triggers the decompression bomb warning:
max_pixels = Image.MAX_IMAGE_PIXELS
Image.MAX_IMAGE_PIXELS = None
2021-01-30 05:01:42 +03:00
# Previously raised an access violation on Windows
with Image.open("Tests/images/l2rgb_read.bmp") as im:
with pytest.raises((ValueError, MemoryError, OSError)):
im.tobytes()
Image.MAX_IMAGE_PIXELS = max_pixels
2021-01-30 05:01:42 +03:00
2023-06-27 07:43:58 +03:00
@pytest.mark.skipif(sys.maxsize <= 2**32, reason="Requires 64-bit system")
2020-02-12 19:29:19 +03:00
def test_ysize():
numpy = pytest.importorskip("numpy", reason="NumPy not installed")
2020-02-12 19:29:19 +03:00
# Should not raise 'Integer overflow in ysize'
arr = numpy.zeros((46341, 46341), dtype=numpy.uint8)
Image.fromarray(arr)