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)
