fix one more test

Author: misrasaurabh1

tests/test_instrument_tests.py

@@ -131,11 +131,11 @@ def codeflash_wrap(wrapped, test_module_name, test_class_name, test_name, functi
     """
     if sys.version_info < (3, 12):
         expected += """test_stdout_tag = f"{{test_module_name}}:{{(test_class_name + '.' if test_class_name else '')}}{{test_name}}:{{function_name}}:{{loop_index}}:{{invocation_id}}"
-    print(f"!$######{{test_stdout_tag}}######$!")"""
+    """
     else:
         expected += """test_stdout_tag = f'{{test_module_name}}:{{(test_class_name + '.' if test_class_name else '')}}{{test_name}}:{{function_name}}:{{loop_index}}:{{invocation_id}}'
-    print(f'!$######{{test_stdout_tag}}######$!')"""
-    expected += """
+    """
+    expected += """print(f'!$######{{test_stdout_tag}}######$!')
     exception = None
     gc.disable()
     try:
@@ -146,12 +146,7 @@ def codeflash_wrap(wrapped, test_module_name, test_class_name, test_name, functi
         codeflash_duration = time.perf_counter_ns() - counter
         exception = e
     gc.enable()
-    """
-    if sys.version_info < (3, 12):
-        expected += """print(f"!######{{test_stdout_tag}}######!")"""
-    else:
-        expected += """print(f'!######{{test_stdout_tag}}######!')"""
-    expected += """
+    print(f'!######{{test_stdout_tag}}######!')
     pickled_return_value = pickle.dumps(exception) if exception else pickle.dumps(return_value)
     codeflash_cur.execute('INSERT INTO test_results VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)', (test_module_name, test_class_name, test_name, function_name, loop_index, invocation_id, codeflash_duration, pickled_return_value, 'function_call'))
     codeflash_con.commit()
@@ -240,11 +235,11 @@ def codeflash_wrap(wrapped, test_module_name, test_class_name, test_name, functi
     """
     if sys.version_info < (3, 12):
         expected += """test_stdout_tag = f"{{test_module_name}}:{{(test_class_name + '.' if test_class_name else '')}}{{test_name}}:{{function_name}}:{{loop_index}}:{{invocation_id}}"
-    print(f"!$######{{test_stdout_tag}}######$!")"""
+    """
     else:
         expected += """test_stdout_tag = f'{{test_module_name}}:{{(test_class_name + '.' if test_class_name else '')}}{{test_name}}:{{function_name}}:{{loop_index}}:{{invocation_id}}'
-    print(f'!$######{{test_stdout_tag}}######$!')"""
-    expected += """
+    """
+    expected += """print(f'!$######{{test_stdout_tag}}######$!')
     exception = None
     gc.disable()
     try:
@@ -255,12 +250,7 @@ def codeflash_wrap(wrapped, test_module_name, test_class_name, test_name, functi
         codeflash_duration = time.perf_counter_ns() - counter
         exception = e
     gc.enable()
-    """
-    if sys.version_info < (3, 12):
-        expected += """print(f"!######{{test_stdout_tag}}######!")"""
-    else:
-        expected += """print(f'!######{{test_stdout_tag}}######!')"""
-    expected += """
+    print(f'!######{{test_stdout_tag}}######!')
     pickled_return_value = pickle.dumps(exception) if exception else pickle.dumps(return_value)
     codeflash_cur.execute('INSERT INTO test_results VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)', (test_module_name, test_class_name, test_name, function_name, loop_index, invocation_id, codeflash_duration, pickled_return_value, 'function_call'))
     codeflash_con.commit()