Handle individual pump unit toggles

tests/test_simulation.py

@@ -101,3 +101,31 @@ def test_toggle_maintenance_progresses_until_restored():
     sim.log()
     assert pump.integrity >= 0.99
     assert "primary_pump" not in reactor.maintenance_active
+
+
+def test_primary_pump_unit_toggle_updates_active_flag():
+    reactor = Reactor.default()
+    state = reactor.initial_state()
+    reactor.primary_pump_active = True
+    reactor.primary_pump_units = [True, True]
+
+    reactor.step(state, dt=1.0, command=ReactorCommand(primary_pumps={1: False}))
+    assert reactor.primary_pump_units == [False, True]
+    assert reactor.primary_pump_active is True
+
+    reactor.step(state, dt=1.0, command=ReactorCommand(primary_pumps={2: False}))
+    assert reactor.primary_pump_units == [False, False]
+    assert reactor.primary_pump_active is False
+
+
+def test_secondary_pump_unit_toggle_can_restart_pump():
+    reactor = Reactor.default()
+    state = reactor.initial_state()
+    reactor.secondary_pump_active = False
+    reactor.secondary_pump_units = [False, False]
+
+    reactor.step(state, dt=1.0, command=ReactorCommand(secondary_pumps={1: True}))
+
+    assert reactor.secondary_pump_units == [True, False]
+    assert reactor.secondary_pump_active is True
+    assert state.secondary_loop.mass_flow_rate > 0.0