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.
44 lines
2.0 KiB
44 lines
2.0 KiB
from model_mommy import mommy
|
|
from rest_framework import status
|
|
from rest_framework.reverse import reverse
|
|
|
|
from api.tests.api.utils import CRUDMixin, prepare_project, make_doc, make_label
|
|
from api.models import DOCUMENT_CLASSIFICATION
|
|
|
|
|
|
class TestMemberProgress(CRUDMixin):
|
|
|
|
def setUp(self):
|
|
self.project = prepare_project(DOCUMENT_CLASSIFICATION)
|
|
self.example = make_doc(self.project.item)
|
|
self.url = reverse(viewname='member_progress', args=[self.project.item.id])
|
|
|
|
def test_fetch_initial_progress(self):
|
|
response = self.assert_fetch(self.project.admin, status.HTTP_200_OK)
|
|
expected_progress = [{'user': member.username, 'done': 0} for member in self.project.members]
|
|
self.assertEqual(response.data, {'total': 1, 'progress': expected_progress})
|
|
|
|
def test_fetch_progress(self):
|
|
mommy.make('ExampleState', example=self.example, confirmed_by=self.project.admin)
|
|
response = self.assert_fetch(self.project.admin, status.HTTP_200_OK)
|
|
expected_progress = [{'user': member.username, 'done': 0} for member in self.project.members]
|
|
expected_progress[0]['done'] = 1
|
|
self.assertEqual(response.data, {'total': 1, 'progress': expected_progress})
|
|
|
|
|
|
class TestCategoryDistribution(CRUDMixin):
|
|
|
|
def setUp(self):
|
|
self.project = prepare_project(DOCUMENT_CLASSIFICATION)
|
|
self.example = make_doc(self.project.item)
|
|
self.label = make_label(self.project.item, text='label')
|
|
mommy.make('Category', example=self.example, label=self.label, user=self.project.admin)
|
|
self.url = reverse(viewname='category_distribution', args=[self.project.item.id])
|
|
|
|
def test_fetch_distribution(self):
|
|
response = self.assert_fetch(self.project.admin, status.HTTP_200_OK)
|
|
expected = {
|
|
member.username: {self.label.text: 0} for member in self.project.members
|
|
}
|
|
expected[self.project.admin.username][self.label.text] = 1
|
|
self.assertEqual(response.data, expected)
|