Skip to content

Commit 6c1a369

Browse files
committed
Merge branch 'main' into codeflash-trace-decorator
2 parents 0146d82 + 1585e4b commit 6c1a369

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

codeflash/optimization/optimizer.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22

33
import ast
44
import os
5+
import time
56
import shutil
67
import tempfile
78
from collections import defaultdict
@@ -143,10 +144,11 @@ def run(self) -> None:
143144
return
144145

145146
console.rule()
147+
start_time = time.time()
146148
function_to_tests: dict[str, list[FunctionCalledInTest]] = discover_unit_tests(self.test_cfg)
147149
num_discovered_tests: int = sum([len(value) for value in function_to_tests.values()])
148150
console.rule()
149-
logger.info(f"Discovered {num_discovered_tests} existing unit tests in {self.test_cfg.tests_root}")
151+
logger.info(f"Discovered {num_discovered_tests} existing unit tests in {(time.time() - start_time):.1f}s at {self.test_cfg.tests_root}")
150152
console.rule()
151153
ph("cli-optimize-discovered-tests", {"num_tests": num_discovered_tests})
152154

0 commit comments

Comments
 (0)