def test_straight_flush(self):
cases = ["2D 3D 4D 5D 6D", "KS QS JS TS 9S"]
for c in cases:
c = classify(_parse_hand(c))
self.assertEqual(constants.hands.STRAIGHT_FLUSH, c[0])
def test_full_house(self):
cases = ["3H 3D 3C KC KS", "4H 4C 4S JC JS"]
for c in cases:
c = classify(_parse_hand(c))
self.assertEqual(constants.hands.FULL_HOUSE, c[0])
def test_four_of_a_kind(self):
cases = ["3H 3D 3C 3S JS", "4H 4C 4S 4D JS"]
for c in cases:
c = classify(_parse_hand(c))
self.assertEqual(constants.hands.FOUR_OF_A_KIND, c[0])
def test_straight(self):
cases = ["2D 3C 4S 5H 6D", "AS KD QH JC TD"]
for c in cases:
c = classify(_parse_hand(c))
self.assertEqual(constants.hands.STRAIGHT, c[0])
def test_flush(self):
cases = ["2D 3D 8D TD 6D", "AC 8C 2C 4C JC"]
for c in cases:
c = classify(_parse_hand(c))
self.assertEqual(constants.hands.FLUSH, c[0])
def test_three_of_a_kind(self):
cases = ["3H 3D 3C KC JS", "4H 4C 4S KC JS"]
for c in cases:
c = classify(_parse_hand(c))
self.assertEqual(constants.hands.THREE_OF_A_KIND, c[0])
def test_two_pair(self):
cases = ["3H 3D 6S JC JS", "3H 4H 3S KC KS"]
for c in cases:
c = classify(_parse_hand(c))
self.assertEqual(constants.hands.TWO_PAIR, c[0])
def test_one_pair(self):
cases = ["3H 7H 6S JC JS", "3H 4H 3S KC TS"]
for c in cases:
c = classify(_parse_hand(c))
self.assertEqual(constants.hands.ONE_PAIR, c[0])
def test_high_card(self):
cases = ["3H 7H 6S KC JS", "3H 4H 5S KC TS"]
for c in cases:
c = classify(_parse_hand(c))
self.assertEqual(constants.hands.HIGH_CARD, c[0])
def test_royal_flush(self):
cases = ["AS KS QS JS TS", "AD TD KD JD QD"]
for c in cases:
c = classify(_parse_hand(c))
self.assertEqual(constants.hands.ROYAL_FLUSH, c[0])