def test_get_optimization_history_plot(direction):
# type: (str) -> None
# Test with no trial.
study = create_study(direction=direction)
figure = _get_optimization_history_plot(study)
assert len(figure.data) == 0
def objective(trial):
# type: (Trial) -> float
if trial.number == 0:
return 1.0
elif trial.number == 1:
return 2.0
elif trial.number == 2:
return 0.0
return 0.0
# Test with a trial.
study = create_study(direction=direction)
study.optimize(objective, n_trials=3)
figure = _get_optimization_history_plot(study)
assert len(figure.data) == 2
assert figure.data[0].x == (0, 1, 2)
assert figure.data[0].y == (1.0, 2.0, 0.0)
assert figure.data[1].x == (0, 1, 2)
if direction == 'minimize':
assert figure.data[1].y == (1.0, 1.0, 0.0)
else:
assert figure.data[1].y == (1.0, 2.0, 2.0)
# Ignore failed trials.
def fail_objective(_):
# type: (Trial) -> float
raise ValueError
study = create_study(direction=direction)
study.optimize(fail_objective, n_trials=1, catch=(ValueError, ))
figure = _get_optimization_history_plot(study)
assert len(figure.data) == 0
def test_get_optimization_history_plot():
# type: () -> None
# Test with no trial.
study = create_study()
figure = _get_optimization_history_plot(study)
assert len(figure.data[0].x) == 0
assert len(figure.data[0].y) == 0
assert len(figure.data[1].x) == 0
assert len(figure.data[1].y) == 0
def objective(trial):
# type: (Trial) -> float
if trial.number == 0:
return 1.0
elif trial.number == 1:
return 2.0
elif trial.number == 2:
return 0.0
return 0.0
# Test with a trial
study = create_study()
study.optimize(objective, n_trials=3)
figure = _get_optimization_history_plot(study)
assert len(figure.data) == 2
assert tuple(figure.data[0].x) == (0, 1, 2)
assert tuple(figure.data[0].y) == (1.0, 2.0, 0.0)
assert tuple(figure.data[1].x) == (0, 1, 2)
assert tuple(figure.data[1].y) == (1.0, 1.0, 0.0)
# Ignore failed trials.
def fail_objective(_):
# type: (Trial) -> float
raise ValueError
study = create_study()
study.optimize(fail_objective, n_trials=1)
figure = _get_optimization_history_plot(study)
assert len(figure.data) == 2