class SudokuTestCase(unittest.TestCase): """ Tests the validity of the solve() method on the challenge problem to check whether the solved puzzle matches the correct solution """ def setUp(self): self.sudoku = Sudoku([ [0, 3, 5, 2, 9, 0, 8, 6, 4], [0, 8, 2, 4, 1, 0, 7, 0, 3], [7, 6, 4, 3, 8, 0, 0, 9, 0], [2, 1, 8, 7, 3, 9, 0, 4, 0], [0, 0, 0, 8, 0, 4, 2, 3, 0], [0, 4, 3, 0, 5, 2, 9, 7, 0], [4, 0, 6, 5, 7, 1, 0, 0, 9], [3, 5, 9, 0, 2, 8, 4, 1, 7], [8, 0, 0, 9, 0, 0, 5, 2, 6] ]) def test_sudoku_puzzle_not_solved(self): self.assertFalse(self.sudoku.isSolved()) def test_sudoku_puzzle_solved(self): self.sudoku.solve() self.assertTrue(self.sudoku.isSolved()) def test_sudoku_valid_solution(self): self.sudoku.solve() solution = np.array([[1, 3, 5, 2, 9, 7, 8, 6, 4], [9, 8, 2, 4, 1, 6, 7, 5, 3], [7, 6, 4, 3, 8, 5, 1, 9, 2], [2, 1, 8, 7, 3, 9, 6, 4, 5], [5, 9, 7, 8, 6, 4, 2, 3, 1], [6, 4, 3, 1, 5, 2, 9, 7, 8], [4, 2, 6, 5, 7, 1, 3, 8, 9], [3, 5, 9, 6, 2, 8, 4, 1, 7], [8, 7, 1, 9, 4, 3, 5, 2, 6]]) equality = {i == j for i, j in zip( self.sudoku.puzzle.flatten(), solution.flatten())} self.assertTrue(True in equality and False not in equality)
class SudokuTestCase2(unittest.TestCase): """ Tests the validity of the solve() method of the Sudoku class to make sure that the ending puzzle matches the correct solution """ def setUp(self): self.sudoku = Sudoku([ [0, 0, 0, 0, 5, 6, 4, 0, 0], [6, 0, 7, 0, 0, 1, 0, 8, 9], [0, 1, 0, 0, 0, 0, 0, 6, 0], [7, 2, 0, 3, 0, 9, 0, 5, 0], [1, 3, 0, 0, 6, 0, 0, 7, 2], [0, 9, 0, 1, 0, 7, 0, 4, 3], [0, 7, 0, 0, 0, 0, 0, 9, 0], [8, 5, 0, 9, 0, 0, 3, 0, 6], [0, 0, 1, 5, 8, 0, 0, 0, 0] ]) def test_sudoku_solved(self): self.sudoku.solve() self.sudoku.isSolved() def test_sudoku_valid_solution(self): self.sudoku.solve() solution = np.array([ [2, 8, 9, 7, 5, 6, 4, 3, 1], [6, 4, 7, 2, 3, 1, 5, 8, 9], [3, 1, 5, 4, 9, 8, 2, 6, 7], [7, 2, 6, 3, 4, 9, 1, 5, 8], [1, 3, 4, 8, 6, 5, 9, 7, 2], [5, 9, 8, 1, 2, 7, 6, 4, 3], [4, 7, 3, 6, 1, 2, 8, 9, 5], [8, 5, 2, 9, 7, 4, 3, 1, 6], [9, 6, 1, 5, 8, 3, 7, 2, 4], ]) equality = {i == j for i, j in zip( self.sudoku.puzzle.flatten(), solution.flatten())} self.assertTrue(True in equality and False not in equality)