wip

Author: aseembits93

code_to_optimize/tests/pytest/benchmarks/test_benchmark_bubble_sort.py

@@ -3,10 +3,29 @@
 from code_to_optimize.bubble_sort import sorter
 
 
+class DummyBenchmark:
+    def __call__(self, func, *args, **kwargs):
+        # Mimic calling benchmark(func, *args, **kwargs)
+        return func(*args, **kwargs)
+
+    def __getattr__(self, name):
+        # Mimic benchmark attributes like .pedantic, .extra_info etc.
+        def dummy(*args, **kwargs):
+            return None
+
+        return dummy
+
+
+@pytest.fixture
+def benchmark():
+    return DummyBenchmark()
+
+
 def test_sort(benchmark):
     result = benchmark(sorter, list(reversed(range(500))))
     assert result == list(range(500))
 
+
 # This should not be picked up as a benchmark test
 def test_sort2():
     result = sorter(list(reversed(range(500))))

code_to_optimize/tests/pytest/conftest.py

@@ -0,0 +1,26 @@
+import pytest
+import time
+
+# @pytest.fixture(autouse=True)
+# def fixture(request):
+#     if request.node.get_closest_marker("no_autouse"):
+#         # Skip the fixture logic
+#         yield
+#     else:
+#         start_time = time.time()
+#         time.sleep(0.1)
+#         yield
+#         print(f"Took {time.time() - start_time} seconds")
+
+
+@pytest.fixture(autouse=True)
+def fixture1(request):  # We don't need this fixture during testing
+    start_time = time.time()
+    time.sleep(0.1)
+    yield
+    print(f"Took {time.time() - start_time} seconds")
+
+
+@pytest.fixture(autouse=True)
+def fixture2(request):  # We need it
+    yield

code_to_optimize/tests/pytest/conftest.py.tmp

@@ -33,6 +33,13 @@ def normalize_code(code: str) -> str:
     return ast.unparse(normalize_node(ast.parse(code)))
 
 
+def modify_autouse_fixture():
+    # find fixutre definition in conftetst.py (the one closest to the test)
+    # get fixtures present in override-fixtures in pyproject.toml
+    # add if marker closest return
+    autousetransformer
+
+
 class OptimFunctionCollector(cst.CSTVisitor):
     METADATA_DEPENDENCIES = (cst.metadata.ParentNodeProvider,)
 

codeflash/code_utils/code_replacer.py

@@ -87,6 +87,48 @@ def add_addopts_to_pyproject() -> None:
             f.write(original_content)
 
 
+@contextmanager
+def add_override_fixtures_to_pyproject() -> None:
+    pyproject_file = find_pyproject_toml()
+    try:
+        # Read original file
+        if pyproject_file.exists():
+            with Path.open(pyproject_file, encoding="utf-8") as f:
+                original_content = f.read()
+                data = tomlkit.parse(original_content)
+            # Backup original markers
+            original_fixtures = data.get("tool", {}).get("codeflash", {}).get("override-fixtures", [])
+            original_fixtures.append("please_put_your_fixtures_here")
+            data["tool"]["pytest"]["override-fixtures"]["markers"] = list(original_fixtures)
+            with Path.open(pyproject_file, "w", encoding="utf-8") as f:
+                f.write(tomlkit.dumps(data))
+        yield
+    finally:
+        with Path.open(pyproject_file, "w", encoding="utf-8") as f:
+            f.write(original_content)
+
+
+@contextmanager
+def add_custom_markers_to_pyproject() -> None:
+    pyproject_file = find_pyproject_toml()
+    try:
+        # Read original file
+        if pyproject_file.exists():
+            with Path.open(pyproject_file, encoding="utf-8") as f:
+                original_content = f.read()
+                data = tomlkit.parse(original_content)
+            # Backup original markers
+            original_markers = data.get("tool", {}).get("pytest", {}).get("ini_options", {}).get("markers", [])
+            original_markers.append("codeflash_no_autouse")
+            data["tool"]["pytest"]["ini_options"]["markers"] = list(original_markers)
+            with Path.open(pyproject_file, "w", encoding="utf-8") as f:
+                f.write(tomlkit.dumps(data))
+        yield
+    finally:
+        with Path.open(pyproject_file, "w", encoding="utf-8") as f:
+            f.write(original_content)
+
+
 @contextmanager
 def rename_conftest(tests_path: Path) -> None:
     conftest_file = find_conftest(tests_path)

codeflash/code_utils/code_utils.py

@@ -238,6 +238,8 @@ formatter-cmds = [
     "uvx ruff check --exit-zero --fix $file",
     "uvx ruff format $file",
 ]
+override-fixtures = ["fixture1"] #autouse fixtures present in conftest.py which have to be disabled during test execution
+
 [tool.pytest.ini_options]
 markers = [
     "no_autouse"