def test_Action_ante(self):
"""Test ante Action"""
action = Action.new_ante(1)
self.assertIsNotNone(action)
self.assertTrue(action.is_ante())
self.assertEqual(action.amount, 1)
self.assertFalse(action.is_all_in())
self.assertEqual(str(action), "ante 1")
def test_ante_ActionRequest(self):
"""Test ActionRequest for ante"""
amount = 20
request = ActionRequest.new_ante_request(amount)
self.assertIsNotNone(request)
self.assertTrue(request.is_ante_request())
self.assertEqual(request.amount, amount)
request.validate_action(Action.new_ante(amount))
request.validate_action(Action.new_fold())
with self.assertRaises(InvalidActionException):
request.validate_action(Action.new_ante(amount+1))
with self.assertRaises(InvalidActionException):
request.validate_action(Action.new_call(amount))
with self.assertRaises(InvalidActionException):
request.validate_action(Action.new_bet(amount))
with self.assertRaises(InvalidActionException):
request.validate_action(Action.new_raise(amount))
def test_ActionRequest(self):
"""Test ActionRequest class"""
request = ActionRequest(ActionRequest.ANTE_REQUEST, 10)
self.assertIsNotNone(request)
self.assertTrue(request.is_ante_request())
self.assertEqual(request.amount, 10)
self.assertEqual(str(request), "ante request for 10")
request.validate_action(Action.new_ante(10))
def test_opening_bet_ActionRequest(self):
"""Test ActionRequest for opening bet"""
amount = 50
request = ActionRequest.new_opening_bet_request(amount)
self.assertIsNotNone(request)
self.assertTrue(request.is_opening_bet_request())
self.assertEqual(request.amount, amount)
request.validate_action(Action.new_check())
request.validate_action(Action.new_bet(amount))
request.validate_action(Action.new_fold())
with self.assertRaises(InvalidActionException):
request.validate_action(Action.new_ante(amount))
with self.assertRaises(InvalidActionException):
request.validate_action(Action.new_blind(amount))
with self.assertRaises(InvalidActionException):
request.validate_action(Action.new_call(amount))
with self.assertRaises(InvalidActionException):
request.validate_action(Action.new_bet(amount+1))
with self.assertRaises(InvalidActionException):
request.validate_action(Action.new_raise(amount))
def test_call_ActionRequest(self):
"""Test ActionRequest for call"""
amount = 50
request = ActionRequest.new_call_request(amount, raise_amount=amount*2)
self.assertIsNotNone(request)
self.assertTrue(request.is_call_request())
self.assertEqual(request.amount, amount)
self.assertEqual(request.raise_amount, amount*2)
self.assertEqual(str(request), "call request for 50 or raise of 100")
request.validate_action(Action.new_call(amount))
request.validate_action(Action.new_raise(amount*2))
request.validate_action(Action.new_fold())
with self.assertRaises(InvalidActionException):
request.validate_action(Action.new_ante(amount))
with self.assertRaises(InvalidActionException):
request.validate_action(Action.new_blind(amount))
with self.assertRaises(InvalidActionException):
request.validate_action(Action.new_call(amount+1))
with self.assertRaises(InvalidActionException):
request.validate_action(Action.new_bet(amount))
with self.assertRaises(InvalidActionException):
request.validate_action(Action.new_raise(amount*2-1))
def test_BettingRound(self):
"""Test BettingRound class"""
players = [ Player(stack=1000),
Player(stack=1000),
Player(stack=1000),
Player(stack=1000) ]
table = Table()
table.seat_players(players, in_order=True)
message_handler = MessageHandler(table, self.console)
pot = Pot(players)
round = BettingRound(table, pot, message_handler)
self.assertIsNotNone(round)
self.assertEqual(round.table, table)
self.assertEqual(round.pot, pot)
self.assertRaises(PokerGameStateException, round.get_action_player)
action_is_on = players[1]
round.set_action(action_is_on)
self.assertEqual(round.get_action_player(), action_is_on)
self.assertIsNone(round.last_to_bet)
self.assertEqual(len(round.action_record), 0)
self.assertEqual(round.action_to_next_player(), players[2])
self.assertEqual(round.action_to_next_player(), players[3])
self.assertEqual(round.action_to_next_player(), players[0])
self.assertEqual(round.action_to_next_player(), players[1])
self.assertEqual(round.required_to_call(players[0]), 0)
self.assertFalse(round.is_pot_good())
# Have all players ante
round.process_action(Action.new_ante(5))
self.assertEqual(players[1].stack, 995)
self.assertEqual(round.get_action_player(), players[2])
self.assertEqual(round.last_to_bet, players[1])
round.process_action(Action.new_ante(5))
self.assertEqual(players[2].stack, 995)
self.assertEqual(round.get_action_player(), players[3])
self.assertEqual(round.last_to_bet, players[1])
round.process_action(Action.new_ante(5))
self.assertEqual(players[3].stack, 995)
self.assertEqual(round.get_action_player(), players[0])
self.assertEqual(round.last_to_bet, players[1])
round.process_action(Action.new_ante(5))
self.assertEqual(players[0].stack, 995)
self.assertEqual(round.get_action_player(), players[1])
self.assertEqual(round.last_to_bet, players[1])
self.assertTrue(round.is_pot_good())
self.assertEqual(round.sweep_bets_into_pot(), pot)
self.assertEqual(pot.amount, 20)
# New round
round = BettingRound(table, pot, message_handler)
self.assertIsNotNone(round)
round.set_action(action_is_on)
# Player 1 blinds, 2 blinds, 3 fold, 0 raises to 300
round.process_action(Action.new_blind(50))
self.assertEqual(players[1].bet, 50)
round.process_action(Action.new_blind(100))
round.process_action(Action.new_fold())
self.assertTrue(players[3].is_folded())
round.process_action(Action.new_raise(300))
self.assertFalse(round.is_pot_good())
self.assertEqual(round.get_action_player(), players[1])
self.assertEqual(len(round.action_record), 4)
self.assertEqual(round.total_pot(), 470)
# Player 1 calls, 2 raises all in, 3 has folded, 0 folds
self.assertEqual(round.required_to_call(), 250)
round.process_action(Action.new_call(250))
self.assertEqual(round.required_to_call(), 200)
round.process_action(Action.new_raise(895, all_in=True))
self.assertTrue(players[2].is_all_in())
# Make sure we skipped player 3 who folded earlier
self.assertEqual(round.get_action_player(), players[0])
self.assertEqual(round.required_to_call(), 695)
round.process_action(Action.new_fold())
self.assertFalse(round.is_pot_good())
self.assertEqual(round.total_pot(), 1615)
# Plater 1 calls and pot should be good
self.assertEqual(round.required_to_call(), 695)
round.process_action(Action.new_call(695, all_in=True))
self.assertTrue(round.is_pot_good())
self.assertEqual(round.total_pot(), 2310)
round.sweep_bets_into_pot()
# Should only be main pot
self.assertIsNone(pot.parent)
# All players bets should have beet swept into pot
for player in players:
self.assertEqual(0, player.bet,
"Bet for %s == %d != 0" % (player, player.bet))
self.assertEqual(pot.amount, 2310)
