fix unit tests

Author: mohammedahmed18

code_to_optimize/bubble_sort.py

@@ -1,5 +1,11 @@
 def sorter(arr):
     print("codeflash stdout: Sorting list")
-    arr.sort()
+    for i in range(len(arr)):
+        for j in range(len(arr) - 1):
+            if arr[j] > arr[j + 1]:
+                temp = arr[j]
+                arr[j] = arr[j + 1]
+                arr[j + 1] = temp
     print(f"result: {arr}")
     return arr
+    

codeflash/models/models.py

@@ -165,6 +165,9 @@ def get_code_block_splitter(file_path: Path) -> str:
     return f"{LINE_SPLITTER_MARKER_PREFIX}{file_path}"
 
 
+splitter_pattern = re.compile(f"^{LINE_SPLITTER_MARKER_PREFIX}([^\n]+)\n", re.MULTILINE | re.DOTALL)
+
+
 class CodeStringsMarkdown(BaseModel):
     code_strings: list[CodeString] = []
     _cache: dict = PrivateAttr(default_factory=dict)
@@ -198,9 +201,7 @@ def file_to_path(self) -> dict[str, str]:
 
     @staticmethod
     def parse_flattened_code(flat_code: str) -> CodeStringsMarkdown:
-        pattern = rf"^{LINE_SPLITTER_MARKER_PREFIX}([^\n]+)\n"
-        matches = list(re.finditer(pattern, flat_code))
-
+        matches = list(splitter_pattern.finditer(flat_code))
         results = CodeStringsMarkdown()
         for i, match in enumerate(matches):
             start = match.end()

codeflash/optimization/function_optimizer.py

@@ -735,7 +735,7 @@ def replace_function_and_helpers_with_optimized_code(
             if scoped_optimized_code is None:
                 logger.warning(
                     f"Optimized code not found for {relative_module_path} In the context\n-------\n{optimized_code}\n-------\n"
-                    "Existing files in the context are: {list(file_to_code_context.keys())}, re-check your 'split markers'"
+                    "re-check your 'split markers'"
                     f"existing files are {file_to_code_context.keys()}"
                 )
                 scoped_optimized_code = ""