Unit Test Generator
Class: UnitTestGenerator
Constructor
__init__(self, verbose=False)
Initializes the UnitTestGenerator with optional verbosity setting.
Methods
generate_tests_from_dir
generate_tests_from_dir(self, dir_path: str, output_path: str = None)
Generates unit tests for all supported files in a given directory.- Parameters:
dir_path
: Path of the directory containing source files.max_critique
: Maximum number of critique iterations.output_path
: (Optional) Custom output path for generated tests.verbose
: Enable verbose logging.enable_critique
: Enable AI critique and improvement of generated tests.
- Returns: A tuple containing an empty dictionary and llm usage statistics.
- Parameters:
generate_tests
generate_tests(self, file_path: str, content: str = None, max_critique: int = 3, output_path: str = None, verbose: bool = False, enable_critique: bool = False)
Generates unit tests for a given file with various configuration options.- Parameters:
file_path
: Path of the file relative to the project root.content
: (Optional) File content.max_critique
: Maximum number of critique iterations.output_path
: (Optional) Custom output path for generated tests.verbose
: Enable verbose logging.enable_critique
: Enable AI critique and improvement of generated tests.
- Returns: A tuple containing an empty dictionary and llm usage statistics.
- Parameters:
run_tests
run_tests(self) -> Dict
Runs the generated unit tests and returns the results.