Coverage for backend/idaes_factory/adapters/enum_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 flowsheetInternals.unitops.models import SimulationObject 

5 

6 

7class EnumAdapter(ValueAdapter): 

8 """ 

9 Returns the enum value of the property. 

10 """ 

11 def __init__(self, toggle_key:str): 

12 self.toggle_key = toggle_key 

13 

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

15 

16 # get the property from the unit model 

17 property_set = model.properties 

18 prop = get_property(property_set, self.toggle_key) 

19 # serialise the property 

20 return prop.values.first().value