from unittest import TestCase
from medacy.pipelines.clinical_pipeline import ClinicalPipeline
from medacy.pipeline_components.feature_overlayers.gold_annotator_component import GoldAnnotatorOverlayer
from medacy.pipeline_components.feature_overlayers.metamap.metamap import MetaMap


class TestClinicalPipeline(TestCase):

    @classmethod
    def setUpClass(cls):
        metamap = MetaMap(metamap_path="/home/share/programs/metamap/2016/public_mm/bin/metamap",
                          cache_output=False)
        cls.pipeline = ClinicalPipeline(metamap)  # Will fail as MetaMap isn't installed

    def test_init(self):
        pipeline = self.pipeline

    def test_contains_gold(self):
        pipeline = self.pipeline
        with self.assertRaises(AssertionError) as context:
            pipeline.add_component(GoldAnnotatorOverlayer)