def test_get_answer_out_of_bounds(self): from sudoku_board import SudokuBoard board = SudokuBoard() with self.assertRaises(IndexError): board.get_answer(0, 0) with self.assertRaises(IndexError): board.get_answer(10, 10)
def test_get_answer(self): from sudoku_board import SudokuBoard board = SudokuBoard() board.add_answer(1, 2, 4) self.assertEqual(board.get_answer(1, 2), 4) with self.assertRaises(IndexError): board.get_answer(0, 0)