def test_ascending_ints(self):
n = 4
mat = MyMatrix.ascending_ints(n)
self.assertEqual(n * n, len(mat))
expected_mat = [
[0, 1, 2, 3],
[4, 5, 6, 7],
[8, 9, 10, 11],
[12, 13, 14, 15],
]
for i in range(n):
for j in range(n):
self.assertEqual(expected_mat[i][j], mat[i, j])
del mat
n = 3
mat = MyMatrix.ascending_ints(n, 1)
expected_mat = [
[1, 2, 3],
[4, 5, 6],
[7, 8, 9],
]
for i in range(n):
for j in range(n):
self.assertEqual(expected_mat[i][j], mat[i, j])
def test_rotate(self):
mat = MyMatrix()
mat.rotate()
self.assertIsNotNone(mat)
n = 1
mat = MyMatrix.ascending_ints(n, 1)
mat.rotate()
expected_mat = [
[1],
]
for i in range(n):
for j in range(n):
self.assertEqual(expected_mat[i][j], mat[i, j])
n = 3
mat = MyMatrix.ascending_ints(n, 1)
mat.rotate()
expected_mat = [
[7, 4, 1],
[8, 5, 2],
[9, 6, 3],
]
for i in range(n):
for j in range(n):
self.assertEqual(expected_mat[i][j], mat[i, j])