def test_check_board_for_r(self):
cells = {
(0, 0): 'K',
(0, 1): None,
(1, 0): None,
(1, 1): None,
}
board = ChessBoard(3, 3, [])
self.assertIsNone(board.check_board_for_r(0, 2, cells))
self.assertIsNone(board.check_board_for_r(2, 0, cells))
self.assertEqual(board.check_board_for_r(2, 1, cells), {(2, 0), (2, 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)
