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

tests/test_component_registry.py

@@ -0,0 +1,115 @@
+from __future__ import annotations
+
+import unittest
+
+from app.simulation.components.experimental.library import LIBRARY
+from app.simulation.components.amesim.library import LIBRARY as AMESIM_LIBRARY
+from app.simulation.core.base import AlgebraicComponent
+from app.simulation.core.catalog import ComponentDisplaySpec, PortDisplaySpec
+from app.simulation.core.medium import IdealGasMedium
+from app.simulation.core.ports import PortDefinition
+from app.simulation.registry import (
+    COMPONENT_LIBRARY_REGISTRY,
+    COMPONENT_MODEL_REGISTRY,
+    discover_component_registries,
+    validate_component_model_class,
+)
+
+
+class ComponentRegistryTests(unittest.TestCase):
+    def test_manifest_is_the_only_public_model_source(self) -> None:
+        libraries, models = discover_component_registries()
+
+        self.assertEqual(libraries, COMPONENT_LIBRARY_REGISTRY)
+        self.assertEqual(models, COMPONENT_MODEL_REGISTRY)
+        self.assertEqual(
+            tuple(models),
+            (
+                "cylinder",
+                "tank",
+                "pipe",
+                "orifice",
+                "tee",
+                "amesim_pnpl01",
+                "amesim_pnch023",
+                "amesim_pnor001",
+                "amesim_pnvo001_fixed",
+                "amesim_pnl00r",
+                "amesim_pn3node2",
+                "amesim_p4node2",
+            ),
+        )
+        self.assertNotIn(
+            "app.simulation.components.experimental.flow.pipe:Pipe",
+            LIBRARY.models,
+        )
+
+
+    def test_amesim_library_is_enabled_as_separate_public_boundary(self) -> None:
+        libraries, models = discover_component_registries()
+
+        self.assertIn("amesim", libraries)
+        self.assertEqual(libraries["amesim"], AMESIM_LIBRARY)
+        self.assertEqual(models["amesim_pnpl01"].library.id, "amesim")
+        self.assertEqual(models["amesim_pnch023"].library.id, "amesim")
+        self.assertEqual(models["amesim_pnor001"].library.id, "amesim")
+        self.assertEqual(models["amesim_pnvo001_fixed"].library.id, "amesim")
+        self.assertEqual(models["amesim_pnl00r"].library.id, "amesim")
+        self.assertEqual(models["amesim_pn3node2"].library.id, "amesim")
+        self.assertEqual(models["amesim_p4node2"].library.id, "amesim")
+
+    def test_every_public_model_owns_its_catalog_contract(self) -> None:
+        for model_type, spec in COMPONENT_MODEL_REGISTRY.items():
+            with self.subTest(model_type=model_type):
+                declarations = spec.component_class.__dict__
+                self.assertIn("MODEL_VERSION", declarations)
+                self.assertIn("DISPLAY", declarations)
+                self.assertIn("create", declarations)
+                self.assertEqual(spec.display.library_id, spec.library.id)
+                self.assertIn(
+                    spec.display.category_id,
+                    spec.library.category_by_id,
+                )
+
+    def test_display_ports_must_match_physical_ports(self) -> None:
+        class BrokenDisplayComponent(AlgebraicComponent):
+            MODEL_TYPE = "broken_display"
+            MODEL_VERSION = "1.0.0"
+            PORTS = (
+                PortDefinition.pneumatic(
+                    "port_a",
+                    nominal_role="bidirectional",
+                ),
+            )
+            PARAMETERS = ()
+            RESULT_VARIABLES = ()
+            DISPLAY = ComponentDisplaySpec(
+                label="错误显示模型",
+                library_id="experimental",
+                category_id="flow",
+                symbol="generic",
+                ports=(PortDisplaySpec("port_b", "right"),),
+            )
+
+            @classmethod
+            def create(
+                cls,
+                *,
+                name: str,
+                medium: IdealGasMedium,
+                parameters,
+            ):
+                return cls(name)
+
+        with self.assertRaisesRegex(
+            ValueError,
+            "display ports must exactly match",
+        ):
+            validate_component_model_class(
+                BrokenDisplayComponent,
+                library=LIBRARY,
+            )
+
+
+if __name__ == "__main__":
+    unittest.main()