diff --git a/crc.py b/crc.py index ad728fa..3720e8d 100644 --- a/crc.py +++ b/crc.py @@ -10,6 +10,7 @@ import sys from dataclasses import dataclass from typing import ( + Any, BinaryIO, ByteString, Iterable, @@ -403,13 +404,13 @@ def verify( def _bytes_generator( - data: Union[int, ByteString, BinaryIO, Iterable[ByteString]] + data: Union[int, ByteString, BinaryIO, Iterable[ByteString]] # type: ignore ) -> Iterable[bytes]: if isinstance(data, int): yield data.to_bytes(1, "big") - elif isinstance(data, ByteString): - yield bytes(data) - elif isinstance(data, (Iterable, BinaryIO)): + elif isinstance(data, ByteString): # type: ignore + yield bytes(data) # type: ignore + elif isinstance(data, (Iterable, BinaryIO)): # type: ignore yield from (bytes(e) for e in data) else: raise TypeError(f"Unsupported parameter type: {type(data)}") diff --git a/test/unit/test_crc.py b/test/unit/test_crc.py index 89247ef..9dcd1f8 100644 --- a/test/unit/test_crc.py +++ b/test/unit/test_crc.py @@ -21,7 +21,7 @@ create_lookup_table, ) -Fixture = namedtuple("TestData", "data checksum") +Fixture = namedtuple("Fixture", "data checksum") class TemplateTest(unittest.TestCase):