best summed runtime helper

Author: mohammedahmed18

codeflash/models/models.py

@@ -10,6 +10,7 @@
 from codeflash.lsp.helpers import is_LSP_enabled, report_to_markdown_table
 from codeflash.lsp.lsp_message import LspMarkdownMessage
 from codeflash.models.test_type import TestType
+from codeflash.result.best_summed_runtime import calculate_best_summed_runtime
 
 if TYPE_CHECKING:
     from collections.abc import Iterator
@@ -817,9 +818,7 @@ def total_passed_runtime(self) -> int:
         :return: The runtime in nanoseconds.
         """
         # TODO this doesn't look at the intersection of tests of baseline and original
-        return sum(
-            [min(usable_runtime_data) for _, usable_runtime_data in self.usable_runtime_data_by_test_case().items()]
-        )
+        return calculate_best_summed_runtime(self.usable_runtime_data_by_test_case())
 
     def file_to_no_of_tests(self, test_functions_to_remove: list[str]) -> Counter[Path]:
         map_gen_test_file_to_no_of_tests = Counter()

codeflash/result/best_summed_runtime.py

@@ -0,0 +1,2 @@
+def calculate_best_summed_runtime(grouped_runtime_info: dict[any, list[int]]) -> int:
+    return sum([min(usable_runtime_data) for _, usable_runtime_data in grouped_runtime_info.items()])

codeflash/verification/pytest_plugin.py

@@ -20,6 +20,8 @@
 import pytest
 from pluggy import HookspecMarker
 
+from codeflash.result.best_summed_runtime import calculate_best_summed_runtime
+
 if TYPE_CHECKING:
     from _pytest.config import Config, Parser
     from _pytest.main import Session
@@ -345,7 +347,8 @@ def dynamic_tolerance(self, avg_ns: float) -> float:  # noqa: PLR0911
 
     @pytest.hookimpl
     def pytest_runtest_logreport(self, report: pytest.TestReport) -> None:
-        if report.when == "call" and report.passed and (duration_ns := get_runtime_from_stdout(report.capstdout)):  # noqa: SIM102
+        if report.when == "call" and report.passed:
+            duration_ns = get_runtime_from_stdout(report.capstdout)
             if duration_ns:
                 clean_id = re.sub(r"\s*\[\s*\d+\s*\]\s*$", "", report.nodeid)
                 self.usable_runtime_data_by_test_case.setdefault(clean_id, []).append(duration_ns)
@@ -384,9 +387,7 @@ def pytest_runtestloop(self, session: Session) -> bool:
                 if session.shouldstop:
                     raise session.Interrupted(session.shouldstop)
 
-            best_runtime_until_now = sum(
-                [min(usable_runtime_data) for _, usable_runtime_data in self.usable_runtime_data_by_test_case.items()]
-            )
+            best_runtime_until_now = calculate_best_summed_runtime(self.usable_runtime_data_by_test_case)
             runtimes.append(best_runtime_until_now)
 
             if should_stop(runtimes):