mirror of https://github.com/doccano/doccano.git
pythonannotation-tooldatasetsactive-learningtext-annotationdatasetnatural-language-processingdata-labelingmachine-learning
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
91 lines
2.9 KiB
91 lines
2.9 KiB
import abc
|
|
|
|
from django.db import IntegrityError
|
|
from django.test import TestCase
|
|
from model_mommy import mommy
|
|
|
|
from api.models import SEQ2SEQ
|
|
from labels.models import TextLabel
|
|
from api.tests.api.utils import prepare_project
|
|
|
|
|
|
class TestTextLabeling(abc.ABC, TestCase):
|
|
collaborative = False
|
|
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
cls.project = prepare_project(
|
|
SEQ2SEQ,
|
|
collaborative_annotation=cls.collaborative
|
|
)
|
|
cls.example = mommy.make('Example', project=cls.project.item)
|
|
cls.user = cls.project.admin
|
|
cls.another_user = cls.project.approver
|
|
cls.text_label = TextLabel(
|
|
example=cls.example,
|
|
user=cls.user,
|
|
text='foo'
|
|
)
|
|
|
|
def test_can_annotate_category_to_unannotated_data(self):
|
|
can_annotate = TextLabel.objects.can_annotate(self.text_label, self.project.item)
|
|
self.assertTrue(can_annotate)
|
|
|
|
def test_uniqueness(self):
|
|
a = mommy.make('TextLabel')
|
|
with self.assertRaises(IntegrityError):
|
|
TextLabel(example=a.example,
|
|
user=a.user,
|
|
text=a.text).save()
|
|
|
|
|
|
class TestNonCollaborativeTextLabeling(TestTextLabeling):
|
|
collaborative = False
|
|
|
|
def test_cannot_annotate_same_text_to_annotated_data(self):
|
|
mommy.make(
|
|
'TextLabel',
|
|
example=self.example,
|
|
user=self.user,
|
|
text=self.text_label.text
|
|
)
|
|
can_annotate = TextLabel.objects.can_annotate(self.text_label, self.project.item)
|
|
self.assertFalse(can_annotate)
|
|
|
|
def test_can_annotate_different_text_to_annotated_data(self):
|
|
mommy.make('TextLabel', example=self.example, user=self.user)
|
|
can_annotate = TextLabel.objects.can_annotate(self.text_label, self.project.item)
|
|
self.assertTrue(can_annotate)
|
|
|
|
def test_allow_another_user_to_annotate_same_text(self):
|
|
mommy.make(
|
|
'TextLabel',
|
|
example=self.example,
|
|
user=self.another_user,
|
|
text=self.text_label.text
|
|
)
|
|
can_annotate = TextLabel.objects.can_annotate(self.text_label, self.project.item)
|
|
self.assertTrue(can_annotate)
|
|
|
|
|
|
class TestCollaborativeTextLabeling(TestTextLabeling):
|
|
collaborative = True
|
|
|
|
def test_deny_another_user_to_annotate_same_text(self):
|
|
mommy.make(
|
|
'TextLabel',
|
|
example=self.example,
|
|
user=self.another_user,
|
|
text=self.text_label.text
|
|
)
|
|
can_annotate = TextLabel.objects.can_annotate(self.text_label, self.project.item)
|
|
self.assertFalse(can_annotate)
|
|
|
|
def test_allow_another_user_to_annotate_different_text(self):
|
|
mommy.make(
|
|
'TextLabel',
|
|
example=self.example,
|
|
user=self.another_user
|
|
)
|
|
can_annotate = TextLabel.objects.can_annotate(self.text_label, self.project.item)
|
|
self.assertTrue(can_annotate)
|