Change assertion for test_gpt

ryansurf · Oct 8, 2024 · 940b2e6 · 940b2e6
1 parent 5c6f39b
commit 940b2e6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_gpt.py b/tests/test_gpt.py
@@ -17,4 +17,6 @@ def test_simple_gpt():
     surf_summary = "Please respond with the exact phrase 'gpt works'. Do not include any additional text or context."
     gpt_prompt = "This is for testing purposes"
     gpt_response = gpt.simple_gpt(surf_summary, gpt_prompt)
-    assert "gpt works" in gpt_response, f"Expected 'gpt works', but got: {gpt_response}"
+    expected_response = "gpt works"
+
+    assert gpt_response == expected_response, f"Expected '{expected_response}', but got: {gpt_response}"