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
|
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_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\((.*)\)>")
class PintOutputChecker(doctest.OutputChecker):
def check_output(self, want, got, optionflags):
check = super().check_output(want, got, optionflags)
if check:
return check
try:
if eval(want) == eval(got):
return True
except Exception:
pass
for regex in (_q_re, _sq_re):
try:
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
except Exception:
pass
cnt = 0
for regex in (_unit_re,):
try:
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
except Exception:
pass
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,
)
requires_babel = pytest.mark.skipif(
not HAS_BABEL, reason="Requires Babel with units support"
)
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."
)
# 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])
|
