timeit.py

@contextlib.contextmanager  # Create factory function for *with* context managers
def timeit():
    """Time execution of code, based on time.clock(). Call using a *with* statement!

    Examples:
        with timeit():
            a @ a
    """
    start = time.clock()
    yield  # The decorated func must return a generator-iterator
    print('Operation took: %0.6f seconds!' % (time.clock() - start))