from docforge import GriffeLoader


def test_load_multiple_modules(temp_package):
    (temp_package / "a.py").write_text('"""A."""\n')
    (temp_package / "b.py").write_text('"""B."""\n')

    loader = GriffeLoader()
    project = loader.load_project(["testpkg.a", "testpkg.b"])

    assert len(project.modules) == 2
    assert set(project.get_module_list()) == {"testpkg.a", "testpkg.b"}