add test for skip api

This commit is contained in:
moririn2528
2023-08-28 14:47:07 +09:00
parent 08e44c7aba
commit ad149f073c
3 changed files with 29 additions and 6 deletions
+24
View File
@@ -151,6 +151,30 @@ class APITestCase(TestCase):
assert better.number == 2
assert worse.number == 1
def test_skip_trial(self) -> None:
storage = optuna.storages.InMemoryStorage()
study = create_study(storage=storage)
trials: list[optuna.Trial] = []
for _ in range(3):
trial = study.ask()
study.mark_comparison_ready(trial)
trials.append(trial)
app = create_app(storage)
study_id = study._study._study_id
status, _, _ = send_request(
app,
f"/api/studies/{study_id}/{trials[1]._trial_id}/skip",
"POST",
content_type="application/json",
)
self.assertEqual(status, 204)
best_trials = study.best_trials
assert len(best_trials) == 2
assert best_trials[0].number == 0
assert best_trials[1].number == 2
def test_create_study(self) -> None:
for name, directions, expected_status in [
("single-objective success", ["minimize"], 201),