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

tests/test_test_mql_pnl0001_segment.py

@@ -76,5 +76,78 @@ class TestMqlPnl0001ChamberSegmentTests(unittest.TestCase):
         self.assertEqual([len(row) for row in solution.y], [3, 3, 3, 3])
 
 
+class TestMqlPnl0001PairChamberSegmentTests(unittest.TestCase):
+    def setUp(self) -> None:
+        self.system = TestMqlSystem()
+        self.spec = self.system.discover_pneumatic_branch_topology().chamber_segment_specs[0]
+
+    def test_pair_closure_uses_both_topology_derived_lines(self) -> None:
+        closure = self.system.pnl0001_pair_chamber_segment_closure_from_spec(
+            self.spec,
+            inlet_node_pressure_pa=16.0e6,
+            outlet_node_pressure_pa=14.0e6,
+        )
+
+        snapshot = closure.snapshot()
+        rhs = closure.rhs(closure.initial_state_vector())
+
+        self.assertEqual(closure.components.inlet_line.name, "pneumatic_96")
+        self.assertEqual(closure.components.outlet_line.name, "pneumatic_97")
+        self.assertEqual(len(closure.initial_state_vector()), 6)
+        self.assertEqual(len(rhs), 6)
+        self.assertGreater(snapshot.inlet_node_to_line_flow, 0.0)
+        self.assertLess(snapshot.outlet_node_to_line_flow, 0.0)
+        self.assertAlmostEqual(
+            rhs[0] + rhs[2] + rhs[4],
+            snapshot.inlet_node_to_line_flow + snapshot.outlet_node_to_line_flow,
+            delta=1.0e-12,
+        )
+
+    def test_both_internal_orifice_flows_cancel_from_total_mass(self) -> None:
+        closure = self.system.pnl0001_pair_chamber_segment_closure_from_spec(
+            self.spec,
+            inlet_node_pressure_pa=15.3e6,
+            outlet_node_pressure_pa=15.3e6,
+        )
+        state = closure.initial_state_vector()
+        state[0] *= 1.01
+        state[1] *= 1.01
+        state[4] *= 0.99
+        state[5] *= 0.99
+
+        snapshot = closure.snapshot(state)
+        rhs = closure.rhs(state)
+
+        self.assertGreater(snapshot.inlet_line_to_chamber_flow, 0.0)
+        self.assertLess(snapshot.outlet_line_to_chamber_flow, 0.0)
+        self.assertAlmostEqual(
+            rhs[0] + rhs[2] + rhs[4],
+            snapshot.inlet_node_to_line_flow + snapshot.outlet_node_to_line_flow,
+            delta=1.0e-12,
+        )
+        self.assertAlmostEqual(
+            closure.components.outlet_line.port_2.m_flow,
+            -snapshot.outlet_line_to_chamber_flow,
+        )
+
+    def test_simulates_six_state_pair_line_chamber_segment(self) -> None:
+        solution = self.system.simulate_pnl0001_pair_chamber_segment_from_spec(
+            self.spec,
+            inlet_node_pressure_pa=15.31e6,
+            outlet_node_pressure_pa=15.29e6,
+            config=SolveIVPConfig(
+                t_start=0.0,
+                t_stop=1.0e-7,
+                max_step=1.0e-8,
+            ),
+            t_eval=[0.0, 5.0e-8, 1.0e-7],
+        )
+
+        self.assertTrue(solution.success)
+        self.assertEqual(len(solution.t), 3)
+        self.assertEqual(len(solution.y), 6)
+        self.assertEqual([len(row) for row in solution.y], [3, 3, 3, 3, 3, 3])
+
+
 if __name__ == "__main__":
     unittest.main()