diff options
Diffstat (limited to 'Lib/timeit.py')
| -rwxr-xr-x | Lib/timeit.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/Lib/timeit.py b/Lib/timeit.py index cf7446d..0b1c601 100755 --- a/Lib/timeit.py +++ b/Lib/timeit.py @@ -65,7 +65,7 @@ default_timer = time.perf_counter # in Timer.__init__() depend on setup being indented 4 spaces and stmt # being indented 8 spaces. template = """ -def inner(_it, _timer): +def inner(_it, _timer{init}): {setup} _t0 = _timer() for _i in _it: @@ -119,9 +119,10 @@ class Timer: stmt = reindent(stmt, 8) if isinstance(setup, str): setup = reindent(setup, 4) - src = template.format(stmt=stmt, setup=setup) + src = template.format(stmt=stmt, setup=setup, init='') elif callable(setup): - src = template.format(stmt=stmt, setup='_setup()') + src = template.format(stmt=stmt, setup='_setup()', + init=', _setup=_setup') ns['_setup'] = setup else: raise ValueError("setup is neither a string nor callable") |