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
|
# This file is part of Hypothesis, which may be found at
# https://github.com/HypothesisWorks/hypothesis/
#
# Copyright the Hypothesis Authors.
# Individual contributors are listed in AUTHORS.rst and the git log.
#
# This Source Code Form is subject to the terms of the Mozilla Public License,
# v. 2.0. If a copy of the MPL was not distributed with this file, You can
# obtain one at https://mozilla.org/MPL/2.0/.
import math
import sys
import pytest
from hypothesis import given, reject, strategies as st
from hypothesis.errors import InvalidArgument
from hypothesis.strategies import complex_numbers
from tests.common.debug import assert_no_examples, find_any, minimal
def test_minimal():
assert minimal(complex_numbers(), lambda x: True) == 0
def test_minimal_nonzero_real():
assert minimal(complex_numbers(), lambda x: x.real != 0) == 1
def test_minimal_nonzero_imaginary():
assert minimal(complex_numbers(), lambda x: x.imag != 0) == 1j
def test_minimal_quadrant1():
assert minimal(complex_numbers(), lambda x: x.imag > 0 and x.real > 0) == 1 + 1j
def test_minimal_quadrant2():
assert minimal(complex_numbers(), lambda x: x.imag > 0 and x.real < 0) == -1 + 1j
def test_minimal_quadrant3():
assert minimal(complex_numbers(), lambda x: x.imag < 0 and x.real < 0) == -1 - 1j
def test_minimal_quadrant4():
assert minimal(complex_numbers(), lambda x: x.imag < 0 and x.real > 0) == 1 - 1j
@given(st.data(), st.integers(-5, 5).map(lambda x: 10**x))
def test_max_magnitude_respected(data, mag):
c = data.draw(complex_numbers(max_magnitude=mag))
assert abs(c) <= mag * (1 + sys.float_info.epsilon)
@given(complex_numbers(max_magnitude=0))
def test_max_magnitude_zero(val):
assert val == 0
@given(st.data(), st.integers(-5, 5).map(lambda x: 10**x))
def test_min_magnitude_respected(data, mag):
c = data.draw(complex_numbers(min_magnitude=mag))
assert (
abs(c.real) >= mag
or abs(c.imag) >= mag
or abs(c) >= mag * (1 - sys.float_info.epsilon)
)
def test_minimal_min_magnitude_zero():
assert minimal(complex_numbers(min_magnitude=0), lambda x: True) == 0
def test_minimal_min_magnitude_positive():
assert minimal(complex_numbers(min_magnitude=0.5), lambda x: True) in (0.5, 1)
def test_minimal_minmax_magnitude():
assert minimal(
complex_numbers(min_magnitude=0.5, max_magnitude=1.5), lambda x: True
) in (0.5, 1)
@given(st.data(), st.floats(0, 10e300, allow_infinity=False, allow_nan=False))
def test_minmax_magnitude_equal(data, mag):
val = data.draw(st.complex_numbers(min_magnitude=mag, max_magnitude=mag))
try:
# Cap magnitude at 10e300 to avoid float overflow, and imprecision
# at very large exponents (which makes math.isclose fail)
assert math.isclose(abs(val), mag)
except OverflowError:
reject()
def _is_subnormal(x):
return 0 < abs(x) < sys.float_info.min
@pytest.mark.parametrize(
"allow_subnormal, min_magnitude, max_magnitude",
[
(True, 0, None),
(True, 1, None),
(False, 0, None),
],
)
def test_allow_subnormal(allow_subnormal, min_magnitude, max_magnitude):
strat = complex_numbers(
min_magnitude=min_magnitude,
max_magnitude=max_magnitude,
allow_subnormal=allow_subnormal,
).filter(lambda x: x.real != 0 and x.imag != 0)
if allow_subnormal:
find_any(strat, lambda x: _is_subnormal(x.real) or _is_subnormal(x.imag))
else:
assert_no_examples(
strat, lambda x: _is_subnormal(x.real) or _is_subnormal(x.imag)
)
@pytest.mark.parametrize("allow_subnormal", [1, 0.0, "False"])
def test_allow_subnormal_validation(allow_subnormal):
with pytest.raises(InvalidArgument):
complex_numbers(allow_subnormal=allow_subnormal).example()
|
