def logging_setup() -> Generator[None, None, None]:
# We need to reconstruct our default handler to properly capture stderr.
logging._reset_library_root_logger()
logging.enable_default_handler()
logging.set_verbosity(logging.INFO)
logging.enable_propagation()
yield
# After testing, restore default propagation setting.
logging.disable_propagation()
def test_log_completed_trial(capsys: _pytest.capture.CaptureFixture) -> None:
# We need to reconstruct our default handler to properly capture stderr.
logging._reset_library_root_logger()
logging.set_verbosity(logging.INFO)
study = create_study()
study.optimize(lambda _: 1.0, n_trials=1)
_, err = capsys.readouterr()
assert "Trial 0" in err
logging.set_verbosity(logging.WARNING)
study.optimize(lambda _: 1.0, n_trials=1)
_, err = capsys.readouterr()
assert "Trial 1" not in err
logging.set_verbosity(logging.DEBUG)
study.optimize(lambda _: 1.0, n_trials=1)
_, err = capsys.readouterr()
assert "Trial 2" in err
