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