def test_preflop_bet_sizes(self):
tiers = BetTiers(pot=500, big_blind=20, is_preflop=True)
def tier_name_for(bet):
return tiers.tier(bet).name
self.assertEqual(tier_name_for(0), "CHECK")
self.assertEqual(tier_name_for(10), "CHECK") # calling from the SB
self.assertEqual(tier_name_for(20), "MIN_RAISE")
self.assertEqual(tier_name_for(55), "RAISE")
self.assertEqual(tier_name_for(90), "BIG_RAISE")
self.assertEqual(tier_name_for(300), "OVERBET")
class BetTierTest(unittest.TestCase):
"""Tests that we put bets in the correct tiers"""
def setUp(self):
self.tiers = BetTiers(pot=500, big_blind=20, is_preflop=False)
def test_random_bets(self):
"""Verifies that random bets all get bucketed to a tier"""
for _ in range(25):
bet = random.randint(20, 1200)
tier = self.tiers.tier(bet)
self.assertTrue(tier)
def test_bet_sizes(self):
"""Verifies that different bet sizes are classified correctly"""
def tier_name_for(bet):
return self.tiers.tier(bet).name
self.assertEqual(tier_name_for(0), "CHECK")
self.assertEqual(tier_name_for(20), "MIN_RAISE")
self.assertEqual(tier_name_for(200), "RAISE")
self.assertEqual(tier_name_for(500), "BIG_RAISE")
self.assertEqual(tier_name_for(750), "OVERBET")
def test_preflop_bet_sizes(self):
tiers = BetTiers(pot=500, big_blind=20, is_preflop=True)
def tier_name_for(bet):
return tiers.tier(bet).name
self.assertEqual(tier_name_for(0), "CHECK")
self.assertEqual(tier_name_for(10), "CHECK") # calling from the SB
self.assertEqual(tier_name_for(20), "MIN_RAISE")
self.assertEqual(tier_name_for(55), "RAISE")
self.assertEqual(tier_name_for(90), "BIG_RAISE")
self.assertEqual(tier_name_for(300), "OVERBET")
def test_all_ins(self):
tiers = BetTiers(pot=500, big_blind=20, opponent_stack=300)
self.assertEqual(tiers.tier(300).name, "ALL_IN")
class BetTierTest(unittest.TestCase):
"""Tests that we put bets in the correct tiers"""
def setUp(self):
self.tiers = BetTiers(pot=500, big_blind=20, is_preflop=False)
def test_random_bets(self):
"""Verifies that random bets all get bucketed to a tier"""
for _ in range(25):
bet = random.randint(20, 1200)
tier = self.tiers.tier(bet)
self.assertTrue(tier)
def test_bet_sizes(self):
"""Verifies that different bet sizes are classified correctly"""
def tier_name_for(bet):
return self.tiers.tier(bet).name
self.assertEqual(tier_name_for(0), "CHECK")
self.assertEqual(tier_name_for(20), "MIN_RAISE")
self.assertEqual(tier_name_for(200), "RAISE")
self.assertEqual(tier_name_for(500), "BIG_RAISE")
self.assertEqual(tier_name_for(750), "OVERBET")
def test_preflop_bet_sizes(self):
tiers = BetTiers(pot=500, big_blind=20, is_preflop=True)
def tier_name_for(bet):
return tiers.tier(bet).name
self.assertEqual(tier_name_for(0), "CHECK")
self.assertEqual(tier_name_for(10), "CHECK") # calling from the SB
self.assertEqual(tier_name_for(20), "MIN_RAISE")
self.assertEqual(tier_name_for(55), "RAISE")
self.assertEqual(tier_name_for(90), "BIG_RAISE")
self.assertEqual(tier_name_for(300), "OVERBET")
def test_all_ins(self):
tiers = BetTiers(pot=500, big_blind=20, opponent_stack=300)
self.assertEqual(tiers.tier(300).name, "ALL_IN")
def setUp(self):
self.tiers = BetTiers(pot=500, big_blind=20, is_preflop=False)
def test_all_ins(self):
tiers = BetTiers(pot=500, big_blind=20, opponent_stack=300)
self.assertEqual(tiers.tier(300).name, "ALL_IN")
def setUp(self):
self.tiers = BetTiers(pot=500, big_blind=20, is_preflop=False)
def test_all_ins(self):
tiers = BetTiers(pot=500, big_blind=20, opponent_stack=300)
self.assertEqual(tiers.tier(300).name, "ALL_IN")
