wassname/optuna-dashboard
1
0
Fork 0
mirror of https://github.com/wassname/optuna-dashboard.git synced 2026-09-09 11:28:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add test case

Browse Source
This commit is contained in:
c-bata
2021-01-12 23:25:21 +09:00
parent c17e9127ca
commit 0a58b1bfe1
1 changed files with 21 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_api.py
+21 -16
View File
@@ -24,23 +24,28 @@ class APITestCase(TestCase):
self.assertEqual(len(study_summaries), 2)
def test_create_study(self) -> None:
storage = optuna.storages.InMemoryStorage()
self.assertEqual(len(storage.get_all_study_summaries()), 0)
for name, directions in [
("single-objective", ["minimize"]),
("multi-objective", ["minimize", "maximize"]),
]:
with self.subTest(name):
storage = optuna.storages.InMemoryStorage()
self.assertEqual(len(storage.get_all_study_summaries()), 0)
app = create_app(storage)
request_body = {
"study_name": "foo",
"direction": "minimize",
}
status, _, _ = send_request(
app,
"/api/studies",
"POST",
content_type="application/json",
body=json.dumps(request_body),
)
self.assertEqual(status, 201)
self.assertEqual(len(storage.get_all_study_summaries()), 1)
app = create_app(storage)
request_body = {
"study_name": "foo",
"directions": directions,
}
status, _, _ = send_request(
app,
"/api/studies",
"POST",
content_type="application/json",
body=json.dumps(request_body),
)
self.assertEqual(status, 201)
self.assertEqual(len(storage.get_all_study_summaries()), 1)
def test_create_study_duplicated(self) -> None:
storage = optuna.storages.InMemoryStorage()