modif du code

juguirlet · Dec 16, 2023 · 9d6aeca · 9d6aeca
1 parent b765c6b
commit 9d6aeca
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test.py b/test.py
@@ -28,13 +28,13 @@ def test_get_features_selected(streamlit_client):
     assert isinstance(features_selected_list, list)
 
 def test_get_customer_shap_values(streamlit_client):
-    data_df = streamlit_client.cache(lambda: app.app_test.head(1))()
+    data_df = streamlit_client.cache(lambda: streamlit_client.head(1))()
     shap_values_list, _, _ = get_customer_shap_values(data_df)
     assert isinstance(shap_values_list, list)
 
 def test_request_prediction(streamlit_client):
     api_url_calc = f'https://juguirlet.pythonanywhere.com/api/v1/predict'
-    data = streamlit_client.cache(lambda: app.app_test.head(1))()
+    data = streamlit_client.cache(lambda: streamlit_client.head(1))()
     response = request_prediction(api_url_calc, data)
     assert "prediction" in response