import time
from contextlib import contextmanager

@contextmanager
def benchmark():
    start = time.time()
    try:
        yield
    finally:
        print(time.time() - start)


with benchmark():
    time.sleep(1)