Apply Ruff Corrections

tests/async/test_performance.py

@@ -1,7 +1,6 @@
 import os
 import sys
 import pytest
-import asyncio
 import time
 
 # Add the parent directory to the Python path
@@ -10,6 +9,7 @@ sys.path.append(parent_dir)
 
 from crawl4ai.async_webcrawler import AsyncWebCrawler
 
+
 @pytest.mark.asyncio
 async def test_crawl_speed():
     async with AsyncWebCrawler(verbose=True) as crawler:
@@ -17,13 +17,14 @@ async def test_crawl_speed():
         start_time = time.time()
         result = await crawler.arun(url=url, bypass_cache=True)
         end_time = time.time()
-        
+
         assert result.success
         crawl_time = end_time - start_time
         print(f"Crawl time: {crawl_time:.2f} seconds")
-        
+
         assert crawl_time < 10, f"Crawl took too long: {crawl_time:.2f} seconds"
 
+
 @pytest.mark.asyncio
 async def test_concurrent_crawling_performance():
     async with AsyncWebCrawler(verbose=True) as crawler:
@@ -32,41 +33,47 @@ async def test_concurrent_crawling_performance():
             "https://www.example.com",
             "https://www.python.org",
             "https://www.github.com",
-            "https://www.stackoverflow.com"
+            "https://www.stackoverflow.com",
         ]
-        
+
         start_time = time.time()
         results = await crawler.arun_many(urls=urls, bypass_cache=True)
         end_time = time.time()
-        
+
         total_time = end_time - start_time
         print(f"Total time for concurrent crawling: {total_time:.2f} seconds")
-        
+
         assert all(result.success for result in results)
         assert len(results) == len(urls)
-        
-        assert total_time < len(urls) * 5, f"Concurrent crawling not significantly faster: {total_time:.2f} seconds"
+
+        assert (
+            total_time < len(urls) * 5
+        ), f"Concurrent crawling not significantly faster: {total_time:.2f} seconds"
+
 
 @pytest.mark.asyncio
 async def test_crawl_speed_with_caching():
     async with AsyncWebCrawler(verbose=True) as crawler:
         url = "https://www.nbcnews.com/business"
-        
+
         start_time = time.time()
         result1 = await crawler.arun(url=url, bypass_cache=True)
         end_time = time.time()
         first_crawl_time = end_time - start_time
-        
+
         start_time = time.time()
         result2 = await crawler.arun(url=url, bypass_cache=False)
         end_time = time.time()
         second_crawl_time = end_time - start_time
-        
+
         assert result1.success and result2.success
         print(f"First crawl time: {first_crawl_time:.2f} seconds")
         print(f"Second crawl time (cached): {second_crawl_time:.2f} seconds")
-        
-        assert second_crawl_time < first_crawl_time / 2, "Cached crawl not significantly faster"
+
+        assert (
+            second_crawl_time < first_crawl_time / 2
+        ), "Cached crawl not significantly faster"
+
 
 if __name__ == "__main__":
-    pytest.main([__file__, "-v"])
+    pytest.main([__file__, "-v"])