Add a basic test for CosseratRod attributes

GazzolaLab · Aug 4, 2024 · 579c700 · 579c700
1 parent 6c47988
commit 579c700
Showing 1 changed file with 34 additions and 0 deletions.
diff --git a/backend/tests/test_rod_initialisation.py b/backend/tests/test_rod_initialisation.py
@@ -0,0 +1,34 @@
+# TODO: implement more tests based on functionality
+
+from elasticapp._PyCosseratRods import CosseratRod
+
+
+def test_attributes():
+    assert {
+        "acceleration_collection",
+        "alpha_collection",
+        "dilatation",
+        "director_collection",
+        "dissipation_constant_for_forces",
+        "dissipation_constant_for_torques",
+        "external_forces",
+        "external_torques",
+        "internal_forces",
+        "internal_torques",
+        "inv_mass_second_moment_of_inertia",
+        "lengths",
+        "loc",
+        "mass",
+        "mass_second_moment_of_inertia",
+        "material",
+        "n_elems",
+        "omega_collection",
+        "position_collection",
+        "radius",
+        "rest_lengths",
+        "rest_voronoi_lengths",
+        "tangents",
+        "velocity_collection",
+        "volume",
+        "voronoi_dilatation",
+    }.issubset(dir(CosseratRod))