mirror of
https://github.com/python-pillow/Pillow.git
synced 2025-07-10 16:22:22 +03:00
Merge cb50421b57 into f2de251c76
This commit is contained in:
Yay295
GitHub
commit
9f3b4a8634
|
|
@ -30,7 +30,6 @@ from .helper import (
|
||||||
assert_image_similar_tofile,
|
assert_image_similar_tofile,
|
||||||
assert_not_all_same,
|
assert_not_all_same,
|
||||||
hopper,
|
hopper,
|
||||||
is_big_endian,
|
|
||||||
is_win32,
|
is_win32,
|
||||||
mark_if_feature_version,
|
mark_if_feature_version,
|
||||||
skip_unless_feature,
|
skip_unless_feature,
|
||||||
|
|
@ -1169,15 +1168,6 @@ class TestImageBytes:
|
||||||
reloaded.frombytes(source_bytes)
|
reloaded.frombytes(source_bytes)
|
||||||
assert reloaded.tobytes() == source_bytes
|
assert reloaded.tobytes() == source_bytes
|
||||||
|
|
||||||
@pytest.mark.parametrize("mode", Image.MODES + ["BGR;15", "BGR;16", "BGR;24"])
|
|
||||||
def test_getdata_putdata(self, mode: str) -> None:
|
|
||||||
if is_big_endian() and mode == "BGR;15":
|
|
||||||
pytest.xfail("Known failure of BGR;15 on big-endian")
|
|
||||||
im = hopper(mode)
|
|
||||||
reloaded = helper_image_new(mode, im.size)
|
|
||||||
reloaded.putdata(im.getdata())
|
|
||||||
assert_image_equal(im, reloaded)
|
|
||||||
|
|
||||||
|
|
||||||
class MockEncoder(ImageFile.PyEncoder):
|
class MockEncoder(ImageFile.PyEncoder):
|
||||||
pass
|
pass
|
||||||
|
|
|
||||||
153
Tests/test_image_data.py
Normal file
153
Tests/test_image_data.py
Normal file
|
|
@ -0,0 +1,153 @@
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import sys
|
||||||
|
from array import array
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
from PIL import Image
|
||||||
|
|
||||||
|
from .helper import assert_image_equal, hopper, is_big_endian
|
||||||
|
|
||||||
|
|
||||||
|
def test_getdata_sanity() -> None:
|
||||||
|
data = hopper().getdata()
|
||||||
|
|
||||||
|
len(data)
|
||||||
|
list(data)
|
||||||
|
|
||||||
|
assert data[0] == (20, 20, 70)
|
||||||
|
|
||||||
|
|
||||||
|
def test_putdata_sanity() -> None:
|
||||||
|
im1 = hopper()
|
||||||
|
|
||||||
|
data = list(im1.getdata())
|
||||||
|
|
||||||
|
im2 = Image.new(im1.mode, im1.size, 0)
|
||||||
|
im2.putdata(data)
|
||||||
|
|
||||||
|
assert_image_equal(im1, im2)
|
||||||
|
|
||||||
|
# readonly
|
||||||
|
im2 = Image.new(im1.mode, im2.size, 0)
|
||||||
|
im2.readonly = 1
|
||||||
|
im2.putdata(data)
|
||||||
|
|
||||||
|
assert not im2.readonly
|
||||||
|
assert_image_equal(im1, im2)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"mode, first_pixel, data_size",
|
||||||
|
(
|
||||||
|
("1", 0, 960),
|
||||||
|
("L", 17, 960),
|
||||||
|
("I", 17, 960),
|
||||||
|
("F", 17.0, 960),
|
||||||
|
("RGB", (11, 13, 52), 960),
|
||||||
|
("RGBA", (11, 13, 52, 255), 960),
|
||||||
|
("CMYK", (244, 242, 203, 0), 960),
|
||||||
|
("YCbCr", (16, 147, 123), 960),
|
||||||
|
),
|
||||||
|
)
|
||||||
|
def test_getdata_roundtrip(
|
||||||
|
mode: str, first_pixel: float | tuple[int, ...], data_size: int
|
||||||
|
) -> None:
|
||||||
|
im = hopper(mode).resize((32, 30), Image.Resampling.NEAREST)
|
||||||
|
data = im.getdata()
|
||||||
|
assert data[0] == first_pixel
|
||||||
|
assert len(data) == data_size
|
||||||
|
assert len(list(data)) == data_size
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"value, expected",
|
||||||
|
(
|
||||||
|
(0xFFFFFFFF, (255, 255, 255, 255)),
|
||||||
|
(-1, (255, 255, 255, 255)),
|
||||||
|
(sys.maxsize, (255, 255, 255, 255 if sys.maxsize > 2**32 else 127)),
|
||||||
|
),
|
||||||
|
)
|
||||||
|
def test_putdata_long_integers(value: int, expected: tuple[int, int, int, int]) -> None:
|
||||||
|
im = Image.new("RGBA", (1, 1))
|
||||||
|
im.putdata([value])
|
||||||
|
assert im.getpixel((0, 0)) == expected
|
||||||
|
|
||||||
|
|
||||||
|
def test_putdata_pypy_performance() -> None:
|
||||||
|
im = Image.new("L", (256, 256))
|
||||||
|
im.putdata(list(range(256)) * 256)
|
||||||
|
|
||||||
|
|
||||||
|
def test_putdata_mode_with_L_with_float() -> None:
|
||||||
|
im = Image.new("L", (1, 1), 0)
|
||||||
|
im.putdata([2.0])
|
||||||
|
assert im.getpixel((0, 0)) == 2
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"mode, scale, offset",
|
||||||
|
(
|
||||||
|
("I", 2, 256),
|
||||||
|
("I;16", 2, 256),
|
||||||
|
("I;16L", 2, 256),
|
||||||
|
("I;16B", 2, 256),
|
||||||
|
("F", 2.0, 256.0),
|
||||||
|
),
|
||||||
|
)
|
||||||
|
def test_putdata_scale_and_offset(mode: str, scale: float, offset: float) -> None:
|
||||||
|
src = hopper("L")
|
||||||
|
data = list(src.getdata())
|
||||||
|
im = Image.new(mode, src.size, 0)
|
||||||
|
im.putdata(data, scale, offset)
|
||||||
|
|
||||||
|
target = [scale * val + offset for val in data]
|
||||||
|
assert list(im.getdata()) == target
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"mode, type, value",
|
||||||
|
(
|
||||||
|
("L", "B", 0), # B = unsigned char
|
||||||
|
("F", "f", 0.0), # f = float
|
||||||
|
),
|
||||||
|
)
|
||||||
|
def test_putdata_array(mode: str, type: str, value: float) -> None:
|
||||||
|
# shouldn't segfault
|
||||||
|
# see https://github.com/python-pillow/Pillow/issues/1008
|
||||||
|
|
||||||
|
im = Image.new(mode, (150, 100))
|
||||||
|
arr = array(type, [value]) * 15000
|
||||||
|
im.putdata(arr)
|
||||||
|
|
||||||
|
assert len(im.getdata()) == len(arr)
|
||||||
|
|
||||||
|
|
||||||
|
def test_putdata_not_flattened() -> None:
|
||||||
|
im = Image.new("L", (1, 1))
|
||||||
|
with pytest.raises(TypeError):
|
||||||
|
im.putdata([[0]])
|
||||||
|
with pytest.raises(TypeError):
|
||||||
|
im.putdata([[0]], 2)
|
||||||
|
|
||||||
|
with pytest.raises(TypeError):
|
||||||
|
im = Image.new("I", (1, 1))
|
||||||
|
im.putdata([[0]])
|
||||||
|
with pytest.raises(TypeError):
|
||||||
|
im = Image.new("F", (1, 1))
|
||||||
|
im.putdata([[0]])
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize("mode", Image.MODES + ["BGR;15", "BGR;16", "BGR;24"])
|
||||||
|
def test_getdata_putdata(mode: str) -> None:
|
||||||
|
if is_big_endian() and mode == "BGR;15":
|
||||||
|
pytest.xfail("Known failure of BGR;15 on big-endian")
|
||||||
|
im = hopper(mode)
|
||||||
|
if mode.startswith("BGR;"):
|
||||||
|
with pytest.warns(DeprecationWarning, match=mode):
|
||||||
|
reloaded = Image.new(mode, im.size)
|
||||||
|
else:
|
||||||
|
reloaded = Image.new(mode, im.size)
|
||||||
|
reloaded.putdata(im.getdata())
|
||||||
|
assert_image_equal(im, reloaded)
|
||||||
|
|
@ -1,30 +0,0 @@
|
||||||
from __future__ import annotations
|
|
||||||
|
|
||||||
from PIL import Image
|
|
||||||
|
|
||||||
from .helper import hopper
|
|
||||||
|
|
||||||
|
|
||||||
def test_sanity() -> None:
|
|
||||||
data = hopper().getdata()
|
|
||||||
|
|
||||||
len(data)
|
|
||||||
list(data)
|
|
||||||
|
|
||||||
assert data[0] == (20, 20, 70)
|
|
||||||
|
|
||||||
|
|
||||||
def test_mode() -> None:
|
|
||||||
def getdata(mode: str) -> tuple[float | tuple[int, ...], int, int]:
|
|
||||||
im = hopper(mode).resize((32, 30), Image.Resampling.NEAREST)
|
|
||||||
data = im.getdata()
|
|
||||||
return data[0], len(data), len(list(data))
|
|
||||||
|
|
||||||
assert getdata("1") == (0, 960, 960)
|
|
||||||
assert getdata("L") == (17, 960, 960)
|
|
||||||
assert getdata("I") == (17, 960, 960)
|
|
||||||
assert getdata("F") == (17.0, 960, 960)
|
|
||||||
assert getdata("RGB") == ((11, 13, 52), 960, 960)
|
|
||||||
assert getdata("RGBA") == ((11, 13, 52, 255), 960, 960)
|
|
||||||
assert getdata("CMYK") == ((244, 242, 203, 0), 960, 960)
|
|
||||||
assert getdata("YCbCr") == ((16, 147, 123), 960, 960)
|
|
||||||
|
|
@ -1,125 +0,0 @@
|
||||||
from __future__ import annotations
|
|
||||||
|
|
||||||
import sys
|
|
||||||
from array import array
|
|
||||||
|
|
||||||
import pytest
|
|
||||||
|
|
||||||
from PIL import Image
|
|
||||||
|
|
||||||
from .helper import assert_image_equal, hopper
|
|
||||||
|
|
||||||
|
|
||||||
def test_sanity() -> None:
|
|
||||||
im1 = hopper()
|
|
||||||
|
|
||||||
data = list(im1.getdata())
|
|
||||||
|
|
||||||
im2 = Image.new(im1.mode, im1.size, 0)
|
|
||||||
im2.putdata(data)
|
|
||||||
|
|
||||||
assert_image_equal(im1, im2)
|
|
||||||
|
|
||||||
# readonly
|
|
||||||
im2 = Image.new(im1.mode, im2.size, 0)
|
|
||||||
im2.readonly = 1
|
|
||||||
im2.putdata(data)
|
|
||||||
|
|
||||||
assert not im2.readonly
|
|
||||||
assert_image_equal(im1, im2)
|
|
||||||
|
|
||||||
|
|
||||||
def test_long_integers() -> None:
|
|
||||||
# see bug-200802-systemerror
|
|
||||||
def put(value: int) -> float | tuple[int, ...] | None:
|
|
||||||
im = Image.new("RGBA", (1, 1))
|
|
||||||
im.putdata([value])
|
|
||||||
return im.getpixel((0, 0))
|
|
||||||
|
|
||||||
assert put(0xFFFFFFFF) == (255, 255, 255, 255)
|
|
||||||
assert put(0xFFFFFFFF) == (255, 255, 255, 255)
|
|
||||||
assert put(-1) == (255, 255, 255, 255)
|
|
||||||
assert put(-1) == (255, 255, 255, 255)
|
|
||||||
if sys.maxsize > 2**32:
|
|
||||||
assert put(sys.maxsize) == (255, 255, 255, 255)
|
|
||||||
else:
|
|
||||||
assert put(sys.maxsize) == (255, 255, 255, 127)
|
|
||||||
|
|
||||||
|
|
||||||
def test_pypy_performance() -> None:
|
|
||||||
im = Image.new("L", (256, 256))
|
|
||||||
im.putdata(list(range(256)) * 256)
|
|
||||||
|
|
||||||
|
|
||||||
def test_mode_with_L_with_float() -> None:
|
|
||||||
im = Image.new("L", (1, 1), 0)
|
|
||||||
im.putdata([2.0])
|
|
||||||
assert im.getpixel((0, 0)) == 2
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize("mode", ("I", "I;16", "I;16L", "I;16B"))
|
|
||||||
def test_mode_i(mode: str) -> None:
|
|
||||||
src = hopper("L")
|
|
||||||
data = list(src.getdata())
|
|
||||||
im = Image.new(mode, src.size, 0)
|
|
||||||
im.putdata(data, 2, 256)
|
|
||||||
|
|
||||||
target = [2 * elt + 256 for elt in data]
|
|
||||||
assert list(im.getdata()) == target
|
|
||||||
|
|
||||||
|
|
||||||
def test_mode_F() -> None:
|
|
||||||
src = hopper("L")
|
|
||||||
data = list(src.getdata())
|
|
||||||
im = Image.new("F", src.size, 0)
|
|
||||||
im.putdata(data, 2.0, 256.0)
|
|
||||||
|
|
||||||
target = [2.0 * float(elt) + 256.0 for elt in data]
|
|
||||||
assert list(im.getdata()) == target
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize("mode", ("BGR;15", "BGR;16", "BGR;24"))
|
|
||||||
def test_mode_BGR(mode: str) -> None:
|
|
||||||
data = [(16, 32, 49), (32, 32, 98)]
|
|
||||||
with pytest.warns(DeprecationWarning, match=mode):
|
|
||||||
im = Image.new(mode, (1, 2))
|
|
||||||
im.putdata(data)
|
|
||||||
|
|
||||||
assert list(im.getdata()) == data
|
|
||||||
|
|
||||||
|
|
||||||
def test_array_B() -> None:
|
|
||||||
# shouldn't segfault
|
|
||||||
# see https://github.com/python-pillow/Pillow/issues/1008
|
|
||||||
|
|
||||||
arr = array("B", [0]) * 15000
|
|
||||||
im = Image.new("L", (150, 100))
|
|
||||||
im.putdata(arr)
|
|
||||||
|
|
||||||
assert len(im.getdata()) == len(arr)
|
|
||||||
|
|
||||||
|
|
||||||
def test_array_F() -> None:
|
|
||||||
# shouldn't segfault
|
|
||||||
# see https://github.com/python-pillow/Pillow/issues/1008
|
|
||||||
|
|
||||||
im = Image.new("F", (150, 100))
|
|
||||||
arr = array("f", [0.0]) * 15000
|
|
||||||
im.putdata(arr)
|
|
||||||
|
|
||||||
assert len(im.getdata()) == len(arr)
|
|
||||||
|
|
||||||
|
|
||||||
def test_not_flattened() -> None:
|
|
||||||
im = Image.new("L", (1, 1))
|
|
||||||
with pytest.raises(TypeError):
|
|
||||||
im.putdata([[0]])
|
|
||||||
with pytest.raises(TypeError):
|
|
||||||
im.putdata([[0]], 2)
|
|
||||||
|
|
||||||
with pytest.raises(TypeError):
|
|
||||||
im = Image.new("I", (1, 1))
|
|
||||||
im.putdata([[0]])
|
|
||||||
with pytest.raises(TypeError):
|
|
||||||
im = Image.new("F", (1, 1))
|
|
||||||
im.putdata([[0]])
|
|
||||||
Loading…
Reference in New Issue
Block a user