Add task test case for gcp image label detection

Hironsan · Hironsan · commit ca138ec4ef30 · 2021-06-08T11:46:17.000+09:00
diff --git a/tests/test_task.py b/tests/test_task.py
@@ -1,15 +1,16 @@
 import pytest
 
 from auto_labeling_pipeline.labels import ClassificationLabels, Seq2seqLabels, SequenceLabels
-from auto_labeling_pipeline.task import DocumentClassification, Seq2seq, SequenceLabeling
+from auto_labeling_pipeline.task import DocumentClassification, ImageClassification, Seq2seq, SequenceLabeling
 
 
 @pytest.mark.parametrize(
     'task, expected',
     [
         (DocumentClassification, ClassificationLabels),
         (SequenceLabeling, SequenceLabels),
-        (Seq2seq, Seq2seqLabels)
+        (Seq2seq, Seq2seqLabels),
+        (ImageClassification, ClassificationLabels)
     ]
 )
 def test_return_correct_label_collection(task, expected):

Original file line number	Diff line number	Diff line change
`@@ -1,15 +1,16 @@`
`1`	`1`	`import pytest`
`2`	`2`
`3`	`3`	`from auto_labeling_pipeline.labels import ClassificationLabels, Seq2seqLabels, SequenceLabels`
`4`		`-from auto_labeling_pipeline.task import DocumentClassification, Seq2seq, SequenceLabeling`
	`4`	`+from auto_labeling_pipeline.task import DocumentClassification, ImageClassification, Seq2seq, SequenceLabeling`
`5`	`5`
`6`	`6`
`7`	`7`	`@pytest.mark.parametrize(`
`8`	`8`	`'task, expected',`
`9`	`9`	`[`
`10`	`10`	`(DocumentClassification, ClassificationLabels),`
`11`	`11`	`(SequenceLabeling, SequenceLabels),`
`12`		`- (Seq2seq, Seq2seqLabels)`
	`12`	`+ (Seq2seq, Seq2seqLabels),`
	`13`	`+ (ImageClassification, ClassificationLabels)`
`13`	`14`	`]`
`14`	`15`	`)`
`15`	`16`	`def test_return_correct_label_collection(task, expected):`