Python 函数计时器

写了一个简单的函数计时器，用来计算函数执行时间。比较适合需要批量计算许多函数的场景，不然写得太乱了。

import time
from typing import Callable


def timer(function: Callable, *args):
    start = time.perf_counter()
    function(*args)
    end = time.perf_counter()
    print(f'Spend time --{round(end - start, 5)}s-- in function --{function.__name__}--.')


def fun1(t):
    print(t)
    a = 1
    for i in range(t):
        a += 1


def fun2(t, y):
    print(t, y)
    a = 1
    for i in range(t):
        a -= 1
    for i in range(y):
        a -= 1


timer(fun1, 10000000)
timer(fun2, 10000000, 10000000)