test.py

import cProfile, pstats, io
from pstats import SortKey


def slow_sum(size):
    x= 0
    for i in range(size):
        for j in range(size):
            x += i + j
    return x

def fast_sum(size):
    return sum( [ (i+j) for j in range(size) for i in range(size)] )

pr = cProfile.Profile()
pr.enable()

size = 2000
slow_val = slow_sum(size)
fast_val = fast_sum(size)

assert slow_val == fast_val, "Values are not equal"



pr.disable()
s = io.StringIO()
sortby = SortKey.CUMULATIVE
ps = pstats.Stats(pr, stream=s).sort_stats(sortby)
ps.print_stats()
fileStream = open("profile_stats.log","w")
fileStream.write(s.getvalue())
fileStream.close()