python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-01-11 17:56:18 +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)
@pytest.mark.parametrize(
"lut_mode, table_size",
"lut_mode, table_channels, table_size",
[
("RGB", (3, 3)),
("CMYK", (4, 3)),
("RGB", (3, (2, 3, 3))),
("RGB", (3, (65, 3, 3))),
("RGB", (3, (3, 65, 3))),
("RGB", (3, (3, 3, 65))),
("RGB", 3, 3),
("CMYK", 4, 3),
("RGB", 3, (2, 3, 3)),
("RGB", 3, (65, 3, 3)),
("RGB", 3, (3, 65, 3)),
("RGB", 3, (2, 3, 65)),
],
)
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:
im = Image.new("RGB", (10, 10), 0)
assert im.im is not None
im.im.color_lut_3d(
lut_mode,
Image.Resampling.BILINEAR,
*self.generate_identity_table(*table_size),
*self.generate_identity_table(table_channels, table_size),
)
@pytest.mark.parametrize(
"image_mode, lut_mode, table_size",
"image_mode, lut_mode, table_channels, table_size",
[
("L", "RGB", (3, 3)),
("RGB", "L", (3, 3)),
("L", "L", (3, 3)),
("RGB", "RGBA", (3, 3)),
("RGB", "RGB", (4, 3)),
("L", "RGB", 3, 3),
("RGB", "L", 3, 3),
("L", "L", 3, 3),
("RGB", "RGBA", 3, 3),
("RGB", "RGB", 4, 3),
],
)
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:
with pytest.raises(ValueError, match="wrong mode"):
im = Image.new(image_mode, (10, 10), 0)
@ -146,27 +146,27 @@ class TestColorLut3DCoreAPI:
im.im.color_lut_3d(
lut_mode,
Image.Resampling.BILINEAR,
*self.generate_identity_table(*table_size),
*self.generate_identity_table(table_channels, table_size),
)
@pytest.mark.parametrize(
"image_mode, lut_mode, table_size",
"image_mode, lut_mode, table_channels, table_size",
[
("RGBA", "RGBA", (3, 3)),
("RGBA", "RGBA", (4, 3)),
("RGB", "HSV", (3, 3)),
("RGB", "RGBA", (4, 3)),
("RGBA", "RGBA", 3, 3),
("RGBA", "RGBA", 4, 3),
("RGB", "HSV", 3, 3),
("RGB", "RGBA", 4, 3),
],
)
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:
im = Image.new(image_mode, (10, 10), 0)
assert im.im is not None
im.im.color_lut_3d(
lut_mode,
Image.Resampling.BILINEAR,
*self.generate_identity_table(*table_size),
*self.generate_identity_table(table_channels, table_size),
)
def test_identities(self) -> None: