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:29:21 +09:00
parent 0a58b1bfe1
commit 745309d66c
1 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_api.py
+10 -5
View File
@@ -24,9 +24,10 @@ class APITestCase(TestCase):
self.assertEqual(len(study_summaries), 2)
def test_create_study(self) -> None:
for name, directions in [
("single-objective", ["minimize"]),
("multi-objective", ["minimize", "maximize"]),
for name, directions, expected_status in [
("single-objective success", ["minimize"], 201),
("multi-objective success", ["minimize", "maximize"], 201),
("invalid direction name", ["invalid-direction", "maximize"], 400),
]:
with self.subTest(name):
storage = optuna.storages.InMemoryStorage()
@@ -44,8 +45,12 @@ class APITestCase(TestCase):
content_type="application/json",
body=json.dumps(request_body),
)
self.assertEqual(status, 201)
self.assertEqual(len(storage.get_all_study_summaries()), 1)
self.assertEqual(status, expected_status)
if expected_status == 201:
self.assertEqual(len(storage.get_all_study_summaries()), 1)
else:
self.assertEqual(len(storage.get_all_study_summaries()), 0)
def test_create_study_duplicated(self) -> None:
storage = optuna.storages.InMemoryStorage()