🚧 Add environment maneuverability

envs/pump_station_env.py

@@ -39,7 +39,7 @@ LEVEL_HIGH = 2.7         # m — high-level switch threshold
 class PumpStationEnv(PlantEnv):
     """Closed-loop tank + pump physics over the pump_node layout."""
 
-    TICK_MS = 9.0        # overrides PlantEnv TICK_MS
+    TICK_MS = 10.0        # overrides PlantEnv TICK_MS
 
     def __init__(self, *args, stop_after_ms: float | None = None, **kw):
         super().__init__(*args, **kw)
@@ -49,7 +49,7 @@ class PumpStationEnv(PlantEnv):
     def model(self, state: PlantState, dt_s: float) -> dict[str, object]:
         dt_h = dt_s / 3600.0
         # what the PLC currently commands (master side: DO/AO)
-        pump_running = bool(state.by_code("DO0.0"))
+        pump_running = bool(state.by_code("DI0.0"))
 
         # speed setpoint from AO0.0, normalized over its display range
         ao_lo, ao_hi = state.channel_range("AO0.0")

envs/test_done_env.py

@@ -0,0 +1,28 @@
+"""Test-only environment: is_done() after 30 ms of sim time.
+
+Used by the web-UI env-control tests (FR-10.2.15): the paced ticker
+must notice is_done(), set state "done" (no auto-restart), and keep
+the instance for inspection.
+"""
+
+from __future__ import annotations
+
+from wago_sim.scenario.plantenv import PlantEnv, PlantState
+
+
+class PumpStationEnv(PlantEnv):
+    """Minimal env: fills AI0.0, done after 30 ms."""
+
+    def __init__(self, *args, stop_after_ms: float | None = 30.0, **kw):
+        super().__init__(*args, **kw)
+        self.stop_after_ms = stop_after_ms
+
+    def model(self, state: PlantState, dt_s: float) -> dict[str, object]:
+        level = float(state.by_code("AI0.0"))
+        return {"AI0.0": level + 1.0}
+
+    def is_done(self, state: PlantState) -> bool:
+        return state.sim_time_ms >= (self.stop_after_ms or 0.0)
+
+
+Env = PumpStationEnv