import unittest

from data_import.pipeline.catalog import Options
from projects.models import (
    DOCUMENT_CLASSIFICATION,
    IMAGE_CLASSIFICATION,
    INTENT_DETECTION_AND_SLOT_FILLING,
    SEQ2SEQ,
    SEQUENCE_LABELING,
    SPEECH2TEXT,
)


class TestOptions(unittest.TestCase):
    def test_return_at_least_one_option(self):
        tasks = [
            DOCUMENT_CLASSIFICATION,
            IMAGE_CLASSIFICATION,
            INTENT_DETECTION_AND_SLOT_FILLING,
            SEQ2SEQ,
            SEQUENCE_LABELING,
            SPEECH2TEXT,
        ]
        for task in tasks:
            with self.subTest(task=task):
                options = Options.filter_by_task(task)
                self.assertGreaterEqual(len(options), 1)