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

1from flowsheetInternals.unitops.models import SimulationObject 

2from .property_info_adapter import ValueAdapter 

3from ..idaes_factory_context import IdaesFactoryContext 

4 

5 

6 

7class ArgAdapter(ValueAdapter): 

8 """ 

9 Represents the methods used to get the arguments of a unit model. 

10 Only serialisation is supported because arguments aren't returned. 

11 """ 

12 def __init__(self, schema: dict[str, ValueAdapter]) -> None: 

13 self.schema = schema 

14 

15 def serialise(self, ctx: IdaesFactoryContext, unit_model: SimulationObject): 

16 return { 

17 key: adapter.serialise(ctx, unit_model) 

18 for key, adapter in self.schema.items() 

19 } 

20 

21 

22class ConstantArg(ValueAdapter): 

23 def __init__(self, value): 

24 self.value = value 

25 

26 def serialise(self, ctx, unit_model): 

27 return self.value