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)