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

parametrize test_histogram()

Browse Source
This commit is contained in:
Yay295 2024-09-18 18:16:06 -05:00
parent dc3751599f
commit 90f709fb79
1 changed files with 30 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
Tests/test_image_histogram.py
View File

@ -1,19 +1,36 @@
from __future__ import annotations
import pytest
from PIL import Image
from .helper import hopper
expected_data = {
"1": (256, 0, 10994),
"CMYK": (1024, 0, 16384),
"F": (256, 0, 662),
"HSV": (768, 0, 1696),
"I": (256, 0, 662),
"I;16": (256, 0, 8192),
"I;16B": (256, 0, 8192),
"I;16L": (256, 0, 8192),
"I;16N": (256, 0, 8192),
"L": (256, 0, 662),
"LA": (512, 0, 662),
"La": (512, 0, 662),
"LAB": (768, 0, 1946),
"P": (256, 0, 1551),
"PA": (512, 0, 1551),
"RGB": (768, 4, 675),
"RGBA": (1024, 0, 16384),
"RGBa": (1024, 0, 16384),
"RGBX": (1024, 0, 16384),
"YCbCr": (768, 0, 1908),
}
def test_histogram() -> None:
def histogram(mode: str) -> tuple[int, int, int]:
h = hopper(mode).histogram()
return len(h), min(h), max(h)
assert histogram("1") == (256, 0, 10994)
assert histogram("L") == (256, 0, 662)
assert histogram("I") == (256, 0, 662)
assert histogram("F") == (256, 0, 662)
assert histogram("P") == (256, 0, 1551)
assert histogram("RGB") == (768, 4, 675)
assert histogram("RGBA") == (1024, 0, 16384)
assert histogram("CMYK") == (1024, 0, 16384)
assert histogram("YCbCr") == (768, 0, 1908)
@pytest.mark.parametrize("mode", Image.MODES)
def test_histogram(mode: str) -> None:
h = hopper(mode).histogram()
assert (len(h), min(h), max(h)) == expected_data[mode]