def test_check_dataframe_column_names_consistency(): err_msg = "Estimator does not have a feature_names_in_" with raises(ValueError, match=err_msg): check_dataframe_column_names_consistency("estimator_name", BaseBadClassifier()) check_dataframe_column_names_consistency("estimator_name", PartialFitChecksName())
def test_pandas_column_name_consistency(estimator): _set_checking_parameters(estimator) with ignore_warnings(category=(FutureWarning)): with pytest.warns(None) as record: check_dataframe_column_names_consistency( estimator.__class__.__name__, estimator) for warning in record: assert "was fitted without feature names" not in str( warning.message)
def test_check_dataframe_column_names_consistency(): err_msg = "Estimator does not have a feature_names_in_" with raises(ValueError, match=err_msg): check_dataframe_column_names_consistency("estimator_name", BaseBadClassifier()) check_dataframe_column_names_consistency("estimator_name", PartialFitChecksName()) lr = LogisticRegression() check_dataframe_column_names_consistency(lr.__class__.__name__, lr) lr.__doc__ = "Docstring that does not document the estimator's attributes" err_msg = ( "Estimator LogisticRegression does not document its feature_names_in_ attribute" ) with raises(ValueError, match=err_msg): check_dataframe_column_names_consistency(lr.__class__.__name__, lr)
def test_pandas_column_name_consistency(estimator): _set_checking_parameters(estimator) with ignore_warnings(category=(FutureWarning)): check_dataframe_column_names_consistency(estimator.__class__.__name__, estimator)