From 8a37354539fdbcd1ea0093486143304817466c20 Mon Sep 17 00:00:00 2001 From: ali Date: Wed, 18 Feb 2026 14:52:59 +0200 Subject: [PATCH] feat: enable coverage check for all languages Remove the Python-only gate on coverage checking so JS/TS optimizations also require adequate test coverage before proceeding. Co-Authored-By: Claude Opus 4.6 --- codeflash/optimization/function_optimizer.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/codeflash/optimization/function_optimizer.py b/codeflash/optimization/function_optimizer.py index 372d7892f..3f5a10667 100644 --- a/codeflash/optimization/function_optimizer.py +++ b/codeflash/optimization/function_optimizer.py @@ -1904,8 +1904,7 @@ def setup_and_establish_baseline( original_code_baseline, test_functions_to_remove = baseline_result.unwrap() # Check test quantity for all languages quantity_ok = quantity_of_tests_critic(original_code_baseline) - # TODO: {Self} Only check coverage for Python - coverage infrastructure not yet reliable for JS/TS - coverage_ok = coverage_critic(original_code_baseline.coverage_results) if is_python() else True + coverage_ok = coverage_critic(original_code_baseline.coverage_results) if isinstance(original_code_baseline, OriginalCodeBaseline) and (not coverage_ok or not quantity_ok): if self.args.override_fixtures: restore_conftest(original_conftest_content) @@ -2348,8 +2347,7 @@ def establish_original_code_baseline( ) console.rule() return Failure("Failed to establish a baseline for the original code - bevhavioral tests failed.") - # Skip coverage check for non-Python languages (coverage not yet supported) - if is_python() and not coverage_critic(coverage_results): + if not coverage_critic(coverage_results): did_pass_all_tests = all(result.did_pass for result in behavioral_results) if not did_pass_all_tests: return Failure("Tests failed to pass for the original code.")