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
|
import sys
from timeit import Timer
DECODE_TESTS = [
('json', 'loads', 'dumps'),
('yajl', 'loads', 'dumps'),
('nssjson', 'loads', 'dumps'),
('simplejson', 'loads', 'dumps'),
('ujson', 'loads', 'dumps'),
('cjson', 'decode', 'encode'),
('mujson', 'loads', 'dumps')
]
ENCODE_TESTS = [
('simplejson', 'loads', 'dumps'),
('nssjson', 'loads', 'dumps'),
('cjson', 'decode', 'encode'),
('json', 'loads', 'dumps'),
('yajl', 'loads', 'dumps'),
('ujson', 'loads', 'dumps'),
('mujson', 'loads', 'dumps')
]
DECODE_ENCODE_TESTS = [
('json', 'loads', 'dumps'),
('yajl', 'loads', 'dumps'),
('nssjson', 'loads', 'dumps'),
('simplejson', 'loads', 'dumps'),
('ujson', 'loads', 'dumps'),
('cjson', 'decode', 'encode'),
('mujson', 'loads', 'dumps')
]
ACTIONS = {
'decoded': 'loads(bin)',
'encoded': 'dumps(obj)',
'de/encoded': 'dumps(loads(bin))'
}
ASCII_BREAK = '\n' + '*' * 75 + '\n'
_import_tpl = 'from {} import ({} as loads, {} as dumps)'
_bin_tmpl = "bin = open('json/{}', 'rb').read()"
_obj = "obj = loads(bin)"
_warm_up = "loads(dumps([]))"
def timeit(module, action, num, jsn, import_stmt):
setup = [import_stmt, _warm_up, _bin_tmpl.format(jsn), _obj]
try:
result = Timer(ACTIONS[action], '; '.join(setup)).timeit(num) * 1000
print('{:25s} {} {} {} times in {} milliseconds.'.format(
module, action, jsn, num, result))
except:
raise
print('{:25s} threw Exception trying to {} {} {} times.'.format(
module, action, jsn, num))
def main(number, jsn):
print(ASCII_BREAK)
for module, decoder, encoder in DECODE_TESTS:
import_stmt = _import_tpl.format(module, decoder, encoder)
timeit(module, 'decoded', number, jsn, import_stmt)
print(ASCII_BREAK)
for module, decoder, encoder in ENCODE_TESTS:
import_stmt = _import_tpl.format(module, decoder, encoder)
timeit(module, 'encoded', number, jsn, import_stmt)
print(ASCII_BREAK)
for module, decoder, encoder in DECODE_ENCODE_TESTS:
import_stmt = _import_tpl.format(module, decoder, encoder)
timeit(module, 'de/encoded', number, jsn, import_stmt)
print(ASCII_BREAK)
if __name__ == '__main__':
if len(sys.argv) == 3:
main(int(sys.argv[1]), sys.argv[2])
|
