导出test_mql机械节点总力

tests/test_test_mql_pnl0001_segment.py

@@ -552,6 +552,18 @@ class TestMqlPn3P4NodeChamberSegmentTests(unittest.TestCase):
             snapshot.mechanical.states[0].alias,
             "mass_friction_endstops_10",
         )
+        self.assertAlmostEqual(
+            snapshot.mechanical_node_total_force_by_alias[
+                "dynamic_mechanical_node_alternative_2"
+            ],
+            0.0,
+        )
+        self.assertAlmostEqual(
+            snapshot.mechanical_node_total_force_by_alias[
+                "dynamic_mechanical_node_alternative_3"
+            ],
+            -326.7174678324367,
+        )
         self.assertAlmostEqual(rhs[112], -0.8167936695810917)
 
     def test_full_state_closure_applies_elastic_endstop_force_to_small_masses(self) -> None:
@@ -563,6 +575,7 @@ class TestMqlPn3P4NodeChamberSegmentTests(unittest.TestCase):
         state = closure.initial_state_vector()
         state[113] = 1.0e-3
 
+        snapshot = closure.snapshot(state)
         rhs = closure.rhs(state)
         initial_rhs = closure.rhs(closure.initial_state_vector())
         endstop = closure.mechanical_closure.assembly.elastic_endstops[
@@ -577,6 +590,12 @@ class TestMqlPn3P4NodeChamberSegmentTests(unittest.TestCase):
             rhs[112],
             initial_rhs[112] - expected_contact_force / mass,
         )
+        self.assertAlmostEqual(
+            snapshot.mechanical_node_total_force_by_alias[
+                "dynamic_mechanical_node_alternative_2"
+            ],
+            expected_contact_force,
+        )
 
     def test_simulates_full_132_state_closure(self) -> None:
         solution = self.system.simulate_full_state_from_spec(