Update test_code_utils.py
Some checks failed
CodeFlash / Optimize new code in this PR (pull_request) Failing after 1s
end-to-end-test / bubble-sort-optimization-pytest-no-git (pull_request) Failing after 2s
end-to-end-test / bubble-sort-optimization-unittest (pull_request) Failing after 1s
Coverage E2E / end-to-end-test-coverage (pull_request) Failing after 2s
end-to-end-test / futurehouse-structure (pull_request) Failing after 1s
end-to-end-test / init-optimization (pull_request) Failing after 2s
end-to-end-test / tracer-replay (pull_request) Failing after 2s
end-to-end-test / topological-sort-optimization (pull_request) Failing after 1s
Mypy Type Checking for CLI / type-check-cli (pull_request) Failing after 2s
unit-tests / unit-tests (3.10.13) (pull_request) Failing after 1s
unit-tests / unit-tests (3.11.6) (pull_request) Failing after 3s
unit-tests / unit-tests (3.12.1) (pull_request) Failing after 2s
unit-tests / unit-tests (3.13.0) (pull_request) Failing after 2s
unit-tests / unit-tests (3.9.18) (pull_request) Failing after 2s

This commit is contained in:
Kevin Turcios 2025-02-28 16:08:40 -08:00
parent 018581f613
commit f1dc490115

View file

@ -17,6 +17,7 @@ from codeflash.code_utils.code_utils import (
is_class_defined_in_file,
module_name_from_file_path,
path_belongs_to_site_packages,
has_any_async_functions,
)
from codeflash.code_utils.concolic_utils import clean_concolic_tests
from codeflash.code_utils.coverage_utils import generate_candidates, prepare_coverage_files
@ -441,3 +442,27 @@ def test_Grammar_copy():
Grammar.copy(Grammar())
"""
assert cleaned_code == expected_cleaned_code.strip()
def test_has_any_async_functions_with_async_code() -> None:
code = """
def normal_function():
pass
async def async_function():
pass
"""
result = has_any_async_functions(code)
assert result is True
def test_has_any_async_functions_without_async_code() -> None:
code = """
def normal_function():
pass
def another_function():
pass
"""
result = has_any_async_functions(code)
assert result is False