This commit is contained in:
Contramundum committed 2023-09-01 14:36:07 +09:00
1 parent 6553bb5fbd
commit b929bc6b36
2 files changed
+6 -3

No files matched your search

+3 -2
View File
@@ -108,6 +108,8 @@ class APITestCase(TestCase):
study.mark_comparison_ready(trial)
study.report_preference(study.trials[0], study.trials[1])
assert len(study.best_trials) == 1
app = create_app(storage)
study_id = study._study._study_id
status, _, body = send_request(
@@ -119,9 +121,8 @@ class APITestCase(TestCase):
self.assertEqual(status, 200)
best_trials = json.loads(body)["best_trials"]
assert len(best_trials) == 2
assert len(best_trials) == 1
assert best_trials[0]["number"] == 0
assert best_trials[1]["number"] == 2
def test_report_preference(self) -> None:
storage = optuna.storages.InMemoryStorage()