接入test_mql P4环网剩余管路

tests/test_test_mql_pnl0001_segment.py

@@ -288,6 +288,16 @@ class TestMqlPn3P4NodeChamberSegmentTests(unittest.TestCase):
             closure.components.p4_port1_far_primary_chamber.name,
             "pn_c1_11",
         )
+        self.assertEqual(closure.components.p4_port1_far_port1_line.name, "pneumatic_85")
+        self.assertEqual(closure.components.p4_port1_next_node.alias, "pnnode4_23")
+        self.assertEqual(
+            closure.components.p4_port1_next_primary_line.name,
+            "pneumatic_71",
+        )
+        self.assertEqual(
+            closure.components.p4_port1_next_primary_chamber.name,
+            "pn_c1_12",
+        )
         self.assertEqual(
             closure.components.p4_port1_remote_orifice.name,
             "pn_morifice_10",
@@ -323,6 +333,16 @@ class TestMqlPn3P4NodeChamberSegmentTests(unittest.TestCase):
             closure.components.p4_port3_far_primary_chamber.name,
             "pn_c1_15",
         )
+        self.assertEqual(closure.components.p4_port3_far_port3_line.name, "pneumatic_82")
+        self.assertEqual(closure.components.p4_port3_next_node.alias, "pnnode4_21")
+        self.assertEqual(
+            closure.components.p4_port3_next_primary_line.name,
+            "pneumatic_73",
+        )
+        self.assertEqual(
+            closure.components.p4_port3_next_primary_chamber.name,
+            "pn_c1_14",
+        )
         self.assertEqual(
             closure.components.p4_port3_remote_orifice.name,
             "pn_morifice_1",
@@ -339,8 +359,8 @@ class TestMqlPn3P4NodeChamberSegmentTests(unittest.TestCase):
             closure.components.p4_port3_remote_orifice_resistance.name,
             "pneumatic_105",
         )
-        self.assertEqual(len(closure.initial_state_vector()), 46)
-        self.assertEqual(len(rhs), 46)
+        self.assertEqual(len(closure.initial_state_vector()), 58)
+        self.assertEqual(len(rhs), 58)
         self.assertAlmostEqual(
             snapshot.p4_balance.pressure_pa,
             snapshot.p4_primary_line.p,
@@ -391,6 +411,24 @@ class TestMqlPn3P4NodeChamberSegmentTests(unittest.TestCase):
                 port1_remote_neighborhood.port_1.remote_node_alias
             ).line_alias,
         )
+        port1_far_port1 = self.system._pnl0002_connection_for_node_port(
+            closure.components.p4_port1_far_node.alias,
+            "port_1",
+        )
+        self.assertEqual(
+            closure.components.p4_port1_far_port1_line.name,
+            port1_far_port1.line_alias,
+        )
+        self.assertEqual(
+            closure.components.p4_port1_next_node.alias,
+            port1_far_port1.remote_node_alias,
+        )
+        self.assertEqual(
+            closure.components.p4_port1_next_primary_line.name,
+            self.system._p4_primary_connection_for_node(
+                port1_far_port1.remote_node_alias
+            ).line_alias,
+        )
         self.assertEqual(
             closure.components.p4_port1_remote_orifice.name,
             port1_remote_neighborhood.port_4.orifice_alias,
@@ -428,6 +466,24 @@ class TestMqlPn3P4NodeChamberSegmentTests(unittest.TestCase):
                 port3_remote_neighborhood.port_3.remote_node_alias
             ).line_alias,
         )
+        port3_far_port3 = self.system._pnl0002_connection_for_node_port(
+            closure.components.p4_port3_far_node.alias,
+            "port_3",
+        )
+        self.assertEqual(
+            closure.components.p4_port3_far_port3_line.name,
+            port3_far_port3.line_alias,
+        )
+        self.assertEqual(
+            closure.components.p4_port3_next_node.alias,
+            port3_far_port3.remote_node_alias,
+        )
+        self.assertEqual(
+            closure.components.p4_port3_next_primary_line.name,
+            self.system._p4_primary_connection_for_node(
+                port3_far_port3.remote_node_alias
+            ).line_alias,
+        )
         self.assertEqual(
             closure.components.p4_port3_remote_orifice.name,
             port3_remote_neighborhood.port_4.orifice_alias,
@@ -491,7 +547,14 @@ class TestMqlPn3P4NodeChamberSegmentTests(unittest.TestCase):
         )
         self.assertAlmostEqual(
             snapshot.p4_port1_far_node_to_primary_line_flow
-            + snapshot.p4_port1_far_node_to_line_flow,
+            + snapshot.p4_port1_far_node_to_line_flow
+            + snapshot.p4_port1_far_to_next_line_flow,
+            0.0,
+            delta=1.0e-12,
+        )
+        self.assertAlmostEqual(
+            snapshot.p4_port1_next_node_to_primary_line_flow
+            + snapshot.p4_port1_next_node_to_line_flow,
             0.0,
             delta=1.0e-12,
         )
@@ -517,18 +580,25 @@ class TestMqlPn3P4NodeChamberSegmentTests(unittest.TestCase):
         )
         self.assertAlmostEqual(
             snapshot.p4_port3_far_node_to_primary_line_flow
-            + snapshot.p4_port3_far_node_to_line_flow,
+            + snapshot.p4_port3_far_node_to_line_flow
+            + snapshot.p4_port3_far_to_next_line_flow,
             0.0,
             delta=1.0e-12,
         )
-        mass_derivative_sum = sum(rhs[index] for index in range(0, 46, 2))
+        self.assertAlmostEqual(
+            snapshot.p4_port3_next_node_to_primary_line_flow
+            + snapshot.p4_port3_next_node_to_line_flow,
+            0.0,
+            delta=1.0e-12,
+        )
+        mass_derivative_sum = sum(rhs[index] for index in range(0, 58, 2))
         self.assertAlmostEqual(
             mass_derivative_sum,
             -snapshot.p4_port3_remote_node_to_resistance_flow,
             delta=1.0e-12,
         )
 
-    def test_simulates_forty_six_state_pn3_p4_node_chamber_segment(self) -> None:
+    def test_simulates_fifty_eight_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,
@@ -543,8 +613,8 @@ class TestMqlPn3P4NodeChamberSegmentTests(unittest.TestCase):
 
         self.assertTrue(solution.success)
         self.assertEqual(len(solution.t), 3)
-        self.assertEqual(len(solution.y), 46)
-        self.assertEqual([len(row) for row in solution.y], [3] * 46)
+        self.assertEqual(len(solution.y), 58)
+        self.assertEqual([len(row) for row in solution.y], [3] * 58)
 
 
 if __name__ == "__main__":