Time python function

function_timeit.py

from functools import partial
import timeit

def measure_time(n, f, *args, **kwargs):
  """
  Args:
    n (int): number of tests
    f (func): function to time
    *args: any arguments to pass to f
  """
  return timeit.timeit(partial(f, *args, **kwargs), number=n)

# measure_time(100, my_func, input_to_my_func, a_kwarg_to_my_func=1)