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
|
import json
import pytest
from dirty_equals import IsStrictDict
from pydantic_core import SchemaSerializer, core_schema
def test_dict_str_int():
v = SchemaSerializer(core_schema.dict_schema(core_schema.str_schema(), core_schema.int_schema()))
assert v.to_python({'a': 1, 'b': 2, 'c': 3}) == {'a': 1, 'b': 2, 'c': 3}
assert v.to_python({'a': 1, 'b': 2, 'c': 3}, mode='json') == {'a': 1, 'b': 2, 'c': 3}
assert v.to_json({'a': 1, 'b': 2, 'c': 3}) == b'{"a":1,"b":2,"c":3}'
assert v.to_json({'a': 1, 'b': 2, 'c': 3}, indent=2) == b'{\n "a": 1,\n "b": 2,\n "c": 3\n}'
def test_dict_any_any():
v = SchemaSerializer(core_schema.dict_schema())
assert v.to_python({'a': 1, b'b': 2, 33: 3}) == {'a': 1, b'b': 2, 33: 3}
assert v.to_python({'a': 1, b'b': 2, 33: 3, True: 4}, mode='json') == {'a': 1, 'b': 2, '33': 3, 'true': 4}
assert v.to_json({'a': 1, b'b': 2, 33: 3, True: 4}) == b'{"a":1,"b":2,"33":3,"true":4}'
assert v.to_python({(1, 2): 3}) == {(1, 2): 3}
assert v.to_python({(1, 2): 3}, mode='json') == {'1,2': 3}
assert v.to_json({(1, 2): 3}) == b'{"1,2":3}'
def test_include():
s = SchemaSerializer(core_schema.dict_schema(serialization=core_schema.filter_dict_schema(include={'a', 'c'})))
assert s.to_python({'a': 1, 'b': 2, 'c': 3, 'd': 4}) == {'a': 1, 'c': 3}
assert s.to_json({'a': 1, 'b': 2, 'c': 3, 'd': 4}) == b'{"a":1,"c":3}'
assert s.to_python({'a': 1, 'b': 2, 'd': 4}, include={'d'}) == {'a': 1, 'd': 4}
assert s.to_python({'a': 1, 'b': 2, 'd': 4}, include={'d': None}) == {'a': 1, 'd': 4}
assert s.to_python({'a': 1, 'b': 2, 'd': 4}, include={'d': {1}}) == {'a': 1, 'd': 4}
assert s.to_python({'a': 1, 'b': 2, 'd': 4, 5: 6}, include={5}) == {'a': 1, 5: 6}
assert s.to_python({'a': 1, 'b': 2, 'd': 4, 5: 6}, mode='json', include={5}) == {'a': 1, '5': 6}
assert s.to_json({'a': 1, 'b': 2, 'd': 4, 5: 6}, include={5}) == b'{"a":1,"5":6}'
def test_exclude():
s = SchemaSerializer(core_schema.dict_schema(serialization=core_schema.filter_dict_schema(exclude={'a', 'c'})))
assert s.to_python({'a': 1, 'b': 2, 'c': 3, 'd': 4}) == {'b': 2, 'd': 4}
assert s.to_json({'a': 1, 'b': 2, 'c': 3, 'd': 4}) == b'{"b":2,"d":4}'
assert s.to_python({'a': 1, 'b': 2, 'c': 3, 'd': 4}, exclude={'d'}) == {'b': 2}
assert s.to_python({'a': 1, 'b': 2, 'c': 3, 'd': 4}, exclude={'__all__'}) == {}
assert s.to_python({'a': 1, 'b': 2, 'c': 3, 'd': 4}, exclude={'d': ...}) == {'b': 2}
assert s.to_python({'a': 1, 'b': 2, 'c': 3, 'd': 4}, exclude={'d': {1}}) == {'b': 2, 'd': 4}
assert s.to_json({'a': 1, 'b': 2, 'c': 3, 'd': 4}, exclude={'d'}) == b'{"b":2}'
def test_filter():
s = SchemaSerializer(
core_schema.dict_schema(
serialization=core_schema.filter_dict_schema(include={'1', '3', '5'}, exclude={'5', '6'})
)
)
assert s.to_python({'0': 0, '1': 1, '2': 2, '3': 3, '4': 4, '5': 5, '6': 6, '7': 7}) == {'1': 1, '3': 3}
@pytest.mark.parametrize(
'params',
[
dict(include=None, exclude=None, expected={'0': 0, '1': 1, '2': 2, '3': 3}),
dict(include={'0', '1'}, exclude=None, expected={'0': 0, '1': 1}),
dict(include={'0': ..., '1': ...}, exclude=None, expected={'0': 0, '1': 1}),
dict(include={'0': {1}, '1': {1}}, exclude=None, expected={'0': 0, '1': 1}),
dict(include=None, exclude={'0', '1'}, expected={'2': 2, '3': 3}),
dict(include=None, exclude={'0': ..., '1': ...}, expected={'2': 2, '3': 3}),
dict(include={'0', '1'}, exclude={'1', '2'}, expected={'0': 0}),
dict(include=None, exclude={'3': {1}}, expected={'0': 0, '1': 1, '2': 2, '3': 3}),
dict(include={'0', '1'}, exclude={'3': {1}}, expected={'0': 0, '1': 1}),
dict(include={'0', '1'}, exclude={'1': {1}}, expected={'0': 0, '1': 1}),
dict(include={'0', '1'}, exclude={'1': ...}, expected={'0': 0}),
dict(include=None, exclude={'__all__'}, expected={}),
],
)
def test_filter_args(params):
s = SchemaSerializer(core_schema.dict_schema())
# user IsStrictDict to check dict order
include, exclude, expected = params['include'], params['exclude'], IsStrictDict(params['expected'])
value = {'0': 0, '1': 1, '2': 2, '3': 3}
assert s.to_python(value, include=include, exclude=exclude) == expected
assert s.to_python(value, mode='json', include=include, exclude=exclude) == expected
assert json.loads(s.to_json(value, include=include, exclude=exclude)) == expected
@pytest.mark.parametrize(
'params',
[
dict(include=None, exclude=None, expected={'0': [0], '1': [0, 1], '2': [0, 1, 2], '3': [0, 1, 2, 3]}),
dict(include=None, exclude={'1': {0}}, expected={'0': [0], '1': [1], '2': [0, 1, 2], '3': [0, 1, 2, 3]}),
dict(include={'1': {0}}, exclude=None, expected={'1': [0]}),
dict(include={'__all__': {0}}, exclude=None, expected={'0': [0], '1': [0], '2': [0], '3': [0]}),
dict(
include=None, exclude={'0': {'__all__'}}, expected={'0': [], '1': [0, 1], '2': [0, 1, 2], '3': [0, 1, 2, 3]}
),
dict(include=None, exclude={'__all__': {'__all__'}}, expected={'0': [], '1': [], '2': [], '3': []}),
dict(include=None, exclude={'__all__': {0}}, expected={'0': [], '1': [1], '2': [1, 2], '3': [1, 2, 3]}),
dict(include=None, exclude={'__all__': {0}, '3': {1}}, expected={'0': [], '1': [1], '2': [1, 2], '3': [2, 3]}),
],
)
def test_filter_args_nested(params):
s = SchemaSerializer(core_schema.dict_schema(core_schema.str_schema(), core_schema.list_schema()))
include, exclude, expected = params['include'], params['exclude'], params['expected']
value = {'0': [0], '1': [0, 1], '2': [0, 1, 2], '3': [0, 1, 2, 3]}
assert s.to_python(value, include=include, exclude=exclude) == expected
assert s.to_python(value, mode='json', include=include, exclude=exclude) == expected
assert json.loads(s.to_json(value, include=include, exclude=exclude)) == expected
def test_filter_int():
s = SchemaSerializer(
core_schema.dict_schema(
core_schema.any_schema(), serialization=core_schema.filter_dict_schema(include={1, 3, 5}, exclude={5, 6})
)
)
assert s.to_python({0: 0, 1: 1, 2: 2, 3: 3, 4: 4, 5: 5, 6: 6, 7: 7}) == {1: 1, 3: 3}
def test_filter_runtime():
s = SchemaSerializer(
core_schema.dict_schema(
core_schema.any_schema(), serialization=core_schema.filter_dict_schema(exclude={'0', '1'})
)
)
assert s.to_python({'0': 0, '1': 1, '2': 2, '3': 3}, include={'1', '2'}) == {'1': 1, '2': 2}
assert s.to_python({'0': 0, '1': 1, '2': 2, '3': 3}, include={'1', '2'}, exclude={'2', '3'}) == {'1': 1}
def test_filter_runtime_int():
s = SchemaSerializer(
core_schema.dict_schema(core_schema.any_schema(), serialization=core_schema.filter_dict_schema(exclude={0, 1}))
)
assert s.to_python({0: 0, 1: 1, 2: 2, 3: 3}, include={1, 2}) == {1: 1, 2: 2}
|
