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
|
from itertools import accumulate
from json import dumps
from math import isnan
from hypothesis import given, settings
from hypothesis.strategies import integers
from pytest import raises
from test_hypotheses import json
from partial_json_parser import *
from partial_json_parser.core.options import *
def test_str():
assert parse_json('"', STR) == ""
with raises(PartialJSON):
parse_json('"', ~STR)
assert parse_json(r'"\\') == "\\"
assert parse_json(r'"\\u') == "\\u"
assert parse_json(r'"\\U\\u') == "\\U\\u"
def test_arr():
assert parse_json('["', ARR) == []
assert parse_json('["', ARR | STR) == [""]
with raises(PartialJSON):
parse_json("[", STR)
with raises(PartialJSON):
parse_json('["', STR)
with raises(PartialJSON):
parse_json('[""', STR)
with raises(PartialJSON):
parse_json('["",', STR)
def test_obj():
assert parse_json('{"": "', OBJ) == {}
assert parse_json('{"": "', OBJ | STR) == {"": ""}
with raises(PartialJSON):
parse_json("{", STR)
with raises(PartialJSON):
parse_json('{"', STR)
with raises(PartialJSON):
parse_json('{""', STR)
with raises(PartialJSON):
parse_json('{"":', STR)
with raises(PartialJSON):
parse_json('{"":"', STR)
with raises(PartialJSON):
parse_json('{"":""', STR)
def test_singletons():
assert parse_json("n", NULL) is None
with raises(PartialJSON):
parse_json("n", ~NULL)
assert parse_json("t", BOOL) == True
with raises(PartialJSON):
parse_json("t", ~BOOL)
assert parse_json("f", BOOL) == False
with raises(PartialJSON):
parse_json("f", ~BOOL)
assert parse_json("I", INF) == float("inf")
with raises(PartialJSON):
parse_json("I", ~INFINITY)
assert parse_json("-I", INF) == float("-inf")
with raises(PartialJSON):
parse_json("-I", ~_INFINITY)
assert isnan(parse_json("N", NAN)) # type: ignore
with raises(PartialJSON):
parse_json("N", ~NAN)
def test_num():
assert parse_json("0", ~NUM) == 0
assert parse_json("-1.25e+4", ~NUM) == -1.25e4
assert parse_json("-1.25e+", NUM) == -1.25
assert parse_json("-1.25e", NUM) == -1.25
def test_error():
with raises(MalformedJSON):
parse_json("a")
with raises(MalformedJSON):
parse_json("{0")
with raises(MalformedJSON):
parse_json("--")
def test_fix():
assert fix("[") == fix_fast("[") == ("[", "]")
assert fix("[0.") == fix_fast("[0.") == ("[0", "]")
assert fix('{"key": ') == fix_fast('{"key": ') == ("{", "}")
assert fix("t") == fix_fast("t") == ("", "true")
assert fix("[1", ~NUM) == fix_fast("[1", ~NUM) == ("[", "]")
assert fix("1", ~NUM) == fix_fast("1", ~NUM) == ("1", "")
with raises(PartialJSON):
fix("-")
def consistent(json_string, allow):
try:
res = fix(json_string, allow)
return res == fix_fast(json_string, allow)
except PartialJSON as err:
with raises(PartialJSON, match=str(err)):
fix_fast(json_string, allow)
return True
def test_consistency():
dict_example = {"key1": 123, "key2": "value"}
list_example = [1, 2, None, float("inf"), float("-inf"), float("nan"), True, False, "string", dict_example]
dict_json = dumps(dict_example)
list_json = dumps(list_example)
for json_string in (*accumulate(dict_json), *accumulate(list_json)):
for allow in range(ALL + 1):
assert consistent(json_string, allow), f"{Allow(allow)!r} - {json_string}"
@settings(deadline=None)
@given(json.map(dumps), integers(0, ALL).map(Allow))
def test_consistencies(json_string, allow):
for json_string in accumulate(json_string):
assert consistent(json_string, allow), f"{Allow(allow)!r} - {json_string}"
|
