Skip to content

Commit 19e5745

Browse files
committed
generated tests as multiple markdown blocks instead of a single block
1 parent c09a334 commit 19e5745

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

codeflash/optimization/function_optimizer.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1379,11 +1379,13 @@ def process_review(
13791379
generated_tests, original_runtime_by_test, optimized_runtime_by_test
13801380
)
13811381

1382-
generated_tests_str = "\n#------------------------------------------------\n".join(
1383-
[test.generated_original_test_source for test in generated_tests.generated_tests]
1384-
)
1382+
generated_tests_str = ""
1383+
for test in generated_tests.generated_tests:
1384+
generated_tests_str += f"```python\n{test.generated_original_test_source}\n```"
1385+
generated_tests_str += "\n\n"
1386+
13851387
if concolic_test_str:
1386-
generated_tests_str += "\n#------------------------------------------------\n" + concolic_test_str
1388+
generated_tests_str += f"```python\n{concolic_test_str}\n```\n\n"
13871389

13881390
existing_tests, replay_tests, concolic_tests = existing_tests_source_for(
13891391
self.function_to_optimize.qualified_name_with_modules_from_root(self.project_root),

0 commit comments

Comments
 (0)