timeit

timeit.py

import time
from pytest import fixture


def timeit(iterations_num=10_000):
    def wrapper(fun):
        def nested_wrapper(*args, **kwargs):
            start = time.time()
            for i in range(iterations_num):
                result = fun(*args, **kwargs)
            stop = time.time()
            total = stop - start
            print(f"total time spend: {total}s")
            return result

        return nested_wrapper

    return wrapper


@timeit()
def fun1(input_list, lambda_fun):
    return list(map(lambda_fun, input_list))


@timeit()
def fun2(input_list, lambda_fun):
    return [str(t + 1) for t in input_list]


@fixture
def input_lambda():
    return lambda t: str(t + 1)


@fixture
def input_list():
    return range(1000)


def test_map(input_list, input_lambda):
    fun1(input_list, input_lambda)
    fun2(input_list, input_lambda)