Python: calcula el tiempo de ejecución de un fragmento de código

import time
'''
# 例一
start = time.time()
x = 0
for i in range(0,199999999):
    x += i
print(x)
end = time.time()
print('例一的代码执行时长为：',round(end-start),'秒')

# 例二
start = time.time()
time.sleep(3)
end = time.time()
print('例二的代码执行时长为：',round(end-start),'秒')
'''
# 优化
def call_time(fn):
    start = time.time()
    fn()
    end = time.time()
    print('例一的代码执行时长为：', round(end - start), '秒')

def test1():
    x = 0
    for i in range(0,199999999):
        x += i
    print(x)

def test2():
    time.sleep(3)

call_time(test1)
call_time(test2)

# 执行结果：
# 19999999700000001
# 例一的代码执行时长为： 10 秒
# 例一的代码执行时长为： 3 秒