From 11030ad074cc09dabbbef1dec81d89017ae56844 Mon Sep 17 00:00:00 2001 From: Hironsan Date: Thu, 24 Feb 2022 07:05:02 +0900 Subject: [PATCH] Add a test case for relation type creation --- backend/labels/tests/test_views.py | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/backend/labels/tests/test_views.py b/backend/labels/tests/test_views.py index 78ac2a38..f32b4d45 100644 --- a/backend/labels/tests/test_views.py +++ b/backend/labels/tests/test_views.py @@ -1,3 +1,4 @@ +from model_mommy import mommy from rest_framework import status from rest_framework.reverse import reverse @@ -127,9 +128,9 @@ class TestDataLabeling: def setUp(self): self.project = prepare_project(task=self.task) self.non_member = make_user() - doc = make_doc(self.project.item) + self.doc = make_doc(self.project.item) self.data = self.create_data() - self.url = reverse(viewname=self.view_name, args=[self.project.item.id, doc.id]) + self.url = reverse(viewname=self.view_name, args=[self.project.item.id, self.doc.id]) def create_data(self): label = make_label(self.project.item) @@ -159,6 +160,17 @@ class TestSpanCreation(TestDataLabeling, CRUDMixin): return {"label": label.id, "start_offset": 0, "end_offset": 1} +class TestRelationCreation(TestDataLabeling, CRUDMixin): + task = SEQUENCE_LABELING + view_name = "relation_list" + + def create_data(self): + relation_type = mommy.make("RelationType", project=self.project.item) + from_id = mommy.make("Span", example=self.doc, start_offset=0, end_offset=1) + to_id = mommy.make("Span", example=self.doc, start_offset=1, end_offset=2) + return {"type": relation_type.id, "from_id": from_id.id, "to_id": to_id.id} + + class TestTextLabelCreation(TestDataLabeling, CRUDMixin): task = SEQ2SEQ view_name = "text_list"