simple test for moit function

cmelab · May 4, 2022 · 2952c5f · 2952c5f
1 parent d3ed9ad
commit 2952c5f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/cmeutils/tests/test_geometry.py b/cmeutils/tests/test_geometry.py
@@ -4,10 +4,14 @@
 import pytest
 
 from base_test import BaseTest
-from cmeutils.geometry import get_plane_normal, angle_between_vectors
+from cmeutils.geometry import get_plane_normal, angle_between_vectors, moit
 
 
 class TestGeometry(BaseTest):
+    def test_moit(self):
+        _moit = moit(points=[(-1, 0, 0), (1, 0, 0)], masses=[1, 1])
+        assert np.array_equal(_moit, np.array([0, 2., 2.]))
+
     def test_get_plane_normal(self):
         points = np.array(
             [[ 1, 0, 0],