def test_empty(self): block = SudokuBlock() for x in range(3): self.assertEqual(block.row(x), [None, None, None]) self.assertEqual(block.col(x), [None, None, None])
def test_row(self): block = SudokuBlock([1, 2, 3, 4, 5, 6, 7, 8, 9]) self.assertEqual(block.row(0), [1, 2, 3]) self.assertEqual(block.row(1), [4, 5, 6]) self.assertEqual(block.row(2), [7, 8, 9])