add more unit tests

Author: KRRT7

tests/test_instrument_all_and_run.py

@@ -193,6 +193,18 @@ def test_sort():
         )
         assert test_results[1].runtime > 0
         assert test_results[1].did_pass
+        results2, _ = func_optimizer.run_and_parse_tests(
+            testing_type=TestingMode.BEHAVIOR,
+            test_env=test_env,
+            test_files=func_optimizer.test_files,
+            optimization_iteration=0,
+            pytest_min_loops=1,
+            pytest_max_loops=1,
+            testing_time=0.1,
+        )
+        assert "codeflash stdout: Sorting list" in results2[0].stdout
+        assert "result: [0, 1, 2, 3, 4, 5]" in results2[0].stdout
+        assert compare_test_results(test_results, results2)
     finally:
         fto_path.write_text(original_code, "utf-8")
         test_path.unlink(missing_ok=True)
@@ -377,6 +389,18 @@ def test_sort():
         assert test_results[3].runtime > 0
         assert test_results[3].did_pass
 
+        results2, _ = func_optimizer.run_and_parse_tests(
+            testing_type=TestingMode.BEHAVIOR,
+            test_env=test_env,
+            test_files=func_optimizer.test_files,
+            optimization_iteration=0,
+            pytest_min_loops=1,
+            pytest_max_loops=1,
+            testing_time=0.1,
+        )
+
+        assert compare_test_results(test_results, results2)
+
         # Replace with optimized code that mutated instance attribute
         optimized_code = """
 class BubbleSorter:

tests/test_instrument_tests.py

@@ -482,6 +482,8 @@ def test_sort():
         )
         assert test_results_perf[1].runtime > 0
         assert test_results_perf[1].did_pass
+        assert "codeflash stdout: Sorting list" in test_results_perf[1].stdout
+        assert "result: [0, 1, 2, 3, 4, 5]" in test_results_perf[1].stdout
     finally:
         test_path.unlink(missing_ok=True)
         test_path_perf.unlink(missing_ok=True)
@@ -693,6 +695,9 @@ def test_sort_parametrized(input, expected_output):
         assert test_results_perf[1].runtime > 0
         assert test_results_perf[1].did_pass
 
+        assert "codeflash stdout: Sorting list" in test_results_perf[1].stdout
+        assert "result: [0.0, 1.0, 2.0, 3.0, 4.0, 5.0]" in test_results_perf[1].stdout
+
         assert test_results_perf[2].id.function_getting_tested == "sorter"
         assert test_results_perf[2].id.iteration_id == "0_2"
         assert test_results_perf[2].id.test_class_name is None
@@ -1230,7 +1235,8 @@ def test_sort():
         assert test_results[0].runtime > 0
         assert test_results[0].did_pass
         assert test_results[0].return_value is None
-
+        assert "codeflash stdout: Sorting list" in test_results[0].stdout
+        assert "result: [0.0, 1.0, 2.0, 3.0, 4.0, 5.0]" in test_results[0].stdout
         assert test_results[1].id.function_getting_tested == "sorter"
         assert test_results[1].id.iteration_id == "2_2_1"
         assert test_results[1].id.test_class_name is None