122 lines
4.4 KiB
Python
122 lines
4.4 KiB
Python
from __future__ import annotations
|
|
|
|
import unittest
|
|
|
|
from app.main import (
|
|
ReactFlowProjectPayload,
|
|
build_reactflow_system_xml,
|
|
compile_reactflow_network,
|
|
run_system_xml_simulation,
|
|
)
|
|
from app.system_xml import validate_system_xml_document
|
|
from tests.test_amesim_pnvo001_signal_xml import signal_edge, signal_port
|
|
from tests.test_generic_system_xml_simulation import component_node, physical_edge
|
|
from tests.test_system_xml_protocol import physical_port
|
|
|
|
|
|
def amesim_ud00_signal_project() -> ReactFlowProjectPayload:
|
|
return ReactFlowProjectPayload(
|
|
projectSchemaVersion=1,
|
|
name="amesim-ud00-signal-smoke",
|
|
nodes=[
|
|
component_node(
|
|
"piecewise_1",
|
|
"amesim_ud00",
|
|
[signal_port("out", "output", "right")],
|
|
{
|
|
"tstart": 0.0,
|
|
"start1": 1.0,
|
|
"end1": 1.0,
|
|
"t1": 1.0,
|
|
"start2": 1.0,
|
|
"end2": 1.0,
|
|
"t2": 0.0,
|
|
"start3": 1.0,
|
|
"end3": 1.0,
|
|
"t3": 0.0,
|
|
"start4": 1.0,
|
|
"end4": 1.0,
|
|
"t4": 0.0,
|
|
"start5": 1.0,
|
|
"end5": 1.0,
|
|
"t5": 0.0,
|
|
"start6": 1.0,
|
|
"end6": 1.0,
|
|
"t6": 0.0,
|
|
"start7": 1.0,
|
|
"end7": 1.0,
|
|
"t7": 0.0,
|
|
"start8": 1.0,
|
|
"end8": 1.0,
|
|
"t8": 0.0,
|
|
"nstages": 1.0,
|
|
"iscyclic": 0.0,
|
|
},
|
|
),
|
|
component_node(
|
|
"cylinder_1",
|
|
"cylinder",
|
|
[physical_port("port_b", "outlet", "right")],
|
|
{"volume": 0.01, "p0": 300000.0, "T0": 300.0},
|
|
),
|
|
component_node(
|
|
"valve_1",
|
|
"amesim_pnvo001",
|
|
[
|
|
signal_port("res", "input", "left"),
|
|
physical_port("port_2", "bidirectional", "left"),
|
|
physical_port("port_3", "bidirectional", "right"),
|
|
],
|
|
{
|
|
"cq": 0.72,
|
|
"area0": 5.0e-6,
|
|
"Cv": 0.5,
|
|
"Kv": 0.4,
|
|
"gi": 0.0,
|
|
"flowset": 1.0,
|
|
"opening0": 0.0,
|
|
},
|
|
),
|
|
component_node(
|
|
"tank_1",
|
|
"tank",
|
|
[physical_port("port_a", "inlet", "left")],
|
|
{"volume": 0.1, "p0": 100000.0, "T0": 300.0},
|
|
),
|
|
],
|
|
edges=[
|
|
signal_edge("signal-1", "piecewise_1", "out", "valve_1", "res"),
|
|
physical_edge("edge-1", "cylinder_1", "port_b", "valve_1", "port_2"),
|
|
physical_edge("edge-2", "valve_1", "port_3", "tank_1", "port_a"),
|
|
],
|
|
simulation={"t_start": 0.0, "t_stop": 0.002, "step": 0.001, "max_step": 0.001, "method": "BDF"},
|
|
)
|
|
|
|
|
|
class AmesimUd00XmlTests(unittest.TestCase):
|
|
def test_signal_project_compiles_with_signal_connection(self) -> None:
|
|
network = compile_reactflow_network(amesim_ud00_signal_project())
|
|
|
|
self.assertIn("piecewise_1", network.components)
|
|
self.assertEqual(network.components["piecewise_1"].model_type, "amesim_ud00")
|
|
self.assertEqual(network.components["valve_1"].model_type, "amesim_pnvo001")
|
|
self.assertTrue(network.pressure_flow_structure_dict()["isSquare"])
|
|
|
|
def test_signal_system_xml_validates_and_simulates(self) -> None:
|
|
xml = build_reactflow_system_xml(amesim_ud00_signal_project())
|
|
report = validate_system_xml_document(xml)
|
|
|
|
self.assertTrue(report.valid, report.as_dict())
|
|
result = run_system_xml_simulation(xml)
|
|
|
|
self.assertTrue(result["success"], result["message"])
|
|
self.assertEqual(result["series"]["time"], [0.0, 0.001, 0.002])
|
|
self.assertEqual(result["series"]["piecewise_1.out.signal"], [1.0, 1.0, 1.0])
|
|
self.assertEqual(result["series"]["valve_1.res.signal"], [1.0, 1.0, 1.0])
|
|
self.assertIn("valve_1.xv", result["series"])
|
|
self.assertGreater(result["diagnostics"]["signal"]["propagations"], 0)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|