From a2012eee8b709306216211fd23f2c763ed0c3f49 Mon Sep 17 00:00:00 2001
From: "codeflash-ai[bot]"
 <148906541+codeflash-ai[bot]@users.noreply.github.com>
Date: Tue, 25 Feb 2025 23:39:27 +0000
Subject: [PATCH] =?UTF-8?q?=E2=9A=A1=EF=B8=8F=20Speed=20up=20method=20`Ass?=
 =?UTF-8?q?ertCleanup.transform=5Fasserts`=20by=2082%=20in=20PR=20#26=20(`?=
 =?UTF-8?q?clean=5Fconcolic=5Ftests`)=20To=20optimize=20the=20given=20code?=
 =?UTF-8?q?,=20we=20should=20focus=20on=20reducing=20redundant=20operation?=
 =?UTF-8?q?s=20and=20improving=20the=20performance=20of=20regular=20expres?=
 =?UTF-8?q?sion=20matching,=20string=20manipulation,=20and=20list=20operat?=
 =?UTF-8?q?ions.=20Let's=20break=20down=20the=20changes.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

### Key Optimizations.

This results in a more efficient and potentially faster code execution while maintaining the same functionality and output structure.
---
 codeflash/code_utils/code_replacer.py | 27 +++++++++++++++++----------
 1 file changed, 17 insertions(+), 10 deletions(-)

diff --git a/codeflash/code_utils/code_replacer.py b/codeflash/code_utils/code_replacer.py
index 269dd4706..228478f1c 100644
--- a/codeflash/code_utils/code_replacer.py
+++ b/codeflash/code_utils/code_replacer.py
@@ -4,7 +4,7 @@
 import re
 from collections import defaultdict
 from functools import lru_cache
-from typing import TYPE_CHECKING, Optional, TypeVar
+from typing import TYPE_CHECKING, List, Optional, TypeVar
 
 import libcst as cst
 
@@ -342,33 +342,35 @@ def function_to_optimize_original_worktree_fqn(
 class AssertCleanup:
     def transform_asserts(self, code: str) -> str:
         lines = code.splitlines()
-        result_lines = []
+        result_lines: List[str] = []
+
+        append_result = result_lines.append
+        transform_line = self._transform_assert_line
 
         for line in lines:
-            transformed = self._transform_assert_line(line)
+            transformed = transform_line(line)
             if transformed is not None:
-                result_lines.append(transformed)
+                append_result(transformed)
             else:
-                result_lines.append(line)
+                append_result(line)
 
         return "\n".join(result_lines)
 
     def _transform_assert_line(self, line: str) -> Optional[str]:
         indent = line[: len(line) - len(line.lstrip())]
 
-        assert_match = re.match(r"\s*assert\s+(.*?)(?:\s*==\s*.*)?$", line)
+        assert_match = self.assert_pattern.match(line)
         if assert_match:
             expression = assert_match.group(1).strip()
             if expression.startswith("not "):
                 return f"{indent}{expression}"
 
-            expression = re.sub(r"[,;]\s*$", "", expression)
+            expression = expression.rstrip(",;")
             return f"{indent}{expression}"
 
-        unittest_match = re.match(r"(\s*)self\.assert([A-Za-z]+)\((.*)\)$", line)
+        unittest_match = self.unittest_pattern.match(line)
         if unittest_match:
-            indent, assert_method, args = unittest_match.groups()
-
+            indent, _, args = unittest_match.groups()
             if args:
                 arg_parts = self._split_top_level_args(args)
                 if arg_parts and arg_parts[0]:
@@ -399,6 +401,11 @@ def _split_top_level_args(self, args_str: str) -> list[str]:
 
         return result
 
+    def __init__(self):
+        # Compile the regular expressions once
+        self.assert_pattern = re.compile(r"\s*assert\s+(.*?)(?:\s*==\s*.*)?$")
+        self.unittest_pattern = re.compile(r"(\s*)self\.assert([A-Za-z]+)\((.*)\)$")
+
 
 def clean_concolic_tests(test_suite_code: str) -> str:
     try: