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

Fixed tests

Browse Source
This commit is contained in:
gen740
2022-12-01 12:28:44 +09:00
parent 91a8de1d55
commit 188b88f26b
2 changed files with 20 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python_tests/test_api.py
+20 -6
View File
@@ -2,9 +2,11 @@ import json
from unittest import TestCase
import optuna
from optuna.study import StudyDirection
from optuna import get_all_study_summaries
from optuna.study import StudyDirection
from optuna.version import __version__ as optuna_ver
from optuna_dashboard._app import create_app
from packaging import version
from .wsgi_client import send_request
@@ -17,8 +19,13 @@ def objective(trial: optuna.trial.Trial) -> float:
class APITestCase(TestCase):
def test_get_study_summaries(self) -> None:
storage = optuna.storages.InMemoryStorage()
storage.create_new_study(study_name="foo1", directions=[StudyDirection.MINIMIZE])
storage.create_new_study(study_name="foo2", directions=[StudyDirection.MINIMIZE])
if version.parse(optuna_ver) >= version.Version("3.1.0.dev"):
storage.create_new_study(study_name="foo1", directions=[StudyDirection.MINIMIZE])
storage.create_new_study(study_name="foo2", directions=[StudyDirection.MINIMIZE])
else:
storage.create_new_study("foo1")
storage.create_new_study("foo2")
app = create_app(storage)
status, _, body = send_request(
@@ -127,7 +134,10 @@ class APITestCase(TestCase):
def test_create_study_duplicated(self) -> None:
storage = optuna.storages.InMemoryStorage()
storage.create_new_study(study_name="foo", directions=[StudyDirection.MINIMIZE])
if version.parse(optuna_ver) >= version.Version("3.1.0.dev"):
storage.create_new_study(study_name="foo", directions=[StudyDirection.MINIMIZE])
else:
storage.create_new_study("foo")
self.assertEqual(len(get_all_study_summaries(storage)), 1)
app = create_app(storage)
@@ -147,8 +157,12 @@ class APITestCase(TestCase):
def test_delete_study(self) -> None:
storage = optuna.storages.InMemoryStorage()
storage.create_new_study(study_name="foo1", directions=[StudyDirection.MINIMIZE])
storage.create_new_study(study_name="foo2", directions=[StudyDirection.MINIMIZE])
if version.parse(optuna_ver) >= version.Version("3.1.0.dev"):
storage.create_new_study(study_name="foo1", directions=[StudyDirection.MINIMIZE])
storage.create_new_study(study_name="foo2", directions=[StudyDirection.MINIMIZE])
else:
storage.create_new_study(study_name="foo1")
storage.create_new_study(study_name="foo2")
self.assertEqual(len(get_all_study_summaries(storage)), 2)
app = create_app(storage)