def test_add_black_cell(self): grid = Grid(9) grid.add_black_cell(1, 5) grid.add_black_cell(4, 9) expected = [(1, 5), (4, 9), (6, 1), (9, 5)] actual = grid.get_black_cells() self.assertEqual(expected, actual)
def __init__(self, grid: Grid, *args, **kwargs): super().__init__(grid.n, *args, **kwargs) self.grid = grid self.black_cells = grid.get_black_cells() self.numbered_cells = grid.get_numbered_cells()