from docforge import GriffeLoader


def test_function_signature(temp_package):
    (temp_package / "fn.py").write_text(
        '''def add(a: int, b: int = 1) -> int:
    """Adds numbers."""
    return a + b
'''
    )

    loader = GriffeLoader()
    project = loader.load_project(["testpkg.fn"])
    module = project.get_module("testpkg.fn")

    fn = module.get_object("add")
    assert fn.kind == "function"
    assert fn.signature is not None
    assert "a" in fn.signature
    assert "b" in fn.signature