def test_study_summary_deprecated() -> None:
with pytest.warns(FutureWarning):
StudySummary(
study_name="test",
direction=StudyDirection.NOT_SET,
best_trial=None,
user_attrs={},
system_attrs={},
n_trials=0,
datetime_start=datetime.datetime.now(),
study_id=0,
)
def test_study_summary_deprecated():
# type: () -> None
with pytest.deprecated_call():
StudySummary(
study_name="test",
direction=StudyDirection.NOT_SET,
best_trial=None,
user_attrs={},
system_attrs={},
n_trials=0,
datetime_start=datetime.datetime.now(),
study_id=0,
)
def test_get_all_study_summaries_with_multiple_studies():
# type: () -> None
storage = create_test_storage()
# Set up a MINIMIZE study.
study_id_1 = storage.create_new_study()
storage.set_study_direction(study_id_1, StudyDirection.MINIMIZE)
trial_id_1_1 = storage.create_new_trial(study_id_1)
trial_id_1_2 = storage.create_new_trial(study_id_1)
storage.set_trial_value(trial_id_1_1, 100)
storage.set_trial_value(trial_id_1_2, 0)
storage.set_trial_state(trial_id_1_1, TrialState.COMPLETE)
storage.set_trial_state(trial_id_1_2, TrialState.COMPLETE)
# Set up a MAXIMIZE study.
study_id_2 = storage.create_new_study()
storage.set_study_direction(study_id_2, StudyDirection.MAXIMIZE)
trial_id_2_1 = storage.create_new_trial(study_id_2)
trial_id_2_2 = storage.create_new_trial(study_id_2)
storage.set_trial_value(trial_id_2_1, -100)
storage.set_trial_value(trial_id_2_2, -200)
storage.set_trial_state(trial_id_2_1, TrialState.COMPLETE)
storage.set_trial_state(trial_id_2_2, TrialState.COMPLETE)
# Set up an empty study.
study_id_3 = storage.create_new_study()
summaries = storage.get_all_study_summaries()
summaries = sorted(summaries)
expected_summary_1 = StudySummary(
study_id=study_id_1,
study_name=storage.get_study_name_from_id(study_id_1),
direction=StudyDirection.MINIMIZE,
user_attrs={},
system_attrs={},
best_trial=summaries[0].best_trial, # This always passes.
n_trials=2,
datetime_start=summaries[0].datetime_start # This always passes.
)
expected_summary_2 = StudySummary(
study_id=study_id_2,
study_name=storage.get_study_name_from_id(study_id_2),
direction=StudyDirection.MAXIMIZE,
user_attrs={},
system_attrs={},
best_trial=summaries[1].best_trial, # This always passes.
n_trials=2,
datetime_start=summaries[1].datetime_start # This always passes.
)
expected_summary_3 = StudySummary(
study_id=study_id_3,
study_name=storage.get_study_name_from_id(study_id_3),
direction=StudyDirection.NOT_SET,
user_attrs={},
system_attrs={},
best_trial=None,
n_trials=0,
datetime_start=None)
assert summaries[0] == expected_summary_1
assert summaries[1] == expected_summary_2
assert summaries[2] == expected_summary_3
assert summaries[0].best_trial is not None
assert summaries[0].best_trial.value == 0
assert summaries[1].best_trial is not None
assert summaries[1].best_trial.value == -100
