linting errors

ryansurf · Nov 9, 2024 · 527dda2 · 527dda2
1 parent 83583f8
commit 527dda2
Showing 1 changed file with 7 additions and 11 deletions.
diff --git a/tests/test_gpt.py b/tests/test_gpt.py
@@ -16,32 +16,28 @@ def test_simple_gpt():
     """
 
     surf_summary = ""
-    gpt_prompt = (
-        """Please output the days of the week in English. What day
+    gpt_prompt = """Please output the days of the week in English. What day
         is your favorite?"""
-        )
 
     gpt_response = gpt.simple_gpt(surf_summary, gpt_prompt).lower()
     expected_response = set([
         "monday",
         "tuesday",
         "wednesday",
         "thursday",
-        "friday"
-        "saturday",
+        "friday" "saturday",
         "sunday",
         "一",
         "二",
         "三",
         "四",
-        "五"
+        "五",
     ])
 
-    # Can case the "gpt_response" string into a list, and 
-    # check for set intersection with the expected response set 
+    # Can case the "gpt_response" string into a list, and
+    # check for set intersection with the expected response set
     gpt_response_set = set(gpt_response.split())
 
-
-    assert (
-        gpt_response_set.intersection(expected_response)
+    assert gpt_response_set.intersection(
+        expected_response
     ), f"Expected '{expected_response}', but got: {gpt_response}"