diff --git a/tests/test_api.py b/tests/test_api.py index de5d97d7..1e545274 100644 --- a/tests/test_api.py +++ b/tests/test_api.py @@ -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()