Fix tests

vintasoftware · Jan 29, 2025 · d917575 · d917575
1 parent 430a007
commit d917575
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/tests/test_helpers/test_use_cases.py b/tests/test_helpers/test_use_cases.py
@@ -153,8 +153,8 @@ def test_create_message():
 
     assert response["input"] == "Hello, will I have to use my umbrella in Lisbon tomorrow?"
     assert response["output"] == (
-        "It looks like Lisbon will have a sunny day with a high of 35 degrees Celsius tomorrow. "
-        "No umbrella needed!"
+        "The forecast for Lisbon tomorrow is hot with a temperature of 35 degrees Celsius, "
+        "and it doesn't suggest rain. Therefore, you probably won't need to use your umbrella."
     )
 
 

diff --git a/tests/test_views.py b/tests/test_views.py
@@ -340,9 +340,7 @@ def test_create_thread_message(authenticated_client):
     ai_message = chat_messages[1]
 
     assert human_message.content == "Hello, what is the temperature in SF right now?"
-    assert (
-        ai_message.content == "The current temperature in San Francisco, CA is 32 degrees Celsius."
-    )
+    assert ai_message.content == "The current temperature in San Francisco, CA is 32°C."
 
 
 # DELETE