Merge pull request #192 from codeflash-ai/aseembits93-patch-1

Prevent large inputs in tracer tests

Author: aseembits93

code_to_optimize/code_directories/simple_tracer_e2e/workload.py

@@ -2,6 +2,7 @@
 
 
 def funcA(number):
+    number = number if number < 1000 else 1000
     k = 0
     for i in range(number * 100):
         k += i

tests/scripts/end_to_end_test_tracer_replay.py

@@ -10,7 +10,7 @@ def run_test(expected_improvement_pct: int) -> bool:
         min_improvement_x=0.1,
         expected_unit_tests=1,
         coverage_expectations=[
-            CoverageExpectation(function_name="funcA", expected_coverage=100.0, expected_lines=[5, 6, 7, 9, 12]),
+            CoverageExpectation(function_name="funcA", expected_coverage=100.0, expected_lines=[5, 6, 7, 8, 10, 13]),
         ],
     )
     cwd = (