set_test.py

import random
import itertools

class game:
    def __init__(self, cards=None):
        if not cards:
            self.cards = [self.random_card() for _ in xrange(12)]
        else:
            self.cards = cards

    def __repr__(self):
        return '\n'.join(str(card) for card in self.cards)

    def random_card(self):
        """Return random tuple (color, shape, number, texture)"""
        return tuple( random.choice([1,2,3]) for _ in xrange(4) )

    def is_set(self, x, y, z):
        """Returns true if card trio is a set at every position based on sum mod 3, which is only 0 IFF all same or all different"""
        return sum( sum(i)%3 for i in zip(x,y,z) ) == 0

    def compare_all(self, idx=False):
        """Compare all combinations of three cards and return list of tuples of winning sets. If idx, also return corresponding list index of each card."""
        sets = []
        self.trios = itertools.combinations( enumerate(self.cards), 3 )
        for x, y, z in self.trios:
            if self.is_set(x[1], y[1], z[1]):
                cards_idx = (x[0], y[0], z[0])
                if idx:
                    sets.append( map(lambda x: (x, self.cards[x]), cards_idx) )
                else:
                    sets.append( map(lambda x: self.cards[x], cards_idx) )
        return sets

    def play(self, idx=False):
        results = self.compare_all(idx)
        if results:
            return results
        return False

def test(n):
    return '{}%'.format(sum( game().play() for _ in xrange(n) )/float(n) * 100.0)