实现test_mql PNL0002管路动态

PythonModels/systems/test_mql_pneumatic_lines.py

@@ -9,14 +9,17 @@ from PythonModels.components.amesim_pneumatic import (
 )
 from PythonModels.components.amesim_pneumatic_line import (
     AmesimPnl0001Pipe,
+    AmesimPnl0002Pipe,
     AmesimPnl0003Pipe,
     AmesimPnl00rPipe,
 )
 from PythonModels.systems.test_mql_line_parameters import (
     TestMqlPnl0001Spec,
+    TestMqlPnl0002Spec,
     TestMqlPnl0003Spec,
     TestMqlPnl00rSpec,
     load_test_mql_pnl0001_specs,
+    load_test_mql_pnl0002_specs,
     load_test_mql_pnl0003_specs,
     load_test_mql_pnl00r_specs,
 )
@@ -34,6 +37,18 @@ class TestMqlPnl0001Assembly:
         raise KeyError(alias)
 
 
+@dataclass(frozen=True)
+class TestMqlPnl0002Assembly:
+    specs: tuple[TestMqlPnl0002Spec, ...]
+    lines: dict[str, AmesimPnl0002Pipe]
+
+    def spec(self, alias: str) -> TestMqlPnl0002Spec:
+        for spec in self.specs:
+            if spec.alias == alias:
+                return spec
+        raise KeyError(alias)
+
+
 @dataclass(frozen=True)
 class TestMqlPnl0003Assembly:
     specs: tuple[TestMqlPnl0003Spec, ...]
@@ -82,6 +97,30 @@ def build_test_mql_pnl0001_assembly(
     return TestMqlPnl0001Assembly(specs=specs, lines=lines)
 
 
+def build_test_mql_pnl0002_assembly(
+    archive_path: str | Path,
+    *,
+    gas: AmesimPneumaticGas = HELIUM_PNEUMATIC_GAS,
+) -> TestMqlPnl0002Assembly:
+    specs = load_test_mql_pnl0002_specs(archive_path)
+    lines = {
+        spec.alias: AmesimPnl0002Pipe(
+            name=spec.alias,
+            diameter_mm=spec.diameter_mm,
+            length_m=spec.length_m,
+            relative_roughness=spec.relative_roughness,
+            polytropic_constant=spec.polytropic_constant,
+            heat_transfer_coefficient=spec.heat_transfer_coefficient,
+            external_temperature_k=spec.external_temperature_k,
+            gas=gas,
+            pctr_0=spec.initial_center_absolute_pressure_pa,
+            Tctr_0=spec.initial_center_temperature_k,
+        )
+        for spec in specs
+    }
+    return TestMqlPnl0002Assembly(specs=specs, lines=lines)
+
+
 def build_test_mql_pnl0003_assembly(
     archive_path: str | Path,
     *,