fix(pc):

tests/executor/test_circuit_breaker.py

@@ -1,8 +1,6 @@
 """Tests for CircuitBreaker."""
 
-import pytest
 import time
-from unittest.mock import patch
 from agentic_security.executor.circuit_breaker import CircuitBreaker
 
 

tests/executor/test_concurrent.py

@@ -2,7 +2,7 @@
 
 import pytest
 import asyncio
-from unittest.mock import Mock, patch, AsyncMock
+from unittest.mock import Mock, patch
 from agentic_security.executor.concurrent import ConcurrentExecutor, ExecutorMetrics
 from agentic_security.probe_actor.state import FuzzerState
 
@@ -268,7 +268,10 @@ class TestConcurrentExecutor:
             # 5 requests with rate=5/s and burst=2
             # First 2 immediate, next 3 should take ~0.6s total
             await executor.execute_batch(
-                request_factory, ["p1", "p2", "p3", "p4", "p5"], "test_module", fuzzer_state
+                request_factory,
+                ["p1", "p2", "p3", "p4", "p5"],
+                "test_module",
+                fuzzer_state,
             )
             elapsed = time.monotonic() - start
 

tests/executor/test_rate_limiter.py

@@ -3,7 +3,6 @@
 import asyncio
 import pytest
 import time
-from unittest.mock import patch
 from agentic_security.executor.rate_limiter import TokenBucketRateLimiter
 
 

tests/probe_actor/test_fuzzer.py

@@ -88,7 +88,9 @@ async def test_perform_many_shot_scan_probe_injection(
     ]
 
     msj_prepare_prompts_mock.return_value = [
-        MagicMock(dataset_name="msj_probe_module", prompts=["msj_probe_prompt"], lazy=False)
+        MagicMock(
+            dataset_name="msj_probe_module", prompts=["msj_probe_prompt"], lazy=False
+        )
     ]
 
     # Mock request_factory

tests/probe_data/test_unified_loader.py

@@ -1,7 +1,7 @@
 """Tests for unified dataset loader."""
 
 import pytest
-from unittest.mock import Mock, patch, AsyncMock
+from unittest.mock import patch
 from agentic_security.probe_data.unified_loader import (
     InputSourceConfig,
     UnifiedDatasetLoader,
@@ -85,12 +85,12 @@ class TestUnifiedDatasetLoader:
             prompts=["prompt1", "prompt2", "prompt3"],
             tokens=10,
             approx_cost=0.0,
-            metadata={}
+            metadata={},
         )
 
         with patch(
             "agentic_security.probe_data.unified_loader.load_csv",
-            return_value=mock_dataset
+            return_value=mock_dataset,
         ):
             result = await loader.load_all()
 
@@ -114,12 +114,12 @@ class TestUnifiedDatasetLoader:
             prompts=["hf_prompt1", "hf_prompt2"],
             tokens=8,
             approx_cost=0.0,
-            metadata={}
+            metadata={},
         )
 
         with patch(
             "agentic_security.probe_data.unified_loader.load_dataset_generic",
-            return_value=mock_dataset
+            return_value=mock_dataset,
         ):
             result = await loader.load_all()
 
@@ -151,19 +151,19 @@ class TestUnifiedDatasetLoader:
             prompts=["prompt1"],
             tokens=5,
             approx_cost=0.0,
-            metadata={}
+            metadata={},
         )
         mock_dataset2 = ProbeDataset(
             dataset_name="csv2",
             prompts=["prompt2", "prompt3"],
             tokens=10,
             approx_cost=0.0,
-            metadata={}
+            metadata={},
         )
 
         with patch(
             "agentic_security.probe_data.unified_loader.load_csv",
-            side_effect=[mock_dataset1, mock_dataset2]
+            side_effect=[mock_dataset1, mock_dataset2],
         ):
             result = await loader.load_all()
 
@@ -199,12 +199,12 @@ class TestUnifiedDatasetLoader:
             prompts=["prompt1"],
             tokens=5,
             approx_cost=0.0,
-            metadata={}
+            metadata={},
         )
 
         with patch(
             "agentic_security.probe_data.unified_loader.load_csv",
-            return_value=mock_dataset
+            return_value=mock_dataset,
         ) as mock_load:
             result = await loader.load_all()
 
@@ -229,12 +229,12 @@ class TestUnifiedDatasetLoader:
             prompts=["prompt1", "prompt2", "prompt3", "prompt4", "prompt5"],
             tokens=20,
             approx_cost=0.0,
-            metadata={}
+            metadata={},
         )
 
         with patch(
             "agentic_security.probe_data.unified_loader.load_csv",
-            return_value=mock_dataset
+            return_value=mock_dataset,
         ):
             result = await loader.load_all()
 
@@ -253,7 +253,7 @@ class TestUnifiedDatasetLoader:
 
         with patch(
             "agentic_security.probe_data.unified_loader.load_csv",
-            side_effect=Exception("File not found")
+            side_effect=Exception("File not found"),
         ):
             result = await loader.load_all()
 
@@ -299,19 +299,19 @@ class TestUnifiedDatasetLoader:
             prompts=["a"],
             tokens=1,
             approx_cost=0.0,
-            metadata={}
+            metadata={},
         )
         mock_dataset2 = ProbeDataset(
             dataset_name="high_weight",
             prompts=["b"],
             tokens=1,
             approx_cost=0.0,
-            metadata={}
+            metadata={},
         )
 
         with patch(
             "agentic_security.probe_data.unified_loader.load_csv",
-            side_effect=[mock_dataset1, mock_dataset2]
+            side_effect=[mock_dataset1, mock_dataset2],
         ):
             result = await loader.load_all()
 
@@ -347,12 +347,12 @@ class TestUnifiedDatasetLoader:
             prompts=["remote_prompt"],
             tokens=5,
             approx_cost=0.0,
-            metadata={"source_type": "csv", "url": "https://example.com/data.csv"}
+            metadata={"source_type": "csv", "url": "https://example.com/data.csv"},
         )
 
         with patch(
             "agentic_security.probe_data.unified_loader.load_dataset_generic",
-            return_value=mock_dataset
+            return_value=mock_dataset,
         ):
             result = await loader.load_all()