def test_latest_bet_raise_amount():
game = Game(starting_stacks=[100, 200, 300])
assert game.view().last_raise_amount() == 0
game.add_action(Action(0, Move.SMALL_BLIND, total_bet=1, amount_added=1))
assert game.view().last_raise_amount() == 1
assert game.view().current_bet_amount() == 1
game.add_action(Action(1, Move.BIG_BLIND, total_bet=2, amount_added=2))
assert game.view().last_raise_amount() == 1
assert game.view().current_bet_amount() == 2
game.add_action(Action(2, Move.BET_RAISE, total_bet=10, amount_added=10))
assert game.view().last_raise_amount() == 8
assert game.view().current_bet_amount() == 10
# Raise to 20 total. This is a raise of 8 on top of the original 10.
# It is a min-raise.
game.add_action(
Action(0, Move.BET_RAISE, total_bet=18, amount_added=18 - 1))
assert game.view().last_raise_amount() == 8
assert game.view().current_bet_amount() == 18
# Riase it to 30 total (up from the previous 18). This is a raise of 12.
game.add_action(
Action(1, Move.BET_RAISE, total_bet=30, amount_added=30 - 2))
assert game.view().last_raise_amount() == 12
assert game.view().current_bet_amount() == 30
def test_fold():
game = GameRunner(starting_stacks=[500, 500])
game.start_game()
# Preflop Action
game.bet_raise(to=230)
game.call()
# Flop Action
game.fold()
assert game.game_view().events() == [
Street.PREFLOP,
Action(player_index=0,
move=Move.SMALL_BLIND,
amount_added=1,
total_bet=1),
Action(player_index=1,
move=Move.BIG_BLIND,
amount_added=2,
total_bet=2),
Action(player_index=0,
move=Move.BET_RAISE,
amount_added=229,
total_bet=230),
Action(player_index=1,
move=Move.CHECK_CALL,
amount_added=228,
total_bet=230),
Street.FLOP,
Action(player_index=0, move=Move.FOLD, amount_added=0, total_bet=0),
Street.HAND_OVER,
]
def test_is_all_in():
game = Game(starting_stacks=[100, 200, 10])
game.add_action(Action(0, Move.SMALL_BLIND, total_bet=1, amount_added=1))
game.add_action(Action(1, Move.BIG_BLIND, total_bet=2, amount_added=2))
game.add_action(Action(2, Move.BET_RAISE, total_bet=10, amount_added=10))
assert game.view().is_all_in() == [False, False, True]
def test_blinds():
game = Game(starting_stacks=[10, 20, 100])
assert game.view().small_blind() == Action(0,
Move.SMALL_BLIND,
total_bet=1,
amount_added=1)
game.add_action(game.view().small_blind())
assert game.view().big_blind() == Action(1,
Move.BIG_BLIND,
total_bet=2,
amount_added=2)
def test_all_in_is_call():
game = GameRunner(starting_stacks=[30, 20])
game.start_game()
game.bet_raise(to=25)
assert game.game_view().go_all_in() == Action(player_index=1,
move=Move.CHECK_CALL,
amount_added=18,
total_bet=25)
def test_raise_all_in():
game = GameRunner(starting_stacks=[200, 250, 100])
game.start_game()
game.bet_raise(to=10)
assert game.game_view().go_all_in() == Action(player_index=0,
move=Move.BET_RAISE,
amount_added=199,
total_bet=200)
def test_amount_bet():
game = Game(starting_stacks=[100, 200, 300])
preflop_action = [
Action(0, Move.SMALL_BLIND, total_bet=1, amount_added=1),
Action(1, Move.BIG_BLIND, total_bet=2, amount_added=2),
Action(2, Move.BET_RAISE, total_bet=6, amount_added=6),
]
for a in preflop_action:
game.add_action(a)
assert game.view().amount_added_in_street() == [1, 2, 6]
game.set_street(Street.FLOP)
flop_action = [
Action(0, Move.CHECK_CALL, total_bet=0, amount_added=0),
Action(1, Move.BET_RAISE, total_bet=10, amount_added=10),
Action(2, Move.FOLD, amount_added=0, total_bet=10),
Action(0, Move.FOLD, amount_added=0, total_bet=10),
]
for a in flop_action:
game.add_action(a)
assert game.view().amount_added_in_street() == [0, 10, 0]
assert game.view().amount_added_total() == [1, 12, 6]
def test_call_all_in():
game = Game(starting_stacks=[10, 20, 100])
game.add_action(Action(0, Move.SMALL_BLIND, total_bet=1, amount_added=1))
game.add_action(Action(1, Move.BIG_BLIND, total_bet=2, amount_added=2))
game.add_action(Action(2, Move.BET_RAISE, total_bet=100, amount_added=100))
call = game.view().call()
assert game.view().call() == Action(player_index=0,
move=Move.CHECK_CALL,
amount_added=9,
total_bet=100)
game.add_action(call)
call = game.view().call()
assert call == Action(player_index=1,
move=Move.CHECK_CALL,
amount_added=18,
total_bet=100)
def test_bet_and_call_amount():
game = Game(starting_stacks=[100, 200, 300])
assert game.view().current_bet_amount() == 0
assert game.view().amount_to_call() == [0, 0, 0]
game.add_action(Action(0, Move.SMALL_BLIND, total_bet=1, amount_added=1))
assert game.view().current_bet_amount() == 1
assert game.view().amount_to_call() == [0, 1, 1]
game.add_action(Action(1, Move.BIG_BLIND, total_bet=2, amount_added=2))
game.add_action(Action(2, Move.BET_RAISE, total_bet=10, amount_added=10))
assert game.view().current_bet_amount() == 10
assert game.view().amount_to_call() == [9, 8, 0]
game.add_action(Action(0, Move.CHECK_CALL, total_bet=10, amount_added=9))
game.add_action(Action(1, Move.CHECK_CALL, total_bet=10, amount_added=8))
assert game.view().current_bet_amount() == 10
assert game.view().amount_to_call() == [0, 0, 0]
def test_stack_sizes():
game = Game(starting_stacks=[100, 200, 300])
assert game.view().current_stack_sizes() == [100, 200, 300]
game.add_action(Action(0, Move.SMALL_BLIND, total_bet=1, amount_added=1))
game.add_action(Action(1, Move.BIG_BLIND, total_bet=2, amount_added=2))
game.add_action(Action(2, Move.BET_RAISE, total_bet=6, amount_added=6))
assert game.view().current_stack_sizes() == [99, 198, 294]
game.add_action(Action(0, Move.CHECK_CALL, total_bet=6, amount_added=5))
game.add_action(Action(1, Move.CHECK_CALL, total_bet=6, amount_added=4))
assert game.view().current_stack_sizes() == [94, 194, 294]
game.set_street(Street.TURN)
game.add_action(Action(0, Move.CHECK_CALL, total_bet=0, amount_added=0))
game.add_action(Action(1, Move.BET_RAISE, total_bet=10, amount_added=10))
assert game.view().current_stack_sizes() == [94, 184, 294]
def test_is_folded():
game = Game(starting_stacks=[100, 200, 300])
game.add_action(Action(0, Move.SMALL_BLIND, total_bet=1, amount_added=1))
game.add_action(Action(1, Move.BIG_BLIND, total_bet=2, amount_added=2))
game.add_action(Action(2, Move.FOLD, total_bet=2, amount_added=0))
assert game.view().is_folded() == [False, False, True]
game.add_action(Action(1, Move.CHECK_CALL, total_bet=2, amount_added=1))
game.add_action(Action(2, Move.CHECK_CALL, total_bet=2, amount_added=0))
assert game.view().is_folded() == [False, False, True]
game.set_street(Street.TURN)
assert game.view().is_folded() == [False, False, True]
game.add_action(Action(0, Move.BET_RAISE, total_bet=10, amount_added=10))
game.add_action(Action(1, Move.FOLD, total_bet=10, amount_added=0))
assert game.view().is_folded() == [False, True, True]
def test_call():
game = GameRunner(starting_stacks=[100, 100, 100])
game.start_game()
assert game.game_view().events() == [
Street.PREFLOP,
Action(player_index=0,
move=Move.SMALL_BLIND,
amount_added=1,
total_bet=1),
Action(player_index=1,
move=Move.BIG_BLIND,
amount_added=2,
total_bet=2),
]
assert game.street() == Street.PREFLOP
assert game.current_player() == 2
# Preflop Action
assert game.bet_raise(to=25) == ActionResult(street=Street.PREFLOP,
current_player=0,
total_bet=25,
amount_to_call=24)
assert game.call() == ActionResult(street=Street.PREFLOP,
current_player=1,
total_bet=25,
amount_to_call=23)
assert game.fold() == ActionResult(street=Street.FLOP,
current_player=0,
total_bet=0,
amount_to_call=0)
# Flop Action
assert game.bet_raise(to=50) == ActionResult(street=Street.FLOP,
current_player=2,
total_bet=50,
amount_to_call=50)
assert game.call() == ActionResult(street=Street.TURN,
current_player=0,
total_bet=0,
amount_to_call=0)
# Turn Action
assert game.check() == ActionResult(street=Street.TURN,
current_player=2,
total_bet=0,
amount_to_call=0)
assert game.check() == ActionResult(street=Street.RIVER,
current_player=0,
total_bet=0,
amount_to_call=0)
# River
assert game.check() == ActionResult(street=Street.RIVER,
current_player=2,
total_bet=0,
amount_to_call=0)
assert game.bet_raise(raise_amount=10) == ActionResult(street=Street.RIVER,
current_player=0,
total_bet=10,
amount_to_call=10)
assert game.call() == ActionResult(street=Street.HAND_OVER)
def test_action_views():
game = Game(starting_stacks=[100, 200, 300])
preflop_action = [
Action(0, Move.SMALL_BLIND, total_bet=1, amount_added=1),
Action(1, Move.BIG_BLIND, total_bet=2, amount_added=2),
Action(2, Move.BET_RAISE, total_bet=6, amount_added=6),
Action(0, Move.CHECK_CALL, total_bet=6, amount_added=5),
Action(1, Move.CHECK_CALL, total_bet=6, amount_added=4),
]
for a in preflop_action:
game.add_action(a)
assert game.current_street() == Street.PREFLOP
assert game.events == preflop_action
preflop_view = game.view()
assert preflop_view.street_action() == preflop_action
assert preflop_view.all_actions() == preflop_action
game.set_street(Street.FLOP)
flop_action = [
Action(0, Move.CHECK_CALL, total_bet=0, amount_added=0),
Action(1, Move.BET_RAISE, total_bet=10, amount_added=10),
Action(2, Move.FOLD, total_bet=10, amount_added=0),
Action(0, Move.FOLD, total_bet=10, amount_added=0),
]
for a in flop_action:
game.add_action(a)
assert game.current_street() == Street.FLOP
assert game.events == preflop_action + [Street.FLOP] + flop_action
# assert preflop_view.timestamp ==
assert preflop_view.street_action() == preflop_action
assert preflop_view.all_actions() == preflop_action
# Recreate the preflop view
preflop_view = game.view(timestamp=5)
assert preflop_view.street_action() == preflop_action
assert preflop_view.all_actions() == preflop_action
flop_view = game.view()
assert flop_view.street_action() == flop_action
assert flop_view.all_actions() == preflop_action + flop_action
# Test a view in the middle of the preflop
mid_preflop_view = game.view(3)
assert mid_preflop_view.street_action() == [
Action(0, Move.SMALL_BLIND, total_bet=1, amount_added=1),
Action(1, Move.BIG_BLIND, total_bet=2, amount_added=2),
Action(2, Move.BET_RAISE, total_bet=6, amount_added=6),
]