Function _sub_function

marimo/_runtime/pytest.py:110–173

(
    old_item: _pytest.Item, parent: Any, fn: Callable[..., Any]
)

def _sub_function(
    old_item: _pytest.Item, parent: Any, fn: Callable[..., Any]
) -> _pytest.Item:
    # Directly execute the cell, since this means it's a toplevel function with no deps.
    # Or a cell where which we already wrapped in skip.
    if isinstance(old_item.obj, Cell):
        return old_item

    import pytest  # type: ignore
    from _pytest.fixtures import FuncFixtureInfo  # type: ignore

    fixtureinfo = old_item._fixtureinfo
    param_names: set[str] = set()

    if hasattr(old_item, "callspec") and old_item.callspec:
        params: dict[str, Any] = old_item.callspec.params
        param_names = set(params.keys())

        def make_test_func(
            func_JYWB: Callable[..., Any], param_dict: dict[str, Any]
        ) -> Callable[[], Any]:
            # note _JYWB is a suffix to easily detect in stack trace for
            # removal.
            # Also no functools.wraps(func) because we need the empty
            # call signature
            def test_wrapper(*args: Any, **kwargs: Any) -> Any:
                return func_JYWB(*args, **kwargs, **param_dict)

            # but copy attributes from the original function
            test_wrapper.__name__ = func_JYWB.__name__
            test_wrapper.__module__ = func_JYWB.__module__
            return test_wrapper

        fn = make_test_func(fn, params)

        # Filter out parametrized args from fixtureinfo since they're baked in
        fixtureinfo = FuncFixtureInfo(
            argnames=tuple(
                a for a in fixtureinfo.argnames if a not in param_names
            ),
            initialnames=tuple(
                a for a in fixtureinfo.initialnames if a not in param_names
            ),
            names_closure=[
                a for a in fixtureinfo.names_closure if a not in param_names
            ],
            name2fixturedefs={
                k: v
                for k, v in fixtureinfo.name2fixturedefs.items()
                if k not in param_names
            },
        )

    pyfn = pytest.Function.from_parent(
        parent,
        name=old_item.name,
        callobj=fn,
        fixtureinfo=fixtureinfo,  # Preserve fixture metadata (minus params)