def setUp(self): rules = RuleFactory() self.Straight = rules.straight() self.Flush = rules.flush() self.StraightFlush = rules.straightFlush() self.HightCards = rules.highCards() self.onePair = rules.onePair() self.twoPairs = rules.twoPairs() self.threeOfAKind = rules.threeOfAKind() self.fullHouse = rules.fullHouse() self.fourOfAKind = rules.fourOfAKind()
def __init__(self): rules = RuleFactory() self.prioritedRankings = [rules.straightFlush(), rules.fourOfAKind(), rules.fullHouse(), rules.flush(), rules.straight(), rules.threeOfAKind(), rules.twoPairs(), rules.onePair(), rules.highCards()]
def testSameSuit(self): hand = Hand([Card('3', 'D'), Card('4', 'D'), Card('T', 'D'), Card('Q', 'D'), Card('J', 'D')]) expectedHand = Hand([Card('Q', 'D'), Card('J', 'D'), Card('T', 'D'), Card('4', 'D'), Card('3', 'D')]) flush = RuleFactory().flush() self.assertEqual(0, flush.sort(hand).__cmp__(expectedHand))
def testStraight(self): hand = Hand([Card('T', 'S'), Card('J', 'H'), Card('7', 'D'), Card('8', 'S'), Card('9', 'H')]) expectedHand = Hand([Card('J', 'H'), Card('T', 'S'), Card('9', 'H'), Card('8', 'S'), Card('7', 'D')]) straightRule = RuleFactory().straight() self.assertEqual(0, straightRule.sort(hand).__cmp__(expectedHand))
def testHighCard(self): hand = Hand([Card('2', 'D'), Card('3', 'H'), Card('T', 'D'), Card('A','H'), Card('7','S')]) expectedHand = Hand([Card('A','H'), Card('T', 'D'), Card('7','S'), Card('3', 'H'), Card('2', 'D')]) highCards = RuleFactory().highCards() self.assertEqual(0, highCards.sort(hand).__cmp__(expectedHand))
def testOnePairRule(self): hand = Hand([Card('5', 'D'), Card('4', 'S'), Card('7', 'H'), Card('4','C'), Card('2', 'H')]) expectedHand = Hand([Card('4', 'S'), Card('4','C'), Card('7', 'H'), Card('5', 'D'), Card('2', 'H')]) onePairRule = RuleFactory().onePair() self.assertEqual(0, onePairRule.sort(hand).__cmp__(expectedHand))