diff --git a/backend/api/tests/data/example.jsonl b/backend/api/tests/data/example.jsonl deleted file mode 100644 index 9bd22c86..00000000 --- a/backend/api/tests/data/example.jsonl +++ /dev/null @@ -1,3 +0,0 @@ -{"text": "example"} -{"text": "example"} -{"text": "example"} diff --git a/backend/api/tests/data/classification.jsonl b/backend/api/tests/data/text_classification/classification.jsonl similarity index 100% rename from backend/api/tests/data/classification.jsonl rename to backend/api/tests/data/text_classification/classification.jsonl diff --git a/backend/api/tests/data/example.csv b/backend/api/tests/data/text_classification/example.csv similarity index 100% rename from backend/api/tests/data/example.csv rename to backend/api/tests/data/text_classification/example.csv diff --git a/backend/api/tests/data/example.json b/backend/api/tests/data/text_classification/example.json similarity index 100% rename from backend/api/tests/data/example.json rename to backend/api/tests/data/text_classification/example.json diff --git a/backend/api/tests/data/example.xlsx b/backend/api/tests/data/text_classification/example.xlsx similarity index 100% rename from backend/api/tests/data/example.xlsx rename to backend/api/tests/data/text_classification/example.xlsx diff --git a/backend/api/tests/data/example_fasttext.txt b/backend/api/tests/data/text_classification/example_fasttext.txt similarity index 100% rename from backend/api/tests/data/example_fasttext.txt rename to backend/api/tests/data/text_classification/example_fasttext.txt diff --git a/backend/api/tests/test_tasks.py b/backend/api/tests/test_tasks.py index 622ee32a..2b994036 100644 --- a/backend/api/tests/test_tasks.py +++ b/backend/api/tests/test_tasks.py @@ -29,28 +29,28 @@ class TestIngestClassificationData(TestCase): self.assertEqual(Category.objects.count(), expected_annotation) def test_jsonl(self): - filename = 'classification.jsonl' + filename = 'text_classification/example.jsonl' file_format = 'JSONL' kwargs = {'column_label': 'labels'} self.assert_count(filename, file_format, kwargs, expected_example=4, expected_label=3, expected_annotation=5) def test_csv(self): - filename = 'example.csv' + filename = 'text_classification/example.csv' file_format = 'CSV' self.assert_count(filename, file_format, expected_example=4, expected_label=2, expected_annotation=2) def test_fasttext(self): - filename = 'example_fasttext.txt' + filename = 'text_classification/example_fasttext.txt' file_format = 'fastText' self.assert_count(filename, file_format, expected_example=4, expected_label=5, expected_annotation=5) def test_excel(self): - filename = 'example.xlsx' + filename = 'text_classification/example.xlsx' file_format = 'Excel' self.assert_count(filename, file_format, expected_example=3, expected_label=2, expected_annotation=3) def test_json(self): - filename = 'example.json' + filename = 'text_classification/example.json' file_format = 'JSON' self.assert_count(filename, file_format, expected_example=4, expected_label=3, expected_annotation=5)