def draw_discards(cards, ranklist): """ Calculates the approprate card to discard for any draw-type hands. """ if len(cards) != 5: raise ValueError('Card list needs to be 5 cards for a valid discard.') suit = ev.dominant_suit(cards) suit_count = ev.count_suit(cards, suit) # Check 4 card draws first # Flush draws if suit_count == 4: return ev.strip_suits(cards, suit) # Test for open-ended straight draw(s) OESD = ev.chk_straight_draw(cards, 4, 0) if OESD is not None: return extract_discards(cards, OESD) # Test for gutshot straight draw(s) GSSD = ev.chk_straight_draw(cards, 4, 1) if GSSD is not None: return extract_discards(cards, GSSD) # Draw to high cards (J+) if card.RANKS[ranklist[2].rank] > 10: highcards = ''.join([ranklist[i].rank for i in range(3)]) return ev.strip_ranks(cards, highcards) elif card.RANKS[ranklist[1].rank] > 10: highcards = ''.join([ranklist[i].rank for i in range(2)]) return ev.strip_ranks(cards, highcards) # Draw to an Ace # We'll generally draw to an Ace over any backdoor draws. if ranklist[0].rank == 'A': return ev.strip_ranks(cards, 'A') if suit_count == 3: # Backdoor flush draw return ev.strip_suits(cards, suit) # Backdoor straight draws are pretty desparate BDSD = ev.chk_straight_draw(cards, 3, 0) if BDSD is not None: return extract_discards(cards, BDSD) # 1-gap Backdoor straight draws are truly desparate! BDSD = ev.chk_straight_draw(cards, 3, 1) if BDSD is not None: return extract_discards(cards, BDSD) # Last ditch effort - just draw to the best 2. highcards = ''.join([ranklist[i].rank for i in range(2)]) return ev.strip_ranks(cards, highcards)
def test_chkstraightdraw_4card_1gap_34589A_returnsA345(self): qty, gap = 4, 1 cards = tools.convert_to_cards(['3h', '4s', '5d', '8h', '9d', 'Ad']) expected = tools.convert_to_cards(['3h', '4s', '5d', 'Ad']) result = evaluator.chk_straight_draw(cards, qty, gap) self.assertEqual(expected, result)
def test_chkstraightdraw_4card_0gap_A2234_returns234A(self): qty, gap = 4, 0 cards = tools.convert_to_cards(['Ah', '2h', '2c', '3d', '4h']) expected = tools.convert_to_cards(['2h', '3d', '4h', 'Ah']) result = evaluator.chk_straight_draw(cards, qty, gap) self.assertEqual(expected, result)
def test_chkstraightdraw_4card_1gap_43674_returns3467(self): qty, gap = 4, 1 cards = tools.convert_to_cards(['4c', '3c', '6h', '7d', '4d']) expected = tools.convert_to_cards(['3c', '4c', '6h', '7d']) result = evaluator.chk_straight_draw(cards, qty, gap) self.assertEqual(expected, result)
def test_chkstraightdraw_4card_0gap_2J5Q6K_returnsNone(self): qty, gap = 4, 0 cards = tools.convert_to_cards(['2c', 'Js', '5h', 'Qd', '6s', 'Kh']) expected = None result = evaluator.chk_straight_draw(cards, qty, gap) self.assertEqual(expected, result)
def test_chkstraightdraw_3card_1gap_JA596_returnsNone(self): qty, gap = 3, 1 cards = tools.convert_to_cards(['Jd', 'As', '5h', '9d', '6s']) expected = None result = evaluator.chk_straight_draw(cards, qty, gap) self.assertEqual(expected, result)
def test_chkstraightdraw_3card_1gap_27T4KA_returns24A(self): qty, gap = 3, 1 cards = tools.convert_to_cards(['2d', '7h', 'Ts', '4s', 'Ks', 'As']) expected = tools.convert_to_cards(['2d', '4s', 'As']) result = evaluator.chk_straight_draw(cards, qty, gap) self.assertEqual(expected, result)
def test_chkstraightdraw_3card_0gap_AKQJ_returnsQKA(self): qty, gap = 3, 0 cards = tools.convert_to_cards(['As', 'Kc', 'Qd', 'Jd']) expected = tools.convert_to_cards(['Qd', 'Kc', 'As']) result = evaluator.chk_straight_draw(cards, qty, gap) self.assertEqual(expected, result)
def test_chkstraightdraw_2card_0gap_2A5Q6_returns56(self): qty, gap = 2, 0 cards = tools.convert_to_cards(['2c', 'As', '5h', 'Qd', '6s']) expected = tools.convert_to_cards(['5h', '6s']) result = evaluator.chk_straight_draw(cards, qty, gap) self.assertEqual(expected, result)