def test_card_ranks(self):
""" Test the card ranks"""
sf = "6C 7C 8C 9C TC".split()
fk = "9D 9H 9S 9C 7D".split()
fh = "TD TC TH 7C 7D".split()
self.assertEqual(poker_game.card_ranks(sf), [10, 9, 8, 7, 6])
self.assertEqual(poker_game.card_ranks(fk), [9, 9, 9, 9, 7])
self.assertEqual(poker_game.card_ranks(fh), [10, 10, 10, 7, 7])
def test_two_pair(self):
""" Test two pairs"""
fk = "9D 9H 9S 9C 7D".split()
tp = "TD 9H TH 7C 7S".split()
fkranks = poker_game.card_ranks(fk)
tpranks = poker_game.card_ranks(tp)
print fkranks
print tpranks
self.assertEqual(poker_game.two_pair(fkranks), None)
self.assertEqual(poker_game.two_pair(tpranks), (10, 7))
def test_straight(self):
""" Test the straight """
al = "AC 2D 4H 3D 5S".split() # Ace-Low Straight
self.assertEqual(poker_game.straight([9, 8, 7, 6, 5]), True)
self.assertEqual(poker_game.straight([9, 8, 7, 4, 5]), False)
self.assertEqual(poker_game.straight(poker_game.card_ranks(al)), True)
def test_kind(self):
""" Test the kind """
fk = "9D 9H 9S 9C 7D".split()
fkranks = poker_game.card_ranks(fk)
self.assertEqual(poker_game.kind(4, fkranks), 9)
self.assertEqual(poker_game.kind(3, fkranks), None)
self.assertEqual(poker_game.kind(2, fkranks), None)
self.assertEqual(poker_game.kind(1, fkranks), 7)
