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

Added type hints

Browse Source
This commit is contained in:
Andrew Murray 2024-06-21 21:38:06 +10:00
parent e0a5d81127
commit 44da8b64c2
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
Tests/test_file_tiff.py
View File

@ -874,16 +874,20 @@ class TestFileTiff:
with Image.open(test_file): with Image.open(test_file):
pass pass
def test_open_tiff_uint16_multiband(self): def test_open_tiff_uint16_multiband(self) -> None:
"""Test opening multiband TIFFs and reading all channels.""" """Test opening multiband TIFFs and reading all channels."""
def check_pixel(im: Image.Image, expected_pixel, pos: tuple[int, int]): def check_pixel(
im: Image.Image, expected_pixel: tuple[int, ...], pos: tuple[int, int]
) -> None:
actual_pixel = im.getpixel((0, 0)) actual_pixel = im.getpixel((0, 0))
if isinstance(actual_pixel, int): if isinstance(actual_pixel, int):
actual_pixel = (actual_pixel,) actual_pixel = (actual_pixel,)
assert actual_pixel == expected_pixel assert actual_pixel == expected_pixel
def check_image(im: Image.Image, width: int, height: int, expected_pixel): def check_image(
im: Image.Image, width: int, height: int, expected_pixel: tuple[int, ...]
) -> None:
assert im.width == width assert im.width == width
assert im.height == height assert im.height == height
for x in range(im.width): for x in range(im.width):
@ -892,7 +896,7 @@ class TestFileTiff:
base_value = 4660 base_value = 4660
for i in range(1, 6): for i in range(1, 6):
pixel = tuple([base_value + j for j in range(0, i)]) pixel = tuple(base_value + j for j in range(i))
infile = f"Tests/images/uint16_{i}_{base_value}.tif" infile = f"Tests/images/uint16_{i}_{base_value}.tif"
im = Image.open(infile) im = Image.open(infile)