@@ -80,9 +80,9 @@ def test_task_handler_creation(self) -> None:
8080 config = self .config ,
8181 )
8282
83- assert task_handler .llm_client is not None # noqa: S101
84- assert task_handler .mcp_clients is not None # noqa: S101
85- assert task_handler .config is not None # noqa: S101
83+ assert task_handler .llm_client is not None
84+ assert task_handler .mcp_clients is not None
85+ assert task_handler .config is not None
8686
8787 def test_sanitize_arguments_dict (self ) -> None :
8888 """Test argument sanitization with dict input."""
@@ -95,7 +95,7 @@ def test_sanitize_arguments_dict(self) -> None:
9595 # Test with valid dict
9696 args_dict = {"owner" : "testorg" , "repo" : "testrepo" , "issue_number" : 1 }
9797 sanitized = task_handler .sanitize_arguments (args_dict )
98- assert sanitized == args_dict # noqa: S101
98+ assert sanitized == args_dict
9999
100100 def test_sanitize_arguments_json_string (self ) -> None :
101101 """Test argument sanitization with JSON string input."""
@@ -109,7 +109,7 @@ def test_sanitize_arguments_json_string(self) -> None:
109109 args_json = '{"owner": "testorg", "repo": "testrepo", "issue_number": 1}'
110110 sanitized = task_handler .sanitize_arguments (args_json )
111111 expected = {"owner" : "testorg" , "repo" : "testrepo" , "issue_number" : 1 }
112- assert sanitized == expected # noqa: S101
112+ assert sanitized == expected
113113
114114 def test_sanitize_arguments_invalid_json (self ) -> None :
115115 """Test argument sanitization with invalid JSON."""
@@ -150,7 +150,7 @@ def test_handle_task_basic_workflow(self) -> None:
150150 result = task_handler .handle (self .github_task )
151151
152152 # Should complete without errors
153- assert result is None # handle() method returns None on completion # noqa: S101
153+ assert result is None # handle() method returns None on completion
154154
155155 def test_handle_task_with_tool_calls (self ) -> None :
156156 """Test task handling with tool calls."""
@@ -167,7 +167,7 @@ def test_handle_task_with_tool_calls(self) -> None:
167167 result = task_handler .handle (self .github_task )
168168
169169 # Should complete after making tool calls
170- assert result is None # noqa: S101
170+ assert result is None
171171
172172 def test_handle_task_with_think_tags (self ) -> None :
173173 """Test handling of <think> tags in LLM responses."""
@@ -237,7 +237,7 @@ def test_handle_task_max_iterations(self) -> None:
237237 result = task_handler .handle (self .github_task )
238238
239239 # Should stop due to max iterations
240- assert result is None # noqa: S101
240+ assert result is None
241241
242242 def test_handle_task_with_invalid_json_responses (self ) -> None :
243243 """Test handling of invalid JSON responses from LLM."""
@@ -327,7 +327,7 @@ def failing_call_tool(tool: str, args: dict[str, Any]) -> dict[str, Any] | None:
327327 task_handler .handle (self .github_task )
328328
329329 # Verify that tool was called multiple times due to retries
330- assert call_count >= MAX_TOOL_FAILURES # noqa: S101
330+ assert call_count >= MAX_TOOL_FAILURES
331331
332332 def test_make_system_prompt (self ) -> None :
333333 """Test system prompt generation."""
@@ -339,8 +339,8 @@ def test_make_system_prompt(self) -> None:
339339
340340 # Test that system prompt is generated (accessing private method for testing)
341341 system_prompt = task_handler ._make_system_prompt () # noqa: SLF001
342- assert isinstance (system_prompt , str ) # noqa: S101
343- assert len (system_prompt ) > 0 # noqa: S101
342+ assert isinstance (system_prompt , str )
343+ assert len (system_prompt ) > 0
344344
345345
346346class TestTaskHandlerWithDifferentTasks (unittest .TestCase ):
@@ -391,7 +391,7 @@ def test_handle_gitlab_task(self) -> None:
391391
392392 # Handle GitLab task
393393 result = task_handler .handle (self .gitlab_task )
394- assert result is None # Should complete successfully # noqa: S101
394+ assert result is None # Should complete successfully
395395
396396 def test_handle_task_with_multiple_mcp_clients (self ) -> None :
397397 """Test task handling with multiple MCP clients."""
@@ -411,7 +411,7 @@ def test_handle_task_with_multiple_mcp_clients(self) -> None:
411411
412412 # Handle task
413413 result = task_handler .handle (self .gitlab_task )
414- assert result is None # noqa: S101
414+ assert result is None
415415
416416
417417if __name__ == "__main__" :
0 commit comments