修复test_mql事件后积分步长控制

tests/test_core_solver.py

@@ -0,0 +1,68 @@
+import sys
+import types
+import unittest
+from unittest.mock import patch
+
+from PythonModels.core.solver import SolveIVPConfig, integrate_ode
+
+
+class IntegrateOdeTests(unittest.TestCase):
+    def test_scipy_solver_receives_step_size_controls(self) -> None:
+        calls: list[dict[str, object]] = []
+
+        def fake_solve_ivp(**kwargs):
+            calls.append(kwargs)
+            return object()
+
+        scipy_module = types.ModuleType("scipy")
+        integrate_module = types.ModuleType("scipy.integrate")
+        integrate_module.solve_ivp = fake_solve_ivp
+        scipy_module.integrate = integrate_module
+
+        with patch.dict(
+            sys.modules,
+            {"scipy": scipy_module, "scipy.integrate": integrate_module},
+        ):
+            integrate_ode(
+                rhs=lambda _time, state: state,
+                initial_state=[1.0],
+                config=SolveIVPConfig(
+                    t_start=0.0,
+                    t_stop=1.0,
+                    max_step=1.0e-4,
+                    first_step=1.0e-8,
+                ),
+                t_eval=[0.0, 1.0],
+            )
+
+        self.assertEqual(calls[0]["max_step"], 1.0e-4)
+        self.assertEqual(calls[0]["first_step"], 1.0e-8)
+
+    def test_scipy_solver_omits_unset_first_step(self) -> None:
+        calls: list[dict[str, object]] = []
+
+        def fake_solve_ivp(**kwargs):
+            calls.append(kwargs)
+            return object()
+
+        scipy_module = types.ModuleType("scipy")
+        integrate_module = types.ModuleType("scipy.integrate")
+        integrate_module.solve_ivp = fake_solve_ivp
+        scipy_module.integrate = integrate_module
+
+        with patch.dict(
+            sys.modules,
+            {"scipy": scipy_module, "scipy.integrate": integrate_module},
+        ):
+            integrate_ode(
+                rhs=lambda _time, state: state,
+                initial_state=[1.0],
+                config=SolveIVPConfig(t_start=0.0, t_stop=1.0),
+            )
+
+        self.assertEqual(calls[0]["max_step"], 1.0e-3)
+        self.assertNotIn("first_step", calls[0])
+
+
+if __name__ == "__main__":
+    unittest.main()