|
|
@ -17,10 +17,6 @@ class TestTextClassificationProject(TestCase): |
|
|
|
def setUpTestData(cls): |
|
|
|
cls.project = mommy.make('TextClassificationProject') |
|
|
|
|
|
|
|
def test_image(self): |
|
|
|
image_url = self.project.image |
|
|
|
self.assertTrue(image_url.endswith('.jpg')) |
|
|
|
|
|
|
|
def test_get_bundle_name(self): |
|
|
|
template = self.project.get_bundle_name() |
|
|
|
self.assertEqual(template, 'document_classification') |
|
|
@ -41,10 +37,6 @@ class TestSequenceLabelingProject(TestCase): |
|
|
|
def setUpTestData(cls): |
|
|
|
cls.project = mommy.make('SequenceLabelingProject') |
|
|
|
|
|
|
|
def test_image(self): |
|
|
|
image_url = self.project.image |
|
|
|
self.assertTrue(image_url.endswith('.jpg')) |
|
|
|
|
|
|
|
def test_get_bundle_name(self): |
|
|
|
template = self.project.get_bundle_name() |
|
|
|
self.assertEqual(template, 'sequence_labeling') |
|
|
@ -65,10 +57,6 @@ class TestSeq2seqProject(TestCase): |
|
|
|
def setUpTestData(cls): |
|
|
|
cls.project = mommy.make('Seq2seqProject') |
|
|
|
|
|
|
|
def test_image(self): |
|
|
|
image_url = self.project.image |
|
|
|
self.assertTrue(image_url.endswith('.jpg')) |
|
|
|
|
|
|
|
def test_get_bundle_name(self): |
|
|
|
template = self.project.get_bundle_name() |
|
|
|
self.assertEqual(template, 'seq2seq') |
|
|
@ -89,10 +77,6 @@ class TestSpeech2textProject(TestCase): |
|
|
|
def setUpTestData(cls): |
|
|
|
cls.project = mommy.make('Speech2textProject') |
|
|
|
|
|
|
|
def test_image(self): |
|
|
|
image_url = self.project.image |
|
|
|
self.assertTrue(image_url.endswith('.jpg')) |
|
|
|
|
|
|
|
def test_get_bundle_name(self): |
|
|
|
template = self.project.get_bundle_name() |
|
|
|
self.assertEqual(template, 'speech2text') |
|
|
|