from typing import Set
import irisclassifier
import pytest

performance_thresholds = {0.50, 0.75, 0.95}

def test_evaluation():
 i = irisclassifier.IrisClassifier()
 i.ingestion()
 i.segregation()
 i.train()
 res = i.evaluation()
 assert res > 0.75

@pytest.mark.parametrize('th', performance_thresholds)
def test_evaluations(th):
 i = irisclassifier.IrisClassifier()
 i.ingestion()
 i.segregation()
 i.train()
 res = i.evaluation()
 assert res > th

