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_i2xy(self):
for idx in range(64):
self.assertIn(Game.i2xy(idx), ALG_POS)
def test_i2xy(self):
for idx in xrange(64):
self.assertIn(Game.i2xy(idx), ALG_POS)
