def nayvy_list_tested_and_untested_functions(
) -> Optional[Tuple[List[str], List[str]]]:
""" Vim interface for listing up
- Already tested function
- Non-tested function
"""
filepath = vim.eval('expand("%")')
loader = SyntacticModuleLoader()
impl_test_paths = get_impl_and_test_paths(filepath)
if impl_test_paths is None:
error('Please check if your python project is created correcty')
return ([], [])
impl_path, test_path = impl_test_paths
# get lines
impl_module_lines = get_latest_lines(impl_path)
test_module_lines = get_latest_lines(test_path)
# create module objects
impl_mod = loader.load_module_from_lines(impl_module_lines)
test_mod = loader.load_module_from_lines(test_module_lines)
if impl_mod is None or test_mod is None:
error('Loading python scripts failed.')
return ([], [])
intersection = impl_mod.to_test().intersect(test_mod)
subtraction = impl_mod.to_test().sub(test_mod)
return (
intersection.to_func_list_lines(),
subtraction.to_func_list_lines(),
)
def nayvy_list_tested_and_untested_functions(
) -> Optional[Tuple[List[str], List[str]]]:
""" Vim interface for listing up
- Already tested function
- Non-tested function
"""
filepath = vim.eval('expand("%")')
if basename(filepath).startswith('test_'):
error('You are already in test script.')
return None
lines = vim.current.buffer[:]
loader = SyntacticModuleLoader()
auto_generator = AutoGenerator(loader)
test_path = auto_generator.touch_test_file(filepath)
if test_path is None:
error('Please check if your python project is created correcty')
return ([], [])
impl_mod = loader.load_module_from_lines(lines)
test_mod = loader.load_module_from_path(test_path)
if impl_mod is None or test_mod is None:
error('Loading python scripts failed.')
return ([], [])
intersection = impl_mod.to_test().intersect(test_mod)
subtraction = impl_mod.to_test().sub(test_mod)
return (
intersection.to_func_list_lines(),
subtraction.to_func_list_lines(),
)
def nayvy_test_generate(func_names: List[str] = []) -> None:
""" Vim interface for jumping or generating unittest.
"""
loader = SyntacticModuleLoader()
auto_generator = AutoGenerator(
loader,
CONFIG.pyproject_root_markers,
)
filepath = vim.eval('expand("%")')
if not func_names and is_test_path(filepath):
# Nothing should be done when already in a test buffer
# and function names were not provided.
error('You are already in test script.')
return
impl_test_paths = get_impl_and_test_paths(filepath)
if impl_test_paths is None:
error('Please check if your python project is created correcty')
return
impl_path, test_path = impl_test_paths
auto_generator.touch_test_file(impl_path)
impl_module_lines = get_latest_lines(impl_path)
test_module_lines = get_latest_lines(test_path)
if func_names == []:
impl_mod = loader.load_module_from_lines(impl_module_lines)
if impl_mod is None:
error('Please check the current buffer is valid')
return
# if `func_names` is not given, infer it as nearest function
func = impl_mod.get_nearest_function(vim.current.window.cursor[0] - 1)
if func is None:
error('The cursor is probably outside the function.')
return
func_names = [func.name]
updated_test_module_lines = test_module_lines
for func_name in func_names:
may_updated_test_module_lines = auto_generator.get_added_test_lines(
func_name,
impl_module_lines,
updated_test_module_lines,
)
if may_updated_test_module_lines:
updated_test_module_lines = may_updated_test_module_lines
test_win = VimWin.find_win(test_path)
# open test script
if not test_win:
vim.command(f'vs {test_path}')
else:
vim.command('{} tabnext'.format(test_win.tabnr))
vim.command('{} wincmd w'.format(test_win.winnr))
if updated_test_module_lines is not None:
# change lines
vim.current.buffer[:] = updated_test_module_lines
else:
info('Test function already exists')
# search lines
row: int
column: int
for i, line in enumerate(test_module_lines):
needle = f'def test_{func_name}'
if needle in line:
vim.current.window.cursor = (
i + 1,
line.index(needle),
)
return
def nayvy_test_generate(func_names: List[str] = []) -> None:
""" Vim interface for jump of generate unittest.
"""
filepath = vim.eval('expand("%")')
# check if alredy in test file
if basename(filepath).startswith('test_'):
error('You are already in test script.')
return
loader = SyntacticModuleLoader()
auto_generator = AutoGenerator(loader)
test_path = auto_generator.touch_test_file(filepath)
if test_path is None:
error('Please check if your python project is created correcty')
return
impl_module_lines = vim.current.buffer[:]
impl_mod = loader.load_module_from_lines(impl_module_lines)
if impl_mod is None:
error('Please check the current buffer is valid')
return
if func_names == []:
func_name = impl_mod.get_function(vim.current.window.cursor[0] - 1)
if func_name is None:
warning('The cursor is probably outside the function.')
vim.command(f'vs {test_path}')
return
func_names = [func_name]
# load test file lines.
with open(test_path) as f:
test_module_lines = [line.strip('\n') for line in f.readlines()]
for func_name in func_names:
tmp = auto_generator.get_added_test_lines(
func_name,
impl_module_lines,
test_module_lines,
)
if tmp is not None:
test_module_lines = tmp
# open test in split buffer
vim.command(f'vs {test_path}')
if test_module_lines is not None:
# change lines
vim.current.buffer[:] = test_module_lines
else:
print('Test function already exists')
# search lines
row: int
column: int
for i, line in enumerate(test_module_lines):
needle = f'def test_{func_name}'
if needle in line:
vim.current.window.cursor = (
i + 1,
line.index(needle),
)
return