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
|
import math
import pytest
from pydantic_core import SchemaSerializer, core_schema
@pytest.mark.parametrize(
'value,expected',
[
(complex(-1.23e-4, 567.89), '-0.000123+567.89j'),
(complex(0, -1.23), '-1.23j'),
(complex(1.5, 0), '1.5+0j'),
(complex(1, 2), '1+2j'),
(complex(0, 1), '1j'),
(complex(0, 1e-500), '0j'),
(complex(-float('inf'), 2), '-inf+2j'),
(complex(float('inf'), 2), 'inf+2j'),
(complex(float('nan'), 2), 'NaN+2j'),
],
)
def test_complex_json(value, expected):
v = SchemaSerializer(core_schema.complex_schema())
c = v.to_python(value)
c_json = v.to_python(value, mode='json')
json_str = v.to_json(value).decode()
assert c_json == expected
assert json_str == f'"{expected}"'
if math.isnan(value.imag):
assert math.isnan(c.imag)
else:
assert c.imag == value.imag
if math.isnan(value.real):
assert math.isnan(c.real)
else:
assert c.imag == value.imag
def test_complex_inference() -> None:
s = SchemaSerializer(core_schema.any_schema())
assert s.to_python(1 + 2j) == 1 + 2j
assert s.to_json(1 + 2j) == b'"1+2j"'
|
