def test_is_straight(self):
label_patterns = [
True, True, False, False, False, False, False, False, False, True,
True
]
for hands, label in zip(self.hands_patterns, label_patterns):
with self.subTest(hands=hands, label=label):
self.assertEqual(poker.is_straight(hands), label)
def test_straight_false(self):
self.assertFalse(is_straight([
Card('T', 'D'), Card('J', 'S'),
Card('2', 'D'), Card('K', 'C'),
Card('A', 'D')
]))
def test_straight_true_ace_at_end(self):
self.assertTrue(is_straight([
Card('T', 'D'), Card('J', 'S'),
Card('Q', 'D'), Card('K', 'C'),
Card('A', 'D')
]))
def test_straight_true_ace_at_start(self):
self.assertTrue(is_straight([
Card('A', 'D'), Card('2', 'D'),
Card('3', 'S'), Card('4', 'H'),
Card('5', 'C'),
]))
def test_is_straight(self):
hand = ["2C", "3H", "4C", "5H", "6C"]
self.assertTrue(is_straight(sort_cards_values(hand)))
