def test_frozen_called_single_methods_when_multi() -> None:
state = TrialState.COMPLETE
values = (0.2, 0.3)
params = {"x": 10}
distributions = {"x": UniformDistribution(5, 12)}
user_attrs = {"foo": "bar"}
system_attrs = {"baz": "qux"}
intermediate_values = {0: 0.0, 1: 0.1, 2: 0.1}
trial = create_trial(
state=state,
values=values,
params=params,
distributions=distributions,
user_attrs=user_attrs,
system_attrs=system_attrs,
intermediate_values=intermediate_values,
)
with pytest.raises(RuntimeError):
trial.value
with pytest.raises(RuntimeError):
trial.value = 0.1
with pytest.raises(RuntimeError):
trial.value = [0.1]
def test_create_trial(state: TrialState) -> None:
value = 0.2
params = {"x": 10}
distributions = {"x": UniformDistribution(5, 12)}
user_attrs = {"foo": "bar"}
system_attrs = {"baz": "qux"}
intermediate_values = {0: 0.0, 1: 0.1, 2: 0.1}
trial = create_trial(
state=state,
value=value,
params=params,
distributions=distributions,
user_attrs=user_attrs,
system_attrs=system_attrs,
intermediate_values=intermediate_values,
)
assert isinstance(trial, FrozenTrial)
assert trial.state == (state if state is not None else TrialState.COMPLETE)
assert trial.value == value
assert trial.params == params
assert trial.distributions == distributions
assert trial.user_attrs == user_attrs
assert trial.system_attrs == system_attrs
assert trial.intermediate_values == intermediate_values
assert trial.datetime_start is not None
assert (trial.datetime_complete is not None) == (state is None or state.is_finished())
def test_relative_parameters(storage_init_func):
# type: (Callable[[], storages.BaseStorage]) -> None
relative_search_space = {
"x": UniformDistribution(low=5, high=6),
"y": UniformDistribution(low=5, high=6),
}
relative_params = {"x": 5.5, "y": 5.5, "z": 5.5}
sampler = DeterministicRelativeSampler(relative_search_space,
relative_params) # type: ignore
study = create_study(storage=storage_init_func(), sampler=sampler)
def create_trial():
# type: () -> Trial
return Trial(study, study._storage.create_new_trial(study._study_id))
# Suggested from `relative_params`.
trial0 = create_trial()
distribution0 = UniformDistribution(low=0, high=100)
assert trial0._suggest("x", distribution0) == 5.5
# Not suggested from `relative_params` (due to unknown parameter name).
trial1 = create_trial()
distribution1 = distribution0
assert trial1._suggest("w", distribution1) != 5.5
# Not suggested from `relative_params` (due to incompatible value range).
trial2 = create_trial()
distribution2 = UniformDistribution(low=0, high=5)
assert trial2._suggest("x", distribution2) != 5.5
# Error (due to incompatible distribution class).
trial3 = create_trial()
distribution3 = IntUniformDistribution(low=1, high=100)
with pytest.raises(ValueError):
trial3._suggest("y", distribution3)
# Error ('z' is included in `relative_params` but not in `relative_search_space`).
trial4 = create_trial()
distribution4 = UniformDistribution(low=0, high=10)
with pytest.raises(ValueError):
trial4._suggest("z", distribution4)
# Error (due to incompatible distribution class).
trial5 = create_trial()
distribution5 = IntLogUniformDistribution(low=1, high=100)
with pytest.raises(ValueError):
trial5._suggest("y", distribution5)