def test_castling_forbidden_rook_moved(self):
'''Test that castling is forbidden once the rook moved.'''
board = Board(10, 10)
king = King(board.white)
rook_l = Rook(board.white)
rook_r = Rook(board.white)
board[0, 7].piece = rook_l
board[7, 7].piece = rook_r
board[4, 7].piece = king
moves = king.get_moves((4, 7), board)
self.assertTrue(Castling("left",board.white) in moves)
self.assertTrue(Castling("right",board.white) in moves)
move = Move((0,7),(1,7))
self.assertTrue(move in rook_l.get_moves((0,7),board))
move.perform(board)
moves = king.get_moves((4, 7), board)
self.assertFalse(Castling("left",board.white) in moves)
self.assertTrue(Castling("right",board.white) in moves)
move = Move((7,7),(6,7))
self.assertTrue(move in rook_r.get_moves((7,7),board))
move.perform(board)
moves = king.get_moves((4, 7), board)
self.assertFalse(Castling("left",board.white) in moves)
self.assertFalse(Castling("right",board.white) in moves)
def test_rook_moves(self): '''Test the rook's moves unblocked''' board = Board(10, 10) rook = Rook(board.white) i, j = 4, 4 board[i, j].piece = rook moves = rook.get_moves((i, j), board) self.assertTrue(Move((i,j),(i,0)) in moves) self.assertTrue(Move((i,j),(i,7)) in moves) self.assertTrue(Move((i,j),(0,j)) in moves) self.assertTrue(Move((i,j),(7,j)) in moves)
def test_castling_permitted_after_undo(self):
board = Board(10, 10)
king = King(board.white)
rook_l = Rook(board.white)
rook_r = Rook(board.white)
board[0, 7].piece = rook_l
board[7, 7].piece = rook_r
board[4, 7].piece = king
moves = king.get_moves((4, 7), board)
self.assertTrue(Castling("left",board.white) in moves)
self.assertTrue(Castling("right",board.white) in moves)
move = Move((0,7),(1,7))
self.assertTrue(move in rook_l.get_moves((0,7),board))
move.perform(board)
move.undo(board)
moves = king.get_moves((4, 7), board)
self.assertTrue(Castling("left",board.white) in moves)
self.assertTrue(Castling("right",board.white) in moves)
