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

parametrize test_putdata_array()

Browse Source
This commit is contained in:
Yay295 2024-04-28 00:16:32 -05:00
parent 99519473a8
commit c9d7beaed0
1 changed files with 10 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
Tests/test_image_data.py
View File

@ -117,23 +117,19 @@ def test_putdata_mode_BGR(mode: str) -> None:
assert list(im.getdata()) == data assert list(im.getdata()) == data
def test_putdata_array_B() -> None: @pytest.mark.parametrize(
"mode, type, value",
(
("L", "B", 0), # B = unsigned char
("F", "f", 0.0), # f = float
),
)
def test_putdata_array(mode: str, type: str, value: float) -> None:
# shouldn't segfault # shouldn't segfault
# see https://github.com/python-pillow/Pillow/issues/1008 # see https://github.com/python-pillow/Pillow/issues/1008
arr = array("B", [0]) * 15000 im = Image.new(mode, (150, 100))
im = Image.new("L", (150, 100)) arr = array(type, [value]) * 15000
im.putdata(arr)
assert len(im.getdata()) == len(arr)
def test_putdata_array_F() -> None:
# shouldn't segfault
# see https://github.com/python-pillow/Pillow/issues/1008
im = Image.new("F", (150, 100))
arr = array("f", [0.0]) * 15000
im.putdata(arr) im.putdata(arr)
assert len(im.getdata()) == len(arr) assert len(im.getdata()) == len(arr)