class TestGrid(unittest.TestCase): def setUp(self): self.grid = Grid() def test_display(self): for x in range(9): for y in range(9): if x==y: self.grid.set(x, y, x+1) self.assertEqual("1........\n.2.......\n..3......\n...4.....\n....5....\n.....6...\n......7..\n.......8.\n........9\n", self.grid.display()) def test_remove_candidate(self): self.assertEqual([1,2,3,4,5,6,7,8,9], self.grid.candidates[0][0]) self.grid.remove_candidate(0, 0, 1) self.assertEqual([2,3,4,5,6,7,8,9], self.grid.candidates[0][0]) def test_remove_candidate_no_exception_if_the_number_is_already_removed(self): self.grid.remove_candidate(0, 0, 1) try: self.grid.remove_candidate(0, 0, 1) except ValueError: self.fail("Removing a number already removed should not raise a ValueError") def test_load(self): stringToLoad = "..1..2..3\n.........\n456......\n.......7.\n.........\n....8....\n.........\n123456789\n..8..2..5\n" self.grid.load(StringIO(stringToLoad)) self.assertEqual(stringToLoad, self.grid.display()) def test_load_without_line_returns(self): self.grid.load(StringIO("..1..2..3.........456.............7..............8.............123456789..8..2..5")) self.assertEqual("..1..2..3\n.........\n456......\n.......7.\n.........\n....8....\n.........\n123456789\n..8..2..5\n", self.grid.display()) def test_load_grid_with_zeroes_and_random_line_returns(self): self.grid.load(StringIO("00100200300000000045\n600000000000007000000000000008\n0000000000000123456789008002005")) self.assertEqual("..1..2..3\n.........\n456......\n.......7.\n.........\n....8....\n.........\n123456789\n..8..2..5\n", self.grid.display()) def test_load_with_a_grid_longer_than_9x9_characters(self): with self.assertRaises(GridLoadingException) as ex: Grid(StringIO("1234567891........................................................................")) self.assertEqual("Loading Error : too many characters", ex.exception.message) def test_load_with_invalid_characters(self): with self.assertRaises(GridLoadingException) as ex: Grid(StringIO("123456789\n.........\n.........\n...x.....\n.........\n.........\n.........\n.........\n.........\n")) self.assertEqual("Loading Error : invalid character on line 4, column 4 : ...x.....", ex.exception.message) def test_load_with_leading_comments_and_empty_lines(self): stringToLoad = "..1..2..3\n.........\n456......\n.......7.\n.........\n....8....\n.........\n123456789\n..8..2..5\n" self.grid.load(StringIO("# comment\n\n\n" + stringToLoad)) self.assertEqual(stringToLoad, self.grid.display()) def test_load_with_more_than_9_lines(self): with self.assertRaises(GridLoadingException) as ex: Grid(StringIO("123456789\n.........\n.........\n.........\n.........\n.........\n.........\n.........\n.........\n\n.........\n")) self.assertEqual("Loading Error : too many characters", ex.exception.message) def test_load_incomplete_grid(self): with self.assertRaises(GridLoadingException) as ex: Grid(StringIO("123456789\n.........\n....\n.........\n.........\n\n.........\n")) self.assertEqual("Loading Error : incomplete grid, missing cells", ex.exception.message) def test_get_solution(self): self.grid.load(StringIO("1........\n.2.......\n..3......\n...4.....\n....5....\n.....6...\n......7..\n.......8.\n........9\n")) self.assertEqual(2, self.grid.get_solution(1,1)) self.assertEqual(None, self.grid.get_solution(2,1)) def test_iter_next_starting_cell_with_blank_grid(self): iterator = iter(self.grid) self.assertEqual([0, 0], iterator.next()) def test_iter_next(self): self.grid.load(StringIO("41.6.8972\n3.2479185\n789215364\n926341758\n138756429\n574982631\n257164893\n843597216\n691823547\n")) iterator = iter(self.grid) self.assertEqual([2, 0], iterator.next(), "first") self.assertEqual([4, 0], iterator.next(), "second") self.assertEqual([1, 1], iterator.next(), "third") def test_iter_next_exit_if_complete_grid(self): self.grid.load(StringIO("415638972\n362479185\n789215364\n926341758\n138756429\n574982631\n257164893\n843597216\n691823547\n")) iterator = iter(self.grid) with self.assertRaises(StopIteration) as ex: iterator.next() def test_iter_next_exit_if_complete_cycle(self): self.grid.load(StringIO("41.638972\n362479185\n789215364\n926341758\n138756429\n574982631\n257164893\n843597216\n691823547\n")) iterator = iter(self.grid) self.assertEqual([2, 0], iterator.next()) with self.assertRaises(StopIteration) as ex: iterator.next() def test_start_coordinates_of_block(self): self.assertEqual((3, 0), startCoordinatesOfBlock(4, 1)) self.assertEqual((0, 0), startCoordinatesOfBlock(0, 0)) self.assertEqual((6, 6), startCoordinatesOfBlock(7, 6)) self.assertEqual((0, 6), startCoordinatesOfBlock(2, 8)) def test_is_solved(self): self.assertFalse(self.grid.is_solved()) self.grid.load(StringIO("415638972\n362479185\n789215364\n926341758\n138756429\n574982631\n257164893\n843597216\n691823547\n")) self.assertTrue(self.grid.is_solved()) self.grid.load(StringIO("41.638972\n362479185\n789215364\n926341758\n138756429\n574982631\n257164893\n843597216\n691823547\n")) self.assertFalse(self.grid.is_solved())