Pillow/Tests/check_large_memory.py

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

55 lines
1.3 KiB
Python
Raw Normal View History

from __future__ import annotations
2024-01-20 14:23:03 +03:00
2023-06-27 07:43:58 +03:00
import sys
from pathlib import PosixPath
2024-01-19 13:50:27 +03:00
from types import ModuleType
2023-06-27 07:43:58 +03:00
2020-03-28 04:51:28 +03:00
import pytest
from PIL import Image
# 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). It does succeed on a 3gb Ubuntu 12.04x64 VM on Python
2016-02-09 14:02:43 +03:00
# 2.7 and 3.2.
2024-01-19 13:50:27 +03:00
numpy: ModuleType | None
try:
import numpy
except ImportError:
numpy = None
YDIM = 32769
XDIM = 48000
2023-06-27 07:43:58 +03:00
pytestmark = pytest.mark.skipif(sys.maxsize <= 2**32, reason="requires 64-bit system")
2024-01-19 13:50:27 +03:00
def _write_png(tmp_path: PosixPath, xdim: int, ydim: int) -> None:
2020-03-28 04:51:28 +03:00
f = str(tmp_path / "temp.png")
im = Image.new("L", (xdim, ydim), 0)
im.save(f)
2024-01-19 13:50:27 +03:00
def test_large(tmp_path: PosixPath) -> None:
2021-08-12 14:50:09 +03:00
"""succeeded prepatch"""
2020-03-28 04:51:28 +03:00
_write_png(tmp_path, XDIM, YDIM)
2024-01-19 13:50:27 +03:00
def test_2gpx(tmp_path: PosixPath) -> None:
2020-03-28 04:51:28 +03:00
"""failed prepatch"""
_write_png(tmp_path, XDIM, XDIM)
2020-03-28 04:51:28 +03:00
@pytest.mark.skipif(numpy is None, reason="Numpy is not installed")
2024-01-19 13:50:27 +03:00
def test_size_greater_than_int() -> None:
assert numpy is not None
2020-03-28 04:51:28 +03:00
arr = numpy.ndarray(shape=(16394, 16394))
Image.fromarray(arr)