def test_rotation2(self):
"""
Test create_rotations with 180 rotation only.
"""
result = Quilt.create_rotations(self.a, 2)
expected = np.asarray([[0, 1, 2, 3], [5, 6, 7, 8], [4, 9, 2, 3],
[0, 0, 0, 0], [3, 2, 9, 4], [8, 7, 6, 5],
[3, 2, 1, 0]])
np.testing.assert_array_equal(expected, result)
def test_rotation4(self):
"""
Test create_rotations with 4 rotations (every 90 degrees)
"""
result = Quilt.create_rotations(self.a, 4)
expected = np.asarray([[0, 1, 2, 3, 0, 3, 8, 3, 0, 4, 5, 0],
[5, 6, 7, 8, 0, 2, 7, 2, 0, 9, 6, 1],
[4, 9, 2, 3, 0, 1, 6, 9, 0, 2, 7, 2],
[0, 0, 0, 0, 0, 0, 5, 4, 0, 3, 8, 3],
[3, 2, 9, 4, 0, 0, 0, 0, 0, 0, 0, 0],
[8, 7, 6, 5, 0, 0, 0, 0, 0, 0, 0, 0],
[3, 2, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0]])
np.testing.assert_array_equal(expected, result)
def test_rotation4_invert(self):
"""
Test create_rotations with 4 rotations (every 90 degrees). Test it with
a matrix that has height>width
"""
a = np.rot90(self.a, -1)
result = Quilt.create_rotations(a, 4)
expected = np.asarray([[0, 1, 2, 3, 0, 3, 8, 3, 0, 4, 5, 0],
[5, 6, 7, 8, 0, 2, 7, 2, 0, 9, 6, 1],
[4, 9, 2, 3, 0, 1, 6, 9, 0, 2, 7, 2],
[0, 0, 0, 0, 0, 0, 5, 4, 0, 3, 8, 3],
[3, 2, 9, 4, 0, 0, 0, 0, 0, 0, 0, 0],
[8, 7, 6, 5, 0, 0, 0, 0, 0, 0, 0, 0],
[3, 2, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0]])
np.testing.assert_array_equal(expected, result)