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()