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)