python flatten test

flatten-test.py

from itertools import chain, product
from time import time

REPEAT = 10000

for s1, s2 in product([10, 100], repeat=2):
    print "%i tuples with %i elements each" % (s2, s1)
    a = [tuple(range(s1)) for i in range(s2)]

    t = time()
    for i in range(REPEAT):
        b = [j for i in a for j in i]
    print "comprehensions:", time() - t

    t = time()
    for i in range(REPEAT):
        b = reduce(lambda x, y: x + list(y), a, [])
    print "reduce:        ", time() - t

    t = time()
    for i in range(REPEAT):
        b = list(chain.from_iterable(a))
    print "chain:         ", time() - t