From c9239fafd2d8494100e428d53b8377476a946434 Mon Sep 17 00:00:00 2001
From: "codeflash-ai[bot]"
 <148906541+codeflash-ai[bot]@users.noreply.github.com>
Date: Fri, 29 Aug 2025 22:23:32 +0000
Subject: [PATCH] =?UTF-8?q?=E2=9A=A1=EF=B8=8F=20Speed=20up=20function=20`e?=
 =?UTF-8?q?xtract=5Ftest=5Fcontext=5Ffrom=5Fframe`=20by=20491%=20in=20PR?=
 =?UTF-8?q?=20#687=20(`granular-async-instrumentation`)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

The optimization replaces expensive `Path` object creation and method calls with direct string manipulation operations, delivering a **491% speedup**.

**Key optimizations:**

1. **Eliminated Path object overhead**: Replaced `Path(filename).stem.startswith("test_")` with `filename.rpartition('/')[-1].rpartition('\\')[-1].rpartition('.')[0].startswith("test_")` - avoiding Path instantiation entirely.

2. **Optimized path parts extraction**: Replaced `Path(filename).parts` with `filename.replace('\\', '/').split('/')` - using simple string operations instead of Path parsing.

**Performance impact analysis:**
- Original profiler shows lines 25-26 (Path operations) consumed **86.3%** of total runtime (44.7% + 41.6%)
- Optimized version reduces these same operations to just **25.4%** of runtime (15% + 10.4%)
- The string manipulation operations are ~6x faster per call than Path object creation

**Test case benefits:**
- **Large-scale tests** see the biggest gains (516% faster for 900-frame stack, 505% faster for 950-frame chain) because the Path overhead multiplies with stack depth
- **Edge cases** with complex paths benefit significantly (182-206% faster for subdirectory and pytest frame tests)
- **Basic tests** show minimal overhead since Path operations weren't the bottleneck in shallow stacks

The optimization maintains identical behavior while eliminating the most expensive operations identified in the profiling data - Path object instantiation and method calls that occurred once per stack frame.
---
 codeflash/code_utils/codeflash_wrap_decorator.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/codeflash/code_utils/codeflash_wrap_decorator.py b/codeflash/code_utils/codeflash_wrap_decorator.py
index de138dee..ccbc5c88 100644
--- a/codeflash/code_utils/codeflash_wrap_decorator.py
+++ b/codeflash/code_utils/codeflash_wrap_decorator.py
@@ -41,8 +41,8 @@ def extract_test_context_from_frame() -> tuple[str, str | None, str]:
 
             if (
                 frame.f_globals.get("__name__", "").startswith("test_")
-                or Path(filename).stem.startswith("test_")
-                or "test" in Path(filename).parts
+                or filename.rpartition("/")[-1].rpartition("\\")[-1].rpartition(".")[0].startswith("test_")
+                or "test" in filename.replace("\\", "/").split("/")
             ):
                 test_module_name = frame.f_globals.get("__name__", "unknown_module")
 
@@ -53,7 +53,9 @@ def extract_test_context_from_frame() -> tuple[str, str | None, str]:
                         if class_name.startswith("Test") or class_name.endswith("Test") or "test" in class_name.lower():
                             potential_tests.append((test_module_name, class_name, function_name))
 
-                elif "test" in test_module_name or Path(filename).stem.startswith("test_"):
+                elif "test" in test_module_name or filename.rpartition("/")[-1].rpartition("\\")[-1].rpartition(".")[
+                    0
+                ].startswith("test_"):
                     potential_tests.append((test_module_name, None, function_name))
 
             if (