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.
 
 
 
 
 
 

164 lines
5.7 KiB

from django.test import TestCase, override_settings
from django.core.exceptions import ValidationError
from django.db.utils import IntegrityError
from model_mommy import mommy
from ..models import Label, DocumentAnnotation, SequenceAnnotation, Seq2seqAnnotation
from ..serializers import DocumentAnnotationSerializer
from ..serializers import SequenceAnnotationSerializer
from ..serializers import Seq2seqAnnotationSerializer
@override_settings(STATICFILES_STORAGE='django.contrib.staticfiles.storage.StaticFilesStorage')
class TestTextClassificationProject(TestCase):
@classmethod
def setUpTestData(cls):
cls.project = mommy.make('TextClassificationProject')
def test_image(self):
image_url = self.project.image
self.assertTrue(image_url.endswith('.jpg'))
def test_get_bundle_name(self):
template = self.project.get_bundle_name()
self.assertEqual(template, 'document_classification')
def test_get_annotation_serializer(self):
serializer = self.project.get_annotation_serializer()
self.assertEqual(serializer, DocumentAnnotationSerializer)
def test_get_annotation_class(self):
klass = self.project.get_annotation_class()
self.assertEqual(klass, DocumentAnnotation)
@override_settings(STATICFILES_STORAGE='django.contrib.staticfiles.storage.StaticFilesStorage')
class TestSequenceLabelingProject(TestCase):
@classmethod
def setUpTestData(cls):
cls.project = mommy.make('SequenceLabelingProject')
def test_image(self):
image_url = self.project.image
self.assertTrue(image_url.endswith('.jpg'))
def test_get_bundle_name(self):
template = self.project.get_bundle_name()
self.assertEqual(template, 'sequence_labeling')
def test_get_annotation_serializer(self):
serializer = self.project.get_annotation_serializer()
self.assertEqual(serializer, SequenceAnnotationSerializer)
def test_get_annotation_class(self):
klass = self.project.get_annotation_class()
self.assertEqual(klass, SequenceAnnotation)
@override_settings(STATICFILES_STORAGE='django.contrib.staticfiles.storage.StaticFilesStorage')
class TestSeq2seqProject(TestCase):
@classmethod
def setUpTestData(cls):
cls.project = mommy.make('Seq2seqProject')
def test_image(self):
image_url = self.project.image
self.assertTrue(image_url.endswith('.jpg'))
def test_get_bundle_name(self):
template = self.project.get_bundle_name()
self.assertEqual(template, 'seq2seq')
def test_get_annotation_serializer(self):
serializer = self.project.get_annotation_serializer()
self.assertEqual(serializer, Seq2seqAnnotationSerializer)
def test_get_annotation_class(self):
klass = self.project.get_annotation_class()
self.assertEqual(klass, Seq2seqAnnotation)
class TestLabel(TestCase):
def test_text_uniqueness(self):
label = mommy.make('Label')
mommy.make('Label', text=label.text)
with self.assertRaises(IntegrityError):
Label(project=label.project, text=label.text).save()
def test_keys_uniqueness(self):
label = mommy.make('Label', prefix_key='ctrl', suffix_key='a')
with self.assertRaises(ValidationError):
Label(project=label.project,
text='example',
prefix_key=label.prefix_key,
suffix_key=label.suffix_key).full_clean()
def test_suffix_key_uniqueness(self):
label = mommy.make('Label', prefix_key=None, suffix_key='a')
with self.assertRaises(ValidationError):
Label(project=label.project,
text='example',
prefix_key=label.prefix_key,
suffix_key=label.suffix_key).full_clean()
def test_cannot_add_label_only_prefix_key(self):
project = mommy.make('Project')
label = Label(project=project,
text='example',
prefix_key='ctrl')
with self.assertRaises(ValidationError):
label.clean()
def test_can_add_label_only_suffix_key(self):
project = mommy.make('Project')
label = Label(project=project,
text='example',
suffix_key='a')
label.full_clean()
def test_can_add_label_suffix_key_with_prefix_key(self):
project = mommy.make('Project')
label = Label(project=project,
text='example',
prefix_key='ctrl',
suffix_key='a')
label.full_clean()
class TestDocumentAnnotation(TestCase):
def test_uniqueness(self):
a = mommy.make('DocumentAnnotation')
with self.assertRaises(IntegrityError):
DocumentAnnotation(document=a.document, user=a.user, label=a.label).save()
class TestSequenceAnnotation(TestCase):
def test_uniqueness(self):
a = mommy.make('SequenceAnnotation')
with self.assertRaises(IntegrityError):
SequenceAnnotation(document=a.document,
user=a.user,
label=a.label,
start_offset=a.start_offset,
end_offset=a.end_offset).save()
def test_position_constraint(self):
with self.assertRaises(ValidationError):
mommy.make('SequenceAnnotation',
start_offset=1, end_offset=0).clean()
class TestSeq2seqAnnotation(TestCase):
def test_uniqueness(self):
a = mommy.make('Seq2seqAnnotation')
with self.assertRaises(IntegrityError):
Seq2seqAnnotation(document=a.document,
user=a.user,
text=a.text).save()