def test_can_reach_castle_queen(self):
rook1 = Rook(self.white, (0, 0))
king = King(self.white, (4, 0))
rook2 = Rook(self.white, (7, 0))
self.board.add_piece(king)
self.board.add_piece(rook1)
self.board.add_piece(rook2)
self.assertTrue(king.can_reach(self.board, (2, 0)))
def test_cant_reach(self):
king = King(self.white, (4, 4))
self.board.add_piece(king)
self.assertFalse(king.can_reach(self.board, (2, 2)))
def test_can_reach_normal(self):
king = King(self.white, (4, 4))
self.board.add_piece(king)
self.assertTrue(king.can_reach(self.board, (3, 3)))
