Coverage for agentlib_flexquant/modules/baseline_mpc.py: 53%

15 statements  

« prev     ^ index     » next       coverage.py v7.4.4, created at 2025-08-01 15:10 +0000

1from agentlib_mpc.modules import mpc_full, minlp_mpc 

2 

3 

4class FlexibilityBaselineMPC(mpc_full.MPC): 

5 config: mpc_full.MPCConfig 

6 

7 def pre_computation_hook(self): 

8 if self.get("in_provision").value: 

9 timestep = (self.get("_P_external").value.index[1] - 

10 self.get("_P_external").value.index[0]) 

11 self.set("rel_start", self.get("_P_external").value.index[0] - 

12 self.env.time) 

13 # the provision profile gives a value for the start of a time step. 

14 # For the end of the flex interval add time step! 

15 self.set("rel_end", self.get("_P_external").value.index[-1] - 

16 self.env.time + timestep) 

17 

18 

19class FlexibilityBaselineMINLPMPC(minlp_mpc.MINLPMPC): 

20 config: minlp_mpc.MINLPMPCConfig 

21 

22 def pre_computation_hook(self): 

23 if self.get("in_provision").value: 

24 timestep = (self.get("_P_external").value.index[1] - 

25 self.get("_P_external").value.index[0]) 

26 self.set("rel_start", self.get("_P_external").value.index[0] - 

27 self.env.time) 

28 # the provision profile gives a value for the start of a time step. 

29 # For the end of the flex interval add time step! 

30 self.set("rel_end", self.get("_P_external").value.index[-1] - 

31 self.env.time + timestep)