def test_init_with_incompatible_storage(comm):
# type: (CommunicatorBase) -> None
study = TestChainerMNStudy._create_shared_study(InMemoryStorage(), comm)
with pytest.raises(ValueError):
ChainerMNStudy(study, comm)
def test_init_with_incompatible_storage(comm: CommunicatorBase) -> None:
study = create_study(storage=InMemoryStorage(),
study_name="in-memory-study")
with pytest.raises(ValueError):
ChainerMNStudy(study, comm)
def test_update_cache(study_direction):
# type: (str) -> None
storage = InMemoryStorage()
assert storage.best_trial_id is None
# If the direction is 'minimize', the objective function is weakly decreasing.
# Otherwise, it is weakly increasing.
sign = -1 if study_direction == 'minimize' else 1
study = optuna.create_study(storage=storage, direction=study_direction)
study.optimize(lambda trial: sign * trial.number, n_trials=1)
assert storage.best_trial_id == 0
study.optimize(lambda trial: sign * trial.number, n_trials=1)
assert storage.best_trial_id == 1
# The objective value is equal to the best value.
study.optimize(lambda trial: sign * (trial.number - 1), n_trials=1)
assert storage.best_trial_id == 1
# The objective value is inferior to the best value.
study.optimize(lambda trial: sign * (trial.number - 2), n_trials=1)
assert storage.best_trial_id == 1
def get_storage_mongo(storage):
if storage is None:
return InMemoryStorage()
if isinstance(storage, str):
if storage.startswith("mongo"):
return MongoStorage(storage)
else:
return RDBStorage(storage)
else:
return storage
def test_update_cache_none_value():
# type: () -> None
storage = InMemoryStorage()
study = optuna.create_study(storage=storage)
study.optimize(lambda trial: -1 * trial.number, n_trials=1)
assert storage.best_trial_id == 0
# The objective value is None.
study.optimize(lambda trial: None, n_trials=1) # type: ignore
assert storage.best_trial_id == 0