python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-02-03 21:24:31 +03:00
Code Issues Packages Projects Releases Wiki Activity

enable bgr modes for image bytes tests

Browse Source
This commit is contained in:
Yay295 2024-03-11 12:00:48 -05:00
parent 12d626a6bb
commit 680f387bc2
1 changed files with 4 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
Tests/test_image.py
View File

@ -1045,26 +1045,18 @@ class TestImage:
class TestImageBytes: class TestImageBytes:
# The BGR modes don't support the methods tested in this class.
image_modes_not_bgr = [mode for mode in image_modes if "BGR" not in mode[0]]
image_mode_names_not_bgr = [
name for name, num_bands, pixelsize in image_modes if "BGR" not in name
]
sample_bytes = bytes( sample_bytes = bytes(
range( range(2 * 2 * max(pixelsize for mode, num_bands, pixelsize in image_modes))
2 * 2 * max(pixelsize for mode, num_bands, pixelsize in image_modes_not_bgr)
)
) )
@pytest.mark.parametrize("mode", image_mode_names_not_bgr) @pytest.mark.parametrize("mode", image_mode_names)
def test_roundtrip_bytes_constructor(self, mode: str): def test_roundtrip_bytes_constructor(self, mode: str):
source_image = hopper(mode) source_image = hopper(mode)
source_bytes = source_image.tobytes() source_bytes = source_image.tobytes()
copy_image = Image.frombytes(mode, source_image.size, source_bytes) copy_image = Image.frombytes(mode, source_image.size, source_bytes)
assert copy_image.tobytes() == source_bytes assert copy_image.tobytes() == source_bytes
@pytest.mark.parametrize("mode", image_mode_names_not_bgr) @pytest.mark.parametrize("mode", image_mode_names)
def test_roundtrip_bytes_method(self, mode: str): def test_roundtrip_bytes_method(self, mode: str):
source_image = hopper(mode) source_image = hopper(mode)
source_bytes = source_image.tobytes() source_bytes = source_image.tobytes()
@ -1072,7 +1064,7 @@ class TestImageBytes:
copy_image.frombytes(source_bytes) copy_image.frombytes(source_bytes)
assert copy_image.tobytes() == source_bytes assert copy_image.tobytes() == source_bytes
@pytest.mark.parametrize(("mode", "num_bands", "pixelsize"), image_modes_not_bgr) @pytest.mark.parametrize(("mode", "num_bands", "pixelsize"), image_modes)
def test_pixels_after_getdata_putdata( def test_pixels_after_getdata_putdata(
self, mode: str, num_bands: int, pixelsize: int self, mode: str, num_bands: int, pixelsize: int
): ):