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