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