inference test

Author: aseembits93

codeflash/code_utils/formatter.py

@@ -96,19 +96,20 @@ def is_diff_line(line: str) -> bool:
     return len(diff_lines)
 
 
-def format_generated_code(generated_test_source: str) -> str:
-    return re.sub(r"\n{2,}", "\n\n", generated_test_source)
-    # formatter_name = formatter_cmds[0].lower() if formatter_cmds else "disabled"
-    # if formatter_name == "disabled":
-    #     return re.sub(r"\n{2,}", "\n\n", generated_test_source)
-    # # try running formatter, if nothing changes (could be due to formatting failing or no actual formatting needed)
-    # original_temp, test_dir_str, exit_on_failure = None, None, True
-    # formatted_temp, formatted_code, changed = apply_formatter_cmds(
-    #     formatter_cmds, original_temp, test_dir_str, print_status=False, exit_on_failure=exit_on_failure
-    # )
-    # if not changed:
-    #     return re.sub(r"\n{2,}", "\n\n", formatted_code)
-    # return formatted_code
+def format_generated_code(generated_test_source: str, formatter_cmds: Union[list[str], None]) -> str:
+    formatter_name = formatter_cmds[0].lower() if formatter_cmds else "disabled"
+    if formatter_name == "disabled":
+        return re.sub(r"\n{2,}", "\n\n", generated_test_source)
+    with tempfile.TemporaryDirectory() as test_dir_str:
+        # try running formatter, if nothing changes (could be due to formatting failing or no actual formatting needed)
+        original_temp = Path(test_dir_str) / "original_temp.py"
+        original_temp.write_text(generated_test_source, encoding="utf8")
+        _, formatted_code, changed = apply_formatter_cmds(
+            formatter_cmds, original_temp, test_dir_str, print_status=False, exit_on_failure=False
+        )
+        if not changed:
+            return re.sub(r"\n{2,}", "\n\n", formatted_code)
+    return formatted_code
 
 
 def format_code(