Update label serializer to include task_type

backend/api/serializers.py

@@ -60,7 +60,15 @@ class LabelSerializer(serializers.ModelSerializer):
 
     class Meta:
         model = Label
-        fields = ('id', 'text', 'prefix_key', 'suffix_key', 'background_color', 'text_color')
+        fields = (
+            'id',
+            'text',
+            'prefix_key',
+            'suffix_key',
+            'background_color',
+            'text_color',
+            'task_type',
+        )
 
 
 class CommentSerializer(serializers.ModelSerializer):

backend/api/tests/test_serializers.py

@@ -0,0 +1,20 @@
+from django.test import TestCase
+
+from ..models import SEQUENCE_LABELING, Label
+from ..serializers import LabelSerializer
+from .api.utils import prepare_project
+
+
+class TestLabelSerializer(TestCase):
+
+    def test_create_label(self):
+        project = prepare_project(SEQUENCE_LABELING)
+        data = {
+            'text': 'example',
+            'task_type': 'Span'
+        }
+        serializer = LabelSerializer(data=data)
+        serializer.is_valid()
+        label = serializer.save(project=project.item)
+        created = Label.objects.get(pk=label.id)
+        self.assertEqual(label, created)