test_args.py

from hypothesis import given
from hypothesis.strategies import booleans, lists, integers


def foo():
    ...  # FIXME


def test_foo_1_2_3():
    assert foo(1, 2, 3) == 6


def test_foo_list_of_many():
    numbers = list(range(99))
    assert foo(*numbers) == sum(range(99))


def bar():
    ...  # FIXME


def test_bar_list_of_3():
    numbers = list(range(99))
    assert bar(numbers) == sum(range(99))


def baz() -> dict:
    return {
        "first": ...,
        "next": [...],
        "depth": ...,
        # FIXME
    }


@given(booleans(), lists(booleans()), integers(), integers())
def test_baz_mixed(first, next, depth, v):
    assert baz(first, *next, depth=depth, k=v) == {
        "first": first,
        "next": next,
        "depth": depth,
        "k": v,
    }