接入test_mql双管路气室与PN3节点

PythonModels/systems/test_mql.py

@@ -3860,6 +3860,7 @@ class TestMqlSystem:
         self.archive_path = archive_path or Path(__file__).resolve().parents[2] / AMESIM_ARCHIVE_RELATIVE_PATH
         self.network = SimulationNetwork(name=MODEL_NAME)
         self.pnl0001_assembly = self._build_pnl0001_assembly()
+        self.node3_assembly = self._build_node3_assembly()
         self.pneumatic_assembly = self._build_pneumatic_assembly()
         pneumatic_components = self._pneumatic_components_by_alias()
         for spec in COMPONENT_SPECS:
@@ -3889,6 +3890,12 @@ class TestMqlSystem:
 
         return build_test_mql_pnl0001_assembly(self.archive_path)
 
+    @staticmethod
+    def _build_node3_assembly():
+        from PythonModels.systems.test_mql_nodes import build_test_mql_node3_assembly
+
+        return build_test_mql_node3_assembly()
+
     def _pneumatic_components_by_alias(self) -> dict[str, Component]:
         return {
             **self.pneumatic_assembly.fixed_chambers,
@@ -3905,6 +3912,10 @@ class TestMqlSystem:
     def typed_pnl0001_line_count(self) -> int:
         return len(self.pnl0001_assembly.lines)
 
+    @property
+    def typed_node3_count(self) -> int:
+        return len(self.node3_assembly)
+
     def pneumatic_state_vector(self) -> list[float]:
         return self.network.initial_state_vector()
 
@@ -4161,6 +4172,86 @@ class TestMqlSystem:
             t_eval=t_eval,
         )
 
+    def pnl0001_pair_chamber_segment_closure_from_spec(
+        self,
+        spec,
+        *,
+        inlet_node_pressure_pa: float,
+        outlet_node_pressure_pa: float,
+        inlet_node_temperature_k: float = 293.15,
+        outlet_node_temperature_k: float = 293.15,
+    ):
+        """Insert both topology-derived PNL0001 states into the segment."""
+        from PythonModels.components.amesim_pneumatic import (
+            AmesimPneumaticOrifice,
+            AmesimPneumaticVolume,
+        )
+        from PythonModels.systems.test_mql_closure import (
+            TestMqlPneumaticBoundaryCondition,
+            TestMqlPnl0001PairChamberSegmentClosure,
+            TestMqlPnl0001PairChamberSegmentComponents,
+        )
+
+        inlet_line = self.pnl0001_assembly.lines[spec.inlet_line_alias]
+        outlet_line = self.pnl0001_assembly.lines[spec.outlet_line_alias]
+        volume = self.network.components[spec.volume_alias]
+        inlet_orifice = self.network.components[spec.inlet_orifice_alias]
+        outlet_orifice = self.network.components[spec.outlet_orifice_alias]
+        if not isinstance(volume, AmesimPneumaticVolume):
+            raise TypeError(f"{spec.volume_alias} is not an AMESim pneumatic volume")
+        if not isinstance(inlet_orifice, AmesimPneumaticOrifice):
+            raise TypeError(
+                f"{spec.inlet_orifice_alias} is not an AMESim pneumatic orifice"
+            )
+        if not isinstance(outlet_orifice, AmesimPneumaticOrifice):
+            raise TypeError(
+                f"{spec.outlet_orifice_alias} is not an AMESim pneumatic orifice"
+            )
+        return TestMqlPnl0001PairChamberSegmentClosure(
+            components=TestMqlPnl0001PairChamberSegmentComponents(
+                inlet_line=inlet_line,
+                outlet_line=outlet_line,
+                volume=volume,
+                inlet_orifice=inlet_orifice,
+                outlet_orifice=outlet_orifice,
+                spec=spec,
+            ),
+            inlet_node=TestMqlPneumaticBoundaryCondition(
+                pressure_pa=inlet_node_pressure_pa,
+                temperature_k=inlet_node_temperature_k,
+            ),
+            outlet_node=TestMqlPneumaticBoundaryCondition(
+                pressure_pa=outlet_node_pressure_pa,
+                temperature_k=outlet_node_temperature_k,
+            ),
+        )
+
+    def simulate_pnl0001_pair_chamber_segment_from_spec(
+        self,
+        spec,
+        *,
+        inlet_node_pressure_pa: float,
+        outlet_node_pressure_pa: float,
+        inlet_node_temperature_k: float = 293.15,
+        outlet_node_temperature_k: float = 293.15,
+        config: SolveIVPConfig | None = None,
+        t_eval: list[float] | None = None,
+    ):
+        closure = self.pnl0001_pair_chamber_segment_closure_from_spec(
+            spec,
+            inlet_node_pressure_pa=inlet_node_pressure_pa,
+            outlet_node_pressure_pa=outlet_node_pressure_pa,
+            inlet_node_temperature_k=inlet_node_temperature_k,
+            outlet_node_temperature_k=outlet_node_temperature_k,
+        )
+        run_config = config or SolveIVPConfig(t_stop=1.0e-4, max_step=1.0e-5)
+        return integrate_ode(
+            rhs=lambda t, state: closure.rhs(state),
+            initial_state=closure.initial_state_vector(),
+            config=run_config,
+            t_eval=t_eval,
+        )
+
     def pneumatic_branch_closure_from_spec(self, spec):
         return self.pneumatic_branch_closure(
             name=spec.name,