from __future__ import annotations import tempfile import unittest from pathlib import Path from PythonModels.core.solver import SolveIVPConfig from PythonModels.scripts.run_test_mql_full_state_comparison import ( TestMqlFullStateComparisonExecutionConfig, TestMqlFullStateComparisonPathConfig, TestMqlFullStateComparisonScriptConfig, TestMqlPnvoEventBoundaryDiagnostic, format_test_mql_full_state_comparison_summary, format_test_mql_pnvo_event_boundary_summary, run_test_mql_full_state_comparison, ) REPO_ROOT = Path(__file__).resolve().parents[1] TEST_MQL_AME = REPO_ROOT / "AmesimModels" / "test_mql.ame" class RunTestMqlFullStateComparisonScriptTests(unittest.TestCase): def test_full_state_script_writes_summary_and_comparison_csv(self) -> None: with tempfile.TemporaryDirectory() as tmpdir: output_dir = Path(tmpdir) run, returned_output_dir = run_test_mql_full_state_comparison( TestMqlFullStateComparisonScriptConfig( paths=TestMqlFullStateComparisonPathConfig( archive_path=TEST_MQL_AME, output_dir=output_dir, ), execution=TestMqlFullStateComparisonExecutionConfig( data_paths=( "press@pn_c1_8", "vol1@pn_brp2_8", "dm1@pneumatic_69", "xv@pn_morifice_1", "dm2@pn_morifice_1", ), solver=SolveIVPConfig(t_start=0.0, t_stop=0.0), t_eval=(0.0,), ), ) ) summary_path = output_dir / "test_mql_full_state_comparison_summary.txt" csv_path = output_dir / "test_mql_amesim_comparison.csv" csv_summary_path = output_dir / "test_mql_amesim_comparison_summary.txt" summary = summary_path.read_text(encoding="utf-8") csv_header = csv_path.read_text(encoding="utf-8").splitlines()[0] csv_summary_exists = csv_summary_path.exists() self.assertEqual(returned_output_dir, output_dir) self.assertEqual(run.sample_count, 1) self.assertEqual(run.signal_count, 5) self.assertAlmostEqual(run.comparison.max_abs_error, 0.0, delta=1.0e-8) self.assertEqual( run.metrics_by_max_abs_error()[0].data_path, "press@pn_c1_8", ) diagnostic = run.signal_diagnostic("press@pn_c1_8") self.assertEqual(diagnostic.data_path, "press@pn_c1_8") self.assertAlmostEqual(diagnostic.initial_python_value, -1300.0) self.assertAlmostEqual(diagnostic.final_python_value, -1300.0) self.assertAlmostEqual(diagnostic.final_abs_error, 0.0, delta=1.0e-8) self.assertEqual( run.diagnostics_by_final_abs_error()[0].data_path, "press@pn_c1_8", ) flow_diagnostic = run.pnl0001_mass_flow_diagnostic() self.assertEqual(flow_diagnostic.data_path, "dm1@pneumatic_69") self.assertAlmostEqual(flow_diagnostic.initial_python_dm1_g_s, 0.0) self.assertAlmostEqual(flow_diagnostic.initial_amesim_dm1_g_s, 0.0) self.assertAlmostEqual( flow_diagnostic.initial_python_canonical_kg_s, -flow_diagnostic.initial_python_dm1_g_s * 1.0e-3, ) pnvo_diagnostic = run.pnvo_diagnostic() self.assertEqual(pnvo_diagnostic.alias, "pn_morifice_1") self.assertAlmostEqual(pnvo_diagnostic.initial_python_opening, 0.0) self.assertAlmostEqual(pnvo_diagnostic.initial_amesim_opening, 0.0) self.assertAlmostEqual(pnvo_diagnostic.initial_python_mass_flow_kg_s, 0.0) self.assertAlmostEqual(pnvo_diagnostic.initial_amesim_mass_flow_kg_s, 0.0) chamber_diagnostic = run.chamber_rhs_diagnostic("pn_c1_8") self.assertEqual(chamber_diagnostic.chamber_alias, "pn_c1_8") self.assertEqual(chamber_diagnostic.piston_alias, "pn_brp2_8") self.assertAlmostEqual(chamber_diagnostic.chamber_volume_m3, 0.015) self.assertAlmostEqual( chamber_diagnostic.energy_derivative_w, chamber_diagnostic.port_a_energy_flow_w + chamber_diagnostic.port_b_energy_flow_w + chamber_diagnostic.boundary_work_w + chamber_diagnostic.thermal_energy_flow_w, ) self.assertTrue(csv_summary_exists) self.assertIn("Mode: Python 132 full-state closure comparison", summary) self.assertIn("Compared signals: 5", summary) self.assertIn("AMESim first saved interval: 0.010000000000000023", summary) self.assertIn("Final time aligns with AMESim sample: True", summary) self.assertIn("Largest absolute error: press@pn_c1_8=", summary) self.assertIn("Largest final endpoint error: press@pn_c1_8=", summary) self.assertIn("Metrics by max absolute error:", summary) self.assertIn("Endpoint diagnostics by final absolute error:", summary) self.assertIn( "PNL0001 canonical mass-flow diagnostic: dm1@pneumatic_69", summary, ) self.assertIn("equals -AMESim dm1 * 1e-3", summary) self.assertIn("PNVO diagnostic: pn_morifice_1", summary) self.assertIn("final_python_opening=", summary) self.assertIn("final_python_mass_flow_kg_s=", summary) self.assertIn("Largest endpoint chamber RHS breakdown: pn_c1_8", summary) self.assertIn("piston_alias=pn_brp2_8", summary) self.assertIn("mass_derivative_kg_s=", summary) self.assertIn("boundary_work_w=", summary) self.assertIn("thermal_energy_flow_w=", summary) self.assertIn("final_python=", summary) self.assertIn("final_amesim=", summary) self.assertIn("python.press@pn_c1_8", csv_header) self.assertIn("amesim.vol1@pn_brp2_8", csv_header) self.assertIn("python.dm1@pneumatic_69", csv_header) self.assertIn("python.xv@pn_morifice_1", csv_header) self.assertIn("python.dm2@pn_morifice_1", csv_header) def test_default_execution_aligns_first_amesim_saved_sample(self) -> None: execution = TestMqlFullStateComparisonExecutionConfig() self.assertEqual(execution.solver.t_stop, 0.01) self.assertEqual(execution.solver.max_step, 1.0e-3) self.assertEqual(execution.t_eval, (0.0, 0.01)) self.assertIn("xv@pn_morifice_1", execution.data_paths) self.assertIn("dm2@pn_morifice_1", execution.data_paths) def test_formats_pnvo_event_boundary_diagnostic(self) -> None: data_paths = ( "press@pn_c1_8", "dm1@pneumatic_69", "xv@pn_morifice_1", "dm2@pn_morifice_1", ) diagnostic = TestMqlPnvoEventBoundaryDiagnostic( orifice_alias="pn_morifice_1", event_time_s=0.04, integration_stop_time_s=0.039999999999999994, data_paths=data_paths, python_values_by_data_path={ "press@pn_c1_8": -1090.9, "dm1@pneumatic_69": -0.00949, "xv@pn_morifice_1": 1.0, "dm2@pn_morifice_1": 502.945, }, amesim_values_by_data_path={ "press@pn_c1_8": -1146.38, "dm1@pneumatic_69": -0.00667, "xv@pn_morifice_1": 1.0, "dm2@pn_morifice_1": 497.824, }, ) self.assertEqual(diagnostic.orifice_alias, "pn_morifice_1") self.assertLess(diagnostic.integration_stop_time_s, diagnostic.event_time_s) self.assertEqual(diagnostic.abs_error("xv@pn_morifice_1"), 0.0) self.assertAlmostEqual( diagnostic.abs_error("dm2@pn_morifice_1"), 5.121, ) summary = format_test_mql_pnvo_event_boundary_summary(diagnostic) self.assertIn("PNVO event boundary diagnostic (pn_morifice_1)", summary) self.assertIn("Observation side: right-continuous STEP0 opening", summary) self.assertTrue(summary.endswith("\n")) def test_full_state_script_can_skip_artifact_files(self) -> None: with tempfile.TemporaryDirectory() as tmpdir: output_dir = Path(tmpdir) run, returned_output_dir = run_test_mql_full_state_comparison( TestMqlFullStateComparisonScriptConfig( paths=TestMqlFullStateComparisonPathConfig( archive_path=TEST_MQL_AME, output_dir=output_dir, ), execution=TestMqlFullStateComparisonExecutionConfig( write_summary=False, write_comparison_csv=False, data_paths=("press@pn_c1_8",), solver=SolveIVPConfig(t_start=0.0, t_stop=0.0), t_eval=(0.0,), ), ) ) self.assertEqual(returned_output_dir, output_dir) self.assertEqual(run.signal_count, 1) self.assertFalse((output_dir / "test_mql_full_state_comparison_summary.txt").exists()) self.assertFalse((output_dir / "test_mql_amesim_comparison.csv").exists()) def test_format_test_mql_full_state_comparison_summary(self) -> None: run, _ = run_test_mql_full_state_comparison( TestMqlFullStateComparisonScriptConfig( paths=TestMqlFullStateComparisonPathConfig(archive_path=TEST_MQL_AME), execution=TestMqlFullStateComparisonExecutionConfig( write_summary=False, write_comparison_csv=False, data_paths=("press@pn_c1_8",), solver=SolveIVPConfig(t_start=0.0, t_stop=0.0), t_eval=(0.0,), ), ) ) summary = format_test_mql_full_state_comparison_summary(run) self.assertIn("Model: test_mql", summary) self.assertIn("Output schema signals: 858", summary) self.assertIn("Compared signals: 1", summary) self.assertIn("Largest absolute error: press@pn_c1_8=", summary) self.assertIn("Largest final endpoint error: press@pn_c1_8=", summary) self.assertIn("Metrics by max absolute error:", summary) self.assertIn("Endpoint diagnostics by final absolute error:", summary) self.assertIn("Largest endpoint chamber RHS breakdown: pn_c1_8", summary) self.assertTrue(summary.endswith("\n")) if __name__ == "__main__": unittest.main()