Clear inmemory chache when terminating test

This commit is contained in:
keisuke-umezawa
2024-01-08 13:37:54 +09:00
parent 6b05c98f35
commit be2c3340e5
4 changed files with 3 additions and 6 deletions
@@ -9,7 +9,6 @@ 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:
@@ -44,7 +43,6 @@ def make_test_storage() -> optuna.storages.InMemoryStorage:
@pytest.fixture
def storage() -> optuna.storages.InMemoryStorage:
clear_inmemory_cache()
storage = make_test_storage()
return storage
@@ -3,7 +3,6 @@ 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:
@@ -24,7 +23,6 @@ def make_test_storage() -> optuna.storages.InMemoryStorage:
@pytest.fixture
def storage() -> optuna.storages.InMemoryStorage:
clear_inmemory_cache()
storage = make_test_storage()
return storage
@@ -5,12 +5,10 @@ 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
+3
View File
@@ -8,6 +8,8 @@ import optuna
from optuna_dashboard import wsgi
import pytest
from .utils import clear_inmemory_cache
def get_free_port() -> int:
tcp = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
@@ -28,6 +30,7 @@ def make_test_server(
thread.start()
def stop_server() -> None:
clear_inmemory_cache()
httpd.shutdown()
httpd.server_close()
thread.join()