style: some minor changes

Signed-off-by: Zvi Grinberg <[email protected]>

Author: zvigrinberg

src/vuln_analysis/tools/tests/test_transitive_code_search.py

@@ -86,6 +86,7 @@ async def test_transitive_search_golang_5():
     assert path_found is False
     assert len(list_path) is 1
 
+
 # Test fix of https://issues.redhat.com/browse/APPENG-3435
 @pytest.mark.asyncio
 async def test_transitive_search_golang_6():
@@ -125,6 +126,7 @@ async def get_transitive_code_runner_function():
     async for function in transitive_code_search.gen:
         return function.single_fn
 
+
 python_dependency_tree_mock_output = (
     'deptree==0.0.12  # deptree\n'
     '  importlib-metadata==8.7.0  # importlib-metadata\n'
@@ -142,6 +144,7 @@ async def get_transitive_code_runner_function():
     '    mock-package==1.1.1'
 )
 
+
 def mock_file_open(*args, **kwargs):
     file_path = args[0] if args else kwargs.get('file', '')
     mock_file = MagicMock()
@@ -174,19 +177,21 @@ def mock_file_open(*args, **kwargs):
         "search_query": "werkzeug,formparser.MultiPartParser.parse",
         "expected_path_found": False,
         "expected_list_length": 0,
-        "mock_documents": [python_script_example, python_init_function_example, python_full_document_example, python_parse_function_example]
+        "mock_documents": [python_script_example, python_init_function_example, python_full_document_example,
+                           python_parse_function_example]
     },
     {
         "name": "python_3",
         "search_query": "mock_package,mock_function_in_use",
         "expected_path_found": True,
         "expected_list_length": 3,
-        "mock_documents": [python_script_example, python_init_function_example, python_full_document_example, python_parse_function_example, python_mock_function_in_use, python_mock_file]
+        "mock_documents": [python_script_example, python_init_function_example, python_full_document_example,
+                           python_parse_function_example, python_mock_function_in_use, python_mock_file]
     }
 ])
 @patch('vuln_analysis.utils.dep_tree.run_command', return_value=python_dependency_tree_mock_output)
 @patch('builtins.open', side_effect=mock_file_open)
-async def test_transitive_search_python_parameterized(mock_open, mock_run_command,test_case):
+async def test_transitive_search_python_parameterized(mock_open, mock_run_command, test_case):
     """Parameterized test that runs all existing test cases with their respective configurations."""
     transitive_code_search_runner_coroutine = await get_transitive_code_runner_function()
     logging.basicConfig(level=logging.DEBUG)
@@ -199,7 +204,7 @@ async def test_transitive_search_python_parameterized(mock_open, mock_run_comman
     )
 
     with patch('vuln_analysis.utils.document_embedding.retrieve_from_cache',
-                  return_value=(test_case["mock_documents"], True)):
+               return_value=(test_case["mock_documents"], True)):
         result = await transitive_code_search_runner_coroutine(test_case["search_query"])
 
     (path_found, list_path) = result
@@ -210,6 +215,7 @@ async def test_transitive_search_python_parameterized(mock_open, mock_run_comman
     assert path_found == test_case["expected_path_found"]
     assert len(list_path) == test_case["expected_list_length"]
 
+
 @pytest.mark.asyncio
 async def test_python_transitive_search():
     """Test that runs with a real repository"""
@@ -229,5 +235,5 @@ async def test_python_transitive_search():
     print(f"DEBUG: path_found = {path_found}")
     print(f"DEBUG: list_path = {list_path}")
     print(f"DEBUG: len(list_path) = {len(list_path)}")
-    assert path_found == True
-    assert len(list_path) == 2
+    assert path_found is True
+    assert len(list_path) == 2