@@ -682,11 +682,7 @@ def generate_and_instrument_tests(
682682 test_ext = self .language_support .get_test_file_suffix ()
683683 # Show the raw LLM output when available, otherwise the post-processed source
684684 display_source = generated_test .raw_generated_test_source or generated_test .generated_original_test_source
685- code_print (
686- display_source ,
687- file_name = f"test_{ i + 1 } { test_ext } " ,
688- language = self .function_to_optimize .language ,
689- )
685+ code_print (display_source , file_name = f"test_{ i + 1 } { test_ext } " , language = self .function_to_optimize .language )
690686 if concolic_test_str :
691687 logger .info (f"Generated test { count_tests } /{ count_tests } :" )
692688 code_print (concolic_test_str , language = self .function_to_optimize .language )
@@ -1900,9 +1896,7 @@ def setup_and_establish_baseline(
19001896 )
19011897 )
19021898
1903- def display_repaired_functions (
1904- self , generated_tests : GeneratedTestsList , reviews : list [TestFileReview ]
1905- ) -> None :
1899+ def display_repaired_functions (self , generated_tests : GeneratedTestsList , reviews : list [TestFileReview ]) -> None :
19061900 """Display only the repaired function bodies, not the full test files."""
19071901 import libcst as cst
19081902
@@ -2001,9 +1995,7 @@ def review_and_repair_tests(
20011995 tests_for_review = []
20021996 for i , gt in enumerate (generated_tests .generated_tests ):
20031997 failed_fns = failed_by_file .get (gt .behavior_file_path , [])
2004- failure_details = {
2005- fn : test_failure_messages [fn ] for fn in failed_fns if fn in test_failure_messages
2006- }
1998+ failure_details = {fn : test_failure_messages [fn ] for fn in failed_fns if fn in test_failure_messages }
20071999 tests_for_review .append (
20082000 {
20092001 "test_source" : gt .raw_generated_test_source or gt .generated_original_test_source ,
0 commit comments