25 lines
462 B
Python
25 lines
462 B
Python
from docforge.cli.main import cli
|
|
|
|
|
|
def test_tree_command(cli_runner, temp_package):
|
|
(temp_package / "mod.py").write_text(
|
|
'''
|
|
class A:
|
|
def f(self): ...
|
|
'''
|
|
)
|
|
|
|
result = cli_runner.invoke(
|
|
cli,
|
|
[
|
|
"tree",
|
|
"--modules",
|
|
"testpkg.mod",
|
|
],
|
|
)
|
|
|
|
assert result.exit_code == 0
|
|
assert "testpkg" in result.output
|
|
assert "mod" in result.output
|
|
assert "A" in result.output
|