Merge pull request #2132 from doccano/enhancement/packages

[Enhancement] Update dependencies

backend/examples/tests/test_models.py

@@ -55,7 +55,8 @@ def test_user_count_after_multiple_user_confirmation(self):
         expected_progress = [{"user": member.username, "done": 0} for member in self.project.members]
         expected_progress[0]["done"] = 1
         expected_progress[1]["done"] = 1
-        self.assertEqual(progress, {"total": 2, "progress": expected_progress})
+        self.assertEqual(progress["total"], 2)
+        self.assertCountEqual(progress["progress"], expected_progress)
 
 
 class TestExample(TestCase):

backend/examples/views/example.py

@@ -21,7 +21,7 @@ class ExampleList(generics.ListCreateAPIView):
     ordering_fields = ("created_at", "updated_at", "score")
     search_fields = ("text", "filename")
     model = Example
-    filter_class = ExampleFilter
+    filterset_class = ExampleFilter
 
     @property
     def project(self):

backend/label_types/tests/utils.py

@@ -1,10 +1,10 @@
 from model_mommy import mommy
 
-from projects.models import BOUNDING_BOX
+from projects.models import BOUNDING_BOX, SEGMENTATION
 
 
 def make_label(project, **kwargs):
-    if project.project_type.endswith("Classification") or project.project_type == BOUNDING_BOX:
+    if project.project_type.endswith("Classification") or project.project_type in {BOUNDING_BOX, SEGMENTATION}:
         return mommy.make("CategoryType", project=project, **kwargs)
     else:
         return mommy.make("SpanType", project=project, **kwargs)

backend/labels/tests/test_span.py

@@ -1,7 +1,6 @@
 import abc
 
 from django.core.exceptions import ValidationError
-from django.db import IntegrityError
 from django.test import TestCase
 from model_mommy import mommy
 
@@ -142,15 +141,15 @@ def setUp(self):
         self.user = self.project.admin
 
     def test_start_offset_is_not_negative(self):
-        with self.assertRaises(IntegrityError):
+        with self.assertRaises(ValidationError):
             mommy.make("Span", start_offset=-1, end_offset=0)
 
     def test_end_offset_is_not_negative(self):
-        with self.assertRaises(IntegrityError):
+        with self.assertRaises(ValidationError):
             mommy.make("Span", start_offset=-2, end_offset=-1)
 
     def test_start_offset_is_less_than_end_offset(self):
-        with self.assertRaises(IntegrityError):
+        with self.assertRaises(ValidationError):
             mommy.make("Span", start_offset=0, end_offset=0)
 
     def test_unique_constraint(self):