python timeit function with arguments code example

Example 1: how to use timeit in python 3

import timeit
import_module = "import random"
testcode = ''' 
def test(): 
    return random.randint(10, 100)
'''
print(timeit.repeat(stmt=testcode, setup=import_module))

Example 2: python timeit function return value

import time

# Function with both args and kwargs ====
def time_fn( fn, *args, **kwargs ):
    start = time.perf_counter()
    results = fn( *args, **kwargs )
    end = time.perf_counter()
    print(fn.__name__ + ": " + str(end-start) + "s")
    return results

res = time_fn(function_name, input, output)

# Function with only arguments ====
def time_fn(fn, *args):
    start = time.perf_counter()
    results = fn(*args)
    end = time.perf_counter()
    print(fn.__name__ + ": " + str(end - start) + "s")
    return results

res = time_fn(function_name, input)