Coverage for backend/django/idaes_factory/adapters/unit_model_adapter.py: 100%

1from ahuora_builder_types import UnitModelSchema 

2from flowsheetInternals.unitops.models import SimulationObject 

3from .property_info_adapter import SerialisePropertiesAdapter 

4from .port_adapter import SerialisePortAdapter 

5from .arg_adapter import ArgAdapter 

6from .port_adapter import PortListAdapter 

7from ..idaes_factory_context import IdaesFactoryContext 

8 

9 

10class UnitModelAdapter(): 

11 def __init__(self, args:ArgAdapter, properties:SerialisePropertiesAdapter, ports:SerialisePortAdapter) -> None: 

12 self.arg_schema: ArgAdapter = args 

13 self.properties_schema : SerialisePropertiesAdapter = properties 

14 self.ports_schema: SerialisePortAdapter = ports 

15 

16 

17 def serialise(self, ctx: IdaesFactoryContext, unit_model:SimulationObject) -> UnitModelSchema: 

18 return UnitModelSchema( 

19 id=unit_model.pk, 

20 type=unit_model.objectType, 

21 name=unit_model.componentName, 

22 args=self.arg_schema.serialise(ctx, unit_model), 

23 properties=self.properties_schema.serialise(ctx, unit_model), 

24 ports=self.ports_schema.serialise(ctx, unit_model), 

25 initial_values=unit_model.initial_values, 

26 )