组合test_mql 132状态总闭包

PythonModels/systems/test_mql_mechanical.py

@@ -188,14 +188,21 @@ class TestMqlMechanicalMassClosure:
         )
         return TestMqlMechanicalMassSnapshot(states=states)
 
-    def rhs(self, state_vector: list[float]) -> list[float]:
+    def rhs(
+        self,
+        state_vector: list[float],
+        *,
+        force_by_mass_alias: dict[str, float] | None = None,
+    ) -> list[float]:
         snapshot = self.snapshot(state_vector)
+        force_by_mass_alias = force_by_mass_alias or {}
         derivatives: list[float] = []
         for state in snapshot.states:
             mass = self.assembly.masses[state.alias].endstop()
             acceleration, velocity = mass.derivatives(
                 velocity_m_s=state.velocity_m_s,
                 displacement_m=state.displacement_m,
+                port_1_force_n=force_by_mass_alias.get(state.alias, 0.0),
             )
             derivatives.extend([acceleration, velocity])
         return derivatives