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)