| | """Simple tools for timing functions' execution, when IPython is not available. """ |
| |
|
| |
|
| | import timeit |
| | import math |
| |
|
| |
|
| | _scales = [1e0, 1e3, 1e6, 1e9] |
| | _units = ['s', 'ms', '\N{GREEK SMALL LETTER MU}s', 'ns'] |
| |
|
| |
|
| | def timed(func, setup="pass", limit=None): |
| | """Adaptively measure execution time of a function. """ |
| | timer = timeit.Timer(func, setup=setup) |
| | repeat, number = 3, 1 |
| |
|
| | for i in range(1, 10): |
| | if timer.timeit(number) >= 0.2: |
| | break |
| | elif limit is not None and number >= limit: |
| | break |
| | else: |
| | number *= 10 |
| |
|
| | time = min(timer.repeat(repeat, number)) / number |
| |
|
| | if time > 0.0: |
| | order = min(-int(math.floor(math.log10(time)) // 3), 3) |
| | else: |
| | order = 3 |
| |
|
| | return (number, time, time*_scales[order], _units[order]) |
| |
|
| |
|
| | |
| |
|
| | def __do_timings(): |
| | import os |
| | res = os.getenv('SYMPY_TIMINGS', '') |
| | res = [x.strip() for x in res.split(',')] |
| | return set(res) |
| |
|
| | _do_timings = __do_timings() |
| | _timestack = None |
| |
|
| |
|
| | def _print_timestack(stack, level=1): |
| | print('-'*level, '%.2f %s%s' % (stack[2], stack[0], stack[3])) |
| | for s in stack[1]: |
| | _print_timestack(s, level + 1) |
| |
|
| |
|
| | def timethis(name): |
| | def decorator(func): |
| | if name not in _do_timings: |
| | return func |
| |
|
| | def wrapper(*args, **kwargs): |
| | from time import time |
| | global _timestack |
| | oldtimestack = _timestack |
| | _timestack = [func.func_name, [], 0, args] |
| | t1 = time() |
| | r = func(*args, **kwargs) |
| | t2 = time() |
| | _timestack[2] = t2 - t1 |
| | if oldtimestack is not None: |
| | oldtimestack[1].append(_timestack) |
| | _timestack = oldtimestack |
| | else: |
| | _print_timestack(_timestack) |
| | _timestack = None |
| | return r |
| | return wrapper |
| | return decorator |
| |
|
