def test_frozen_trial_suggest_uniform() -> None:
trial = FrozenTrial(
number=0,
trial_id=0,
state=TrialState.COMPLETE,
value=0.2,
datetime_start=datetime.datetime.now(),
datetime_complete=datetime.datetime.now(),
params={"x": 0.2},
distributions={"x": UniformDistribution(0.0, 1.0)},
user_attrs={},
system_attrs={},
intermediate_values={},
)
assert trial.suggest_uniform("x", 0.0, 1.0) == 0.2
with pytest.raises(ValueError):
trial.suggest_uniform("y", 0.0, 1.0)
def test_frozen_trial_params() -> None:
params = {"x": 1}
trial = FrozenTrial(
number=0,
trial_id=0,
state=TrialState.COMPLETE,
value=0.2,
datetime_start=datetime.datetime.now(),
datetime_complete=datetime.datetime.now(),
params=params,
distributions={"x": UniformDistribution(0, 10)},
user_attrs={},
system_attrs={},
intermediate_values={},
)
assert trial.suggest_uniform("x", 0, 10) == 1
assert trial.params == params
params = {"x": 2}
trial.params = params
assert trial.suggest_uniform("x", 0, 10) == 2
assert trial.params == params
