diff --git a/backend/api/tests/api/utils.py b/backend/api/tests/api/utils.py index 60a32234..9dc65dcf 100644 --- a/backend/api/tests/api/utils.py +++ b/backend/api/tests/api/utils.py @@ -102,17 +102,6 @@ def make_tag(project): return mommy.make('Tag', project=project) -def make_label(project, **kwargs): - if project.project_type.endswith('Classification'): - return mommy.make('CategoryType', project=project, **kwargs) - else: - return mommy.make('SpanType', project=project, **kwargs) - - -def make_auto_labeling_config(project): - return mommy.make('AutoLabelingConfig', project=project) - - def prepare_project(task: str = 'Any', collaborative_annotation=False, **kwargs): return make_project( task=task, diff --git a/backend/label_types/tests/test_views.py b/backend/label_types/tests/test_views.py index 289f519f..bb05f737 100644 --- a/backend/label_types/tests/test_views.py +++ b/backend/label_types/tests/test_views.py @@ -6,7 +6,8 @@ from rest_framework.reverse import reverse from rest_framework.test import APITestCase from projects.models import DOCUMENT_CLASSIFICATION -from api.tests.api.utils import (CRUDMixin, make_label, make_project, make_user, prepare_project) +from api.tests.api.utils import (CRUDMixin, make_project, make_user, prepare_project) +from .utils import make_label DATA_DIR = os.path.join(os.path.dirname(__file__), 'data') diff --git a/backend/label_types/tests/utils.py b/backend/label_types/tests/utils.py new file mode 100644 index 00000000..4eb41d39 --- /dev/null +++ b/backend/label_types/tests/utils.py @@ -0,0 +1,8 @@ +from model_mommy import mommy + + +def make_label(project, **kwargs): + if project.project_type.endswith('Classification'): + return mommy.make('CategoryType', project=project, **kwargs) + else: + return mommy.make('SpanType', project=project, **kwargs) diff --git a/backend/labels/tests/test_views.py b/backend/labels/tests/test_views.py index 7d912b93..28d0b9ba 100644 --- a/backend/labels/tests/test_views.py +++ b/backend/labels/tests/test_views.py @@ -3,9 +3,9 @@ from rest_framework.reverse import reverse from projects.models import DOCUMENT_CLASSIFICATION, SEQUENCE_LABELING, SEQ2SEQ from labels.models import Category, Span, TextLabel -from api.tests.api.utils import (CRUDMixin, make_label, - make_user, prepare_project) +from api.tests.api.utils import (CRUDMixin, make_user, prepare_project) from examples.tests.utils import make_doc +from label_types.tests.utils import make_label from .utils import make_annotation diff --git a/backend/metrics/tests.py b/backend/metrics/tests.py index da1d75da..2757b700 100644 --- a/backend/metrics/tests.py +++ b/backend/metrics/tests.py @@ -2,8 +2,9 @@ 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_label +from api.tests.api.utils import CRUDMixin, prepare_project from examples.tests.utils import make_doc +from label_types.tests.utils import make_label from projects.models import DOCUMENT_CLASSIFICATION