From db9fb8f1d3aec1b76aaed3aab6d31a3913d0cadb Mon Sep 17 00:00:00 2001 From: Hironsan Date: Mon, 30 May 2022 10:35:31 +0900 Subject: [PATCH] Fix test_getitem test in TestExamples --- backend/data_import/tests/test_examples.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/backend/data_import/tests/test_examples.py b/backend/data_import/tests/test_examples.py index fc8bb2de..4ce574d3 100644 --- a/backend/data_import/tests/test_examples.py +++ b/backend/data_import/tests/test_examples.py @@ -8,12 +8,12 @@ from projects.models import DOCUMENT_CLASSIFICATION from projects.tests.utils import prepare_project -class TestCategories(TestCase): +class TestExamples(TestCase): def setUp(self): self.project = prepare_project(DOCUMENT_CLASSIFICATION) self.example_uuid = uuid.uuid4() - self.example = Example(uuid=self.example_uuid, text="A", project=self.project.item) - self.examples = Examples([self.example]) + example = Example(uuid=self.example_uuid, text="A", project=self.project.item) + self.examples = Examples([example]) def test_save(self): self.examples.save() @@ -22,4 +22,4 @@ class TestCategories(TestCase): def test_getitem(self): self.examples.save() example = self.examples[self.example_uuid] - self.assertEqual(example, self.example) + self.assertEqual(example.uuid, self.example_uuid)