

def check_iris_criterion(name, criterion):
    ForestClassifier = FOREST_CLASSIFIERS[name]
    clf = ForestClassifier(n_estimators=10, criterion=criterion, random_state=1)
    clf.fit(iris.data, iris.target)
    score = clf.score(iris.data, iris.target)
    assert_greater(score, 0.9, ('Failed with criterion %s and score = %f' % (criterion, score)))
    clf = ForestClassifier(n_estimators=10, criterion=criterion, max_features=2, random_state=1)
    clf.fit(iris.data, iris.target)
    score = clf.score(iris.data, iris.target)
    assert_greater(score, 0.5, ('Failed with criterion %s and score = %f' % (criterion, score)))

dataset/ETHPy150Open scikit-learn/scikit-learn/sklearn/ensemble/tests/test_forest.py check_iris_criterion/original