From 8cd9c0d2923020f040e904b760b425c90147ecc6 Mon Sep 17 00:00:00 2001 From: Saurabh Misra Date: Wed, 16 Apr 2025 15:33:26 -0700 Subject: [PATCH 1/2] measure test discovery time --- codeflash/optimization/optimizer.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/codeflash/optimization/optimizer.py b/codeflash/optimization/optimizer.py index 8eae1014a..a8df4b2a7 100644 --- a/codeflash/optimization/optimizer.py +++ b/codeflash/optimization/optimizer.py @@ -2,6 +2,7 @@ import ast import os +import time import shutil import tempfile from pathlib import Path @@ -94,10 +95,11 @@ def run(self) -> None: return console.rule() + start_time = time.time() function_to_tests: dict[str, list[FunctionCalledInTest]] = discover_unit_tests(self.test_cfg) num_discovered_tests: int = sum([len(value) for value in function_to_tests.values()]) console.rule() - logger.info(f"Discovered {num_discovered_tests} existing unit tests in {self.test_cfg.tests_root}") + logger.info(f"Discovered {num_discovered_tests} existing unit tests in {time.time() - start_time}s at {self.test_cfg.tests_root}") console.rule() ph("cli-optimize-discovered-tests", {"num_tests": num_discovered_tests}) From 974b1f1061c3349b058fb84d249eb19581cddaeb Mon Sep 17 00:00:00 2001 From: Saurabh Misra Date: Wed, 16 Apr 2025 15:40:20 -0700 Subject: [PATCH 2/2] one digit of decimal --- codeflash/optimization/optimizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codeflash/optimization/optimizer.py b/codeflash/optimization/optimizer.py index a8df4b2a7..67fada646 100644 --- a/codeflash/optimization/optimizer.py +++ b/codeflash/optimization/optimizer.py @@ -99,7 +99,7 @@ def run(self) -> None: function_to_tests: dict[str, list[FunctionCalledInTest]] = discover_unit_tests(self.test_cfg) num_discovered_tests: int = sum([len(value) for value in function_to_tests.values()]) console.rule() - logger.info(f"Discovered {num_discovered_tests} existing unit tests in {time.time() - start_time}s at {self.test_cfg.tests_root}") + logger.info(f"Discovered {num_discovered_tests} existing unit tests in {(time.time() - start_time):.1f}s at {self.test_cfg.tests_root}") console.rule() ph("cli-optimize-discovered-tests", {"num_tests": num_discovered_tests})