mirror of https://github.com/doccano/doccano.git
pythondatasetsactive-learningtext-annotationdatasetnatural-language-processingdata-labelingmachine-learningannotation-tool
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.
25 lines
799 B
25 lines
799 B
import uuid
|
|
|
|
from django.test import TestCase
|
|
|
|
from data_import.pipeline.examples import Examples
|
|
from examples.models import Example
|
|
from projects.models import DOCUMENT_CLASSIFICATION
|
|
from projects.tests.utils import prepare_project
|
|
|
|
|
|
class TestExamples(TestCase):
|
|
def setUp(self):
|
|
self.project = prepare_project(DOCUMENT_CLASSIFICATION)
|
|
self.example_uuid = uuid.uuid4()
|
|
example = Example(uuid=self.example_uuid, text="A", project=self.project.item)
|
|
self.examples = Examples([example])
|
|
|
|
def test_save(self):
|
|
self.examples.save()
|
|
self.assertEqual(Example.objects.count(), 1)
|
|
|
|
def test_getitem(self):
|
|
self.examples.save()
|
|
example = self.examples[self.example_uuid]
|
|
self.assertEqual(example.uuid, self.example_uuid)
|