fix test for toasts

MannLabs · Oct 31, 2024 · e2803df · e2803df
1 parent b0afa49
commit e2803df
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/alphastats/gui/pages/03_Preprocessing.py b/alphastats/gui/pages/03_Preprocessing.py
@@ -1,6 +1,6 @@
 import streamlit as st
-from DataSet_Preprocess import PreprocessingStateKeys
 
+from alphastats.DataSet_Preprocess import PreprocessingStateKeys
 from alphastats.gui.utils.preprocessing_helper import (
     configure_preprocessing,
     display_preprocessing_info,

diff --git a/tests/llm/test_llm_helper.py b/tests/llm/test_llm_helper.py
@@ -16,12 +16,13 @@ def mock_streamlit():
     with patch("streamlit.info") as mock_info, patch(
         "streamlit.error"
     ) as mock_error, patch("streamlit.success") as mock_success, patch(
-        "streamlit.session_state", {}
-    ) as mock_session_state:
+        "streamlit.toast"
+    ) as mock_toast, patch("streamlit.session_state", {}) as mock_session_state:
         yield {
             "info": mock_info,
             "error": mock_error,
             "success": mock_success,
+            "toast": mock_toast,
             "session_state": mock_session_state,
         }
 
@@ -92,8 +93,8 @@ def test_set_api_key_from_secrets(mock_exists, mock_st_secrets, mock_streamlit):
 
     set_api_key()
 
-    mock_streamlit["info"].assert_called_with(
-        "OpenAI API key loaded from secrets.toml."
+    mock_streamlit["toast"].assert_called_with(
+        "OpenAI API key loaded from secrets.toml.", icon="✅"
     )
     assert (
         mock_streamlit["session_state"][StateKeys.OPENAI_API_KEY]