from __future__ import annotations

import unittest
from pathlib import Path

from PythonModels.reporting.amesim_results import load_test_mql_amesim_results
from PythonModels.systems.test_mql import TestMqlSystem
from PythonModels.systems.test_mql_nodes import build_test_mql_node3_assembly


REPO_ROOT = Path(__file__).resolve().parents[1]
TEST_MQL_AME = REPO_ROOT / "AmesimModels" / "test_mql.ame"


class TestMqlPneumaticNode3Tests(unittest.TestCase):
    @classmethod
    def setUpClass(cls) -> None:
        cls.nodes = build_test_mql_node3_assembly()
        cls.results = load_test_mql_amesim_results(TEST_MQL_AME)

    def test_builds_all_eight_pn3node2_instances(self) -> None:
        self.assertEqual(len(self.nodes), 8)
        self.assertIn("pn_node3_9", self.nodes)
        system = TestMqlSystem()
        self.assertEqual(system.typed_node3_count, 8)
        self.assertEqual(system.node3_assembly["pn_node3_9"].alias, "pn_node3_9")

    def test_port_two_flow_is_exact_sum_of_ports_one_and_three(self) -> None:
        balance = self.nodes["pn_node3_9"].balance(
            port_2_temperature_k=281.0,
            port_2_pressure_pa=4.4e6,
            port_1_enthalpy_flow_w=2.0,
            port_1_mass_flow_g_s=3.0,
            port_3_enthalpy_flow_w=-0.5,
            port_3_mass_flow_g_s=-1.0,
            port_1_volume_derivative_l_min=4.0,
            port_1_volume_cm3=5.0,
            port_3_volume_derivative_l_min=-1.0,
            port_3_volume_cm3=6.0,
        )

        self.assertEqual(balance.temperature_k, 281.0)
        self.assertEqual(balance.pressure_pa, 4.4e6)
        self.assertEqual(balance.port_2_enthalpy_flow_w, 1.5)
        self.assertEqual(balance.port_2_mass_flow_g_s, 2.0)
        self.assertEqual(balance.port_2_volume_derivative_l_min, 3.0)
        self.assertEqual(balance.port_2_volume_cm3, 11.0)

    def test_pn_node3_9_baseline_matches_adjacent_line_flow_sum(self) -> None:
        index = 500
        balance = self.nodes["pn_node3_9"].balance(
            port_2_temperature_k=self.results.series("t1@pneumatic_88")[index],
            port_2_pressure_pa=(
                self.results.series("p1@pneumatic_88")[index] + 101_300.0
            ),
            port_1_enthalpy_flow_w=self.results.series("dh1@pneumatic_100")[index],
            port_1_mass_flow_g_s=self.results.series("dm1@pneumatic_100")[index],
            port_3_enthalpy_flow_w=self.results.series("dh1@pneumatic_97")[index],
            port_3_mass_flow_g_s=self.results.series("dm1@pneumatic_97")[index],
        )

        self.assertAlmostEqual(
            balance.port_2_mass_flow_g_s,
            self.results.series("dm2@pn_node3_9")[index],
        )
        self.assertAlmostEqual(
            balance.port_2_enthalpy_flow_w,
            self.results.series("dh2@pn_node3_9")[index],
        )
        self.assertEqual(self.results.series("vol2@pn_node3_9")[index], 0.0)
        self.assertEqual(self.results.series("dvol2@pn_node3_9")[index], 0.0)

    def test_rejects_nonphysical_primary_pressure_or_temperature(self) -> None:
        node = self.nodes["pn_node3_9"]
        with self.assertRaises(ValueError):
            node.balance(
                port_2_temperature_k=0.0,
                port_2_pressure_pa=1.0e5,
                port_1_enthalpy_flow_w=0.0,
                port_1_mass_flow_g_s=0.0,
                port_3_enthalpy_flow_w=0.0,
                port_3_mass_flow_g_s=0.0,
            )


if __name__ == "__main__":
    unittest.main()