python - timeit 吃掉返回值

Question

我想以便宜的方式测量函数的执行时间，如下所示:

def my_timeit(func, *args, **kwargs):
    t0 = time.time()
    result = func(*args, **kwargs)
    delta = time.time() - t0
    return delta, result

def foo():
    time.sleep(1.23)
    return 'potato'

delta, result = my_timeit(foo)

但我想使用 timeit、profile 或其他内置功能来处理由于平台差异而导致的任何常见陷阱，而且可能更好获取实际执行时间而不是墙上时间。

我尝试使用 timeit.Timer(foo).timeit(number=1) 但界面似乎模糊了返回值。

Answer 1

这是我目前的尝试。但我欢迎任何建议，因为这感觉太老套了，可能需要改进。

import time
from timeit import Timer

def my_timeit(func, *args, **kwargs):
    output_container = []
    def wrapper():
        output_container.append(func(*args, **kwargs))
    timer = Timer(wrapper)
    delta = timer.timeit(1)
    return delta, output_container.pop()

def foo():
    time.sleep(1.111)
    return 'potato'

delta, result = my_timeit(foo)

---

编辑: 适合作为以下装饰器工作:

def timeit_decorator(the_func):

    @functools.wraps(the_func)
    def my_timeit(*args, **kwargs):
        output_container = []
        def wrapper():
            output_container.append(the_func(*args, **kwargs))
        timer = Timer(wrapper)
        delta = timer.timeit(1)
        my_timeit.last_execution_time = delta
        return output_container.pop()

    return my_timeit