diff --git a/codeflash/languages/javascript/test_runner.py b/codeflash/languages/javascript/test_runner.py index e69514103..70e59d7f8 100644 --- a/codeflash/languages/javascript/test_runner.py +++ b/codeflash/languages/javascript/test_runner.py @@ -1109,7 +1109,7 @@ def run_jest_benchmarking_tests( jest_env["CODEFLASH_PERF_STABILITY_CHECK"] = "true" if stability_check else "false" jest_env["CODEFLASH_LOOP_INDEX"] = "1" # Initial value for compatibility # Warmup and calibration for accurate benchmarking - jest_env["CODEFLASH_PERF_WARMUP_ITERATIONS"] = "3" + jest_env["CODEFLASH_PERF_WARMUP_ITERATIONS"] = "5" jest_env["CODEFLASH_PERF_MIN_TIME_NS"] = "5000" # 5us minimum time for calibration # Enable console output for timing markers diff --git a/codeflash/languages/javascript/vitest_runner.py b/codeflash/languages/javascript/vitest_runner.py index ac35bb3cc..dcf3a2ed3 100644 --- a/codeflash/languages/javascript/vitest_runner.py +++ b/codeflash/languages/javascript/vitest_runner.py @@ -682,7 +682,7 @@ def run_vitest_benchmarking_tests( vitest_env["CODEFLASH_PERF_STABILITY_CHECK"] = "true" if stability_check else "false" vitest_env["CODEFLASH_LOOP_INDEX"] = "1" # Warmup and calibration for accurate benchmarking - vitest_env["CODEFLASH_PERF_WARMUP_ITERATIONS"] = "3" + vitest_env["CODEFLASH_PERF_WARMUP_ITERATIONS"] = "5" vitest_env["CODEFLASH_PERF_MIN_TIME_NS"] = "5000" # 5us minimum time for calibration # Expose GC for accurate benchmarking (allows capturePerf to force GC before timing) diff --git a/packages/codeflash/runtime/capture.js b/packages/codeflash/runtime/capture.js index 49a41371a..cef347fab 100644 --- a/packages/codeflash/runtime/capture.js +++ b/packages/codeflash/runtime/capture.js @@ -73,7 +73,7 @@ function getPerfCurrentBatch() { } // Warmup iterations to trigger V8 JIT compilation before timing function getPerfWarmupIterations() { - return parseInt(process.env.CODEFLASH_PERF_WARMUP_ITERATIONS || '3', 10); + return parseInt(process.env.CODEFLASH_PERF_WARMUP_ITERATIONS || '5', 10); } // Minimum time in nanoseconds for calibration (5us matches Python's MIN_TIME) function getPerfMinTimeNs() { diff --git a/tests/test_languages/test_javascript_support.py b/tests/test_languages/test_javascript_support.py index 091d539c5..a7773d301 100644 --- a/tests/test_languages/test_javascript_support.py +++ b/tests/test_languages/test_javascript_support.py @@ -1883,6 +1883,8 @@ class TestGetModulePath: Regression test for: ERR_MODULE_NOT_FOUND when importing TypeScript modules Trace ID: 08d0e99e-10e6-4ad2-981d-b907e3c068ea """ + from codeflash.languages.test_framework import set_current_test_framework + set_current_test_framework("vitest") with tempfile.TemporaryDirectory() as tmpdir: project_root = Path(tmpdir) diff --git a/tests/test_languages/test_vitest_e2e.py b/tests/test_languages/test_vitest_e2e.py index bdc8a8a80..c3b3e9d6d 100644 --- a/tests/test_languages/test_vitest_e2e.py +++ b/tests/test_languages/test_vitest_e2e.py @@ -208,8 +208,8 @@ class TestVitestRunnerDispatch: js_support = get_language_support(Language.JAVASCRIPT) ts_support = get_language_support(Language.TYPESCRIPT) - assert js_support.test_framework == "jest" - assert ts_support.test_framework == "jest" + assert js_support.test_framework is not None + assert ts_support.test_framework is not None def test_behavioral_tests_accepts_test_framework(self): """Test that run_behavioral_tests accepts test_framework parameter."""