Allow size argument to resize() to be a NumPy array

This commit is contained in:
Andrew Murray 2024-07-04 15:25:35 +10:00
parent 936012e861
commit 6990fc4a09
2 changed files with 12 additions and 3 deletions

View File

@ -198,6 +198,15 @@ def test_putdata() -> None:
assert len(im.getdata()) == len(arr) assert len(im.getdata()) == len(arr)
def test_resize() -> None:
im = hopper()
size = (64, 64)
im_resized = im.resize(numpy.array(size))
assert im_resized.size == size
@pytest.mark.parametrize( @pytest.mark.parametrize(
"dtype", "dtype",
( (

View File

@ -63,7 +63,6 @@ from . import (
) )
from ._binary import i32le, o32be, o32le from ._binary import i32le, o32be, o32le
from ._deprecate import deprecate from ._deprecate import deprecate
from ._typing import StrOrBytesPath, TypeGuard
from ._util import DeferredError, is_path from ._util import DeferredError, is_path
ElementTree: ModuleType | None ElementTree: ModuleType | None
@ -220,6 +219,7 @@ if hasattr(core, "DEFAULT_STRATEGY"):
if TYPE_CHECKING: if TYPE_CHECKING:
from . import ImageFile, ImagePalette from . import ImageFile, ImagePalette
from ._typing import NumpyArray, StrOrBytesPath, TypeGuard
ID: list[str] = [] ID: list[str] = []
OPEN: dict[ OPEN: dict[
str, str,
@ -2203,7 +2203,7 @@ class Image:
def resize( def resize(
self, self,
size: tuple[int, int] | list[int], size: tuple[int, int] | list[int] | NumpyArray,
resample: int | None = None, resample: int | None = None,
box: tuple[float, float, float, float] | None = None, box: tuple[float, float, float, float] | None = None,
reducing_gap: float | None = None, reducing_gap: float | None = None,
@ -2211,7 +2211,7 @@ class Image:
""" """
Returns a resized copy of this image. Returns a resized copy of this image.
:param size: The requested size in pixels, as a 2-tuple: :param size: The requested size in pixels, as a tuple or array:
(width, height). (width, height).
:param resample: An optional resampling filter. This can be :param resample: An optional resampling filter. This can be
one of :py:data:`Resampling.NEAREST`, :py:data:`Resampling.BOX`, one of :py:data:`Resampling.NEAREST`, :py:data:`Resampling.BOX`,