timeit decorator

timeit.py

from contextlib import contextmanager
import time

@contextmanager
def timeit():
    """
    Decorator that prints out how long a function took to run.
    """

    now = time.monotonic()

    try:
        yield
    finally:
        print(f"Function took {time.monotonic() - now}s.")