Coverage for backend/django/PinchAnalysis/serializers/PinchOptionsSerializers.py: 89%

1from core.serializer_base import StateOwnedModelSerializer 

2from rest_framework import serializers 

3from core.auxiliary.serializers.PropertyInfoSerializer import PropertySetSerializer 

4from PinchAnalysis.models.OptionModels import MainOptions, TurbineOptions 

5 

6class TurbineOptionsSerializer(StateOwnedModelSerializer): 

7 

8 properties = PropertySetSerializer(required=False) 

9 

10 class Meta: 

11 model = TurbineOptions 

12 fields = '__all__' 

13 

14 def create(self, validated_data): 

15 return TurbineOptions.create(**validated_data) 

16 

17class MainOptionsSerializer(StateOwnedModelSerializer): 

18 selections = PropertySetSerializer(required=False) 

19 turbine_options = TurbineOptionsSerializer(required=False) 

20 

21 class Meta: 

22 model = MainOptions 

23 fields = '__all__' 

24 

25 def create(self, validated_data): 

26 return MainOptions.create(**validated_data)