class GameTest(unittest.TestCase): def setUp(self): self.game = Game() def test_i2xy(self): for idx in range(64): self.assertIn(self.game.i2xy(idx), ALG_POS) def test_xy2i(self): for pos in ALG_POS: self.assertIn(self.game.xy2i(pos), IDX_POS) def test_str(self): self.game.reset() # reset board to starting position self.assertEqual(str(self.game), START_POS) def test_fen_history(self): self.game.reset() self.assertEqual(self.game.fen_history, [START_POS]) self.game.apply_move('e2e4') hist = [ START_POS, 'rnbqkbnr/pppppppp/8/8/4P3/8/PPPP1PPP/RNBQKBNR b KQkq e3 0 1' ] self.assertEqual(self.game.fen_history, hist) def test_move_history(self): self.game = None self.game = Game() self.assertEqual(self.game.move_history, []) self.game.apply_move('e2e4') self.assertEqual(self.game.move_history, ['e2e4']) def test_get_moves(self): # legal openings self.game.reset() self.assertEqual(sorted(self.game.get_moves()), LEGAL_OPENINGS) # en passant fen = 'rnbqkbnr/ppp2ppp/4p3/3pP3/8/8/PPPP1PPP/RNBQKBNR w KQkq d6 0 1' self.game = Game(fen=fen) self.assertEqual(self.game.get_moves(idx_list=[28]), ['e5d6']) # checkmate fen = '8/p5kp/1p6/2p5/P5P1/2n4P/r2p4/1K6 w - - 2 37' self.game = Game(fen=fen) self.assertEqual(self.game.get_moves(), []) # castling into occupied space & king moving into check fen = '8/2p5/5p2/7r/7P/3R1P2/P3R1P1/2k1K3 b - - 13 46' self.game = Game(fen=fen) self.assertIn('c7c6', self.game.get_moves()) fen = '8/8/2p2p2/7r/7P/3R1P2/P3R1P1/2k1K3 w - - 0 47' self.game = Game(fen=fen, validate=False) self.assertNotIn('e1c1', self.game.get_moves()) # castling through check fen = 'r3k2r/8/8/8/8/8/8/3RKR2 b kq - 0 1' self.game = Game(fen=fen) self.assertNotIn('e8c8', self.game.get_moves(idx_list=[4])) self.assertNotIn('e8g8', self.game.get_moves(idx_list=[4])) # castling out of check fen = 'r3rk2/8/8/8/8/8/8/R3K2R w KQ - 0 1' self.game = Game(fen=fen) self.assertNotIn('e1c8', self.game.get_moves(idx_list=[60])) # Github Issue 9 fen = 'r3kb1r/p1p2pp1/2p4p/3Pp3/6b1/2P5/PP1NN2P/R2QK1q1 w Qkq - 0 16' self.game = Game(fen=fen) self.assertEqual(['d2f1', 'e2g1'], self.game.get_moves()) def test_apply_move(self): # pawn promotion fen = '3qk1b1/P7/8/8/8/8/7P/4K3 w - - 0 1' self.game = Game(fen=fen) self.game.apply_move('a7a8q') self.assertEqual(str(self.game), 'Q2qk1b1/8/8/8/8/8/7P/4K3 b - - 0 1') # apply moves self.game.apply_move('g8h7') self.assertEqual(str(self.game), 'Q2qk3/7b/8/8/8/8/7P/4K3 w - - 1 2') self.game.apply_move('h2h4') self.assertEqual(str(self.game), 'Q2qk3/7b/8/8/7P/8/8/4K3 b - h3 0 2') # block vector tail with capture moves = [ 'd8c8', 'd8b8', 'd8a8', 'e8f8', 'e8d7', 'e8e7', 'e8f7', 'g8f7', 'g8e6', 'g8d5', 'g8c4', 'g8b3', 'g8a2', 'g8h7' ] self.game.reset(fen='Q2qk1b1/8/8/8/8/8/7P/4K3 b - - 0 1') self.assertEqual(self.game.get_moves(), moves) self.game.apply_move('d8b8') moves = set(self.game.get_moves()) self.assertIn('a8b8', moves) self.assertNotIn('a8c8', moves) # white castling fen = 'r3k2r/pppqbppp/3pb3/8/8/3PB3/PPPQBPPP/R3K2R w KQkq - 0 7' new_fen = 'r3k2r/pppqbppp/3pb3/8/8/3PB3/PPPQBPPP/R4RK1 b kq - 1 7' self.game = Game(fen=fen) self.game.apply_move('e1g1') self.assertEqual(str(self.game), new_fen) new_fen = 'r3k2r/pppqbppp/3pb3/8/8/3PB3/PPPQBPPP/2KR3R b kq - 1 7' self.game = Game(fen=fen) self.game.apply_move('e1c1') self.assertEqual(str(self.game), new_fen) # black castling fen = 'r3k2r/pppqbppp/3pb3/8/8/3PB3/PPPQBPPP/R3K2R b KQkq - 0 7' new_fen = 'r4rk1/pppqbppp/3pb3/8/8/3PB3/PPPQBPPP/R3K2R w KQ - 1 8' self.game = Game(fen=fen) self.game.apply_move('e8g8') self.assertEqual(str(self.game), new_fen) new_fen = '2kr3r/pppqbppp/3pb3/8/8/3PB3/PPPQBPPP/R3K2R w KQ - 1 8' self.game = Game(fen=fen) self.game.apply_move('e8c8') self.assertEqual(str(self.game), new_fen) # Disable castling on capture fen = '1r2k2r/3nb1Qp/p1pp4/3p4/3P4/P1N2P2/1PP3PP/R1B3K1 w k - 0 22' new_fen = '1r2k2Q/3nb2p/p1pp4/3p4/3P4/P1N2P2/1PP3PP/R1B3K1 b - - 0 22' self.game = Game(fen=fen) self.game.apply_move('g7h8') self.assertEqual(str(self.game), new_fen) # white en passant fen = 'rnbqkbnr/ppp2ppp/4p3/3pP3/8/8/PPPP1PPP/RNBQKBNR w KQkq d6 0 1' self.game = Game(fen=fen) self.game.apply_move('e5d6') new_fen = 'rnbqkbnr/ppp2ppp/3Pp3/8/8/8/PPPP1PPP/RNBQKBNR b KQkq - 0 1' self.assertEqual(str(self.game), new_fen) # black en passant fen = 'rnbqkbnr/ppp1pppp/8/8/3pP3/8/PPPP1PPP/RNBQKBNR b KQkq e3 0 1' self.game = Game(fen=fen) self.game.apply_move('d4e3') new_fen = 'rnbqkbnr/ppp1pppp/8/8/8/4p3/PPPP1PPP/RNBQKBNR w KQkq - 0 2' self.assertEqual(str(self.game), new_fen) # capture fen = 'r2q1rk1/pppbbppp/2n5/3p4/3PN3/4PN2/1PPBBPPP/R2Q1RK1 b - - 0 9' self.game = Game(fen=fen) self.game.apply_move('d5e4') n_fen = 'r2q1rk1/pppbbppp/2n5/8/3Pp3/4PN2/1PPBBPPP/R2Q1RK1 w - - 0 10' self.assertEqual(str(self.game), n_fen) # invalid move self.game.reset() with self.assertRaises(InvalidMove): self.game.apply_move('e2e2') def test_status(self): # NORMAL game = Game() self.assertEqual(game.status, Game.NORMAL) # CHECK game = Game(fen='r3rk2/8/8/8/8/8/8/R3K2R w KQ - 0 1') self.assertEqual(game.status, game.CHECK) # CHECKMATE game = Game(fen='8/p5kp/1p6/2p5/P5P1/2n4P/r2p4/1K6 w - - 2 37') self.assertEqual(game.status, Game.CHECKMATE) # STALEMATE game = Game(fen='8/8/8/8/8/7k/5q2/7K w - - 0 37') self.assertEqual(game.status, Game.STALEMATE)
def test_xy2i(self): for pos in ALG_POS: self.assertIn(Game.xy2i(pos), IDX_POS)