misc: formatting

Author: hezyin

unit_tests/main.py

@@ -5,7 +5,11 @@
 import click
 from find_reference_tests import find_reference_tests
 from i18n import TUILanguage, get_translation
-from model import FuncToTest, TokenBudgetExceededException, UserCancelledException
+from model import (
+    FuncToTest,
+    TokenBudgetExceededException,
+    UserCancelledException,
+)
 from propose_test import propose_test
 from tools.file_util import retrieve_file_content
 from write_tests import write_and_print_tests
@@ -39,7 +43,9 @@ def run(self):
 
         self.step3_write_and_print_tests(cases, files)
 
-    def step1_propose_cases_and_reference_files(self) -> Tuple[List[str], List[str]]:
+    def step1_propose_cases_and_reference_files(
+        self,
+    ) -> Tuple[List[str], List[str]]:
         """
         Propose test cases and reference files for a specified function.
 
@@ -60,7 +66,9 @@ def step1_propose_cases_and_reference_files(self) -> Tuple[List[str], List[str]]
             )
 
             ref_files = find_reference_tests(
-                self.repo_root, self.func_to_test.func_name, self.func_to_test.file_path
+                self.repo_root,
+                self.func_to_test.func_name,
+                self.func_to_test.file_path,
             )
 
         if ref_files:
@@ -122,7 +130,7 @@ def step2_edit_cases_and_reference_files(
             try:
                 retrieve_file_content(file_path=ref_file, root_path=self.repo_root)
                 valid_files.append(ref_file)
-            except Exception as e:
+            except Exception:
                 invalid_files.append(ref_file)
 
         # Print summary

unit_tests/prompts.py

@@ -41,7 +41,6 @@
 """
 
 
-
 FIND_REFERENCE_PROMPT = """
 Identify a suitable reference test file that can be used as a guide for writing test cases
 for the function {function_name}, located in the file {file_path}.