def test_multiply(self):
m1 = Matrix44(np.arange(self._size))
m2 = Matrix44(np.arange(self._size)[::-1])
m = m1 * m2
self.assertTrue(np.array_equal(m, matrix44.multiply(m2, m1)))
m1 = Matrix44(np.arange(self._size))
m2 = Matrix33(np.arange(9))
m = m1 * m2
self.assertTrue(np.array_equal(m, matrix44.multiply(matrix44.create_from_matrix33(m2), m1)))
def test_decompose(self):
# define expectations for multiple cases
testsets = [
(Vector3([1, 1, 2],
dtype='f4'), Quaternion.from_y_rotation(np.pi,
dtype='f4'),
Vector3([10, 0, -5], dtype='f4'),
Matrix44([
[-1, 0, 0, 0],
[0, 1, 0, 0],
[0, 0, -2, 0],
[10, 0, -5, 1],
],
dtype='f4')),
(Vector3([-1, 3, .5], dtype='f4'),
Quaternion.from_axis_rotation(Vector3([.75, .75, 0],
dtype='f4').normalized,
np.pi,
dtype='f4').normalized,
Vector3([1, -1, 1], dtype='f4'),
Matrix44([
[0, -1, 0, 0],
[3, 0, 0, 0],
[0, 0, -.5, 0],
[1, -1, 1, 1],
],
dtype='f4')),
]
for expected_scale, expected_rotation, expected_translation, expected_model in testsets:
# compose model matrix using original inputs
s = Matrix44.from_scale(expected_scale, dtype='f4')
r = Matrix44.from_quaternion(expected_rotation, dtype='f4')
t = Matrix44.from_translation(expected_translation, dtype='f4')
m = t * r * s
# check that it's the same as the expected matrix
np.testing.assert_almost_equal(np.array(m),
np.array(expected_model))
self.assertTrue(m.dtype == expected_model.dtype)
self.assertTrue(isinstance(m, expected_model.__class__))
# decompose this matrix and recompose the model matrix from the decomposition
ds, dr, dt = m.decompose()
ds = Matrix44.from_scale(ds, dtype='f4')
dr = Matrix44.from_quaternion(dr, dtype='f4')
dt = Matrix44.from_translation(dt, dtype='f4')
dm = dt * dr * ds
# check that it's the same as the original matrix
np.testing.assert_almost_equal(np.array(m), np.array(dm))
self.assertTrue(m.dtype == dm.dtype)
self.assertTrue(isinstance(dm, m.__class__))
def test_create(self):
m = Matrix44()
self.assertTrue(np.array_equal(m, np.zeros(self._shape)))
self.assertEqual(m.shape, self._shape)
m = Matrix44(np.arange(self._size))
self.assertEqual(m.shape, self._shape)
m = Matrix44([[1,2,3,4],[5,6,7,8],[9,10,11,12],[13,14,15,16]])
self.assertEqual(m.shape, self._shape)
m = Matrix44(Matrix44())
self.assertTrue(np.array_equal(m, np.zeros(self._shape)))
self.assertEqual(m.shape, self._shape)
def test_operators_matrix44(self):
m1 = Matrix44.identity()
m2 = Matrix44.from_x_rotation(0.5)
# add
self.assertTrue(np.array_equal(m1 + m2, matrix44.create_identity() + matrix44.create_from_x_rotation(0.5)))
# subtract
self.assertTrue(np.array_equal(m1 - m2, matrix44.create_identity() - matrix44.create_from_x_rotation(0.5)))
# multiply
self.assertTrue(np.array_equal(m1 * m2, matrix44.multiply(matrix44.create_from_x_rotation(0.5), matrix44.create_identity())))
# divide
self.assertRaises(ValueError, lambda: m1 / m2)
# inverse
self.assertTrue(np.array_equal(~m2, matrix44.inverse(matrix44.create_from_x_rotation(0.5))))
# ==
self.assertTrue(Matrix44() == Matrix44())
self.assertFalse(Matrix44() == Matrix44([1. for n in range(16)]))
# !=
self.assertTrue(Matrix44() != Matrix44([1. for n in range(16)]))
self.assertFalse(Matrix44() != Matrix44())
def test_create_from_matrix44(self):
m1 = Matrix33.identity()
m = Matrix44.from_matrix33(m1)
self.assertTrue(np.array_equal(m, np.eye(4)))
m = Matrix44(m1)
self.assertTrue(np.array_equal(m, np.eye(4)))
def test_create_from_quaternion(self):
q = Quaternion()
m = Matrix44.from_quaternion(q)
self.assertTrue(np.array_equal(m, np.eye(4)))
self.assertTrue(np.array_equal(m.quaternion, q))
m = Matrix44(q)
self.assertTrue(np.array_equal(m, np.eye(4)))
def test_accessors(self):
m = Matrix44(np.arange(self._size))
self.assertTrue(np.array_equal(m.m1, [0, 1, 2, 3]))
self.assertTrue(np.array_equal(m.m2, [4, 5, 6, 7]))
self.assertTrue(np.array_equal(m.m3, [8, 9, 10, 11]))
self.assertTrue(np.array_equal(m.m4, [12, 13, 14, 15]))
self.assertTrue(np.array_equal(m.r1, [0, 1, 2, 3]))
self.assertTrue(np.array_equal(m.r2, [4, 5, 6, 7]))
self.assertTrue(np.array_equal(m.r3, [8, 9, 10, 11]))
self.assertTrue(np.array_equal(m.r4, [12, 13, 14, 15]))
self.assertTrue(np.array_equal(m.c1, [0, 4, 8, 12]))
self.assertTrue(np.array_equal(m.c2, [1, 5, 9, 13]))
self.assertTrue(np.array_equal(m.c3, [2, 6, 10, 14]))
self.assertTrue(np.array_equal(m.c4, [3, 7, 11, 15]))
self.assertEqual(m.m11, 0)
self.assertEqual(m.m12, 1)
self.assertEqual(m.m13, 2)
self.assertEqual(m.m14, 3)
self.assertEqual(m.m21, 4)
self.assertEqual(m.m22, 5)
self.assertEqual(m.m23, 6)
self.assertEqual(m.m24, 7)
self.assertEqual(m.m31, 8)
self.assertEqual(m.m32, 9)
self.assertEqual(m.m33, 10)
self.assertEqual(m.m34, 11)
self.assertEqual(m.m41, 12)
self.assertEqual(m.m42, 13)
self.assertEqual(m.m43, 14)
self.assertEqual(m.m44, 15)
m.m11 = 1
self.assertEqual(m.m11, 1)
self.assertEqual(m[0, 0], 1)
m.m11 += 1
self.assertEqual(m.m11, 2)
self.assertEqual(m[0, 0], 2)
