51 lines
1.9 KiB
Python
51 lines
1.9 KiB
Python
from __future__ import annotations
|
|
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
from app.simulation.examples.test_mql.baseline import run_test_mql_baseline_passthrough
|
|
|
|
|
|
REPO_ROOT = Path(__file__).resolve().parents[1]
|
|
TEST_MQL_AME = REPO_ROOT / "AmesimModels" / "test_mql.ame"
|
|
|
|
|
|
class TestMqlBaselinePassthroughTests(unittest.TestCase):
|
|
def test_full_baseline_passthrough_satisfies_output_contract_with_zero_error(self) -> None:
|
|
run = run_test_mql_baseline_passthrough(TEST_MQL_AME)
|
|
|
|
self.assertEqual(run.sample_count, 1002)
|
|
self.assertEqual(run.signal_count, 858)
|
|
self.assertEqual(run.output_schema.signal_count, 858)
|
|
self.assertEqual(run.observation_catalog.binding_count, 114)
|
|
self.assertEqual(len(run.comparison.metrics), 858)
|
|
self.assertEqual(run.comparison.max_abs_error, 0.0)
|
|
self.assertEqual(run.comparison.max_rel_error, 0.0)
|
|
self.assertEqual(run.output.data_paths, run.output_schema.data_paths())
|
|
self.assertAlmostEqual(run.output.series("press@pn_c1_8")[-1], 4310310.457796034)
|
|
|
|
def test_selected_baseline_passthrough_preserves_requested_order(self) -> None:
|
|
data_paths = (
|
|
"vol1@pn_brp2_8",
|
|
"press@pn_c1_8",
|
|
"dm1@pneumatic_69",
|
|
)
|
|
|
|
run = run_test_mql_baseline_passthrough(TEST_MQL_AME, data_paths=data_paths)
|
|
|
|
self.assertEqual(run.signal_count, 3)
|
|
self.assertEqual(run.output.data_paths, data_paths)
|
|
self.assertEqual(tuple(metric.data_path for metric in run.comparison.metrics), data_paths)
|
|
self.assertEqual(run.comparison.max_abs_error, 0.0)
|
|
|
|
def test_baseline_passthrough_rejects_unobserved_data_path(self) -> None:
|
|
with self.assertRaises(KeyError):
|
|
run_test_mql_baseline_passthrough(
|
|
TEST_MQL_AME,
|
|
data_paths=("not_a_signal@not_an_owner",),
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|