Coverage for backend/idaes_service/solver/custom/custom_cooler.py: 100%

1from idaes.models.unit_models.heater import HeaterData 

2from .inverted import add_inverted, initialise_inverted, enable_inverted, disable_inverted 

3from idaes.core import declare_process_block_class 

4 

5@declare_process_block_class("CustomCooler") 

6class CustomCoolerData(HeaterData): 

7 """ 

8 Custom Cooler model that includes inverted deltaP and Heat Added properties. 

9 """ 

10 

11 def build(self, *args, **kwargs): 

12 super().build(*args, **kwargs) 

13 

14 # add deltaP_inverted as a property 

15 add_inverted(self, "heat_duty") 

16 if hasattr(self,"deltaP"): 

17 add_inverted(self, "deltaP") 

18 

19 def initialize_build( 

20 self,*args,**kwargs, 

21 ): 

22 initialise_inverted(self, "heat_duty") 

23 disable_inverted(self, "heat_duty") 

24 

25 if hasattr(self,"deltaP"): 

26 initialise_inverted(self, "deltaP") 

27 disable_inverted(self, "deltaP") 

28 

29 super().initialize_build(*args, **kwargs) 

30 

31 enable_inverted(self, "heat_duty") 

32 if hasattr(self,"deltaP"): 

33 enable_inverted(self, "deltaP")