一个简单的装饰器算出函数运行时间

程序猿在做优化时经常会计算一个函数的用时，从而针对某几个函数进行优化，下面我就用一个简单的装饰器来实现计算函数运行时间：

import time


def set_fun(func):
    def call_fun(*args, **kwargs):
        start_time = time.time()
        func(*args, **kwargs)
        end_time = time.time()
        print('程序用时：%s秒' % int(end_time - start_time))
    return call_fun


@set_fun
def test(*args, **kwargs):
    sum = 0
    for i in range(1000):
        for j in range(100):
            for k in range(100):
                sum += i * j * k

print('test')

test()

看一下运行的界面：