Fix squares; add dimension property; add some tests

test.py

@@ -0,0 +1,53 @@
+#!env python3
+# Eryn Wells <eryn@erynwells.me>
+'''
+Unit tests for the Sudoku module.
+'''
+
+import unittest
+import sudoku
+
+class Sudoku4Tests(unittest.TestCase):
+    def setUp(self):
+        self.board = sudoku.Sudoku(size=4)
+
+    def test_that_board_is_sane(self):
+        self.assertEqual(self.board.size, 4)
+        self.assertEqual(len(self.board.board), 4**2)
+        self.assertEqual(self.board.dimension, 2)
+
+    def test_rows(self):
+        expected_rows = [
+            [ 0,  1,  2,  3],
+            [ 4,  5,  6,  7],
+            [ 8,  9, 10, 11],
+            [12, 13, 14, 15]
+        ]
+        for (row, exrow) in zip(self.board.rows, expected_rows):
+            row_list = list(row)
+            with self.subTest(row=row_list, ex=exrow):
+                self.assertEqual(row_list, exrow)
+
+    def test_columns(self):
+        expected_columns = [
+            [0, 4,  8, 12],
+            [1, 5,  9, 13],
+            [2, 6, 10, 14],
+            [3, 7, 11, 15]
+        ]
+        for (col, excol) in zip(self.board.columns, expected_columns):
+            col_list = list(col)
+            with self.subTest(col=col_list, ex=excol):
+                self.assertEqual(col_list, excol)
+
+    def test_squares(self):
+        expected_squares = [
+            [ 0,  1,  4,  5],
+            [ 2,  3,  6,  7],
+            [ 8,  9, 12, 13],
+            [10, 11, 14, 15]
+        ]
+        for (sq, exsq) in zip(self.board.squares, expected_squares):
+            sq_list = list(sq)
+            with self.subTest(sq=sq_list, ex=exsq):
+                self.assertEqual(sq_list, exsq)