compile regex pattern

Author: aseembits93

codeflash/code_utils/code_utils.py

@@ -15,6 +15,8 @@
 from codeflash.cli_cmds.console import logger
 from codeflash.code_utils.config_parser import find_pyproject_toml
 
+ImportErrorPattern = re.compile(r"^.*ModuleNotFoundError.*$", re.MULTILINE)
+
 
 @contextmanager
 def custom_addopts() -> None:

codeflash/optimization/function_optimizer.py

@@ -3,7 +3,6 @@
 import ast
 import concurrent.futures
 import os
-import re
 import subprocess
 import time
 import uuid
@@ -24,6 +23,7 @@
 from codeflash.code_utils import env_utils
 from codeflash.code_utils.code_replacer import replace_function_definitions_in_module
 from codeflash.code_utils.code_utils import (
+    ImportErrorPattern,
     cleanup_paths,
     file_name_from_test_module_name,
     get_run_tmp_file,
@@ -1196,7 +1196,7 @@ def run_and_parse_tests(
             if "ModuleNotFoundError" in run_result.stdout:
                 from rich.text import Text
 
-                match = re.search(r"^.*ModuleNotFoundError.*$", run_result.stdout, re.MULTILINE).group()
+                match = ImportErrorPattern.search(run_result.stdout).group()
                 panel = Panel(Text.from_markup(f"⚠️  {match} ", style="bold red"), expand=False)
                 console.print(panel)
         if testing_type in {TestingMode.BEHAVIOR, TestingMode.PERFORMANCE}:

tests/test_test_runner.py

@@ -4,6 +4,7 @@
 import tempfile
 from pathlib import Path
 
+from codeflash.code_utils.code_utils import ImportErrorPattern
 from codeflash.models.models import TestFile, TestFiles, TestType
 from codeflash.verification.parse_test_output import parse_test_xml
 from codeflash.verification.test_runner import run_behavioral_tests
@@ -143,6 +144,6 @@ def test_sort():
         results = parse_test_xml(
             test_xml_file_path=result_file, test_files=test_files, test_config=config, run_result=process
         )
-    match = re.search(r"^.*ModuleNotFoundError.*$", process.stdout, re.MULTILINE).group()
+    match = ImportErrorPattern.search(process.stdout).group()
     assert match=="E   ModuleNotFoundError: No module named 'torch'"
     result_file.unlink(missing_ok=True)