同步仿真框架并接入AMESim气动组件

tests/test_component_catalog.py

@@ -0,0 +1,138 @@
+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,
+)
+
+
+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"]], ["storage", "flow", "junctions", "boundary"])
+        self.assertEqual(set(components), {"amesim_pnpl01", "amesim_pnch023", "amesim_pnor001", "amesim_pnvo001_fixed", "amesim_pnl00r", "amesim_pn3node2", "amesim_p4node2"})
+        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")
+
+        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"])
+
+        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")
+
+    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_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"])
+
+
+if __name__ == "__main__":
+    unittest.main()