✅ Enhance test readability and modify image file extension in tests

shroominic · Apr 4, 2024 · 0953050 · 0953050
1 parent bc22070
commit 0953050
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/general_test.py b/tests/general_test.py
@@ -26,7 +26,8 @@ def run_sync(session: CodeInterpreterSession) -> bool:
         assert (
             "3.1"
             in session.generate_response(
-                "Compute pi using Monte Carlo simulation in Python and show me the result."
+                "Compute pi using Monte Carlo simulation in "
+                "Python and show me the result."
             ).content
         )
 
@@ -63,7 +64,8 @@ async def run_async(session: CodeInterpreterSession) -> bool:
             "3.1"
             in (
                 await session.agenerate_response(
-                    "Compute pi using Monte Carlo simulation in Python and show me the result."
+                    "Compute pi using Monte Carlo simulation in "
+                    "Python and show me the result."
                 )
             ).content
         )
@@ -81,7 +83,7 @@ async def run_async(session: CodeInterpreterSession) -> bool:
         )
 
         assert (
-            ".jpeg"
+            ".png"
             in (
                 await session.agenerate_response(
                     "Plot the current stock price of Tesla.",