Coverage for backend/idaes_factory/adapters/toggle_adapter.py: 100%

1from .property_info_adapter import ValueAdapter 

2from ..idaes_factory_context import IdaesFactoryContext 

3from ..queryset_lookup import get_property, get_value_object 

4from common.models.idaes import PropertySchema, PropertiesSchema, PropertyValueSchema 

5from flowsheetInternals.unitops.models import SimulationObject 

6 

7 

8class ToggleAdapter(ValueAdapter): 

9 """ 

10 Returns True if the unit model has dynamics enabled, otherwise False. 

11 """ 

12 def __init__(self, toggle_key:str): 

13 self.toggle_key = toggle_key 

14 

15 def serialise(self, ctx:IdaesFactoryContext, model: SimulationObject) -> bool: 

16 

17 # get the property from the unit model 

18 property_set = model.properties 

19 prop = get_property(property_set, self.toggle_key) 

20 # serialise the property 

21 return prop.values.first().value == True