def test_check_class_weight_balanced_linear_classifier(): # check that ill-computed balanced weights raises an exception msg = "Classifier estimator_name is not computing class_weight=balanced properly" with raises(AssertionError, match=msg): check_class_weight_balanced_linear_classifier( "estimator_name", BadBalancedWeightsClassifier )
def test_class_weight_balanced_linear_classifiers(name, Classifier): check_class_weight_balanced_linear_classifier(name, Classifier)