from __future__ import annotations import unittest from app.simulation.components.amesim.flow.pipes import AmesimPnl0002, AmesimPnl0003 from app.simulation.core.medium import IdealGasMedium from app.simulation.registry import COMPONENT_MODEL_REGISTRY class AmesimPnl0002ComponentTests(unittest.TestCase): def setUp(self) -> None: self.medium = IdealGasMedium() def test_default_create_preserves_contract(self) -> None: pipe = COMPONENT_MODEL_REGISTRY["amesim_pnl0002"].create("pnl_2", self.medium, {}) self.assertIsInstance(pipe, AmesimPnl0002) self.assertEqual(set(pipe.ports), {"port_1", "port_2"}) self.assertEqual(len(pipe.get_state_vector()), 2) self.assertAlmostEqual(pipe.resistance_length, pipe.le / 2.0) def test_center_compliance_initial_state(self) -> None: pipe = AmesimPnl0002( "pnl_2", self.medium, diam=0.02, le=2.0, rr=0.045 / 20.0, p0=100000.0, T0=293.15, ) props = pipe.properties() self.assertAlmostEqual(pipe.volume, 6.283185307179586e-4) self.assertAlmostEqual(props.p, 100000.0, delta=1.0e-6) self.assertAlmostEqual(props.T, 293.15) def test_port_flows_enter_center_from_higher_external_pressure(self) -> None: pipe = AmesimPnl0002("pnl_2", self.medium, p0=100000.0, T0=293.15) center = pipe.properties() forward = pipe.port_mass_flow(101000.0, center.p, center.T) reverse = pipe.port_mass_flow(99000.0, center.p, center.T) self.assertGreater(forward, 0.0) self.assertLess(reverse, 0.0) def test_external_upstream_flow_uses_connected_stream_temperature(self) -> None: pipe = AmesimPnl0002("pnl_2", self.medium, p0=100000.0, T0=350.0) center = pipe.properties() external_pressure = 15.3e6 external_temperature = 293.15 external_h = self.medium.specific_enthalpy_at_pressure( external_pressure, external_temperature, ) pipe.update_stream_outflows( {"port_1": external_h, "port_2": center.h} ) flow = pipe.port_mass_flow( external_pressure, center.p, center.T, port_name="port_1", ) expected = pipe.mass_flow( external_pressure, center.p, external_temperature, ) center_temperature_flow = pipe.mass_flow( external_pressure, center.p, center.T, ) self.assertAlmostEqual(flow, expected) self.assertNotAlmostEqual(flow, center_temperature_flow) def test_missing_stream_cache_preserves_center_temperature_fallback(self) -> None: pipe = AmesimPnl0002("pnl_2", self.medium, p0=100000.0, T0=350.0) center = pipe.properties() external_pressure = 15.3e6 flow = pipe.port_mass_flow( external_pressure, center.p, center.T, port_name="port_1", ) self.assertAlmostEqual( flow, pipe.mass_flow(external_pressure, center.p, center.T), ) def test_center_upstream_flow_keeps_center_temperature(self) -> None: pipe = AmesimPnl0002("pnl_2", self.medium, p0=15.3e6, T0=350.0) center = pipe.properties() external_pressure = 1.0e5 external_h = self.medium.specific_enthalpy_at_pressure( external_pressure, 293.15, ) pipe.update_stream_outflows( {"port_1": external_h, "port_2": center.h} ) flow = pipe.port_mass_flow( external_pressure, center.p, center.T, port_name="port_1", ) self.assertAlmostEqual( flow, pipe.mass_flow(external_pressure, center.p, center.T), ) def test_pressure_flow_residuals_use_two_port_resistances(self) -> None: pipe = AmesimPnl0002("pnl_2", self.medium, p0=100000.0, T0=293.15) center = pipe.properties() pipe.port_1.p = 101000.0 pipe.port_2.p = 99000.0 pipe.port_1.m_flow = pipe.port_mass_flow(pipe.port_1.p, center.p, center.T) pipe.port_2.m_flow = pipe.port_mass_flow(pipe.port_2.p, center.p, center.T) residuals = { residual.id.rsplit(":", 1)[1]: residual for residual in pipe.pressure_flow_equation_residuals() } self.assertEqual( set(residuals), {"port_1_pressure_flow_relation", "port_2_pressure_flow_relation"}, ) self.assertAlmostEqual(residuals["port_1_pressure_flow_relation"].value, 0.0) self.assertAlmostEqual(residuals["port_2_pressure_flow_relation"].value, 0.0) def test_connection_derivative_accumulates_two_external_flows(self) -> None: pipe = AmesimPnl0002("pnl_2", self.medium) props = pipe.properties() pipe.port_1.m_flow = 0.2 pipe.port_2.m_flow = -0.1 derivative = pipe.state_derivative_from_ports( {"port_1": props.h + 1000.0, "port_2": props.h - 1000.0} ) self.assertAlmostEqual(derivative[0], 0.1) def test_connection_derivative_uses_energy_balanced_node_enthalpy(self) -> None: pipe = AmesimPnl0002("pnl_2", self.medium) props = pipe.properties() pipe.port_1.m_flow = 0.2 pipe.port_2.m_flow = 0.1 reference_h = props.h + 10_000.0 pipe.update_flow_temperature_references( {"port_1": reference_h, "port_2": reference_h} ) derivative = pipe.state_derivative_from_ports( {"port_1": props.h, "port_2": props.h} ) self.assertAlmostEqual(derivative[0], 0.3) self.assertAlmostEqual(derivative[1], 0.3 * props.h) class AmesimPnl0003ComponentTests(unittest.TestCase): def setUp(self) -> None: self.medium = IdealGasMedium() def test_default_create_preserves_contract(self) -> None: pipe = COMPONENT_MODEL_REGISTRY["amesim_pnl0003"].create("pnl_3", self.medium, {}) self.assertIsInstance(pipe, AmesimPnl0003) self.assertEqual(set(pipe.ports), {"port_1", "port_2"}) self.assertEqual(len(pipe.get_state_vector()), 4) def test_initial_state_uses_two_half_volume_compliances(self) -> None: pipe = AmesimPnl0003( "pnl_3", self.medium, diam=0.02, le=0.3, rr=0.045 / 20.0, p1_0=15.3e6, T1_0=293.15, p2_0=15.3e6, T2_0=293.15, ) port_1 = pipe.properties_1() port_2 = pipe.properties_2() self.assertAlmostEqual(pipe.volume, 9.424777960769381e-5) self.assertAlmostEqual(pipe.compliance_volume, pipe.volume / 2.0) self.assertAlmostEqual(port_1.p, 15.3e6, delta=1.0e-5) self.assertAlmostEqual(port_2.p, 15.3e6, delta=1.0e-5) def test_center_resistance_flow_follows_end_pressure_gradient(self) -> None: pipe = AmesimPnl0003("pnl_3", self.medium, p1_0=101000.0, p2_0=100000.0) forward = pipe.resistance_mass_flow() pipe = AmesimPnl0003("pnl_3", self.medium, p1_0=100000.0, p2_0=101000.0) reverse = pipe.resistance_mass_flow() self.assertGreater(forward, 0.0) self.assertLess(reverse, 0.0) def test_pressure_flow_residuals_bind_both_port_pressures_to_states(self) -> None: pipe = AmesimPnl0003("pnl_3", self.medium) pipe.properties_1() pipe.properties_2() residuals = { residual.id.rsplit(":", 1)[1]: residual for residual in pipe.pressure_flow_equation_residuals() } self.assertEqual(set(residuals), {"port_1_pressure_state", "port_2_pressure_state"}) self.assertAlmostEqual(residuals["port_1_pressure_state"].value, 0.0) self.assertAlmostEqual(residuals["port_2_pressure_state"].value, 0.0) def test_connection_derivative_conserves_external_and_center_flows(self) -> None: pipe = AmesimPnl0003("pnl_3", self.medium, p1_0=101000.0, p2_0=100000.0) port_1 = pipe.properties_1() port_2 = pipe.properties_2() pipe.port_1.m_flow = 0.2 pipe.port_2.m_flow = -0.1 derivative = pipe.state_derivative_from_ports( {"port_1": port_1.h + 1000.0, "port_2": port_2.h - 1000.0} ) self.assertAlmostEqual(derivative[0] + derivative[2], 0.1) def test_results_include_two_thermodynamic_sides_and_center_flow(self) -> None: pipe = AmesimPnl0003("pnl_3", self.medium, p1_0=101000.0, p2_0=100000.0) values = pipe.component_result_values() self.assertIn("m1", values) self.assertIn("m2", values) self.assertIn("dmctr", values) self.assertIn("re", values) if __name__ == "__main__": unittest.main()