Skip to content

Commit 12d626a

Browse files
committed
add typing to image bytes tests
1 parent c56c890 commit 12d626a

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

Tests/test_image.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1058,22 +1058,24 @@ class TestImageBytes:
10581058
)
10591059

10601060
@pytest.mark.parametrize("mode", image_mode_names_not_bgr)
1061-
def test_roundtrip_bytes_constructor(self, mode):
1061+
def test_roundtrip_bytes_constructor(self, mode: str):
10621062
source_image = hopper(mode)
10631063
source_bytes = source_image.tobytes()
10641064
copy_image = Image.frombytes(mode, source_image.size, source_bytes)
10651065
assert copy_image.tobytes() == source_bytes
10661066

10671067
@pytest.mark.parametrize("mode", image_mode_names_not_bgr)
1068-
def test_roundtrip_bytes_method(self, mode):
1068+
def test_roundtrip_bytes_method(self, mode: str):
10691069
source_image = hopper(mode)
10701070
source_bytes = source_image.tobytes()
10711071
copy_image = Image.new(mode, source_image.size)
10721072
copy_image.frombytes(source_bytes)
10731073
assert copy_image.tobytes() == source_bytes
10741074

10751075
@pytest.mark.parametrize(("mode", "num_bands", "pixelsize"), image_modes_not_bgr)
1076-
def test_pixels_after_getdata_putdata(self, mode, num_bands, pixelsize):
1076+
def test_pixels_after_getdata_putdata(
1077+
self, mode: str, num_bands: int, pixelsize: int
1078+
):
10771079
image_byte_size = 2 * 2 * pixelsize
10781080
start_bytes = self.sample_bytes[:image_byte_size]
10791081
image = Image.frombytes(mode, (2, 2), start_bytes)

0 commit comments

Comments
 (0)