def test_left_diagonal(self): self.board.update(converter.long_alg("b2b3", self.board)) moves = list( self.board.piece_at_square( Location.from_string("c1")).possible_moves(self.board)) self.assertEqual(len(moves), 2) self.assertEqual(moves[0], converter.long_alg("c1b2", self.board)) self.assertEqual(moves[1], converter.long_alg("c1a3", self.board))
def test_update_pawn_moves_two_steps(self): pawn = self.board.piece_at_square(Location.from_string("e2")) self.board.update(converter.long_alg("e2e4", self.board)) self.assertIsInstance(pawn, Pawn) self.assertEqual(self.board.piece_at_square(Location.from_string("e4")), pawn) self.assertIsNone(self.board.piece_at_square(Location.from_string("e2"))) self.assertIsNone(self.board.piece_at_square(Location.from_string("e3"))) self.assertTrue(pawn.just_moved_two_steps) self.board.update(converter.long_alg("d2d4", self.board)) self.assertFalse(pawn.just_moved_two_steps)
def test_advantage_as_result(self): self.assertEqual(self.board.advantage_as_result(converter.long_alg("e2e4", self.board), piece_const.PieceValues()), 0) self.board.position[1][3] = None self.assertEqual( self.board.advantage_as_result(converter.long_alg("d1d7", self.board), piece_const.PieceValues()), 0) self.board.update(converter.short_alg("e3", color.white, self.board)) self.assertEqual( self.board.advantage_as_result( converter.short_alg("Bd2", color.white, self.board), piece_const.PieceValues()), -1)
def test_update_pawn_moves_two_steps(self): pawn = self.board.piece_at_square(Location.from_string("e2")) self.board.update(converter.long_alg("e2e4", self.board)) self.assertIsInstance(pawn, Pawn) self.assertEqual( self.board.piece_at_square(Location.from_string("e4")), pawn) self.assertIsNone( self.board.piece_at_square(Location.from_string("e2"))) self.assertIsNone( self.board.piece_at_square(Location.from_string("e3"))) self.assertTrue(pawn.just_moved_two_steps) self.board.update(converter.long_alg("d2d4", self.board)) self.assertFalse(pawn.just_moved_two_steps)
def test_all_possible_moves_2(self): self.board.update(converter.long_alg("e2e4", self.board)) moves = {"a7a6", "a7a5", "b7b6", "b7b5", "c7c6", "c7c5", "d7d6", "d7d5", "e7e6", "e7e5", "f7f6", "f7f5", "g7g6", "g7g5", "h7h6", "h7h5", "b8a6", "b8c6", "g8f6", "g8h6"} self.assertEqual(moves, {str(move) for move in self.board.all_possible_moves(color.black)})
def test_possible_moves(self): self.board = Board([[None for _ in range(8)] for _ in range(8)]) my_king = King(color.white, Location.from_string("f3")) self.board.place_piece_at_square(my_king, Location.from_string("f3")) moves = ['f3f4', 'f3g3', 'f3f2', 'f3e3', 'f3g4', 'f3e4', 'f3g2', 'f3e2'] for i, move in enumerate(my_king.possible_moves(self.board)): self.assertEqual(move, converter.long_alg(moves[i], self.board))
def test_capture(self): self.board.move_piece(Location.from_string("g1"), Location.from_string("g7")) moves = list( self.board.piece_at_square( Location.from_string("f8")).possible_moves(self.board)) self.assertEqual(len(moves), 1) self.assertEqual(moves[0], converter.long_alg("f8g7", self.board))
def test_advantage_as_result(self): self.assertEqual( self.board.advantage_as_result( converter.long_alg("e2e4", self.board), piece_const.PieceValues()), 0) self.board.position[1][3] = None self.assertEqual( self.board.advantage_as_result( converter.long_alg("d1d7", self.board), piece_const.PieceValues()), 0) self.board.update(converter.short_alg("e3", color.white, self.board)) self.assertEqual( self.board.advantage_as_result( converter.short_alg("Bd2", color.white, self.board), piece_const.PieceValues()), -1)
def test_all_possible_moves_2(self): self.board.update(converter.long_alg("e2e4", self.board)) moves = { "a7a6", "a7a5", "b7b6", "b7b5", "c7c6", "c7c5", "d7d6", "d7d5", "e7e6", "e7e5", "f7f6", "f7f5", "g7g6", "g7g5", "h7h6", "h7h5", "b8a6", "b8c6", "g8f6", "g8h6" } self.assertEqual( moves, {str(move) for move in self.board.all_possible_moves(color.black)})
def test_add(self): self.assertEqual( len(list(self.board.get_king(color.white).add(lambda x: x.shift_up(), self.board))), 0) self.board.update(converter.long_alg("e2e4", self.board)) # King should be able to move up self.assertEqual( len(list(self.board.get_king(color.white).add(lambda x: x.shift_up(), self.board))), 1) # King should not be able to move down self.assertEqual( len(list(self.board.get_king(color.white).add(lambda x: x.shift_down(), self.board))), 0) # King should not be able to move left self.assertEqual( len(list(self.board.get_king(color.white).add(lambda x: x.shift_left(), self.board))), 0) # King should not be able to move right self.assertEqual( len(list(self.board.get_king(color.white).add(lambda x: x.shift_right(), self.board))), 0) # King should not be able to move up left self.assertEqual( len(list(self.board.get_king(color.white).add(lambda x: x.shift_up_left(), self.board))), 0) # King should not be able to move down right self.assertEqual( len(list(self.board.get_king(color.white).add(lambda x: x.shift_down_right(), self.board))), 0) # King should not be able to move down left self.assertEqual( len(list(self.board.get_king(color.white).add(lambda x: x.shift_down_left(), self.board))), 0) # King should not be able to move up right self.assertEqual( len(list(self.board.get_king(color.white).add(lambda x: x.shift_up_right(), self.board))), 0)
def test_in_check(self): self.board = Board([[None for _ in range(8)] for _ in range(8)]) my_king = King(color.white, Location.from_string("f3")) self.board.place_piece_at_square(my_king, Location.from_string("f3")) self.board.place_piece_at_square(Rook(color.black, Location.from_string("f1")), Location.from_string("f1")) print(self.board.piece_at_square(Location.from_string("f1")).color) print(self.board) print(my_king.color) print(color.white == color.black) self.assertTrue(my_king.in_check(self.board)) self.board = Board.init_default() self.board.update(converter.long_alg("f2f3", self.board)) self.board.move_piece(Location.from_string("d8"), Location.from_string("g3")) self.assertTrue(self.board.get_king(color.white).in_check(self.board))
def test_in_check_as_result(self): self.assertFalse(self.board.get_king(color.white).in_check_as_result(self.board, converter.long_alg("e2e4", self.board))) self.board.move_piece(Location.from_string("e1"), Location.from_string("e3")) self.board.move_piece(Location.from_string("e8"), Location.from_string("e5"))