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
|
from __future__ import division, print_function, absolute_import
from numpy.testing import *
from io import BytesIO
import numpy as np
import scipy.io as sio
def make_structarr(n_vars, n_fields, n_structs):
var_dict = {}
for vno in range(n_vars):
vname = 'var%00d' % vno
end_dtype = [('f%d' % d, 'i4', 10) for d in range(n_fields)]
s_arrs = np.zeros((n_structs,), dtype=end_dtype)
var_dict[vname] = s_arrs
return var_dict
def bench_run():
str_io = BytesIO()
print()
print('Read / writing matlab structs')
print('='*60)
print(' write | read | vars | fields | structs | compressed')
print('-'*60)
print()
for n_vars, n_fields, n_structs in (
(10, 10, 20), (20, 20, 40), (30, 30, 50)):
var_dict = make_structarr(n_vars, n_fields, n_structs)
for compression in (False, True):
str_io = BytesIO()
write_time = measure('sio.savemat(str_io, var_dict, do_compression=%r)' % compression)
read_time = measure('sio.loadmat(str_io)')
print('%.5f | %.5f | %5d | %5d | %5d | %r' % (
write_time,
read_time,
n_vars,
n_fields,
n_structs,
compression))
if __name__ == '__main__':
bench_run()
|
