python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-01-26 01:04:29 +03:00
Code Issues Packages Projects Releases Wiki Activity

Added channels parameter

Browse Source
This commit is contained in:
Andrew Murray 2024-08-07 18:28:51 +10:00 committed by Aarni Koskela
parent 5c4aeaa329
commit a06529a3a8
1 changed files with 24 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
Tests/test_color_lut.py
View File

@ -106,39 +106,39 @@ class TestColorLut3DCoreAPI:
im.im.color_lut_3d("RGB", Image.Resampling.BILINEAR, 3, 2, 2, 2, 16) im.im.color_lut_3d("RGB", Image.Resampling.BILINEAR, 3, 2, 2, 2, 16)
@pytest.mark.parametrize( @pytest.mark.parametrize(
"lut_mode, table_size", "lut_mode, table_channels, table_size",
[ [
("RGB", (3, 3)), ("RGB", 3, 3),
("CMYK", (4, 3)), ("CMYK", 4, 3),
("RGB", (3, (2, 3, 3))), ("RGB", 3, (2, 3, 3)),
("RGB", (3, (65, 3, 3))), ("RGB", 3, (65, 3, 3)),
("RGB", (3, (3, 65, 3))), ("RGB", 3, (3, 65, 3)),
("RGB", (3, (3, 3, 65))), ("RGB", 3, (2, 3, 65)),
], ],
) )
def test_correct_args( def test_correct_args(
self, lut_mode: str, table_size: tuple[int, int | tuple[int, int, int]] self, lut_mode: str, table_channels: int, table_size: int | tuple[int, int, int]
) -> None: ) -> None:
im = Image.new("RGB", (10, 10), 0) im = Image.new("RGB", (10, 10), 0)
assert im.im is not None assert im.im is not None
im.im.color_lut_3d( im.im.color_lut_3d(
lut_mode, lut_mode,
Image.Resampling.BILINEAR, Image.Resampling.BILINEAR,
*self.generate_identity_table(*table_size), *self.generate_identity_table(table_channels, table_size),
) )
@pytest.mark.parametrize( @pytest.mark.parametrize(
"image_mode, lut_mode, table_size", "image_mode, lut_mode, table_channels, table_size",
[ [
("L", "RGB", (3, 3)), ("L", "RGB", 3, 3),
("RGB", "L", (3, 3)), ("RGB", "L", 3, 3),
("L", "L", (3, 3)), ("L", "L", 3, 3),
("RGB", "RGBA", (3, 3)), ("RGB", "RGBA", 3, 3),
("RGB", "RGB", (4, 3)), ("RGB", "RGB", 4, 3),
], ],
) )
def test_wrong_mode( def test_wrong_mode(
self, image_mode: str, lut_mode: str, table_size: tuple[int, int] self, image_mode: str, lut_mode: str, table_channels: int, table_size: int
) -> None: ) -> None:
with pytest.raises(ValueError, match="wrong mode"): with pytest.raises(ValueError, match="wrong mode"):
im = Image.new(image_mode, (10, 10), 0) im = Image.new(image_mode, (10, 10), 0)
@ -146,27 +146,27 @@ class TestColorLut3DCoreAPI:
im.im.color_lut_3d( im.im.color_lut_3d(
lut_mode, lut_mode,
Image.Resampling.BILINEAR, Image.Resampling.BILINEAR,
*self.generate_identity_table(*table_size), *self.generate_identity_table(table_channels, table_size),
) )
@pytest.mark.parametrize( @pytest.mark.parametrize(
"image_mode, lut_mode, table_size", "image_mode, lut_mode, table_channels, table_size",
[ [
("RGBA", "RGBA", (3, 3)), ("RGBA", "RGBA", 3, 3),
("RGBA", "RGBA", (4, 3)), ("RGBA", "RGBA", 4, 3),
("RGB", "HSV", (3, 3)), ("RGB", "HSV", 3, 3),
("RGB", "RGBA", (4, 3)), ("RGB", "RGBA", 4, 3),
], ],
) )
def test_correct_mode( def test_correct_mode(
self, image_mode: str, lut_mode: str, table_size: tuple[int, int] self, image_mode: str, lut_mode: str, table_channels: int, table_size: int
) -> None: ) -> None:
im = Image.new(image_mode, (10, 10), 0) im = Image.new(image_mode, (10, 10), 0)
assert im.im is not None assert im.im is not None
im.im.color_lut_3d( im.im.color_lut_3d(
lut_mode, lut_mode,
Image.Resampling.BILINEAR, Image.Resampling.BILINEAR,
*self.generate_identity_table(*table_size), *self.generate_identity_table(table_channels, table_size),
) )
def test_identities(self) -> None: def test_identities(self) -> None: