def test_step_back(self):
env = Env(allow_step_back=True)
_, player_id = env.init_game()
env.step(2)
_, back_player_id = env.step_back()
self.assertEqual(player_id, back_player_id)
self.assertEqual(env.step_back(), False)
env = Env()
with self.assertRaises(Exception):
env.step_back()
def test_get_legal_actions(self):
env = Env()
env.set_agents([RandomAgent(309), RandomAgent(309), RandomAgent(309)])
env.init_game()
legal_actions = env.get_legal_actions()
for legal_action in legal_actions:
self.assertLessEqual(legal_action, 308)
def test_step(self):
env = Env()
_, player_id = env.init_game()
player = env.game.players[player_id]
_, next_player_id = env.step(308)
self.assertEqual(next_player_id,
get_downstream_player_id(player, env.game.players))
def test_decode_action(self):
env = Env()
env.init_game()
env.game.state['actions'] = ['33366', '33355']
env.game.judger.playable_cards[0] = [
'5', '6', '55', '555', '33366', '33355'
]
decoded = env.decode_action(54)
self.assertEqual(decoded, '33366')
env.game.state['actions'] = ['444', '44466', '44455']
decoded = env.decode_action(29)
self.assertEqual(decoded, '444')
def test_decode_action(self):
env = Env()
state, _ = env.init_game()
for action in state['legal_actions']:
decoded = env.decode_action(action)
self.assertIn(decoded, env.game.state['actions'])
state, _ = env.step(0)
for action in range(309):
if action not in state['legal_actions']:
decoded = env.decode_action(action)
self.assertEqual(decoded, 'pass')
break
def test_run(self):
env = Env()
env.set_agents([RandomAgent(309), RandomAgent(309), RandomAgent(309)])
trajectories, payoffs = env.run(is_training=False)
self.assertEqual(len(trajectories), 3)
win = []
for player_id, payoff in enumerate(payoffs.items()):
if payoff == 1:
win.append(player_id)
if len(win) == 1:
self.assertEqual(env.game.players[win[0]].role, 'landlord')
if len(win) == 2:
self.assertEqual(env.game.players[win[0]].role, 'peasant')
self.assertEqual(env.game.players[win[1]].role, 'peasant')
def test_init_game_and_extract_state(self):
env = Env()
state, _ = env.init_game()
self.assertEqual(state.size, 450)