Persist manual control and pump states

This commit is contained in:
Codex Agent
2025-11-22 18:47:19 +01:00
parent f626c99f44
commit 511544c2eb
3 changed files with 21 additions and 0 deletions

View File

@@ -20,6 +20,7 @@ def test_reactor_initial_state_is_cold():
def test_state_save_and_load_roundtrip(tmp_path: Path):
reactor = Reactor.default()
reactor.control.manual_control = True
sim = ReactorSimulation(reactor, timestep=5.0, duration=15.0)
sim.log()
save_path = tmp_path / "plant_state.json"
@@ -32,6 +33,9 @@ def test_state_save_and_load_roundtrip(tmp_path: Path):
sim.last_state.core.fuel_temperature
)
assert restored_reactor.control.rod_fraction == reactor.control.rod_fraction
assert restored_reactor.control.manual_control == reactor.control.manual_control
assert len(restored_state.primary_pumps) == 2
assert len(restored_state.secondary_pumps) == 2
def test_health_monitor_flags_core_failure():