class LocalGameSenseTest(unittest.TestCase):
def setUp(self):
self.game = LocalGame()
def test_senses_actions_content(self):
sense_actions = self.game.sense_actions()
for square in SQUARES:
self.assertIn(square, sense_actions)
def test_sense_invalid(self):
for square in [-1, 65, 66, 1023730, -2]:
with self.assertRaises(ValueError):
self.game.sense(square)
def test_sense_squares(self):
for square in SQUARES:
sense_result = self.game.sense(square)
squares = [s for s, p in sense_result]
self.assertEqual(squares, SENSE_BY_SQUARE[square])
def test_sense_pieces(self):
for sense_square in SQUARES:
sense_result = self.game.sense(sense_square)
for square, piece in sense_result:
self.assertEqual(piece, self.game.board.piece_at(square))
def test_sense_pass(self):
sense_result = self.game.sense(None)
self.assertEqual(sense_result, [])
def playback(game_history: GameHistory, player: Player, color: Color):
game = LocalGame()
opponent_name = game_history.get_white_player_name()
if color == chess.WHITE:
opponent_name = game_history.get_black_player_name()
player.handle_game_start(color, game.board.copy(), opponent_name)
game.start()
turn = game_history.first_turn()
while not game.is_over() and turn < game_history.last_turn():
opt_capture_square = game.opponent_move_results()
if game.turn == color:
player.handle_opponent_move_result(opt_capture_square is not None,
opt_capture_square)
sense_actions = game.sense_actions()
move_actions = game.move_actions()
sense = game_history.sense(turn)
player_sense = player.choose_sense(sense_actions, move_actions,
game.get_seconds_left())
if game.turn == color and sense != player_sense:
print(
'Warning: Sense action did not match history on turn {}. Using the sense action from history.'
.format(turn))
sense_result = game.sense(sense)
if game.turn == color:
player.handle_sense_result(sense_result)
move = game_history.requested_move(turn)
player_move = player.choose_move(move_actions, game.get_seconds_left())
if game.turn == color and move != player_move:
print(
'Warning: Move action did not match history on turn {}. Using the move action from history.'
.format(turn))
requested_move, taken_move, opt_enemy_capture_square = game.move(move)
if game.turn == color:
player.handle_move_result(requested_move, taken_move,
opt_enemy_capture_square is not None,
opt_enemy_capture_square)
game.end_turn()
turn = turn.next
game.end()
winner_color = game.get_winner_color()
win_reason = game.get_win_reason()
game_history = game.get_game_history()
player.handle_game_end(winner_color, win_reason, game_history)