slight fixes

Author: alvin-r

codeflash/optimization/function_optimizer.py

@@ -134,7 +134,7 @@ def optimize_function(self) -> Result[BestOptimization, str]:
             with helper_function_path.open(encoding="utf8") as f:
                 helper_code = f.read()
                 original_helper_code[helper_function_path] = helper_code
-        if has_any_async_functions(code_context.code_to_optimize_with_helpers):
+        if has_any_async_functions(code_context.read_writable_code):
             return Failure("Codeflash does not support async functions in the code to optimize.")
         code_print(code_context.read_writable_code)
 

tests/test_get_helper_code.py

@@ -217,6 +217,7 @@ def __call__(self, *args: _P.args, **kwargs: _P.kwargs) -> _R:
         f.write(code)
         f.flush()
         file_path = Path(f.name).resolve()
+        project_root_path = file_path.parent.resolve()
         function_to_optimize = FunctionToOptimize(
             function_name="__call__",
             file_path=file_path,
@@ -227,7 +228,7 @@ def __call__(self, *args: _P.args, **kwargs: _P.kwargs) -> _R:
         test_config = TestConfig(
             tests_root="tests",
             tests_project_rootdir=Path.cwd(),
-            project_root_path=file_path.parent.resolve(),
+            project_root_path=project_root_path,
             test_framework="pytest",
             pytest_cmd="pytest",
         )
@@ -241,7 +242,7 @@ def __call__(self, *args: _P.args, **kwargs: _P.kwargs) -> _R:
         assert code_context.helper_functions[0].qualified_name == "AbstractCacheBackend.get_cache_or_call"
         assert (
                 code_context.testgen_context_code
-                == f'''```python:{file_path.name}
+                == f'''```python:{file_path.relative_to(project_root_path)}
 _P = ParamSpec("_P")
 _KEY_T = TypeVar("_KEY_T")
 _STORE_T = TypeVar("_STORE_T")