22 lines
509 B
Python
22 lines
509 B
Python
from docforge import GriffeLoader
|
|
|
|
|
|
def test_load_single_module(temp_package):
|
|
(temp_package / "mod.py").write_text(
|
|
'''"""Module docstring."""\n
|
|
def foo():
|
|
"""Foo docstring."""
|
|
pass
|
|
'''
|
|
)
|
|
|
|
loader = GriffeLoader()
|
|
project = loader.load_project(["testpkg.mod"])
|
|
|
|
assert project.name == "testpkg"
|
|
assert "testpkg.mod" in project.modules
|
|
|
|
module = project.get_module("testpkg.mod")
|
|
assert module.docstring == "Module docstring."
|
|
assert "foo" in module.members
|