def test_first_legal_bids(self):
game = Game()
_, current_player_id = game.init_game()
for _ in range(3):
legal_actions = game.judger.get_legal_actions()
self.assertEqual(len(legal_actions), 36)
action = PassAction()
self.assertTrue(action in legal_actions)
_, next_player_id = game.step(action)
legal_actions = game.judger.get_legal_actions()
self.assertEqual(len(legal_actions), 36)
self.assertTrue(PassAction() in legal_actions)
def test_step(self):
game = Game()
_, current_player_id = game.init_game()
legal_actions = game.judger.get_legal_actions()
action = np.random.choice(legal_actions)
print(
f'test_step current_player_id={current_player_id} action={action} legal_actions={[str(action) for action in legal_actions]}'
)
_, next_player_id = game.step(action)
next_legal_actions = game.judger.get_legal_actions()
print(
f'test_step next_player_id={next_player_id} next_legal_actions={[str(action) for action in next_legal_actions]}'
)
def test_init_game(self):
player_ids = list(range(4))
game = Game()
state, current_player = game.init_game()
self.assertEqual(len(game.round.move_sheet), 1)
self.assertIn(current_player, player_ids)
self.assertEqual(len(game.actions), 0)
self.assertEqual(len(game.round.players[current_player].hand),
13) # current_player has 13 cards
self.assertEqual(len(game.round.dealer.shuffled_deck), 52)
self.assertEqual(len(game.round.dealer.stock_pile), 0)
self.assertEqual(state['player_id'], current_player)
self.assertEqual(len(state['hand']), 13)
def test_pass_out_hand(self):
game = Game()
_, current_player_id = game.init_game()
for _ in range(4):
legal_actions = game.judger.get_legal_actions()
action = PassAction()
self.assertTrue(action in legal_actions)
_, next_player_id = game.step(action)
self.assertTrue(not game.round.contract_bid_move)
self.assertTrue(game.round.is_bidding_over())
self.assertTrue(game.is_over())
self.assertTrue(not game.round.get_declarer())
self.assertTrue(not game.round.get_dummy())
self.assertTrue(not game.round.get_left_defender())
self.assertTrue(not game.round.get_right_defender())
def test_print_scene(self):
game = Game()
next_state, next_player_id = game.init_game()
deal_hand_move = game.round.move_sheet[0]
self.assertTrue(isinstance(deal_hand_move, DealHandMove))
while not game.is_over():
current_player_id = game.round.current_player_id
self.assertEqual(current_player_id, next_player_id)
legal_actions = game.judger.get_legal_actions()
action = np.random.choice(legal_actions)
game.round.print_scene()
next_state, next_player_id = game.step(action)
game.round.print_scene()
for player_id in range(4):
player = game.round.players[player_id]
hand = player.hand
self.assertTrue(not hand)
def test_get_num_actions(self):
game = Game()
num_actions = game.get_num_actions()
self.assertEqual(num_actions, 91)
def test_get_num_players(self):
game = Game()
num_players = game.get_num_players()
self.assertEqual(num_players, 4)