1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
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
|
