python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-02-04 13:40:54 +03:00
Code Issues Packages Projects Releases Wiki Activity
89bafd3098
Pillow/Tests/test_lib_pack.py

864 lines
31 KiB
Python
Raw Blame History

from __future__ import annotations
from typing import Union
import pytest
from PIL import Image
from .helper import is_big_endian
X = 255
UnPackParamTypes = list[
tuple[str, str, Union[bytes, int], list[Union[float, tuple[int, ...]]]]
]
def get_pack_parameters() -> UnPackParamTypes:
params: UnPackParamTypes = []
# Mode 1
params += [
("1", "1", b"\x01", [0, 0, 0, 0, 0, 0, 0, X]),
("1", "1;I", b"\x01", [X, X, X, X, X, X, X, 0]),
("1", "1;R", b"\x01", [X, 0, 0, 0, 0, 0, 0, 0]),
("1", "1;IR", b"\x01", [0, X, X, X, X, X, X, X]),
("1", "1", b"\xaa", [X, 0, X, 0, X, 0, X, 0]),
("1", "1;I", b"\xaa", [0, X, 0, X, 0, X, 0, X]),
("1", "1;R", b"\xaa", [0, X, 0, X, 0, X, 0, X]),
("1", "1;IR", b"\xaa", [X, 0, X, 0, X, 0, X, 0]),
("1", "L", b"\xff\x00\x00\xff\x00\x00", [X, 0, 0, X, 0, 0]),
]
# Mode L
params += [
("L", "L", 1, [1, 2, 3, 4]),
("L", "L;16", b"\x00\xc6\x00\xaf", [198, 175]),
("L", "L;16B", b"\xc6\x00\xaf\x00", [198, 175]),
]
# Mode LA
params += [
("LA", "LA", 2, [(1, 2), (3, 4), (5, 6)]),
("LA", "LA;L", 2, [(1, 4), (2, 5), (3, 6)]),
]
# Mode La
params += [("La", "La", 2, [(1, 2), (3, 4), (5, 6)])]
# Mode P
params += [
("P", "P;1", b"\xe4", [1, 1, 1, 0, 0, 255, 0, 0]),
("P", "P;2", b"\xe4", [3, 2, 1, 0]),
("P", "P;4", b"\x02\xef", [0, 2, 14, 15]),
("P", "P", 1, [1, 2, 3, 4]),
]
# Mode PA
params += [
("PA", "PA", 2, [(1, 2), (3, 4), (5, 6)]),
("PA", "PA;L", 2, [(1, 4), (2, 5), (3, 6)]),
]
# Mode RGB
params += [
("RGB", "XRGB;1555", b"\x00\x86\x01\x8c", [(8, 131, 0), (24, 0, 8)]),
("RGB", "RGB;565", b"\x00\n\x00\x1c", [(8, 64, 0), (24, 129, 0)]),
("RGB", "RGB", 3, [(1, 2, 3), (4, 5, 6), (7, 8, 9)]),
("RGB", "RGBX", b"\x01\x02\x03\xff\x05\x06\x07\xff", [(1, 2, 3), (5, 6, 7)]),
("RGB", "XRGB", b"\x00\x02\x03\x04\x00\x06\x07\x08", [(2, 3, 4), (6, 7, 8)]),
("RGB", "XBGR;1555", b"\x00\x86\x01\x8c", [(0, 131, 8), (8, 0, 24)]),
("RGB", "BGR;565", b"\x00\n\x00\x1c", [(0, 64, 8), (0, 129, 24)]),
("RGB", "BGR", 3, [(3, 2, 1), (6, 5, 4), (9, 8, 7)]),
("RGB", "BGRX", b"\x01\x02\x03\x00\x05\x06\x07\x00", [(3, 2, 1), (7, 6, 5)]),
("RGB", "XBGR", b"\x00\x02\x03\x04\x00\x06\x07\x08", [(4, 3, 2), (8, 7, 6)]),
("RGB", "RGB;L", 3, [(1, 4, 7), (2, 5, 8), (3, 6, 9)]),
("RGB", "R", 1, [(1, 9, 9), (2, 9, 9), (3, 9, 9)]),
("RGB", "G", 1, [(9, 1, 9), (9, 2, 9), (9, 3, 9)]),
("RGB", "B", 1, [(9, 9, 1), (9, 9, 2), (9, 9, 3)]),
]
# Mode RGBA
params += [
("RGBA", "RGBA", 4, [(1, 2, 3, 4), (5, 6, 7, 8), (9, 10, 11, 12)]),
("RGBA", "RGBA;L", 4, [(1, 4, 7, 10), (2, 5, 8, 11), (3, 6, 9, 12)]),
("RGBA", "RGB", 3, [(1, 2, 3, 14), (4, 5, 6, 15), (7, 8, 9, 16)]),
("RGBA", "BGR", 3, [(3, 2, 1, 14), (6, 5, 4, 15), (9, 8, 7, 16)]),
("RGBA", "BGRA", 4, [(3, 2, 1, 4), (7, 6, 5, 8), (11, 10, 9, 12)]),
("RGBA", "ABGR", 4, [(4, 3, 2, 1), (8, 7, 6, 5), (12, 11, 10, 9)]),
(
"RGBA",
"BGRa",
4,
[(191, 127, 63, 4), (223, 191, 159, 8), (233, 212, 191, 12)],
),
("RGBA", "R", 1, [(1, 0, 8, 9), (2, 0, 8, 9), (3, 0, 8, 0)]),
("RGBA", "G", 1, [(6, 1, 8, 9), (6, 2, 8, 9), (6, 3, 8, 9)]),
("RGBA", "B", 1, [(6, 7, 1, 9), (6, 7, 2, 0), (6, 7, 3, 9)]),
("RGBA", "A", 1, [(6, 7, 0, 1), (6, 7, 0, 2), (0, 7, 0, 3)]),
]
# Mode RGBa
params += [
("RGBa", "RGBa", 4, [(1, 2, 3, 4), (5, 6, 7, 8), (9, 10, 11, 12)]),
("RGBa", "BGRa", 4, [(3, 2, 1, 4), (7, 6, 5, 8), (11, 10, 9, 12)]),
("RGBa", "aBGR", 4, [(4, 3, 2, 1), (8, 7, 6, 5), (12, 11, 10, 9)]),
]
# Mode RGBX
params += [
("RGBX", "RGBX", 4, [(1, 2, 3, 4), (5, 6, 7, 8), (9, 10, 11, 12)]),
("RGBX", "RGBX;L", 4, [(1, 4, 7, 10), (2, 5, 8, 11), (3, 6, 9, 12)]),
("RGBX", "RGB", 3, [(1, 2, 3, X), (4, 5, 6, X), (7, 8, 9, X)]),
("RGBX", "BGR", 3, [(3, 2, 1, X), (6, 5, 4, X), (9, 8, 7, X)]),
(
"RGBX",
"BGRX",
b"\x01\x02\x03\x00\x05\x06\x07\x00\t\n\x0b\x00",
[(3, 2, 1, X), (7, 6, 5, X), (11, 10, 9, X)],
),
(
"RGBX",
"XBGR",
b"\x00\x02\x03\x04\x00\x06\x07\x08\x00\n\x0b\x0c",
[(4, 3, 2, X), (8, 7, 6, X), (12, 11, 10, X)],
),
("RGBX", "R", 1, [(1, 0, 8, 9), (2, 0, 8, 9), (3, 0, 8, 0)]),
("RGBX", "G", 1, [(6, 1, 8, 9), (6, 2, 8, 9), (6, 3, 8, 9)]),
("RGBX", "B", 1, [(6, 7, 1, 9), (6, 7, 2, 0), (6, 7, 3, 9)]),
("RGBX", "X", 1, [(6, 7, 0, 1), (6, 7, 0, 2), (0, 7, 0, 3)]),
]
# Mode CMYK
params += [
("CMYK", "CMYK", 4, [(1, 2, 3, 4), (5, 6, 7, 8), (9, 10, 11, 12)]),
(
"CMYK",
"CMYK;I",
4,
[(254, 253, 252, 251), (250, 249, 248, 247), (246, 245, 244, 243)],
),
("CMYK", "CMYK;L", 4, [(1, 4, 7, 10), (2, 5, 8, 11), (3, 6, 9, 12)]),
("CMYK", "K", 1, [(6, 7, 0, 1), (6, 7, 0, 2), (0, 7, 0, 3)]),
]
# Mode YCbCr
params += [
("YCbCr", "YCbCr", 3, [(1, 2, 3), (4, 5, 6), (7, 8, 9)]),
("YCbCr", "YCbCr;L", 3, [(1, 4, 7), (2, 5, 8), (3, 6, 9)]),
(
"YCbCr",
"YCbCrX",
b"\x01\x02\x03\xff\x05\x06\x07\xff\t\n\x0b\xff",
[(1, 2, 3), (5, 6, 7), (9, 10, 11)],
),
(
"YCbCr",
"YCbCrK",
b"\x01\x02\x03\xff\x05\x06\x07\xff\t\n\x0b\xff",
[(1, 2, 3), (5, 6, 7), (9, 10, 11)],
),
("YCbCr", "Y", 1, [(1, 0, 8, 9), (2, 0, 8, 9), (3, 0, 8, 0)]),
("YCbCr", "Cb", 1, [(6, 1, 8, 9), (6, 2, 8, 9), (6, 3, 8, 9)]),
("YCbCr", "Cr", 1, [(6, 7, 1, 9), (6, 7, 2, 0), (6, 7, 3, 9)]),
]
# Mode LAB
params += [
("LAB", "LAB", 3, [(1, 130, 131), (4, 133, 134), (7, 136, 137)]),
("LAB", "L", 1, [(1, 9, 9), (2, 9, 9), (3, 9, 9)]),
("LAB", "A", 1, [(9, 1, 9), (9, 2, 9), (9, 3, 9)]),
("LAB", "B", 1, [(9, 9, 1), (9, 9, 2), (9, 9, 3)]),
]
# Mode HSV
params += [
("HSV", "HSV", 3, [(1, 2, 3), (4, 5, 6), (7, 8, 9)]),
("HSV", "H", 1, [(1, 9, 9), (2, 9, 9), (3, 9, 9)]),
("HSV", "S", 1, [(9, 1, 9), (9, 2, 9), (9, 3, 9)]),
("HSV", "V", 1, [(9, 9, 1), (9, 9, 2), (9, 9, 3)]),
]
# Mode I
params += [
("I", "I;16B", 2, [0x0102, 0x0304]),
("I", "I;32S", b"\x83\x00\x00\x01\x01\x00\x00\x83", [0x01000083, -2097151999]),
]
if is_big_endian():
params += [
("I", "I", 4, [0x01020304, 0x05060708]),
(
"I",
"I;32NS",
b"\x83\x00\x00\x01\x01\x00\x00\x83",
[-2097151999, 0x01000083],
),
]
else:
params += [
("I", "I", 4, [0x04030201, 0x08070605]),
(
"I",
"I;32NS",
b"\x83\x00\x00\x01\x01\x00\x00\x83",
[0x01000083, -2097151999],
),
]
# Mode I;16
if is_big_endian():
params += [("I;16N", "I;16N", 2, [0x0102, 0x0304, 0x0506])]
else:
params += [("I;16N", "I;16N", 2, [0x0201, 0x0403, 0x0605])]
# Mode F float
params += [
("F", "F;32F", 4, [1.539989614439558e-36, 4.063216068939723e-34]),
]
if is_big_endian():
params += [
("F", "F", 4, [2.387939260590663e-38, 6.301941157072183e-36]),
("F", "F;32NF", 4, [2.387939260590663e-38, 6.301941157072183e-36]),
]
else:
params += [
("F", "F", 4, [1.539989614439558e-36, 4.063216068939723e-34]),
("F", "F;32NF", 4, [1.539989614439558e-36, 4.063216068939723e-34]),
]
return params
@pytest.mark.parametrize("mode, rawmode, data, pixels", get_pack_parameters())
def test_pack(
mode: str, rawmode: str, data: bytes | int, pixels: list[float | tuple[int, ...]]
) -> None:
"""
Test packing from {mode} to {rawmode}.
data - Either the number of bytes in a pixel of the rawmode, or a byte string.
If a number is given, bytes will be generated starting at 1.
The byte string will be compared against the pack result.
pixels - The pixels to populate the starting image with.
"""
im = Image.new(mode, (len(pixels), 1))
for x, pixel in enumerate(pixels):
im.putpixel((x, 0), pixel)
if isinstance(data, int):
data_len = data * len(pixels)
data = bytes(range(1, data_len + 1))
assert data == im.tobytes("raw", rawmode)
def get_unpack_parameters() -> UnPackParamTypes:
params: UnPackParamTypes = []
# Mode 1
params += [
("1", "1", b"\x01", [0, 0, 0, 0, 0, 0, 0, X]),
("1", "1;I", b"\x01", [X, X, X, X, X, X, X, 0]),
("1", "1;R", b"\x01", [X, 0, 0, 0, 0, 0, 0, 0]),
("1", "1;IR", b"\x01", [0, X, X, X, X, X, X, X]),
("1", "1", b"\xaa", [X, 0, X, 0, X, 0, X, 0]),
("1", "1;I", b"\xaa", [0, X, 0, X, 0, X, 0, X]),
("1", "1;R", b"\xaa", [0, X, 0, X, 0, X, 0, X]),
("1", "1;IR", b"\xaa", [X, 0, X, 0, X, 0, X, 0]),
("1", "1;8", b"\x00\x01\x02\xff", [0, X, X, X]),
]
# Mode L
params += [
("L", "L;2", b"\xe4", [255, 170, 85, 0]),
("L", "L;2I", b"\xe4", [0, 85, 170, 255]),
("L", "L;2R", b"\xe4", [0, 170, 85, 255]),
("L", "L;2IR", b"\xe4", [255, 85, 170, 0]),
("L", "L;4", b"\x02\xef", [0, 34, 238, 255]),
("L", "L;4I", b"\x02\xef", [255, 221, 17, 0]),
("L", "L;4R", b"\x02\xef", [68, 0, 255, 119]),
("L", "L;4IR", b"\x02\xef", [187, 255, 0, 136]),
("L", "L", 1, [1, 2, 3, 4]),
("L", "L;I", 1, [254, 253, 252, 251]),
("L", "L;R", 1, [128, 64, 192, 32]),
("L", "L;16", 2, [2, 4, 6, 8]),
("L", "L;16B", 2, [1, 3, 5, 7]),
("L", "L;16", b"\x00\xc6\x00\xaf", [198, 175]),
("L", "L;16B", b"\xc6\x00\xaf\x00", [198, 175]),
]
# Mode LA
params += [
("LA", "LA", 2, [(1, 2), (3, 4), (5, 6)]),
("LA", "LA;L", 2, [(1, 4), (2, 5), (3, 6)]),
]
# Mode La
params += [
("La", "La", 2, [(1, 2), (3, 4), (5, 6)]),
]
# Mode P
params += [
("P", "P;1", b"\xe4", [1, 1, 1, 0, 0, 1, 0, 0]),
("P", "P;2", b"\xe4", [3, 2, 1, 0]),
# erroneous?
# ("P", "P;2L", b"\xe4", [1, 1, 1, 0]),
("P", "P;4", b"\x02\xef", [0, 2, 14, 15]),
# erroneous?
# ("P", "P;4L", b"\x02\xef", [2, 10, 10, 0]),
("P", "P", 1, [1, 2, 3, 4]),
("P", "P;R", 1, [128, 64, 192, 32]),
]
# Mode PA
params += [
("PA", "PA", 2, [(1, 2), (3, 4), (5, 6)]),
("PA", "PA;L", 2, [(1, 4), (2, 5), (3, 6)]),
]
# Mode RGB
params += [
("RGB", "RGB", 3, [(1, 2, 3), (4, 5, 6), (7, 8, 9)]),
("RGB", "RGB;L", 3, [(1, 4, 7), (2, 5, 8), (3, 6, 9)]),
("RGB", "RGB;R", 3, [(128, 64, 192), (32, 160, 96)]),
("RGB", "RGB;16L", 6, [(2, 4, 6), (8, 10, 12)]),
("RGB", "RGB;16B", 6, [(1, 3, 5), (7, 9, 11)]),
("RGB", "BGR", 3, [(3, 2, 1), (6, 5, 4), (9, 8, 7)]),
("RGB", "XRGB;1555", 2, [(0, 132, 8), (8, 0, 24)]),
("RGB", "ARGB;1555", 2, [(0, 132, 8), (8, 0, 24)]),
("RGB", "ARGB;1555Z", 2, [(0, 132, 8), (8, 0, 24)]),
("RGB", "XBGR;1555", 2, [(8, 132, 0), (24, 0, 8)]),
("RGB", "ABGR;1555", 2, [(8, 132, 0), (24, 0, 8)]),
("RGB", "RGB;565", 2, [(0, 65, 8), (0, 130, 24)]),
("RGB", "BGR;565", 2, [(8, 65, 0), (24, 130, 0)]),
("RGB", "XBGR;4", 2, [(17, 0, 34), (51, 0, 68)]),
("RGB", "ABGR;4", 2, [(17, 0, 34), (51, 0, 68)]),
("RGB", "RGBX", 4, [(1, 2, 3), (5, 6, 7), (9, 10, 11)]),
("RGB", "RGBX;L", 4, [(1, 4, 7), (2, 5, 8), (3, 6, 9)]),
("RGB", "BGRX", 4, [(3, 2, 1), (7, 6, 5), (11, 10, 9)]),
("RGB", "XRGB", 4, [(2, 3, 4), (6, 7, 8), (10, 11, 12)]),
("RGB", "XBGR", 4, [(4, 3, 2), (8, 7, 6), (12, 11, 10)]),
(
"RGB",
"YCC;P",
b"D]\x9c\x82\x1a\x91\xfaOC\xe7J\x12", # random data
[(127, 102, 0), (192, 227, 0), (213, 255, 170), (98, 255, 133)],
),
("RGB", "R", 1, [(1, 0, 0), (2, 0, 0), (3, 0, 0)]),
("RGB", "G", 1, [(0, 1, 0), (0, 2, 0), (0, 3, 0)]),
("RGB", "B", 1, [(0, 0, 1), (0, 0, 2), (0, 0, 3)]),
("RGB", "R;16B", 2, [(1, 0, 0), (3, 0, 0), (5, 0, 0)]),
("RGB", "G;16B", 2, [(0, 1, 0), (0, 3, 0), (0, 5, 0)]),
("RGB", "B;16B", 2, [(0, 0, 1), (0, 0, 3), (0, 0, 5)]),
("RGB", "R;16L", 2, [(2, 0, 0), (4, 0, 0), (6, 0, 0)]),
("RGB", "G;16L", 2, [(0, 2, 0), (0, 4, 0), (0, 6, 0)]),
("RGB", "B;16L", 2, [(0, 0, 2), (0, 0, 4), (0, 0, 6)]),
]
if is_big_endian():
params += [
("RGB", "R;16N", 2, [(1, 0, 0), (3, 0, 0), (5, 0, 0)]),
("RGB", "G;16N", 2, [(0, 1, 0), (0, 3, 0), (0, 5, 0)]),
("RGB", "B;16N", 2, [(0, 0, 1), (0, 0, 3), (0, 0, 5)]),
]
else:
params += [
("RGB", "R;16N", 2, [(2, 0, 0), (4, 0, 0), (6, 0, 0)]),
("RGB", "G;16N", 2, [(0, 2, 0), (0, 4, 0), (0, 6, 0)]),
("RGB", "B;16N", 2, [(0, 0, 2), (0, 0, 4), (0, 0, 6)]),
]
params += [
("RGB", "CMYK", 4, [(250, 249, 248), (242, 241, 240), (234, 233, 233)]),
]
# Mode RGBA
params += [
("RGBA", "LA", 2, [(1, 1, 1, 2), (3, 3, 3, 4), (5, 5, 5, 6)]),
("RGBA", "LA;16B", 4, [(1, 1, 1, 3), (5, 5, 5, 7), (9, 9, 9, 11)]),
("RGBA", "RGBA", 4, [(1, 2, 3, 4), (5, 6, 7, 8), (9, 10, 11, 12)]),
("RGBA", "RGBAX", 5, [(1, 2, 3, 4), (6, 7, 8, 9), (11, 12, 13, 14)]),
("RGBA", "RGBAXX", 6, [(1, 2, 3, 4), (7, 8, 9, 10), (13, 14, 15, 16)]),
(
"RGBA",
"RGBa",
4,
[
(63, 127, 191, 4),
(159, 191, 223, 8),
(191, 212, 233, 12),
],
),
(
"RGBA",
"RGBa",
b"\x01\x02\x03\x00\x10\x20\x30\x7f\x10\x20\x30\xff",
[
(0, 0, 0, 0),
(32, 64, 96, 127),
(16, 32, 48, 255),
],
),
(
"RGBA",
"RGBaX",
b"\x01\x02\x03\x00-\x10\x20\x30\x7f-\x10\x20\x30\xff-",
[
(0, 0, 0, 0),
(32, 64, 96, 127),
(16, 32, 48, 255),
],
),
(
"RGBA",
"RGBaXX",
b"\x01\x02\x03\x00==\x10\x20\x30\x7f!!\x10\x20\x30\xff??",
[
(0, 0, 0, 0),
(32, 64, 96, 127),
(16, 32, 48, 255),
],
),
(
"RGBA",
"RGBa;16L",
8,
[
(63, 127, 191, 8),
(159, 191, 223, 16),
(191, 212, 233, 24),
],
),
(
"RGBA",
"RGBa;16L",
b"\x88\x01\x88\x02\x88\x03\x88\x00\x88\x10\x88\x20\x88\x30\x88\xff",
[
(0, 0, 0, 0),
(16, 32, 48, 255),
],
),
(
"RGBA",
"RGBa;16B",
8,
[
(36, 109, 182, 7),
(153, 187, 221, 15),
(188, 210, 232, 23),
],
),
(
"RGBA",
"RGBa;16B",
b"\x01\x88\x02\x88\x03\x88\x00\x88\x10\x88\x20\x88\x30\x88\xff\x88",
[
(0, 0, 0, 0),
(16, 32, 48, 255),
],
),
(
"RGBA",
"BGRa",
4,
[
(191, 127, 63, 4),
(223, 191, 159, 8),
(233, 212, 191, 12),
],
),
(
"RGBA",
"BGRa",
b"\x01\x02\x03\x00\x10\x20\x30\xff",
[
(0, 0, 0, 0),
(48, 32, 16, 255),
],
),
(
"RGBA",
"RGBA;I",
4,
[
(254, 253, 252, 4),
(250, 249, 248, 8),
(246, 245, 244, 12),
],
),
("RGBA", "RGBA;L", 4, [(1, 4, 7, 10), (2, 5, 8, 11), (3, 6, 9, 12)]),
("RGBA", "ARGB;1555", 2, [(0, 132, 8, 0), (8, 0, 24, 0)]),
("RGBA", "ARGB;1555Z", 2, [(0, 132, 8, X), (8, 0, 24, X)]),
("RGBA", "ABGR;1555", 2, [(8, 132, 0, 0), (24, 0, 8, 0)]),
("RGBA", "ABGR;4", 2, [(17, 0, 34, 0), (51, 0, 68, 0)]),
("RGBA", "RGBA;16L", 8, [(2, 4, 6, 8), (10, 12, 14, 16)]),
("RGBA", "RGBA;16B", 8, [(1, 3, 5, 7), (9, 11, 13, 15)]),
("RGBA", "BGRA;16L", 8, [(6, 4, 2, 8), (14, 12, 10, 16)]),
("RGBA", "BGRA;16B", 8, [(5, 3, 1, 7), (13, 11, 9, 15)]),
("RGBA", "BGRA", 4, [(3, 2, 1, 4), (7, 6, 5, 8), (11, 10, 9, 12)]),
("RGBA", "ARGB", 4, [(2, 3, 4, 1), (6, 7, 8, 5), (10, 11, 12, 9)]),
("RGBA", "ABGR", 4, [(4, 3, 2, 1), (8, 7, 6, 5), (12, 11, 10, 9)]),
(
"RGBA",
"YCCA;P",
b"]bE\x04\xdd\xbej\xed57T\xce\xac\xce:\x11", # random data
[
(0, 161, 0, 4),
(255, 255, 255, 237),
(27, 158, 0, 206),
(0, 118, 0, 17),
],
),
("RGBA", "R", 1, [(1, 0, 0, 0), (2, 0, 0, 0), (3, 0, 0, 0)]),
("RGBA", "G", 1, [(0, 1, 0, 0), (0, 2, 0, 0), (0, 3, 0, 0)]),
("RGBA", "B", 1, [(0, 0, 1, 0), (0, 0, 2, 0), (0, 0, 3, 0)]),
("RGBA", "A", 1, [(0, 0, 0, 1), (0, 0, 0, 2), (0, 0, 0, 3)]),
("RGBA", "R;16B", 2, [(1, 0, 0, 0), (3, 0, 0, 0), (5, 0, 0, 0)]),
("RGBA", "G;16B", 2, [(0, 1, 0, 0), (0, 3, 0, 0), (0, 5, 0, 0)]),
("RGBA", "B;16B", 2, [(0, 0, 1, 0), (0, 0, 3, 0), (0, 0, 5, 0)]),
("RGBA", "A;16B", 2, [(0, 0, 0, 1), (0, 0, 0, 3), (0, 0, 0, 5)]),
("RGBA", "R;16L", 2, [(2, 0, 0, 0), (4, 0, 0, 0), (6, 0, 0, 0)]),
("RGBA", "G;16L", 2, [(0, 2, 0, 0), (0, 4, 0, 0), (0, 6, 0, 0)]),
("RGBA", "B;16L", 2, [(0, 0, 2, 0), (0, 0, 4, 0), (0, 0, 6, 0)]),
("RGBA", "A;16L", 2, [(0, 0, 0, 2), (0, 0, 0, 4), (0, 0, 0, 6)]),
]
if is_big_endian():
params += [
("RGBA", "R;16N", 2, [(1, 0, 0, 0), (3, 0, 0, 0), (5, 0, 0, 0)]),
("RGBA", "G;16N", 2, [(0, 1, 0, 0), (0, 3, 0, 0), (0, 5, 0, 0)]),
("RGBA", "B;16N", 2, [(0, 0, 1, 0), (0, 0, 3, 0), (0, 0, 5, 0)]),
("RGBA", "A;16N", 2, [(0, 0, 0, 1), (0, 0, 0, 3), (0, 0, 0, 5)]),
]
else:
params += [
("RGBA", "R;16N", 2, [(2, 0, 0, 0), (4, 0, 0, 0), (6, 0, 0, 0)]),
("RGBA", "G;16N", 2, [(0, 2, 0, 0), (0, 4, 0, 0), (0, 6, 0, 0)]),
("RGBA", "B;16N", 2, [(0, 0, 2, 0), (0, 0, 4, 0), (0, 0, 6, 0)]),
("RGBA", "A;16N", 2, [(0, 0, 0, 2), (0, 0, 0, 4), (0, 0, 0, 6)]),
]
# Mode RGBa
params += [
("RGBa", "RGBa", 4, [(1, 2, 3, 4), (5, 6, 7, 8), (9, 10, 11, 12)]),
("RGBa", "BGRa", 4, [(3, 2, 1, 4), (7, 6, 5, 8), (11, 10, 9, 12)]),
("RGBa", "aRGB", 4, [(2, 3, 4, 1), (6, 7, 8, 5), (10, 11, 12, 9)]),
("RGBa", "aBGR", 4, [(4, 3, 2, 1), (8, 7, 6, 5), (12, 11, 10, 9)]),
]
# Mode RGBX
params += [
("RGBX", "RGB", 3, [(1, 2, 3, X), (4, 5, 6, X), (7, 8, 9, X)]),
("RGBX", "RGB;L", 3, [(1, 4, 7, X), (2, 5, 8, X), (3, 6, 9, X)]),
("RGBX", "RGB;16B", 6, [(1, 3, 5, X), (7, 9, 11, X)]),
("RGBX", "BGR", 3, [(3, 2, 1, X), (6, 5, 4, X), (9, 8, 7, X)]),
("RGBX", "XRGB;1555", 2, [(0, 132, 8, X), (8, 0, 24, X)]),
("RGBX", "RGB;565", 2, [(0, 65, 8, X), (0, 130, 24, X)]),
("RGBX", "XBGR;1555", 2, [(8, 132, 0, X), (24, 0, 8, X)]),
("RGBX", "BGR;565", 2, [(8, 65, 0, X), (24, 130, 0, X)]),
("RGBX", "XBGR;4", 2, [(17, 0, 34, X), (51, 0, 68, X)]),
("RGBX", "RGBX", 4, [(1, 2, 3, 4), (5, 6, 7, 8), (9, 10, 11, 12)]),
("RGBX", "RGBXX", 5, [(1, 2, 3, 4), (6, 7, 8, 9), (11, 12, 13, 14)]),
("RGBX", "RGBXXX", 6, [(1, 2, 3, 4), (7, 8, 9, 10), (13, 14, 15, 16)]),
("RGBX", "RGBX;L", 4, [(1, 4, 7, 10), (2, 5, 8, 11), (3, 6, 9, 12)]),
("RGBX", "RGBX;16L", 8, [(2, 4, 6, 8), (10, 12, 14, 16)]),
("RGBX", "RGBX;16B", 8, [(1, 3, 5, 7), (9, 11, 13, 15)]),
("RGBX", "BGRX", 4, [(3, 2, 1, X), (7, 6, 5, X), (11, 10, 9, X)]),
("RGBX", "XRGB", 4, [(2, 3, 4, X), (6, 7, 8, X), (10, 11, 12, X)]),
("RGBX", "XBGR", 4, [(4, 3, 2, X), (8, 7, 6, X), (12, 11, 10, X)]),
(
"RGBX",
"YCC;P",
b"D]\x9c\x82\x1a\x91\xfaOC\xe7J\x12", # random data
[
(127, 102, 0, X),
(192, 227, 0, X),
(213, 255, 170, X),
(98, 255, 133, X),
],
),
("RGBX", "R", 1, [(1, 0, 0, 0), (2, 0, 0, 0), (3, 0, 0, 0)]),
("RGBX", "G", 1, [(0, 1, 0, 0), (0, 2, 0, 0), (0, 3, 0, 0)]),
("RGBX", "B", 1, [(0, 0, 1, 0), (0, 0, 2, 0), (0, 0, 3, 0)]),
("RGBX", "X", 1, [(0, 0, 0, 1), (0, 0, 0, 2), (0, 0, 0, 3)]),
]
# Mode CMYK
params += [
("CMYK", "CMYK", 4, [(1, 2, 3, 4), (5, 6, 7, 8), (9, 10, 11, 12)]),
("CMYK", "CMYKX", 5, [(1, 2, 3, 4), (6, 7, 8, 9), (11, 12, 13, 14)]),
("CMYK", "CMYKXX", 6, [(1, 2, 3, 4), (7, 8, 9, 10), (13, 14, 15, 16)]),
(
"CMYK",
"CMYK;I",
4,
[
(254, 253, 252, 251),
(250, 249, 248, 247),
(246, 245, 244, 243),
],
),
("CMYK", "CMYK;L", 4, [(1, 4, 7, 10), (2, 5, 8, 11), (3, 6, 9, 12)]),
("CMYK", "C", 1, [(1, 0, 0, 0), (2, 0, 0, 0), (3, 0, 0, 0)]),
("CMYK", "M", 1, [(0, 1, 0, 0), (0, 2, 0, 0), (0, 3, 0, 0)]),
("CMYK", "Y", 1, [(0, 0, 1, 0), (0, 0, 2, 0), (0, 0, 3, 0)]),
("CMYK", "K", 1, [(0, 0, 0, 1), (0, 0, 0, 2), (0, 0, 0, 3)]),
("CMYK", "C;I", 1, [(254, 0, 0, 0), (253, 0, 0, 0), (252, 0, 0, 0)]),
("CMYK", "M;I", 1, [(0, 254, 0, 0), (0, 253, 0, 0), (0, 252, 0, 0)]),
("CMYK", "Y;I", 1, [(0, 0, 254, 0), (0, 0, 253, 0), (0, 0, 252, 0)]),
("CMYK", "K;I", 1, [(0, 0, 0, 254), (0, 0, 0, 253), (0, 0, 0, 252)]),
]
# Mode YCbCr
params += [
("YCbCr", "YCbCr", 3, [(1, 2, 3), (4, 5, 6), (7, 8, 9)]),
("YCbCr", "YCbCr;L", 3, [(1, 4, 7), (2, 5, 8), (3, 6, 9)]),
("YCbCr", "YCbCrK", 4, [(1, 2, 3), (5, 6, 7), (9, 10, 11)]),
("YCbCr", "YCbCrX", 4, [(1, 2, 3), (5, 6, 7), (9, 10, 11)]),
]
# Mode LAB
params += [
("LAB", "LAB", 3, [(1, 130, 131), (4, 133, 134), (7, 136, 137)]),
("LAB", "L", 1, [(1, 0, 0), (2, 0, 0), (3, 0, 0)]),
("LAB", "A", 1, [(0, 1, 0), (0, 2, 0), (0, 3, 0)]),
("LAB", "B", 1, [(0, 0, 1), (0, 0, 2), (0, 0, 3)]),
]
# Mode HSV
params += [
("HSV", "HSV", 3, [(1, 2, 3), (4, 5, 6), (7, 8, 9)]),
("HSV", "H", 1, [(1, 0, 0), (2, 0, 0), (3, 0, 0)]),
("HSV", "S", 1, [(0, 1, 0), (0, 2, 0), (0, 3, 0)]),
("HSV", "V", 1, [(0, 0, 1), (0, 0, 2), (0, 0, 3)]),
]
# Mode I
params += [
("I", "I;8", 1, [0x01, 0x02, 0x03, 0x04]),
("I", "I;8S", b"\x01\x83", [1, -125]),
("I", "I;16", 2, [0x0201, 0x0403]),
("I", "I;16S", b"\x83\x01\x01\x83", [0x0183, -31999]),
("I", "I;16B", 2, [0x0102, 0x0304]),
("I", "I;16BS", b"\x83\x01\x01\x83", [-31999, 0x0183]),
("I", "I;32", 4, [0x04030201, 0x08070605]),
("I", "I;32S", b"\x83\x00\x00\x01\x01\x00\x00\x83", [0x01000083, -2097151999]),
("I", "I;32B", 4, [0x01020304, 0x05060708]),
("I", "I;32BS", b"\x83\x00\x00\x01\x01\x00\x00\x83", [-2097151999, 0x01000083]),
]
if is_big_endian():
params += [
("I", "I", 4, [0x01020304, 0x05060708]),
("I", "I;16N", 2, [0x0102, 0x0304]),
("I", "I;16NS", b"\x83\x01\x01\x83", [-31999, 0x0183]),
("I", "I;32N", 4, [0x01020304, 0x05060708]),
(
"I",
"I;32NS",
b"\x83\x00\x00\x01\x01\x00\x00\x83",
[
-2097151999,
0x01000083,
],
),
]
else:
params += [
("I", "I", 4, [0x04030201, 0x08070605]),
("I", "I;16N", 2, [0x0201, 0x0403]),
("I", "I;16NS", b"\x83\x01\x01\x83", [0x0183, -31999]),
("I", "I;32N", 4, [0x04030201, 0x08070605]),
(
"I",
"I;32NS",
b"\x83\x00\x00\x01\x01\x00\x00\x83",
[
0x01000083,
-2097151999,
],
),
]
# Mode F int
params += [
("F", "F;8", 1, [0x01, 0x02, 0x03, 0x04]),
("F", "F;8S", b"\x01\x83", [1, -125]),
("F", "F;16", 2, [0x0201, 0x0403]),
("F", "F;16S", b"\x83\x01\x01\x83", [0x0183, -31999]),
("F", "F;16B", 2, [0x0102, 0x0304]),
("F", "F;16BS", b"\x83\x01\x01\x83", [-31999, 0x0183]),
("F", "F;32", 4, [67305984, 134678016]),
("F", "F;32S", b"\x83\x00\x00\x01\x01\x00\x00\x83", [16777348, -2097152000]),
("F", "F;32B", 4, [0x01020304, 0x05060708]),
("F", "F;32BS", b"\x83\x00\x00\x01\x01\x00\x00\x83", [-2097152000, 16777348]),
]
if is_big_endian():
params += [
("F", "F;16N", 2, [0x0102, 0x0304]),
("F", "F;16NS", b"\x83\x01\x01\x83", [-31999, 0x0183]),
("F", "F;32N", 4, [0x01020304, 0x05060708]),
(
"F",
"F;32NS",
b"\x83\x00\x00\x01\x01\x00\x00\x83",
[-2097152000, 16777348],
),
]
else:
params += [
("F", "F;16N", 2, [0x0201, 0x0403]),
("F", "F;16NS", b"\x83\x01\x01\x83", [0x0183, -31999]),
("F", "F;32N", 4, [67305984, 134678016]),
(
"F",
"F;32NS",
b"\x83\x00\x00\x01\x01\x00\x00\x83",
[16777348, -2097152000],
),
]
# Mode F float
params += [
("F", "F;32F", 4, [1.539989614439558e-36, 4.063216068939723e-34]),
("F", "F;32BF", 4, [2.387939260590663e-38, 6.301941157072183e-36]),
(
"F",
"F;64F",
b"333333\xc3?\x00\x00\x00\x00\x00J\x93\xc0", # by struct.pack
[0.15000000596046448, -1234.5],
),
(
"F",
"F;64BF",
b"?\xc3333333\xc0\x93J\x00\x00\x00\x00\x00", # by struct.pack
[0.15000000596046448, -1234.5],
),
]
if is_big_endian():
params += [
("F", "F", 4, [2.387939260590663e-38, 6.301941157072183e-36]),
("F", "F;32NF", 4, [2.387939260590663e-38, 6.301941157072183e-36]),
(
"F",
"F;64NF",
b"?\xc3333333\xc0\x93J\x00\x00\x00\x00\x00",
[0.15000000596046448, -1234.5],
),
]
else:
params += [
("F", "F", 4, [1.539989614439558e-36, 4.063216068939723e-34]),
("F", "F;32NF", 4, [1.539989614439558e-36, 4.063216068939723e-34]),
(
"F",
"F;64NF",
b"333333\xc3?\x00\x00\x00\x00\x00J\x93\xc0",
[0.15000000596046448, -1234.5],
),
]
# Mode I;16
params += [
("I;16", "I;16", 2, [0x0201, 0x0403, 0x0605]),
("I;16", "I;16B", 2, [0x0102, 0x0304, 0x0506]),
("I;16B", "I;16B", 2, [0x0102, 0x0304, 0x0506]),
("I;16L", "I;16L", 2, [0x0201, 0x0403, 0x0605]),
("I;16", "I;12", 2, [0x0010, 0x0203, 0x0040]),
]
if is_big_endian():
params += [
("I;16", "I;16N", 2, [0x0102, 0x0304, 0x0506]),
("I;16B", "I;16N", 2, [0x0102, 0x0304, 0x0506]),
("I;16L", "I;16N", 2, [0x0102, 0x0304, 0x0506]),
("I;16N", "I;16N", 2, [0x0102, 0x0304, 0x0506]),
]
else:
params += [
("I;16", "I;16N", 2, [0x0201, 0x0403, 0x0605]),
("I;16B", "I;16N", 2, [0x0201, 0x0403, 0x0605]),
("I;16L", "I;16N", 2, [0x0201, 0x0403, 0x0605]),
("I;16N", "I;16N", 2, [0x0201, 0x0403, 0x0605]),
]
# Mode CMYK
params += [
("CMYK", "CMYK;16L", 8, [(2, 4, 6, 8), (10, 12, 14, 16)]),
("CMYK", "CMYK;16B", 8, [(1, 3, 5, 7), (9, 11, 13, 15)]),
]
if is_big_endian():
params += [
("CMYK", "CMYK;16N", 8, [(1, 3, 5, 7), (9, 11, 13, 15)]),
]
else:
params += [
("CMYK", "CMYK;16N", 8, [(2, 4, 6, 8), (10, 12, 14, 16)]),
]
return params
@pytest.mark.parametrize("mode, rawmode, data, pixels", get_unpack_parameters())
def test_unpack(
mode: str, rawmode: str, data: bytes | int, pixels: list[float | tuple[int, ...]]
) -> None:
"""
Test unpacking from {rawmode} to {mode}.
data - Either the number of bytes in a pixel of the rawmode, or a byte string.
If a number is given, bytes will be generated starting at 1.
The byte string will be used to populate the starting image.
pixels - The pixels to be compared against the image pixels after unpacking.
"""
if isinstance(data, int):
data_len = data * len(pixels)
data = bytes(range(1, data_len + 1))
im = Image.frombytes(mode, (len(pixels), 1), data, "raw", rawmode, 0, 1)
for x, pixel in enumerate(pixels):
assert pixel == im.getpixel((x, 0))
@pytest.mark.parametrize(
"mode, rawmode, data, pixels",
(
# Mode BGR
("BGR;15", "BGR;15", 3, [(8, 131, 0), (24, 0, 8), (41, 131, 8)]),
("BGR;16", "BGR;16", 3, [(8, 64, 0), (24, 129, 0), (41, 194, 0)]),
("BGR;24", "BGR;24", 3, [(1, 2, 3), (4, 5, 6), (7, 8, 9)]),
# Rawmode RGB/BGR/RGBA/BGRA 15/16/4B
("RGB", "RGB;15", 2, [(8, 131, 0), (24, 0, 8)]),
("RGB", "BGR;15", 2, [(0, 131, 8), (8, 0, 24)]),
("RGB", "RGB;16", 2, [(8, 64, 0), (24, 129, 0)]),
("RGB", "BGR;16", 2, [(0, 64, 8), (0, 129, 24)]),
("RGB", "RGB;4B", 2, [(17, 0, 34), (51, 0, 68)]),
("RGBA", "RGBA;15", 2, [(8, 131, 0, 0), (24, 0, 8, 0)]),
("RGBA", "BGRA;15", 2, [(0, 131, 8, 0), (8, 0, 24, 0)]),
("RGBA", "BGRA;15Z", 2, [(0, 131, 8, X), (8, 0, 24, X)]),
("RGBA", "RGBA;4B", 2, [(17, 0, 34, 0), (51, 0, 68, 0)]),
("RGBX", "RGB;15", 2, [(8, 131, 0, X), (24, 0, 8, X)]),
("RGBX", "BGR;15", 2, [(0, 131, 8, X), (8, 0, 24, X)]),
("RGBX", "RGB;4B", 2, [(17, 0, 34, X), (51, 0, 68, X)]),
),
)
def test_unpack_deprecated(
mode: str, rawmode: str, data: int | bytes, pixels: list[float | tuple[int, ...]]
) -> None:
with pytest.warns(DeprecationWarning):
test_unpack(mode, rawmode, data, pixels)
@pytest.mark.parametrize(
"mode, rawmode, data, pixels",
(
("L", "L", 0, [0]),
("RGB", "RGB", 2, [0]),
("CMYK", "CMYK", 2, [0]),
),
)
def test_unpack_valueerror(
mode: str, rawmode: str, data: int | bytes, pixels: list[float | tuple[int, ...]]
) -> None:
with pytest.raises(ValueError):
test_unpack(mode, rawmode, data, pixels)
Reference in New Issue View Git Blame Copy Permalink