rename conftest

Author: aseembits93

codeflash/code_utils/code_utils.py

@@ -13,7 +13,7 @@
 import tomlkit
 
 from codeflash.cli_cmds.console import logger
-from codeflash.code_utils.config_parser import find_pyproject_toml, find_conftest
+from codeflash.code_utils.config_parser import find_conftest, find_pyproject_toml
 
 
 @contextmanager
@@ -84,6 +84,7 @@ def add_addopts_to_pyproject() -> None:
         with Path.open(pyproject_file, "w", encoding="utf-8") as f:
             f.write(original_content)
 
+
 @contextmanager
 def rename_conftest() -> None:
     conftest_file = find_conftest()
@@ -98,6 +99,7 @@ def rename_conftest() -> None:
         if conftest_file.exists():
             tmp_conftest_file.rename(conftest_file)
 
+
 def encoded_tokens_len(s: str) -> int:
     """Return the approximate length of the encoded tokens.
 

codeflash/code_utils/config_parser.py

@@ -30,6 +30,7 @@ def find_pyproject_toml(config_file: Path | None = None) -> Path:
 
     raise ValueError(msg)
 
+
 def find_conftest() -> Union[Path, None]:
     # Find the conftest file on the root of the project
     dir_path = Path.cwd()

codeflash/discovery/discover_unit_tests.py

@@ -16,8 +16,12 @@
 from pydantic.dataclasses import dataclass
 
 from codeflash.cli_cmds.console import console, logger, test_files_progress_bar
-from codeflash.code_utils.code_utils import custom_addopts, get_run_tmp_file, module_name_from_file_path, \
-    rename_conftest
+from codeflash.code_utils.code_utils import (
+    custom_addopts,
+    get_run_tmp_file,
+    module_name_from_file_path,
+    rename_conftest,
+)
 from codeflash.code_utils.compat import SAFE_SYS_EXECUTABLE, codeflash_cache_db
 from codeflash.models.models import CodePosition, FunctionCalledInTest, TestsInFile, TestType