1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
# -*- coding: utf-8 -*-
"""These test the utils.py functions."""
from typing import Callable, Union
import pytest
from hypothesis import example, given
from hypothesis.strategies import floats, integers, text
from natsort.ns_enum import NSType, NS_DUMB, ns
from natsort.utils import final_data_transform_factory
@pytest.mark.parametrize("alg", [ns.DEFAULT, ns.UNGROUPLETTERS, ns.LOCALE])
@given(x=text(), y=floats(allow_nan=False, allow_infinity=False) | integers())
@pytest.mark.usefixtures("with_locale_en_us")
def test_final_data_transform_factory_default(
x: str, y: Union[int, float], alg: NSType
) -> None:
final_data_transform_func = final_data_transform_factory(alg, "", "::")
value = (x, y)
original_value = "".join(map(str, value))
result = final_data_transform_func(value, original_value)
assert result == value
@pytest.mark.parametrize(
"alg, func",
[
(ns.UNGROUPLETTERS | ns.LOCALE, lambda x: x),
(ns.LOCALE | ns.UNGROUPLETTERS | NS_DUMB, lambda x: x),
(ns.LOCALE | ns.UNGROUPLETTERS | ns.LOWERCASEFIRST, lambda x: x),
(
ns.LOCALE | ns.UNGROUPLETTERS | NS_DUMB | ns.LOWERCASEFIRST,
lambda x: x.swapcase(),
),
],
)
@given(x=text(), y=floats(allow_nan=False, allow_infinity=False) | integers())
@example(x="İ", y=0)
@pytest.mark.usefixtures("with_locale_en_us")
def test_final_data_transform_factory_ungroup_and_locale(
x: str, y: Union[int, float], alg: NSType, func: Callable[[str], str]
) -> None:
final_data_transform_func = final_data_transform_factory(alg, "", "::")
value = (x, y)
original_value = "".join(map(str, value))
result = final_data_transform_func(value, original_value)
if x:
expected = ((func(original_value[:1]),), value)
else:
expected = (("::",), value)
assert result == expected
def test_final_data_transform_factory_ungroup_and_locale_empty_tuple() -> None:
final_data_transform_func = final_data_transform_factory(ns.UG | ns.L, "", "::")
assert final_data_transform_func((), "") == ((), ())
|