def test_check_board_for_k(self):
cells = {
(0, 0): 'R',
(0, 1): None,
(0, 2): None,
(1, 0): None,
(2, 0): None,
}
board = ChessBoard(3, 3, [])
self.assertIsNone(board.check_board_for_k(1, 1, cells))
self.assertEqual(board.check_board_for_k(2, 2, cells), {(1, 1), (2, 1), (1, 2)})
def test_check_board_general(self):
cells = {
(0, 0): 'K',
(0, 1): None,
(1, 0): None,
(1, 1): None,
}
board = ChessBoard(3, 3, [])
br1 = board.check_board(1, 1, cells, 'R')
br2 = board.check_board_for_r(1, 1, cells)
bk1 = board.check_board(1, 1, cells, 'K')
bk2 = board.check_board_for_k(1, 1, cells)
self.assertEqual(br1, br2)
self.assertEqual(bk1, bk2)
self.assertNotEqual(br2, bk2)
