47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
|
import sys
|
||
|
|
import re
|
||
|
|
|
||
|
|
from numpy.testing import jiffies, memusage
|
||
|
|
|
||
|
|
|
||
|
|
def cmdline():
|
||
|
|
m = re.compile(r'\A\d+\Z')
|
||
|
|
args = []
|
||
|
|
repeat = 1
|
||
|
|
for a in sys.argv[1:]:
|
||
|
|
if m.match(a):
|
||
|
|
repeat = eval(a)
|
||
|
|
else:
|
||
|
|
args.append(a)
|
||
|
|
f2py_opts = ' '.join(args)
|
||
|
|
return repeat, f2py_opts
|
||
|
|
|
||
|
|
|
||
|
|
def run(runtest, test_functions, repeat=1):
|
||
|
|
l = [(t, repr(t.__doc__.split('\n')[1].strip())) for t in test_functions]
|
||
|
|
start_memusage = memusage()
|
||
|
|
diff_memusage = None
|
||
|
|
start_jiffies = jiffies()
|
||
|
|
i = 0
|
||
|
|
while i < repeat:
|
||
|
|
i += 1
|
||
|
|
for t, fname in l:
|
||
|
|
runtest(t)
|
||
|
|
if start_memusage is None:
|
||
|
|
continue
|
||
|
|
if diff_memusage is None:
|
||
|
|
diff_memusage = memusage() - start_memusage
|
||
|
|
else:
|
||
|
|
diff_memusage2 = memusage() - start_memusage
|
||
|
|
if diff_memusage2 != diff_memusage:
|
||
|
|
print('memory usage change at step %i:' % i,
|
||
|
|
diff_memusage2 - diff_memusage,
|
||
|
|
fname)
|
||
|
|
diff_memusage = diff_memusage2
|
||
|
|
current_memusage = memusage()
|
||
|
|
print('run', repeat * len(test_functions), 'tests',
|
||
|
|
'in %.2f seconds' % ((jiffies() - start_jiffies) / 100.0))
|
||
|
|
if start_memusage:
|
||
|
|
print('initial virtual memory size:', start_memusage, 'bytes')
|
||
|
|
print('current virtual memory size:', current_memusage, 'bytes')
|