diff options
Diffstat (limited to 'tests/test_natsorted_convenience.py')
| -rw-r--r-- | tests/test_natsorted_convenience.py | 39 |
1 files changed, 24 insertions, 15 deletions
diff --git a/tests/test_natsorted_convenience.py b/tests/test_natsorted_convenience.py index cdc2c50..599c4ba 100644 --- a/tests/test_natsorted_convenience.py +++ b/tests/test_natsorted_convenience.py @@ -5,6 +5,7 @@ See the README or the natsort homepage for more details. """ from operator import itemgetter +from typing import List import pytest from natsort import ( @@ -23,21 +24,21 @@ from natsort import ( @pytest.fixture -def version_list(): +def version_list() -> List[str]: return ["1.9.9a", "1.11", "1.9.9b", "1.11.4", "1.10.1"] @pytest.fixture -def float_list(): +def float_list() -> List[str]: return ["a50", "a51.", "a50.31", "a-50", "a50.4", "a5.034e1", "a50.300"] @pytest.fixture -def fruit_list(): +def fruit_list() -> List[str]: return ["Apple", "corn", "Corn", "Banana", "apple", "banana"] -def test_decoder_returns_function_that_can_decode_bytes_but_return_non_bytes_as_is(): +def test_decoder_returns_function_that_can_decode_bytes_but_return_non_bytes_as_is() -> None: func = decoder("latin1") str_obj = "bytes" int_obj = 14 @@ -46,24 +47,28 @@ def test_decoder_returns_function_that_can_decode_bytes_but_return_non_bytes_as_ assert func(str_obj) is str_obj # same object returned b/c only bytes has decode -def test_as_ascii_converts_bytes_to_ascii(): +def test_as_ascii_converts_bytes_to_ascii() -> None: assert decoder("ascii")(b"bytes") == as_ascii(b"bytes") -def test_as_utf8_converts_bytes_to_utf8(): +def test_as_utf8_converts_bytes_to_utf8() -> None: assert decoder("utf8")(b"bytes") == as_utf8(b"bytes") -def test_realsorted_is_identical_to_natsorted_with_real_alg(float_list): +def test_realsorted_is_identical_to_natsorted_with_real_alg( + float_list: List[str], +) -> None: assert realsorted(float_list) == natsorted(float_list, alg=ns.REAL) @pytest.mark.usefixtures("with_locale_en_us") -def test_humansorted_is_identical_to_natsorted_with_locale_alg(fruit_list): +def test_humansorted_is_identical_to_natsorted_with_locale_alg( + fruit_list: List[str], +) -> None: assert humansorted(fruit_list) == natsorted(fruit_list, alg=ns.LOCALE) -def test_index_natsorted_returns_integer_list_of_sort_order_for_input_list(): +def test_index_natsorted_returns_integer_list_of_sort_order_for_input_list() -> None: given = ["num3", "num5", "num2"] other = ["foo", "bar", "baz"] index = index_natsorted(given) @@ -72,27 +77,31 @@ def test_index_natsorted_returns_integer_list_of_sort_order_for_input_list(): assert [other[i] for i in index] == ["baz", "foo", "bar"] -def test_index_natsorted_reverse(): +def test_index_natsorted_reverse() -> None: given = ["num3", "num5", "num2"] assert index_natsorted(given, reverse=True) == index_natsorted(given)[::-1] -def test_index_natsorted_applies_key_function_before_sorting(): +def test_index_natsorted_applies_key_function_before_sorting() -> None: given = [("a", "num3"), ("b", "num5"), ("c", "num2")] expected = [2, 0, 1] assert index_natsorted(given, key=itemgetter(1)) == expected -def test_index_realsorted_is_identical_to_index_natsorted_with_real_alg(float_list): +def test_index_realsorted_is_identical_to_index_natsorted_with_real_alg( + float_list: List[str], +) -> None: assert index_realsorted(float_list) == index_natsorted(float_list, alg=ns.REAL) @pytest.mark.usefixtures("with_locale_en_us") -def test_index_humansorted_is_identical_to_index_natsorted_with_locale_alg(fruit_list): +def test_index_humansorted_is_identical_to_index_natsorted_with_locale_alg( + fruit_list: List[str], +) -> None: assert index_humansorted(fruit_list) == index_natsorted(fruit_list, alg=ns.LOCALE) -def test_order_by_index_sorts_list_according_to_order_of_integer_list(): +def test_order_by_index_sorts_list_according_to_order_of_integer_list() -> None: given = ["num3", "num5", "num2"] index = [2, 0, 1] expected = [given[i] for i in index] @@ -100,7 +109,7 @@ def test_order_by_index_sorts_list_according_to_order_of_integer_list(): assert order_by_index(given, index) == expected -def test_order_by_index_returns_generator_with_iter_true(): +def test_order_by_index_returns_generator_with_iter_true() -> None: given = ["num3", "num5", "num2"] index = [2, 0, 1] assert order_by_index(given, index, True) != [given[i] for i in index] |