接入test_mql pn_general_chamber_2出口支路

tests/test_test_mql_pnl0001_segment.py

@@ -488,8 +488,28 @@ class TestMqlPn3P4NodeChamberSegmentTests(unittest.TestCase):
             closure.components.p4_port1_next_orifice_resistance_orifice.name,
             "pn_morifice_15",
         )
-        self.assertEqual(len(closure.initial_state_vector()), 106)
-        self.assertEqual(len(rhs), 106)
+        self.assertEqual(
+            closure.components.p4_port1_next_orifice_resistance_chamber_inlet_line.name,
+            "pneumatic_98",
+        )
+        self.assertEqual(
+            closure.components.p4_port1_next_orifice_resistance_chamber.name,
+            "pn_general_chamber_2",
+        )
+        self.assertEqual(
+            closure.components.p4_port1_next_orifice_resistance_chamber_outlet_line.name,
+            "pneumatic_99",
+        )
+        self.assertEqual(
+            closure.components.p4_port1_next_orifice_resistance_chamber_inlet_orifice.name,
+            "pn_orifice_20",
+        )
+        self.assertEqual(
+            closure.components.p4_port1_next_orifice_resistance_chamber_outlet_orifice.name,
+            "pn_orifice_21",
+        )
+        self.assertEqual(len(closure.initial_state_vector()), 112)
+        self.assertEqual(len(rhs), 112)
         self.assertAlmostEqual(
             snapshot.p4_balance.pressure_pa,
             snapshot.p4_primary_line.p,
@@ -601,6 +621,29 @@ class TestMqlPn3P4NodeChamberSegmentTests(unittest.TestCase):
             closure.components.p4_port1_next_orifice_resistance_orifice.name,
             "pn_morifice_15",
         )
+        resistance_chamber_segment = (
+            self.system.discover_pneumatic_branch_topology().chamber_segment_specs[1]
+        )
+        self.assertEqual(
+            closure.components.p4_port1_next_orifice_resistance_chamber_spec,
+            resistance_chamber_segment,
+        )
+        self.assertEqual(
+            closure.components.p4_port1_next_orifice_resistance_chamber_inlet_line.name,
+            resistance_chamber_segment.inlet_line_alias,
+        )
+        self.assertEqual(
+            closure.components.p4_port1_next_orifice_resistance_chamber_outlet_line.name,
+            resistance_chamber_segment.outlet_line_alias,
+        )
+        self.assertEqual(
+            closure.components.p4_port1_next_orifice_resistance_chamber_inlet_orifice.name,
+            resistance_chamber_segment.inlet_orifice_alias,
+        )
+        self.assertEqual(
+            closure.components.p4_port1_next_orifice_resistance_chamber_outlet_orifice.name,
+            resistance_chamber_segment.outlet_orifice_alias,
+        )
         self.assertEqual(
             closure.components.p4_bridge_node.alias,
             port1_next_port1.remote_node_alias,
@@ -943,7 +986,8 @@ class TestMqlPn3P4NodeChamberSegmentTests(unittest.TestCase):
         )
         self.assertAlmostEqual(
             snapshot.p4_port1_next_orifice_resistance_node_to_line_flow
-            + snapshot.p4_port1_next_orifice_resistance_node_to_next_flow,
+            + snapshot.p4_port1_next_orifice_resistance_node_to_next_flow
+            + snapshot.p4_port1_next_orifice_resistance_chamber_outlet_node_to_line_flow,
             0.0,
             delta=1.0e-12,
         )
@@ -991,14 +1035,25 @@ class TestMqlPn3P4NodeChamberSegmentTests(unittest.TestCase):
             ),
             delta=1.0e-12,
         )
-        mass_derivative_sum = sum(rhs[index] for index in range(0, 106, 2))
+        p4_port1_next_orifice_resistance_chamber_mass_sum = (
+            rhs[106] + rhs[108] + rhs[110]
+        )
+        self.assertAlmostEqual(
+            p4_port1_next_orifice_resistance_chamber_mass_sum,
+            (
+                snapshot.p4_port1_next_orifice_resistance_chamber_inlet_boundary_to_line_flow
+                + snapshot.p4_port1_next_orifice_resistance_chamber_outlet_node_to_line_flow
+            ),
+            delta=1.0e-12,
+        )
+        mass_derivative_sum = sum(rhs[index] for index in range(0, 112, 2))
         self.assertAlmostEqual(
             mass_derivative_sum,
-            0.0,
+            snapshot.p4_port1_next_orifice_resistance_chamber_inlet_boundary_to_line_flow,
             delta=1.0e-12,
         )
 
-    def test_simulates_one_hundred_six_state_pn3_p4_node_chamber_segment(self) -> None:
+    def test_simulates_one_hundred_twelve_state_pn3_p4_node_chamber_segment(self) -> None:
         solution = self.system.simulate_pn3_p4_node_chamber_segment_from_spec(
             self.spec,
             inlet_node_pressure_pa=15.31e6,
@@ -1013,8 +1068,8 @@ class TestMqlPn3P4NodeChamberSegmentTests(unittest.TestCase):
 
         self.assertTrue(solution.success)
         self.assertEqual(len(solution.t), 3)
-        self.assertEqual(len(solution.y), 106)
-        self.assertEqual([len(row) for row in solution.y], [3] * 106)
+        self.assertEqual(len(solution.y), 112)
+        self.assertEqual([len(row) for row in solution.y], [3] * 112)
 
 
 if __name__ == "__main__":