Repository navigation
Expand file tree
/
Copy pathtest_base_types.py
More file actions
241 lines (176 loc) · 6.04 KB
/
Copy pathtest_base_types.py
File metadata and controls
241 lines (176 loc) · 6.04 KB
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
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
# -*- coding: utf-8 -*-
# :Project: python-rapidjson -- Basic tests
# :Author: John Anderson <sontek@gmail.com>
# :License: MIT License
# :Copyright: © 2015 John Anderson
# :Copyright: © 2016, 2017, 2018, 2019, 2020, 2021, 2024, 2026 Lele Gaifax
#
import random
import sys
import pytest
import rapidjson as rj
@pytest.mark.parametrize(
'value', (
'A', 'cruel\x00world',
1, -1,
2.3, -36.973846435546875, 1514893636.276703,
{'foo': 'bar', '\x00': 'issue57', 'issue57': '\x00'},
[1, 2, 'a', 1.2, {'foo': 'bar'},],
sys.maxsize, sys.maxsize**2, 10**1000,
))
def test_base_values(value, dumps, loads):
dumped = dumps(value)
loaded = loads(dumped)
assert loaded == value and type(loaded) is type(value)
def test_float(dumps):
value = 0.1 + 0.2
dumped = dumps(value)
assert dumped == '0.30000000000000004'
def test_tuple(dumps):
obj = [1, 2, 'a', 1.2, {'foo': 'bar'},]
assert dumps(obj) == dumps(tuple(obj))
def test_bytes_value(dumps):
value = b'cruel\x00world'
dumped = dumps(value)
assert dumped == r'"cruel\u0000world"'
dumped = dumps(bytearray(value))
assert dumped == r'"cruel\u0000world"'
def test_larger_structure(dumps, loads):
value = {
'words': """
Lorem ipsum dolor sit amet, consectetur adipiscing
elit. Mauris adipiscing adipiscing placerat.
Vestibulum augue augue,
pellentesque quis sollicitudin id, adipiscing.
""",
'list': list(range(200)),
'dict': dict((str(i),'a') for i in range(200)),
'int': 100100100,
'float': 100999.123456
}
dumped = dumps(value)
loaded = loads(dumped)
assert loaded == value
def test_object_hook():
def as_complex(dct):
if '__complex__' in dct:
return complex(dct['real'], dct['imag'])
return dct
result = rj.loads(
'{"__complex__": true, "real": 1, "imag": 2}',
object_hook=as_complex
)
assert result == (1+2j)
def test_end_object():
class ComplexDecoder(rj.Decoder):
def end_object(self, dct):
if '__complex__' in dct:
return complex(dct['real'], dct['imag'])
return dct
loads = ComplexDecoder()
result = loads('{"__complex__": true, "real": 1, "imag": 2}')
assert result == (1+2j)
def test_dumps_default():
def encode_complex(obj):
if isinstance(obj, complex):
return [obj.real, obj.imag]
raise TypeError(repr(obj) + " is not JSON serializable")
result = rj.dumps(2 + 1j, default=encode_complex)
assert result == '[2.0,1.0]'
def test_dumps_default_issue238():
lst = [object() for _ in range(256)]
def default(o):
del lst[128:]
return None
result = rj.dumps(lst, default=default)
assert len(rj.loads(result)) == 128
lst = [object() for _ in range(128)]
def default(o):
if len(lst) < 256:
lst.append(object())
return None
result = rj.dumps(lst, default=default)
assert len(rj.loads(result)) == 256
def test_encoder_default():
class ComplexEncoder(rj.Encoder):
def default(self, obj):
if isinstance(obj, complex):
return [obj.real, obj.imag]
raise TypeError(repr(obj) + " is not JSON serializable")
dumps = ComplexEncoder()
result = dumps(2 + 1j)
assert result == '[2.0,1.0]'
def test_doubles(dumps, loads):
for _ in range(100000):
d = sys.maxsize * random.random()
dumped = dumps(d)
loaded = loads(dumped)
assert loaded == d
def test_unicode(dumps, loads):
arabic='بينهم ان يكون مسلما رشيدا عاقلا ًوابنا شرعيا لابوين عمانيين'
chinese='本站所提供的資料和服務都不收費，因此網站所需要的資金全來自廣告及捐款。若您願意捐款補助'
for text in (arabic, chinese):
dumped = dumps(text)
loaded = loads(dumped)
assert text == loaded
def test_serialize_sets_dumps():
def default_iterable(obj):
if isinstance(obj, set):
return list(obj)
raise TypeError(repr(obj) + " is not JSON serializable")
rj.dumps([set()], default=default_iterable)
with pytest.raises(TypeError):
rj.dumps([set()])
def test_serialize_sets_encoder():
class SetsEncoder(rj.Encoder):
def default(self, obj):
if isinstance(obj, set):
return list(obj)
raise TypeError(repr(obj) + " is not JSON serializable")
dumps = SetsEncoder()
dumps([set()])
with pytest.raises(TypeError):
rj.Encoder()([set()])
def test_constants(dumps, loads):
for c in [None, True, False]:
assert loads(dumps(c)) is c
assert loads(dumps([c]))[0] is c
assert loads(dumps({'a': c}))['a'] is c
def test_iterables(dumps):
assert dumps(iter("abc")) == '["a","b","c"]'
def gen():
yield 1
yield 2
yield 3
assert dumps(gen()) == '[1,2,3]'
def test_decode_error(loads):
pytest.raises(rj.JSONDecodeError, loads, '{')
def test_decode_bytes_and_bytearray():
s = 'FòBàr'
j = f'"{s}"'
utf32 = j.encode('utf-32')
for loader in (rj.loads, rj.Decoder()):
pytest.raises(UnicodeDecodeError, loader, utf32)
utf8 = j.encode('utf-8')
for loader in (rj.loads, rj.Decoder()):
for data in (utf8, bytearray(utf8)):
assert loader(data) == s
def test_shared_keys(loads):
res = loads('[{"key": "value1"}, {"key": "value2"}]')
key1, = res[0].keys()
key2, = res[1].keys()
assert key1 is key2
# TODO: Figure out what we want to do here
bad_tests = """
def test_true_false():
dumped1 = sorted(rj.dumps({True: False, False: True}))
dumped2 = sorted(rj.dumps({
2: 3.0,
4.0: long_type(5),
False: 1,
long_type(6): True,
"7": 0
}))
assert dumped1 == '{"false": true, "true": false}'
expected = '{"2": 3.0, "4.0": 5, "6": true, "7": 0, "false": 1}'
assert dumped2 == expected"""