source/tests/bl_pyapi_mathutils.py

   1 import unittest
   2 from test import support
   3 from mathutils import Matrix, Vector
   4
   5
   6 class MatrixTesting(unittest.TestCase):
   7     def test_matrix_column_access(self):
   8         #mat =
   9         #[ 1  2  3  4 ]
  10         #[ 1  2  3  4 ]
  11         #[ 1  2  3  4 ]
  12         mat = Matrix(((1, 11, 111),
  13                       (2, 22, 222),
  14                       (3, 33, 333),
  15                       (4, 44, 444)))
  16
  17         self.assertEqual(mat[0], Vector((1, 11, 111)))
  18         self.assertEqual(mat[1], Vector((2, 22, 222)))
  19         self.assertEqual(mat[2], Vector((3, 33, 333)))
  20         self.assertEqual(mat[3], Vector((4, 44, 444)))
  21
  22     def test_item_access(self):
  23         args = ((1, 4, 0, -1),
  24                 (2, -1, 2, -2),
  25                 (0, 3, 8, 3),
  26                 (-2, 9, 1, 0))
  27
  28         mat = Matrix(args)
  29
  30         for row in range(4):
  31             for col in range(4):
  32                 self.assertEqual(mat[row][col], args[row][col])
  33
  34         self.assertEqual(mat[0][2], 0)
  35         self.assertEqual(mat[3][1], 9)
  36         self.assertEqual(mat[2][3], 3)
  37         self.assertEqual(mat[0][0], 1)
  38         self.assertEqual(mat[3][3], 0)
  39
  40     def test_item_assignment(self):
  41         mat = Matrix() - Matrix()
  42         indices = (0, 0), (1, 3), (2, 0), (3, 2), (3, 1)
  43         checked_indices = []
  44         for row, col in indices:
  45             mat[row][col] = 1
  46
  47         for row in range(4):
  48             for col in range(4):
  49                 if mat[row][col]:
  50                     checked_indices.append((row, col))
  51
  52         for item in checked_indices:
  53             self.assertIn(item, indices)
  54
  55     def test_matrix_to_3x3(self):
  56         #mat =
  57         #[ 1  2  3  4  ]
  58         #[ 2  4  6  8  ]
  59         #[ 3  6  9  12 ]
  60         #[ 4  8  12 16 ]
  61         mat = Matrix(tuple((i, 2 * i, 3 * i, 4 * i) for i in range(1, 5)))
  62         mat_correct = Matrix(((1, 2, 3), (2, 4, 6), (3, 6, 9)))
  63         self.assertEqual(mat.to_3x3(), mat_correct)
  64
  65     def test_matrix_to_translation(self):
  66         mat = Matrix()
  67         mat[0][3] = 1
  68         mat[1][3] = 2
  69         mat[2][3] = 3
  70         self.assertEqual(mat.to_translation(), Vector((1, 2, 3)))
  71
  72     def test_matrix_translation(self):
  73         mat = Matrix()
  74         mat.translation = Vector((1, 2, 3))
  75         self.assertEqual(mat[0][3], 1)
  76         self.assertEqual(mat[1][3], 2)
  77         self.assertEqual(mat[2][3], 3)
  78
  79     def test_non_square_mult(self):
  80         mat1 = Matrix(((1, 2, 3),
  81                        (4, 5, 6)))
  82         mat2 = Matrix(((1, 2),
  83                        (3, 4),
  84                        (5, 6)))
  85
  86         prod_mat1 = Matrix(((22, 28),
  87                             (49, 64)))
  88         prod_mat2 = Matrix(((9, 12, 15),
  89                             (19, 26, 33),
  90                             (29, 40, 51)))
  91
  92         self.assertEqual(mat1*mat2, prod_mat1)
  93         self.assertEqual(mat2 * mat1, prod_mat2)
  94
  95     def test_mat4x4_vec3D_mult(self):
  96         mat = Matrix(((1, 0, 2, 0),
  97                       (0, 6, 0, 0),
  98                       (0, 0, 1, 1),
  99                       (0, 0, 0, 1)))
 100
 101         vec = Vector((1, 2, 3))
 102
 103         prod_mat_vec = Vector((7, 12, 4))
 104         prod_vec_mat = Vector((1, 12, 5))
 105
 106         self.assertEqual(mat * vec, prod_mat_vec)
 107         self.assertEqual(vec * mat, prod_vec_mat)
 108
 109     def test_mat_vec_mult(self):
 110         mat1 = Matrix()
 111
 112         vec = Vector((1, 2))
 113
 114         self.assertRaises(ValueError, mat1.__mul__, vec)
 115         self.assertRaises(ValueError, vec.__mul__, mat1)
 116
 117         mat2 = Matrix(((1, 2),
 118                        (-2, 3)))
 119
 120         prod = Vector((5, 4))
 121
 122         self.assertEqual(mat2 * vec, prod)
 123
 124     def test_matrix_inverse(self):
 125         mat = Matrix(((1, 4, 0, -1),
 126                       (2, -1, 2, -2),
 127                       (0, 3, 8, 3),
 128                       (-2, 9, 1, 0)))
 129
 130         inv_mat = (1 / 285) * Matrix(((195, -57, 27, -102),
 131                                       (50, -19, 4, 6),
 132                                       (-60, 57, 18, 27),
 133                                       (110, -133, 43, -78)))
 134
 135         self.assertEqual(mat.inverted(), inv_mat)
 136
 137     def test_matrix_mult(self):
 138         mat = Matrix(((1, 4, 0, -1),
 139                       (2, -1, 2, -2),
 140                       (0, 3, 8, 3),
 141                       (-2, 9, 1, 0)))
 142
 143         prod_mat = Matrix(((11, -9, 7, -9),
 144                            (4, -3, 12, 6),
 145                            (0, 48, 73, 18),
 146                            (16, -14, 26, -13)))
 147
 148         self.assertEqual(mat * mat, prod_mat)
 149
 150
 151 def test_main():
 152     try:
 153         support.run_unittest(MatrixTesting)
 154     except:
 155         import traceback
 156         traceback.print_exc()
 157
 158         # alert CTest we failed
 159         import sys
 160         sys.exit(1)
 161
 162 if __name__ == '__main__':
 163     test_main()