From c0df7dc3d8ea792fb9b20f751e26a9206d21ef68 Mon Sep 17 00:00:00 2001
From: nabenabe0928 <shuhei.watanabe.utokyo@gmail.com>
Date: Thu, 7 Dec 2023 07:46:52 +0100
Subject: [PATCH] Clear cache before going to another tests in unit tests

---
 e2e_tests/test_dashboard/visual_regression_test.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/e2e_tests/test_dashboard/visual_regression_test.py b/e2e_tests/test_dashboard/visual_regression_test.py
index ea6ff930..876c585b 100644
--- a/e2e_tests/test_dashboard/visual_regression_test.py
+++ b/e2e_tests/test_dashboard/visual_regression_test.py
@@ -4,11 +4,22 @@ import optuna
 from playwright.sync_api import Page
 import pytest
 
+from optuna_dashboard._storage import trials_cache
+from optuna_dashboard._storage import trials_cache_lock
+from optuna_dashboard._storage import trials_last_fetched_at
+
 from ..test_server import make_test_server
 
 
+def clear_inmemory_cache() -> None:
+    with trials_cache_lock:
+        trials_cache.clear()
+        trials_last_fetched_at.clear()
+
+
 @pytest.fixture
 def storage() -> optuna.storages.InMemoryStorage:
+    clear_inmemory_cache()
     storage = optuna.storages.InMemoryStorage()
     return storage