simple memory+time benchmark

bench.py

def bench(function, iterations):
    import time
    elapsed_times = []
    
    import tracemalloc
    tracemalloc.start()

    for _ in range(iterations):
        prev_snap = tracemalloc.take_snapshot()
        prev_time = time.time()
        function()
        next_time = time.time()
        next_snap = tracemalloc.take_snapshot()

        elapsed_times.append(next_time - prev_time)

        top_stat = next_snap.compare_to(prev_snap, 'lineno')[0]
        print()
        print(top_stat)
        for line in top_stat.traceback.format():
            print(line)
    
    import matplotlib.pyplot as plt
    plt.bar(range(iterations), elapsed_times)
    plt.xlabel("iteration")
    plt.ylabel("seconds elapsed")
    
bench(lambda: print("X"), iterations=5)