diff --git a/app/api/tests/test_models.py b/app/api/tests/test_models.py index 35f10393..781392ca 100644 --- a/app/api/tests/test_models.py +++ b/app/api/tests/test_models.py @@ -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')