Use matrix multiplication.

chrisdembia · Jun 6, 2024 · e5f9532 · e5f9532
1 parent b3dac86
commit e5f9532
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/yeadon/tests/test_solid.py b/yeadon/tests/test_solid.py
@@ -285,8 +285,8 @@ def test_solid():
     sol.set_orientation(position, C, True)
     testing.assert_allclose(sol.pos, position)
     testing.assert_allclose(sol._rot_mat, C)
-    testing.assert_allclose(sol.end_pos, position + (height * C * array([[0],
-        [0], [1]])))
+    testing.assert_allclose(sol.end_pos, position + (height * C @
+                                                     array([[0], [0], [1]])))
     testing.assert_allclose(sol.inertia, zeros((3, 3)))
 
     #TODO: complete tests for solid and the remaining classes in solid.py