fix: update tests to work with tool decorator and fix import sorting

- Reorganized imports in bright_data.py to follow PEP 8 standards
- Revised tests to call bright_data function directly with parameters
- Removed mock tool interface approach in favor of direct function testing
- All tests now pass with the current tool decorator implementation

Author: meirk-brd

src/strands_tools/bright_data.py

@@ -73,14 +73,16 @@
 import logging
 import os
 import time
-from typing import Any, Dict, Optional
+from typing import Dict, Optional
 from urllib.parse import quote
+
 import requests
 from rich.panel import Panel
 from rich.text import Text
-from strands_tools.utils import console_util
 from strands import tool
 
+from strands_tools.utils import console_util
+
 logger = logging.getLogger(__name__)
 
 console = console_util.create()
@@ -141,7 +143,7 @@ def scrape_as_markdown(self, url: str, zone: Optional[str] = None) -> str:
 
         Args:
             url (str): URL to scrape
-            zone: Override default Web Unlocker zone name (optional). 
+            zone: Override default Web Unlocker zone name (optional).
             Must be a Web Unlocker zone - datacenter/residential zones will fail.
             Default: "web_unlocker"
 
@@ -201,7 +203,7 @@ def search_engine(
         Args:
             query (str): Search query
             engine (str): Search engine - 'google', 'bing', or 'yandex'
-            zone: Override default Web Unlocker zone name (optional). 
+            zone: Override default Web Unlocker zone name (optional).
             Must be a Web Unlocker zone - datacenter/residential zones will fail.
             Default: "web_unlocker"
 
@@ -275,7 +277,6 @@ def search_engine(
             if return_json:
                 params.append("brd_json=1")
 
-
             if params:
                 search_url += "&" + "&".join(params)
 
@@ -421,8 +422,8 @@ def bright_data(
     action: The action to perform (scrape_as_markdown, get_screenshot, search_engine, web_data_feed)
     url: URL to scrape or extract data from (for scrape_as_markdown, get_screenshot, web_data_feed)
     output_path: Path to save the screenshot (for get_screenshot)
-    zone: Web Unlocker zone name (optional). If not provided, uses BRIGHTDATA_ZONE environment 
-          variable, or defaults to "web_unlocker1". Set BRIGHTDATA_ZONE in your .env file to 
+    zone: Web Unlocker zone name (optional). If not provided, uses BRIGHTDATA_ZONE environment
+          variable, or defaults to "web_unlocker1". Set BRIGHTDATA_ZONE in your .env file to
           configure your specific Web Unlocker zone name (e.g., BRIGHTDATA_ZONE=web_unlocker_12345)
     query: Search query (for search_engine)
     engine: Search engine to use (google, bing, yandex, default: google)
@@ -504,4 +505,4 @@ def bright_data(
             border_style="red",
         )
         console.print(error_panel)
-        raise
+        raise