121 lines
3.9 KiB
Python
121 lines
3.9 KiB
Python
from __future__ import annotations
|
|
|
|
import unittest
|
|
|
|
from app.simulation.components.amesim.storage.chambers import AmesimPnch012
|
|
from app.simulation.core.medium import IdealGasMedium
|
|
from app.simulation.registry import COMPONENT_MODEL_REGISTRY
|
|
|
|
|
|
class AmesimPnch012ComponentTests(unittest.TestCase):
|
|
def setUp(self) -> None:
|
|
self.medium = IdealGasMedium()
|
|
|
|
def test_default_create_preserves_parameterized_volume_contract(self) -> None:
|
|
chamber = COMPONENT_MODEL_REGISTRY["amesim_pnch012"].create(
|
|
"chamber_1",
|
|
self.medium,
|
|
{},
|
|
)
|
|
|
|
self.assertIsInstance(chamber, AmesimPnch012)
|
|
self.assertEqual(set(chamber.ports), {"port_1", "port_2", "port_3", "port_4"})
|
|
self.assertEqual(len(chamber.get_state_vector()), 2)
|
|
self.assertEqual(chamber.parameter_values["cvol0"], 0.015)
|
|
self.assertAlmostEqual(chamber.total_volume(), 0.015)
|
|
|
|
def test_total_volume_adds_four_external_volume_parameters(self) -> None:
|
|
chamber = AmesimPnch012(
|
|
"chamber_1",
|
|
self.medium,
|
|
cvol0=0.015,
|
|
vol1=0.001,
|
|
vol2=0.002,
|
|
vol3=0.003,
|
|
vol4=0.004,
|
|
dvol1=1.0e-6,
|
|
dvol2=2.0e-6,
|
|
dvol3=-1.0e-6,
|
|
dvol4=0.0,
|
|
)
|
|
|
|
self.assertAlmostEqual(chamber.total_volume(), 0.025)
|
|
self.assertAlmostEqual(chamber.total_volume_rate(), 2.0e-6)
|
|
|
|
def test_initial_state_uses_total_volume_pressure_and_temperature(self) -> None:
|
|
chamber = AmesimPnch012(
|
|
"chamber_1",
|
|
self.medium,
|
|
cvol0=0.015,
|
|
vol1=0.010,
|
|
p0=200000.0,
|
|
T0=300.0,
|
|
)
|
|
props = chamber.properties()
|
|
|
|
self.assertAlmostEqual(props.p, 200000.0, delta=1.0e-8)
|
|
self.assertAlmostEqual(props.T, 300.0)
|
|
self.assertAlmostEqual(chamber.port_1.p, props.p)
|
|
self.assertAlmostEqual(chamber.port_4.p, props.p)
|
|
|
|
def test_pressure_flow_residuals_bind_all_ports_to_chamber_state(self) -> None:
|
|
chamber = AmesimPnch012("chamber_1", self.medium)
|
|
chamber.properties()
|
|
|
|
residuals = {
|
|
residual.id.rsplit(":", 1)[1]: residual
|
|
for residual in chamber.pressure_flow_equation_residuals()
|
|
}
|
|
|
|
self.assertEqual(
|
|
set(residuals),
|
|
{
|
|
"port_1_pressure_state",
|
|
"port_2_pressure_state",
|
|
"port_3_pressure_state",
|
|
"port_4_pressure_state",
|
|
},
|
|
)
|
|
self.assertTrue(all(abs(residual.value) < 1.0e-9 for residual in residuals.values()))
|
|
|
|
def test_derivative_sums_four_mass_flows_and_boundary_work(self) -> None:
|
|
chamber = AmesimPnch012(
|
|
"chamber_1",
|
|
self.medium,
|
|
dvol1=1.0e-6,
|
|
kth=2.0,
|
|
sth=0.5,
|
|
extemp=310.0,
|
|
)
|
|
props = chamber.properties()
|
|
chamber.port_1.m_flow = 0.2
|
|
chamber.port_2.m_flow = -0.1
|
|
chamber.port_3.m_flow = 0.05
|
|
chamber.port_4.m_flow = 0.0
|
|
|
|
derivative = chamber.state_derivative_from_ports(
|
|
{
|
|
"port_1": props.h + 1000.0,
|
|
"port_2": props.h - 1000.0,
|
|
"port_3": props.h + 500.0,
|
|
"port_4": props.h,
|
|
}
|
|
)
|
|
|
|
self.assertAlmostEqual(derivative[0], 0.15)
|
|
self.assertLess(derivative[1], 0.2 * (props.h + 1000.0) + 0.05 * (props.h + 500.0))
|
|
|
|
def test_results_include_total_volume_and_volume_rate(self) -> None:
|
|
chamber = AmesimPnch012("chamber_1", self.medium, vol1=0.001, dvol1=1.0e-6)
|
|
|
|
values = chamber.component_result_values()
|
|
|
|
self.assertIn("vol", values)
|
|
self.assertIn("dvol", values)
|
|
self.assertAlmostEqual(values["vol"], 0.016)
|
|
self.assertAlmostEqual(values["dvol"], 1.0e-6)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|