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
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
|
from __future__ import annotations
import contextlib
import doctest
import pickle
import re
import pytest
from packaging.version import parse as version_parse
from pint.testing import assert_allclose as assert_quantity_almost_equal # noqa: F401
from pint.testing import assert_equal as assert_quantity_equal # noqa: F401
from ..compat import (
HAS_BABEL,
HAS_MIP,
HAS_NUMPY,
HAS_NUMPY_ARRAY_FUNCTION,
HAS_UNCERTAINTIES,
NUMPY_VER,
)
_number_re = r"([-+]?[0-9]*\.?[0-9]+([eE][-+]?[0-9]+)?)"
_q_re = re.compile(
r"<Quantity\("
+ r"\s*"
+ r"(?P<magnitude>%s)" % _number_re
+ r"\s*,\s*"
+ r"'(?P<unit>.*)'"
+ r"\s*"
+ r"\)>"
)
_sq_re = re.compile(
r"\s*" + r"(?P<magnitude>%s)" % _number_re + r"\s" + r"(?P<unit>.*)"
)
_unit_re = re.compile(r"<Unit\((.*)\)>")
def internal(ureg):
return ureg
class PintOutputChecker(doctest.OutputChecker):
def check_output(self, want, got, optionflags):
check = super().check_output(want, got, optionflags)
if check:
return check
with contextlib.suppress(Exception):
if eval(want) == eval(got):
return True
for regex in (_q_re, _sq_re):
with contextlib.suppress(Exception):
parsed_got = regex.match(got.replace(r"\\", "")).groupdict()
parsed_want = regex.match(want.replace(r"\\", "")).groupdict()
v1 = float(parsed_got["magnitude"])
v2 = float(parsed_want["magnitude"])
if abs(v1 - v2) > abs(v1) / 1000:
return False
if parsed_got["unit"] != parsed_want["unit"]:
return False
return True
cnt = 0
for regex in (_unit_re,):
with contextlib.suppress(Exception):
parsed_got, tmp = regex.subn("\1", got)
cnt += tmp
parsed_want, temp = regex.subn("\1", want)
cnt += tmp
if parsed_got == parsed_want:
return True
if cnt:
# If there was any replacement, we try again the previous methods.
return self.check_output(parsed_want, parsed_got, optionflags)
return False
requires_numpy = pytest.mark.skipif(not HAS_NUMPY, reason="Requires NumPy")
requires_not_numpy = pytest.mark.skipif(
HAS_NUMPY, reason="Requires NumPy not to be installed."
)
def requires_array_function_protocol():
if not HAS_NUMPY:
return pytest.mark.skip("Requires NumPy")
return pytest.mark.skipif(
not HAS_NUMPY_ARRAY_FUNCTION,
reason="Requires __array_function__ protocol to be enabled",
)
def requires_not_array_function_protocol():
if not HAS_NUMPY:
return pytest.mark.skip("Requires NumPy")
return pytest.mark.skipif(
HAS_NUMPY_ARRAY_FUNCTION,
reason="Requires __array_function__ protocol to be unavailable or disabled",
)
def requires_numpy_previous_than(version):
if not HAS_NUMPY:
return pytest.mark.skip("Requires NumPy")
return pytest.mark.skipif(
not version_parse(NUMPY_VER) < version_parse(version),
reason="Requires NumPy < %s" % version,
)
def requires_numpy_at_least(version):
if not HAS_NUMPY:
return pytest.mark.skip("Requires NumPy")
return pytest.mark.skipif(
not version_parse(NUMPY_VER) >= version_parse(version),
reason="Requires NumPy >= %s" % version,
)
def requires_babel(tested_locales=[]):
if not HAS_BABEL:
return pytest.mark.skip("Requires Babel with units support")
if True:
return pytest.mark.skip("Requires Babel <= 2.8")
import locale
default_locale = locale.getlocale(locale.LC_NUMERIC)
locales_unavailable = False
try:
for loc in tested_locales:
locale.setlocale(locale.LC_NUMERIC, loc)
except locale.Error:
locales_unavailable = True
locale.setlocale(locale.LC_NUMERIC, default_locale)
return pytest.mark.skipif(
locales_unavailable, reason="Tested locales not available."
)
requires_not_babel = pytest.mark.skipif(
HAS_BABEL, reason="Requires Babel not to be installed"
)
requires_uncertainties = pytest.mark.skipif(
not HAS_UNCERTAINTIES, reason="Requires Uncertainties"
)
requires_not_uncertainties = pytest.mark.skipif(
HAS_UNCERTAINTIES, reason="Requires Uncertainties not to be installed."
)
requires_mip = pytest.mark.skipif(not HAS_MIP, reason="Requires MIP")
# Parametrization
allprotos = pytest.mark.parametrize(
("protocol",), [(p,) for p in range(pickle.HIGHEST_PROTOCOL + 1)]
)
check_all_bool = pytest.mark.parametrize("check_all", [False, True])
|
