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

Merge pull request #727 from nabenabe0928/bug-fix/clear-cache-of-previous-tests

Browse Source 
Clear cache before going to another tests in unit tests
This commit is contained in:
keisuke umezawa
2023-12-14 18:38:02 +09:00
committed by GitHub
parent d627bce9ed aa0fccc65e
commit 9d4c2d5d07
4 changed files with 15 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
e2e_tests/test_dashboard/test_usecases/test_preferential_optimization.py
+2
View File
@@ -9,6 +9,7 @@ from playwright.sync_api import Page
import pytest
from ...test_server import make_test_server
from ...utils import clear_inmemory_cache
def make_test_storage() -> optuna.storages.InMemoryStorage:
@@ -43,6 +44,7 @@ def make_test_storage() -> optuna.storages.InMemoryStorage:
@pytest.fixture
def storage() -> optuna.storages.InMemoryStorage:
clear_inmemory_cache()
storage = make_test_storage()
return storage
e2e_tests/test_dashboard/test_usecases/test_study_history.py
+2
View File
@@ -3,6 +3,7 @@ from playwright.sync_api import Page
import pytest
from ...test_server import make_test_server
from ...utils import clear_inmemory_cache
def make_test_storage() -> optuna.storages.InMemoryStorage:
@@ -23,6 +24,7 @@ def make_test_storage() -> optuna.storages.InMemoryStorage:
@pytest.fixture
def storage() -> optuna.storages.InMemoryStorage:
clear_inmemory_cache()
storage = make_test_storage()
return storage
e2e_tests/test_dashboard/visual_regression_test.py
+2
View File
@@ -5,10 +5,12 @@ from playwright.sync_api import Page
import pytest
from ..test_server import make_test_server
from ..utils import clear_inmemory_cache
@pytest.fixture
def storage() -> optuna.storages.InMemoryStorage:
clear_inmemory_cache()
storage = optuna.storages.InMemoryStorage()
return storage
e2e_tests/utils.py
+9
View File
@@ -0,0 +1,9 @@
from optuna_dashboard._storage import trials_cache
from optuna_dashboard._storage import trials_cache_lock
from optuna_dashboard._storage import trials_last_fetched_at
def clear_inmemory_cache() -> None:
with trials_cache_lock:
trials_cache.clear()
trials_last_fetched_at.clear()