324 lines
15 KiB
Python
324 lines
15 KiB
Python
from __future__ import annotations
|
||
|
||
import json
|
||
from pathlib import Path
|
||
import unittest
|
||
|
||
from app.main import get_component_catalog
|
||
from app.simulation.registry import (
|
||
COMPONENT_MODEL_REGISTRY,
|
||
validate_component_registries,
|
||
)
|
||
from app.simulation.core.metadata import ParameterDefinition
|
||
|
||
|
||
class ComponentCatalogTests(unittest.TestCase):
|
||
def setUp(self) -> None:
|
||
self.catalog = get_component_catalog()
|
||
self.libraries = {
|
||
library["id"]: library for library in self.catalog["libraries"]
|
||
}
|
||
self.library = self.libraries["experimental"]
|
||
self.components = {
|
||
component["modelType"]: component
|
||
for library in self.catalog["libraries"]
|
||
for component in library["components"]
|
||
}
|
||
|
||
def test_experimental_library_is_exposed_as_temporary(self) -> None:
|
||
self.assertEqual(self.catalog["schemaVersion"], 1)
|
||
self.assertEqual(self.library["id"], "experimental")
|
||
self.assertEqual(self.library["label"], "临时测试组件库")
|
||
self.assertEqual(self.library["version"], "0.1.0")
|
||
self.assertTrue(self.library["temporary"])
|
||
self.assertEqual(
|
||
self.library["sourcePackage"],
|
||
"app.simulation.components.experimental",
|
||
)
|
||
self.assertEqual(
|
||
[category["id"] for category in self.library["categories"]],
|
||
["storage", "flow", "junctions"],
|
||
)
|
||
|
||
|
||
def test_amesim_library_exposes_pneumatic_models(self) -> None:
|
||
library = self.libraries["amesim"]
|
||
components = {
|
||
component["modelType"]: component for component in library["components"]
|
||
}
|
||
|
||
self.assertEqual(library["label"], "AMESim 组件库")
|
||
self.assertEqual([category["id"] for category in library["categories"]], ["media", "storage", "flow", "junctions", "boundary", "signals", "mechanical"])
|
||
self.assertEqual(set(components), {"amesim_ideal_air_medium", "amesim_helium_medium", "amesim_pnpl01", "amesim_step0", "amesim_ud00", "amesim_f000", "amesim_forc", "amesim_mecmas21", "amesim_lstp00a", "amesim_lmechn1", "amesim_pnch023", "amesim_pnch012", "amesim_pnor001", "amesim_pnvo001_fixed", "amesim_pnvo001", "amesim_pnl00r", "amesim_pnl0001", "amesim_pnl0002", "amesim_pnl0003", "amesim_pn3node2", "amesim_p4node2"})
|
||
for model_type, component in components.items():
|
||
with self.subTest(model_type=model_type):
|
||
self.assertEqual(component["symbol"], model_type)
|
||
self.assertEqual(
|
||
components["amesim_ideal_air_medium"]["role"],
|
||
"amesimGasMediumDefinition",
|
||
)
|
||
self.assertEqual(
|
||
components["amesim_ideal_air_medium"]["category"]["id"],
|
||
"media",
|
||
)
|
||
self.assertEqual(
|
||
components["amesim_ideal_air_medium"]["label"],
|
||
"空气介质定义",
|
||
)
|
||
self.assertEqual(
|
||
components["amesim_ideal_air_medium"]["modelVersion"],
|
||
"0.2.0",
|
||
)
|
||
medium_parameters = {
|
||
parameter["name"]: parameter
|
||
for parameter in components["amesim_ideal_air_medium"]["parameters"]
|
||
}
|
||
self.assertEqual(
|
||
list(medium_parameters),
|
||
["gi", "property_model"],
|
||
)
|
||
self.assertEqual(
|
||
medium_parameters["property_model"]["editor"],
|
||
"amesimGasPropertyModel",
|
||
)
|
||
self.assertEqual(
|
||
medium_parameters["property_model"]["options"],
|
||
[{"value": 0, "label": "理想气体"}],
|
||
)
|
||
helium_medium = components["amesim_helium_medium"]
|
||
self.assertEqual(helium_medium["role"], "amesimGasMediumDefinition")
|
||
self.assertEqual(helium_medium["category"]["id"], "media")
|
||
self.assertEqual(helium_medium["label"], "氦气介质定义")
|
||
self.assertEqual(helium_medium["modelVersion"], "0.1.0")
|
||
helium_parameters = {
|
||
parameter["name"]: parameter
|
||
for parameter in helium_medium["parameters"]
|
||
}
|
||
self.assertEqual(list(helium_parameters), ["gi", "property_model"])
|
||
self.assertEqual(
|
||
helium_parameters["property_model"]["editor"],
|
||
"amesimGasPropertyModel",
|
||
)
|
||
self.assertEqual(
|
||
helium_parameters["property_model"]["options"],
|
||
[{"value": 0, "label": "Peng–Robinson"}],
|
||
)
|
||
self.assertEqual(
|
||
[port["name"] for port in components["amesim_p4node2"]["ports"]],
|
||
["port_1", "port_2", "port_3", "port_4"],
|
||
)
|
||
self.assertEqual(components["amesim_pnpl01"]["category"]["id"], "boundary")
|
||
self.assertEqual([port["name"] for port in components["amesim_pnpl01"]["ports"]], ["port_1"])
|
||
|
||
pnch_parameters = {
|
||
parameter["name"]: parameter
|
||
for parameter in components["amesim_pnch023"]["parameters"]
|
||
}
|
||
self.assertEqual(components["amesim_pnch023"]["category"]["id"], "storage")
|
||
self.assertEqual(pnch_parameters["cvol"]["quantity"], "volume")
|
||
self.assertEqual(pnch_parameters["kth"]["quantity"], "heat_transfer_coefficient")
|
||
|
||
pnch012_parameters = {
|
||
parameter["name"]: parameter
|
||
for parameter in components["amesim_pnch012"]["parameters"]
|
||
}
|
||
self.assertEqual(components["amesim_pnch012"]["category"]["id"], "storage")
|
||
self.assertEqual([port["name"] for port in components["amesim_pnch012"]["ports"]], ["port_1", "port_2", "port_3", "port_4"])
|
||
self.assertEqual(pnch012_parameters["cvol0"]["quantity"], "volume")
|
||
self.assertEqual(pnch012_parameters["dvol1"]["unit"], "m3/s")
|
||
|
||
pnor_parameters = {
|
||
parameter["name"]: parameter
|
||
for parameter in components["amesim_pnor001"]["parameters"]
|
||
}
|
||
self.assertEqual(components["amesim_pnor001"]["category"]["id"], "flow")
|
||
self.assertEqual(pnor_parameters["area"]["quantity"], "area")
|
||
self.assertEqual(pnor_parameters["area"]["unit"], "m2")
|
||
pnvo_parameters = {
|
||
parameter["name"]: parameter
|
||
for parameter in components["amesim_pnvo001_fixed"]["parameters"]
|
||
}
|
||
self.assertEqual(components["amesim_pnvo001_fixed"]["category"]["id"], "flow")
|
||
self.assertEqual(pnvo_parameters["opening"]["maximum"], 1.0)
|
||
self.assertEqual([port["name"] for port in components["amesim_pnvo001_fixed"]["ports"]], ["port_2", "port_3"])
|
||
|
||
self.assertEqual(components["amesim_step0"]["category"]["id"], "signals")
|
||
self.assertEqual([port["name"] for port in components["amesim_step0"]["ports"]], ["out"])
|
||
ud00_parameters = {
|
||
parameter["name"]: parameter
|
||
for parameter in components["amesim_ud00"]["parameters"]
|
||
}
|
||
self.assertEqual(components["amesim_ud00"]["category"]["id"], "signals")
|
||
self.assertEqual([port["name"] for port in components["amesim_ud00"]["ports"]], ["out"])
|
||
self.assertEqual(ud00_parameters["tstart"]["quantity"], "time")
|
||
self.assertEqual(ud00_parameters["nstages"]["maximum"], 8.0)
|
||
self.assertEqual(components["amesim_f000"]["category"]["id"], "mechanical")
|
||
self.assertEqual([port["name"] for port in components["amesim_f000"]["ports"]], ["port_1"])
|
||
self.assertEqual(components["amesim_f000"]["ports"][0]["domain"], "mechanical")
|
||
self.assertEqual(components["amesim_forc"]["category"]["id"], "mechanical")
|
||
self.assertEqual([port["name"] for port in components["amesim_forc"]["ports"]], ["res", "port_2"])
|
||
self.assertEqual(components["amesim_forc"]["ports"][1]["domain"], "mechanical")
|
||
mecmas_parameters = {
|
||
parameter["name"]: parameter
|
||
for parameter in components["amesim_mecmas21"]["parameters"]
|
||
}
|
||
self.assertEqual(components["amesim_mecmas21"]["category"]["id"], "mechanical")
|
||
self.assertEqual([port["name"] for port in components["amesim_mecmas21"]["ports"]], ["port_1", "port_2"])
|
||
self.assertEqual(mecmas_parameters["mass"]["unit"], "kg")
|
||
self.assertEqual(mecmas_parameters["Kbmin"]["unit"], "N/m")
|
||
lstp_parameters = {
|
||
parameter["name"]: parameter
|
||
for parameter in components["amesim_lstp00a"]["parameters"]
|
||
}
|
||
self.assertEqual(components["amesim_lstp00a"]["category"]["id"], "mechanical")
|
||
self.assertEqual([port["name"] for port in components["amesim_lstp00a"]["ports"]], ["port_1", "port_2"])
|
||
self.assertEqual(lstp_parameters["gap0"]["unit"], "m")
|
||
self.assertEqual(lstp_parameters["kcont"]["unit"], "N/m")
|
||
lmechn_parameters = {
|
||
parameter["name"]: parameter
|
||
for parameter in components["amesim_lmechn1"]["parameters"]
|
||
}
|
||
self.assertEqual(components["amesim_lmechn1"]["category"]["id"], "mechanical")
|
||
self.assertEqual([port["name"] for port in components["amesim_lmechn1"]["ports"]], [f"port_{index}" for index in range(1, 10)])
|
||
self.assertEqual(lmechn_parameters["v1"]["maximum"], 8.0)
|
||
self.assertEqual(components["amesim_pnvo001"]["category"]["id"], "flow")
|
||
self.assertEqual([port["name"] for port in components["amesim_pnvo001"]["ports"]], ["res", "port_2", "port_3"])
|
||
|
||
pnl_parameters = {
|
||
parameter["name"]: parameter
|
||
for parameter in components["amesim_pnl00r"]["parameters"]
|
||
}
|
||
self.assertEqual(components["amesim_pnl00r"]["category"]["id"], "flow")
|
||
self.assertEqual(pnl_parameters["diam"]["quantity"], "length")
|
||
self.assertEqual(pnl_parameters["le"]["unit"], "m")
|
||
|
||
pnl0001_parameters = {
|
||
parameter["name"]: parameter
|
||
for parameter in components["amesim_pnl0001"]["parameters"]
|
||
}
|
||
self.assertEqual(components["amesim_pnl0001"]["category"]["id"], "flow")
|
||
self.assertEqual(pnl0001_parameters["kth"]["quantity"], "heat_transfer_coefficient")
|
||
self.assertEqual(pnl0001_parameters["p0"]["unit"], "Pa")
|
||
|
||
self.assertEqual(components["amesim_pnl0002"]["category"]["id"], "flow")
|
||
self.assertEqual(components["amesim_pnl0003"]["category"]["id"], "flow")
|
||
pnl0003_parameters = {
|
||
parameter["name"]: parameter
|
||
for parameter in components["amesim_pnl0003"]["parameters"]
|
||
}
|
||
self.assertEqual(pnl0003_parameters["p1_0"]["unit"], "Pa")
|
||
self.assertEqual(pnl0003_parameters["T2_0"]["quantity"], "temperature")
|
||
|
||
def test_every_registered_component_is_in_the_catalog(self) -> None:
|
||
self.assertEqual(
|
||
set(self.components),
|
||
set(COMPONENT_MODEL_REGISTRY),
|
||
)
|
||
|
||
def test_catalog_reuses_model_parameter_contracts(self) -> None:
|
||
cylinder = self.components["cylinder"]
|
||
parameters = {
|
||
parameter["name"]: parameter for parameter in cylinder["parameters"]
|
||
}
|
||
|
||
self.assertEqual(cylinder["label"], "气瓶")
|
||
self.assertEqual(cylinder["modelVersion"], "1.0.0")
|
||
self.assertEqual(parameters["volume"]["default"], 0.01)
|
||
self.assertEqual(parameters["volume"]["quantity"], "volume")
|
||
self.assertEqual(parameters["p0"]["unit"], "Pa")
|
||
self.assertTrue(parameters["p0"]["minimumExclusive"])
|
||
|
||
def test_parameter_description_is_optional_interface_metadata(self) -> None:
|
||
described = ParameterDefinition(
|
||
"diam",
|
||
0.01,
|
||
label="管径",
|
||
quantity="length",
|
||
unit="m",
|
||
minimum=0.0,
|
||
minimum_exclusive=True,
|
||
description="用于计算流通面积。",
|
||
).as_interface_dict()
|
||
undescribed = ParameterDefinition(
|
||
"gain",
|
||
1.0,
|
||
label="增益",
|
||
).as_interface_dict()
|
||
|
||
self.assertEqual(described["description"], "用于计算流通面积。")
|
||
self.assertEqual(described["default"], 0.01)
|
||
self.assertEqual(described["unit"], "m")
|
||
self.assertEqual(described["minimum"], 0.0)
|
||
self.assertTrue(described["minimumExclusive"])
|
||
self.assertNotIn("description", undescribed)
|
||
|
||
def test_catalog_exposes_descriptions_for_core_amesim_pneumatic_parameters(
|
||
self,
|
||
) -> None:
|
||
medium_parameters = {
|
||
parameter["name"]: parameter
|
||
for parameter in self.components["amesim_helium_medium"]["parameters"]
|
||
}
|
||
chamber_parameters = {
|
||
parameter["name"]: parameter
|
||
for parameter in self.components["amesim_pnch023"]["parameters"]
|
||
}
|
||
pipe_parameters = {
|
||
parameter["name"]: parameter
|
||
for parameter in self.components["amesim_pnl00r"]["parameters"]
|
||
}
|
||
orifice_parameters = {
|
||
parameter["name"]: parameter
|
||
for parameter in self.components["amesim_pnor001"]["parameters"]
|
||
}
|
||
|
||
self.assertIn("自动分配", medium_parameters["gi"]["description"])
|
||
self.assertIn(
|
||
"Peng–Robinson",
|
||
medium_parameters["property_model"]["description"],
|
||
)
|
||
self.assertIn("固定有效容积", chamber_parameters["cvol"]["description"])
|
||
self.assertIn("Darcy", pipe_parameters["rr"]["description"])
|
||
self.assertIn("有效孔口面积", orifice_parameters["area"]["description"])
|
||
self.assertEqual(chamber_parameters["cvol"]["default"], 0.057)
|
||
self.assertEqual(pipe_parameters["diam"]["unit"], "m")
|
||
|
||
def test_catalog_adds_port_layout_without_changing_physical_contract(self) -> None:
|
||
tee_ports = {
|
||
port["name"]: port for port in self.components["tee"]["ports"]
|
||
}
|
||
|
||
self.assertEqual(tee_ports["port_in"]["side"], "left")
|
||
self.assertEqual(tee_ports["port_out1"]["side"], "right")
|
||
self.assertEqual(tee_ports["port_out2"]["side"], "right")
|
||
self.assertEqual(tee_ports["port_in"]["kind"], "physical")
|
||
self.assertEqual(tee_ports["port_in"]["domain"], "pneumatic")
|
||
self.assertEqual(tee_ports["port_in"]["nominalRole"], "bidirectional")
|
||
|
||
def test_registry_can_repeat_discovery_and_validation(self) -> None:
|
||
validate_component_registries()
|
||
|
||
def test_catalog_schema_is_committed_and_targets_version_one(self) -> None:
|
||
schema_path = (
|
||
Path(__file__).resolve().parent.parent
|
||
/ "schemas"
|
||
/ "component-catalog-v1.schema.json"
|
||
)
|
||
schema = json.loads(schema_path.read_text(encoding="utf-8"))
|
||
|
||
self.assertEqual(schema["properties"]["schemaVersion"]["const"], 1)
|
||
self.assertIn("modelVersion", schema["$defs"]["component"]["required"])
|
||
self.assertIn("version", schema["$defs"]["library"]["required"])
|
||
parameter_schema = schema["$defs"]["parameter"]
|
||
self.assertEqual(
|
||
parameter_schema["properties"]["description"],
|
||
{"type": "string", "minLength": 1},
|
||
)
|
||
self.assertNotIn("description", parameter_schema["required"])
|
||
self.assertFalse(parameter_schema["additionalProperties"])
|
||
self.assertNotIn("unknownField", parameter_schema["properties"])
|
||
|
||
|
||
if __name__ == "__main__":
|
||
unittest.main()
|