73 lines
2.8 KiB
Python
73 lines
2.8 KiB
Python
from __future__ import annotations
|
|
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
from PythonModels.reporting.amesim_results import load_test_mql_amesim_results
|
|
from PythonModels.reporting.test_mql_variables import build_test_mql_variable_catalog
|
|
from PythonModels.systems.test_mql_lines import build_test_mql_line_assembly
|
|
|
|
|
|
REPO_ROOT = Path(__file__).resolve().parents[1]
|
|
TEST_MQL_AME = REPO_ROOT / "AmesimModels" / "test_mql.ame"
|
|
|
|
|
|
class TestMqlLineAssemblyTests(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls) -> None:
|
|
cls.amesim_results = load_test_mql_amesim_results(TEST_MQL_AME)
|
|
cls.variable_catalog = build_test_mql_variable_catalog(cls.amesim_results)
|
|
cls.assembly = build_test_mql_line_assembly(
|
|
cls.amesim_results,
|
|
cls.variable_catalog,
|
|
)
|
|
|
|
def test_assembles_expected_line_counts(self) -> None:
|
|
self.assertEqual(self.assembly.line_count, 40)
|
|
self.assertEqual(
|
|
self.assembly.counts_by_submodel(),
|
|
{"PNL0001": 20, "PNL0003": 8, "PNL0002": 8, "PNL00R": 4},
|
|
)
|
|
self.assertIn("pneumatic_69", self.assembly.aliases())
|
|
self.assertIn("pneumatic_100", self.assembly.aliases())
|
|
|
|
def test_preserves_connection_topology(self) -> None:
|
|
line = self.assembly.by_alias("pneumatic_69")
|
|
|
|
self.assertEqual(line.index, 37)
|
|
self.assertEqual(line.submodel, "PNL0001")
|
|
self.assertEqual(line.pattern, "C-R")
|
|
self.assertEqual(line.source_component, "pnnode4_16")
|
|
self.assertEqual(line.source_port, "port_2")
|
|
self.assertEqual(line.target_component, "pn_c1_8")
|
|
self.assertEqual(line.target_port, "port_1")
|
|
self.assertTrue(line.has_compliance)
|
|
self.assertTrue(line.has_resistance)
|
|
|
|
def test_line_data_paths_are_attached_from_amesim_results(self) -> None:
|
|
line = self.assembly.by_alias("pneumatic_69")
|
|
|
|
self.assertIn("dh1@pneumatic_69", line.data_paths)
|
|
self.assertIn("dm1@pneumatic_69", line.data_paths)
|
|
self.assertIn("dh1", line.signal_names)
|
|
self.assertIn("dm1", line.signal_names)
|
|
|
|
def test_resistance_only_line_keeps_r_pattern(self) -> None:
|
|
line = self.assembly.by_alias("pneumatic_100")
|
|
|
|
self.assertEqual(line.submodel, "PNL00R")
|
|
self.assertEqual(line.pattern, "R")
|
|
self.assertFalse(line.has_compliance)
|
|
self.assertTrue(line.has_resistance)
|
|
self.assertIn("dm1@pneumatic_100", line.data_paths)
|
|
|
|
def test_rcr_and_crc_patterns_are_distinct(self) -> None:
|
|
self.assertEqual(self.assembly.by_alias("pneumatic_80").pattern, "R-C-R")
|
|
self.assertEqual(self.assembly.by_alias("pneumatic_75").pattern, "C-R-C")
|
|
self.assertEqual(len(self.assembly.by_submodel("PNL0002")), 8)
|
|
self.assertEqual(len(self.assembly.by_submodel("PNL0003")), 8)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|