def test_debug_callback():
X, y = load_iris(return_X_y=True)
callback = DebugCallback(verbose=False)
est = LogisticRegression(max_iter=3)
est._set_callbacks(callback)
with warnings.catch_warnings():
warnings.filterwarnings("ignore", category=ConvergenceWarning)
est.fit(X, y)
log_expected = [
r"fit_begin LogisticRegression\(max_iter=3\)",
"iter_end coef=.*",
"iter_end coef=.*",
"iter_end coef=.*",
]
callback.check_log_expected(log_expected)
