python练习十：装饰器

python练习十：装饰器

题目

请设计一个decorator，它可作用于任何函数上，并打印该函数的执行时间

代码

import time, functools

# 打印执行时间
def metric(fn): # 装饰器
    def wrapper(*args): # 可变参数
        print('%s executed in %s ms' % (fn.__name__, 10.24))
        return fn(*args)
    return wrapper

# 测试
@metric
def fast(x, y):
    time.sleep(0.0012)
    return x + y;

@metric
def slow(x, y, z):
    time.sleep(0.1234)
    return x * y * z;

f = fast(11, 22)
s = slow(11, 22, 33)
if f != 33:
    print('测试失败!')
elif s != 7986:
    print('测试失败!')
else:
    print('测试成功!')