fix one more test

Author: misrasaurabh1

tests/test_instrumentation_run_results_aiservice.py

@@ -323,30 +323,27 @@ def test_single_element_list():
         # Verify instance_state result, which checks instance state right after __init__, using  codeflash_capture
 
         # Verify function_to_optimize result
-        assert test_results[0].id.function_getting_tested == "sorter"
+        assert test_results[0].id.function_getting_tested == "BubbleSorter.__init__"
         assert test_results[0].id.test_function_name == "test_single_element_list"
         assert test_results[0].did_pass
+        assert test_results[0].return_value[0] == {"x": 0}
+        assert test_results[0].stdout == ""
+        assert test_results[1].id.function_getting_tested == "sorter"
+        assert test_results[1].id.test_function_name == "test_single_element_list"
+        assert test_results[1].did_pass
 
         # Checks input values to the function to see if they have mutated
         # assert comparator(test_results[1].return_value[1]["self"], BubbleSorter()) TODO: add self as input
-        assert test_results[0].return_value[1]["arr"] == [1, 2, 3]
+        assert test_results[1].return_value[1]["arr"] == [1, 2, 3]
 
         # Check function return value
-        assert test_results[0].return_value[2] == [1, 2, 3]
-        assert (
-            test_results[0].stdout
-            == """codeflash stdout : BubbleSorter.sorter() called
-"""
-        )
-        assert test_results[1].id.function_getting_tested == "BubbleSorter.__init__"
-        assert test_results[1].id.test_function_name == "test_single_element_list"
-        assert test_results[1].did_pass
-        assert test_results[1].return_value[0] == {"x": 0}
+        assert test_results[1].return_value[2] == [1, 2, 3]
         assert (
             test_results[1].stdout
             == """codeflash stdout : BubbleSorter.sorter() called
 """
         )
+
         # Replace with optimized code that mutated instance attribute
         optimized_code_mutated_attr = """
 import sys
@@ -401,10 +398,10 @@ def sorter(self, arr):
             testing_time=0.1,
         )
         # assert test_results_mutated_attr[0].return_value[0]["self"].x == 1 TODO: add self as input
-        assert test_results_mutated_attr[1].id.function_getting_tested == "BubbleSorter.__init__"
-        assert test_results_mutated_attr[1].return_value[0] == {"x": 1}
-        assert test_results_mutated_attr[1].verification_type == VerificationType.INIT_STATE_FTO
-        assert test_results_mutated_attr[1].stdout == "codeflash stdout : BubbleSorter.sorter() called\n"
+        assert test_results_mutated_attr[0].id.function_getting_tested == "BubbleSorter.__init__"
+        assert test_results_mutated_attr[0].return_value[0] == {"x": 1}
+        assert test_results_mutated_attr[0].verification_type == VerificationType.INIT_STATE_FTO
+        assert test_results_mutated_attr[0].stdout == ""
         assert not compare_test_results(
             test_results, test_results_mutated_attr
         )  # The test should fail because the instance attribute was mutated
@@ -453,10 +450,10 @@ def sorter(self, arr):
             pytest_max_loops=1,
             testing_time=0.1,
         )
-        assert test_results_new_attr[1].id.function_getting_tested == "BubbleSorter.__init__"
-        assert test_results_new_attr[1].return_value[0] == {"x": 0, "y": 2}
-        assert test_results_new_attr[1].verification_type == VerificationType.INIT_STATE_FTO
-        assert test_results_new_attr[1].stdout == "codeflash stdout : BubbleSorter.sorter() called\n"
+        assert test_results_new_attr[0].id.function_getting_tested == "BubbleSorter.__init__"
+        assert test_results_new_attr[0].return_value[0] == {"x": 0, "y": 2}
+        assert test_results_new_attr[0].verification_type == VerificationType.INIT_STATE_FTO
+        assert test_results_new_attr[0].stdout == ""
         # assert test_results_new_attr[1].return_value[1]["self"].x == 0 TODO: add self as input
         # assert test_results_new_attr[1].return_value[1]["self"].y == 2 TODO: add self as input
         assert compare_test_results(