35 lines
1.3 KiB
Python
35 lines
1.3 KiB
Python
|
from typing import List
|
||
|
|
|
||
|
|
from chat.ask_codebase.assistants.directory_structure.relevant_file_finder import (
|
||
|
|
RelevantFileFinder,
|
||
|
|
)
|
||
|
|
from chat.ask_codebase.tools.file_list_util import verify_file_list
|
||
|
|
|
||
|
|
|
||
|
|
OBJECTIVE_PROMPT = """
|
||
|
|
Identify a suitable reference test file that can be used as a guide for writing test cases
|
||
|
|
for the function {function_name}, located in the file {file_path}.
|
||
|
|
The reference should provide a clear example of best practices in testing functions of a similar nature.
|
||
|
|
"""
|
||
|
|
|
||
|
|
|
||
|
|
def find_reference_tests(
|
||
|
|
root_path: str, function_name: str, file_path: str
|
||
|
|
) -> List[str]:
|
||
|
|
"""Find reference tests for a specified function
|
||
|
|
|
||
|
|
Args:
|
||
|
|
root_path (str): The path to the root directory of the codebase.
|
||
|
|
function_name (str): The name of the function to generate test cases for.
|
||
|
|
file_path (str): The path to the file containing the target function for which test cases will be generated.
|
||
|
|
|
||
|
|
Returns:
|
||
|
|
List[str]: A list of paths to files that may contain a reference test for the specified function.
|
||
|
|
"""
|
||
|
|
finder = RelevantFileFinder(root_path=root_path)
|
||
|
|
objective = OBJECTIVE_PROMPT.format(
|
||
|
|
function_name=function_name, file_path=file_path
|
||
|
|
)
|
||
|
|
test_paths = finder.analyze(objective)
|
||
|
|
return verify_file_list(test_paths, root_path)
|