def test_player_1_big(self): board = np.array([ [0, 0, 0, 1, 0, -1, 0, 0, 0,], [0, 0, 0, 1, 0, -1, 0, 0, 0,], [0, 0, 0, 1, 0, -1, 0, 0, 0,], [0, 0, 0, 1, -1, 0, 0, -1, 0,], [1, 1, 1, -1, 0, -1, -1, 0, 0,], [0, 0, 0, 1, -1, 0, 0, -1, -1,], [0, 0, 0, 1, 0, -1, 0, 0, 0,], [0, 0, 0, 1, 0, -1, 0, 1, 0,], [0, 0, 0, 0, 0, -1, 0, 0, 0,], ]) target = np.array([ [1, 1, 1, 2, 0, -2, -1, -1, -1,], [1, 1, 1, 2, 0, -2, -1, -1, -1,], [1, 1, 1, 2, 0, -2, -1, -1, -1,], [1, 1, 1, 2, -2, -1, -1, -2, -1,], [2, 2, 2, -2, -1, -2, -2, -1, -1,], [0, 0, 0, 2, -2, 0, 0, -2, -2,], [0, 0, 0, 2, 0, -2, 0, 0, 0,], [0, 0, 0, 2, 0, -2, 0, 2, 0,], [0, 0, 0, 0, 0, -2, 0, 0, 0,], ]) colored1 = color_board(board, 1) colored2 = color_board(board, -1) total = colored1 + colored2 self.assertEqualList(total, target)
def test_coloring_player_1(self): board = np.array([[1, 1, 1], [1, 0, 1], [1, 1, 1]]) target = np.array([[1, 1, 1], [1, 1, 1], [1, 1, 1]]) self.assertEqualList(color_board(board, 1), target) board = np.array([[1, 1, 1, -1, -1, -1], [1, 0, 1, -1, 0, -1], [1, 1, 1, -1, -1, -1]]) target = np.array([[1, 1, 1, -1, -1, -1], [1, 1, 1, -1, 0, -1], [1, 1, 1, -1, -1, -1]]) self.assertEqualList(color_board(board, 1), target)