attila_vs_damon.py

import random
import timeit


ITERATIONS = 10000
COMPLEXITY = 3
SIZE = 1000


def filter_zero(row):
    return 0 in row

def filter_one(row):
    return 1 in row

def attila(data):
    for i, d in enumerate(data):
        if not filter_zero(d):
            data.pop(i)

    for i, d in enumerate(data):
        if not filter_one(d):
            data.pop(i)

    return list(data)

def damon(data):
    for i, d in enumerate(data):
        if not filter_zero(d) or filter_one(d):
            data.pop(i)
    return list(data)


if __name__ == '__main__':
    choices = range(9)
    data_set = [
        (random.choice(choices) for c in range(COMPLEXITY))
        for _ in range(SIZE)
    ]

    data_str = "[%s]" % ', '.join(
        ["(%s)" % ','.join(str(a) for a in d) for d in data_set]
    )

    attila_result = timeit.timeit(
        'attila(%s)' % data_str,
        number=ITERATIONS,
        setup='from __main__ import attila',
    )

    print "Finished with Attila..."

    damon_result = timeit.timeit(
        'damon(%s)' % data_str,
        number=ITERATIONS,
        setup='from __main__ import damon',
    )

    print "Finished with Damon..."
    print ""

    print "Attila:\t", attila_result
    print "Damon:\t", damon_result
    print "Winner", "Damon" if damon_result < attila_result else "Attila"