Python decorators

Timeit decorator

import time

def timeit(method):
    def timed(*args, **kw):
        start_time = time.time()
        result = method(*args, **kw)
        end_time = time.time()

        print(f'{method.__name__}  {(end_time - start_time) * 1000} ms')
        return result
    return timed

Usage

@timeit
def test_func(arg1: int)->None:
    time.sleep(arg1)