fix: formatting for test cases.

Author: Khurdhula-Harshavardhan

tests/test_file_store.py

@@ -32,7 +32,7 @@
         "name": "upload_image_with_temp_url",
         "file": BINARY_FILE_CONTENT,
         "options": {
-            "key": f"test_image.jpg",
+            "key": "test_image.jpg",
             "content_type": "image/jpeg",
             "overwrite": True,
             "temp_public_url": True,

tests/test_image_generation.py

@@ -5,7 +5,6 @@
 import logging
 from dotenv import load_dotenv
 import os
-import base64
 
 load_dotenv()
 logging.basicConfig(level=logging.INFO)
@@ -166,7 +165,7 @@ def test_image_to_image_generation(self, test_case):
             assert result is not None
 
             if type(result) is dict:
-                assert result.get("success") == True
+                assert result.get("success")
                 assert result.get("url") is not None
             elif type(result) is bytes:
                 assert isinstance(result, bytes)
@@ -217,7 +216,7 @@ async def test_image_to_image_generation_async(self, test_case):
 
             assert result is not None
             if type(result) is dict:
-                assert result.get("success") == True
+                assert result.get("success")
                 assert result.get("url") is not None
             elif type(result) is bytes:
                 assert isinstance(result, bytes)

tests/test_web.py

@@ -10,12 +10,8 @@
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-jigsaw = jigsawstack.JigsawStack(
-    api_url="http://localhost:3000/api/", api_key=os.getenv("JIGSAWSTACK_API_KEY")
-)
-async_jigsaw = jigsawstack.AsyncJigsawStack(
-    api_url="http://localhost:3000/api/", api_key=os.getenv("JIGSAWSTACK_API_KEY")
-)
+jigsaw = jigsawstack.JigsawStack(api_key=os.getenv("JIGSAWSTACK_API_KEY"))
+async_jigsaw = jigsawstack.AsyncJigsawStack(api_key=os.getenv("JIGSAWSTACK_API_KEY"))
 
 URL = "https://jigsawstack.com"
 
@@ -40,6 +36,7 @@
         "name": "scrape_with_features",
         "params": {
             "url": URL,
+            "element_prompts": ["title"],
             "features": ["meta", "link"],
         },
     },
@@ -51,14 +48,6 @@
             "root_element_selector": "main",
         },
     },
-    {
-        "name": "scrape_with_wait_for_selector",
-        "params": {
-            "url": URL,
-            "element_prompts": ["dynamic content"],
-            "wait_for": {"mode": "selector", "value": ".loaded-content"},
-        },
-    },
     {
         "name": "scrape_with_wait_for_timeout",
         "params": {
@@ -186,13 +175,6 @@
             "query": "artificial intelligence news",
         },
     },
-    {
-        "name": "search_with_max_results",
-        "params": {
-            "query": "python programming",
-            "max_results": 5,
-        },
-    },
     {
         "name": "search_specific_site",
         "params": {
@@ -203,7 +185,7 @@
         "name": "search_ai_mode",
         "params": {
             "query": "explain quantum computing",
-            "ai": True,
+            "ai_overview": True,
         },
     },
 ]
@@ -248,7 +230,7 @@
         "name": "deep_research_with_depth",
         "params": {
             "query": "renewable energy sources",
-            "depth": 3,
+            "depth": 2,
         },
     },
 ]
@@ -381,9 +363,9 @@ def test_deep_research(self, test_case):
             result = jigsaw.web.deep_research(test_case["params"])
 
             assert result["success"]
-            assert "report" in result
-            assert isinstance(result["report"], str)
-            assert len(result["report"]) > 0
+            assert "results" in result
+            assert isinstance(result["results"], str)
+            assert len(result["results"]) > 0
 
             # Check for sources
             if "sources" in result:
@@ -483,7 +465,7 @@ async def test_search_async(self, test_case):
                 assert "description" in item
 
             # Check AI mode response
-            if test_case["params"].get("ai"):
+            if test_case["params"].get("ai_overview"):
                 assert "ai_overview" in result
 
         except JigsawStackError as e:
@@ -528,9 +510,9 @@ async def test_deep_research_async(self, test_case):
             result = await async_jigsaw.web.deep_research(test_case["params"])
 
             assert result["success"]
-            assert "report" in result
-            assert isinstance(result["report"], str)
-            assert len(result["report"]) > 0
+            assert "results" in result
+            assert isinstance(result["results"], str)
+            assert len(result["results"]) > 0
 
             # Check for sources
             if "sources" in result: