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

1from rest_framework import serializers 

2from core.auxiliary.serializers.PropertyInfoSerializer import PropertySetSerializer 

3from PinchAnalysis.models.OptionModels import MainOptions, TurbineOptions 

4 

5class TurbineOptionsSerializer(serializers.ModelSerializer): 

6 

7 properties = PropertySetSerializer(required=False) 

8 

9 class Meta: 

10 model = TurbineOptions 

11 fields = '__all__' 

12 

13 def create(self, validated_data): 

14 return TurbineOptions.create(**validated_data) 

15 

16class MainOptionsSerializer(serializers.ModelSerializer): 

17 selections = PropertySetSerializer(required=False) 

18 turbine_options = TurbineOptionsSerializer(required=False) 

19 

20 class Meta: 

21 model = MainOptions 

22 fields = '__all__' 

23 

24 def create(self, validated_data): 

25 return MainOptions.create(**validated_data)