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))
