diff options
Diffstat (limited to 'tests/test_parse_bytes_function.py')
| -rw-r--r-- | tests/test_parse_bytes_function.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/tests/test_parse_bytes_function.py b/tests/test_parse_bytes_function.py index 6637cbd..318c4aa 100644 --- a/tests/test_parse_bytes_function.py +++ b/tests/test_parse_bytes_function.py @@ -4,8 +4,8 @@ import pytest from hypothesis import given from hypothesis.strategies import binary -from natsort.ns_enum import ns -from natsort.utils import parse_bytes_factory +from natsort.ns_enum import NSType, ns +from natsort.utils import BytesTransformer, parse_bytes_factory @pytest.mark.parametrize( @@ -19,6 +19,8 @@ from natsort.utils import parse_bytes_factory ], ) @given(x=binary()) -def test_parse_bytest_factory_makes_function_that_returns_tuple(x, alg, example_func): +def test_parse_bytest_factory_makes_function_that_returns_tuple( + x: bytes, alg: NSType, example_func: BytesTransformer +) -> None: parse_bytes_func = parse_bytes_factory(alg) assert parse_bytes_func(x) == example_func(x) |