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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
# -*- coding: utf-8 -*-
"""\
Test the fake fastnumbers module.
"""
import unicodedata
from math import isinf
from typing import Union, cast
from hypothesis import given
from hypothesis.strategies import floats, integers, text
from natsort.compat.fake_fastnumbers import fast_float, fast_int
def is_float(x: str) -> bool:
try:
float(x)
except ValueError:
try:
unicodedata.numeric(x)
except (ValueError, TypeError):
return False
else:
return True
else:
return True
def not_a_float(x: str) -> bool:
return not is_float(x)
def is_int(x: Union[str, float]) -> bool:
try:
return cast(float, x).is_integer()
except AttributeError:
try:
int(x)
except ValueError:
try:
unicodedata.digit(cast(str, x))
except (ValueError, TypeError):
return False
else:
return True
else:
return True
def not_an_int(x: Union[str, float]) -> bool:
return not is_int(x)
# Each test has an "example" version for demonstrative purposes,
# and a test that uses the hypothesis module.
def test_fast_float_returns_nan_alternate_if_nan_option_is_given() -> None:
assert fast_float("nan", nan=7) == 7
def test_fast_float_converts_float_string_to_float_example() -> None:
assert fast_float("45.8") == 45.8
assert fast_float("-45") == -45.0
assert fast_float("45.8e-2", key=lambda x: x.upper()) == 45.8e-2
assert isinf(cast(float, fast_float("nan")))
assert isinf(cast(float, fast_float("+nan")))
assert isinf(cast(float, fast_float("-NaN")))
assert fast_float("۱۲.۱۲") == 12.12
assert fast_float("-۱۲.۱۲") == -12.12
@given(floats(allow_nan=False))
def test_fast_float_converts_float_string_to_float(x: float) -> None:
assert fast_float(repr(x)) == x
def test_fast_float_leaves_string_as_is_example() -> None:
assert fast_float("invalid") == "invalid"
@given(text().filter(not_a_float).filter(bool))
def test_fast_float_leaves_string_as_is(x: str) -> None:
assert fast_float(x) == x
def test_fast_float_with_key_applies_to_string_example() -> None:
assert fast_float("invalid", key=lambda x: x.upper()) == "INVALID"
@given(text().filter(not_a_float).filter(bool))
def test_fast_float_with_key_applies_to_string(x: str) -> None:
assert fast_float(x, key=lambda x: x.upper()) == x.upper()
def test_fast_int_leaves_float_string_as_is_example() -> None:
assert fast_int("45.8") == "45.8"
assert fast_int("nan") == "nan"
assert fast_int("inf") == "inf"
@given(floats().filter(not_an_int))
def test_fast_int_leaves_float_string_as_is(x: float) -> None:
assert fast_int(repr(x)) == repr(x)
def test_fast_int_converts_int_string_to_int_example() -> None:
assert fast_int("-45") == -45
assert fast_int("+45") == 45
assert fast_int("۱۲") == 12
assert fast_int("-۱۲") == -12
@given(integers())
def test_fast_int_converts_int_string_to_int(x: int) -> None:
assert fast_int(repr(x)) == x
def test_fast_int_leaves_string_as_is_example() -> None:
assert fast_int("invalid") == "invalid"
@given(text().filter(not_an_int).filter(bool))
def test_fast_int_leaves_string_as_is(x: str) -> None:
assert fast_int(x) == x
def test_fast_int_with_key_applies_to_string_example() -> None:
assert fast_int("invalid", key=lambda x: x.upper()) == "INVALID"
@given(text().filter(not_an_int).filter(bool))
def test_fast_int_with_key_applies_to_string(x: str) -> None:
assert fast_int(x, key=lambda x: x.upper()) == x.upper()
|
