test_functional.py

from typing import Any

import pytest

import mobase_tests.functional as m


def test_guessed_string():
    # available functions:
    # - fn_0_arg, fn_1_arg, fn_2_arg
    # - fn_0_or_1_arg, fn_1_or_2_or_3_arg

    def no_args() -> int:
        return 0

    def len_of_args(*args: int) -> int:
        return len(args)

    def len_of_args_tweaked(x: int, *args: int, **kwargs: Any) -> int:
        return x + len(args)

    def sum_of_args(*args: int) -> int:
        return sum(args)

    assert m.fn_0_arg(lambda: 0) == 0
    assert m.fn_0_arg(lambda: 5) == 5
    assert m.fn_0_arg(lambda x=2: x) == 2
    assert m.fn_0_arg(len_of_args) == 0

    assert m.fn_1_arg(lambda x: x, 4) == 4
    assert m.fn_1_arg(sum_of_args, 8) == 8
    assert m.fn_1_arg(lambda x=2, y=4: x + y, 3) == 7
    assert m.fn_1_arg(len_of_args_tweaked, 5) == 5

    assert m.fn_2_arg(lambda x, y: x * y, 4, 5) == 20
    assert m.fn_2_arg(lambda x, y=3: x * y, 4, 2) == 8
    assert m.fn_2_arg(sum_of_args, 8, 9) == 17
    assert m.fn_2_arg(lambda x=2, y=4: x + y, 3, 3) == 6
    assert m.fn_2_arg(len_of_args_tweaked, 5, 2) == 6

    assert m.fn_0_or_1_arg(lambda: 3) == 3
    assert m.fn_0_or_1_arg(lambda x: x) == 1

    # the 0 arg is bound first, both are possible, the 0 arg is chosen
    assert m.fn_0_or_1_arg(lambda x=2: x) == 2

    with pytest.raises(TypeError):
        m.fn_1_or_2_or_3_arg(no_args)  # pyright: ignore[reportArgumentType, reportCallIssue]

    assert m.fn_1_or_2_or_3_arg(lambda x=4: x) == 1
    assert m.fn_1_or_2_or_3_arg(lambda x, y: x + y) == 3  # 1 + 2
    assert m.fn_1_or_2_or_3_arg(lambda x, y, z: x * y * z) == 6  # 1 * 2 * 3

    # the 1 arg is bound first
    assert m.fn_1_or_2_or_3_arg(sum_of_args) == 1