Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 1 | import pytest |
Dean Moldovan | 16db1bf | 2016-09-03 17:25:40 +0200 | [diff] [blame] | 2 | from pybind11_tests import (kw_func0, kw_func1, kw_func2, kw_func3, kw_func4, args_function, |
| 3 | args_kwargs_function, kw_func_udl, kw_func_udl_z, KWClass) |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 4 | |
| 5 | |
| 6 | def test_function_signatures(doc): |
Dean Moldovan | 665e880 | 2016-08-12 22:28:31 +0200 | [diff] [blame] | 7 | assert doc(kw_func0) == "kw_func0(arg0: int, arg1: int) -> str" |
| 8 | assert doc(kw_func1) == "kw_func1(x: int, y: int) -> str" |
| 9 | assert doc(kw_func2) == "kw_func2(x: int=100, y: int=200) -> str" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 10 | assert doc(kw_func3) == "kw_func3(data: str='Hello world!') -> None" |
Dean Moldovan | 665e880 | 2016-08-12 22:28:31 +0200 | [diff] [blame] | 11 | assert doc(kw_func4) == "kw_func4(myList: List[int]=[13, 17]) -> str" |
| 12 | assert doc(kw_func_udl) == "kw_func_udl(x: int, y: int=300) -> str" |
| 13 | assert doc(kw_func_udl_z) == "kw_func_udl_z(x: int, y: int=0) -> str" |
Dean Moldovan | 99dbdc1 | 2016-08-19 13:45:36 +0200 | [diff] [blame] | 14 | assert doc(args_function) == "args_function(*args) -> tuple" |
| 15 | assert doc(args_kwargs_function) == "args_kwargs_function(*args, **kwargs) -> tuple" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 16 | assert doc(KWClass.foo0) == "foo0(self: m.KWClass, arg0: int, arg1: float) -> None" |
| 17 | assert doc(KWClass.foo1) == "foo1(self: m.KWClass, x: int, y: float) -> None" |
| 18 | |
| 19 | |
Dean Moldovan | 665e880 | 2016-08-12 22:28:31 +0200 | [diff] [blame] | 20 | def test_named_arguments(msg): |
| 21 | assert kw_func0(5, 10) == "x=5, y=10" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 22 | |
Dean Moldovan | 665e880 | 2016-08-12 22:28:31 +0200 | [diff] [blame] | 23 | assert kw_func1(5, 10) == "x=5, y=10" |
| 24 | assert kw_func1(5, y=10) == "x=5, y=10" |
| 25 | assert kw_func1(y=10, x=5) == "x=5, y=10" |
| 26 | |
| 27 | assert kw_func2() == "x=100, y=200" |
| 28 | assert kw_func2(5) == "x=5, y=200" |
| 29 | assert kw_func2(x=5) == "x=5, y=200" |
| 30 | assert kw_func2(y=10) == "x=100, y=10" |
| 31 | assert kw_func2(5, 10) == "x=5, y=10" |
| 32 | assert kw_func2(x=5, y=10) == "x=5, y=10" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 33 | |
| 34 | with pytest.raises(TypeError) as excinfo: |
| 35 | # noinspection PyArgumentList |
| 36 | kw_func2(x=5, y=10, z=12) |
Dean Moldovan | d47febc | 2017-03-10 15:42:42 +0100 | [diff] [blame] | 37 | assert excinfo.match( |
| 38 | r'(?s)^kw_func2\(\): incompatible.*Invoked with: kwargs: ((x=5|y=10|z=12)(, |$))' + '{3}$') |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 39 | |
Dean Moldovan | 665e880 | 2016-08-12 22:28:31 +0200 | [diff] [blame] | 40 | assert kw_func4() == "{13 17}" |
| 41 | assert kw_func4(myList=[1, 2, 3]) == "{1 2 3}" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 42 | |
Dean Moldovan | 665e880 | 2016-08-12 22:28:31 +0200 | [diff] [blame] | 43 | assert kw_func_udl(x=5, y=10) == "x=5, y=10" |
| 44 | assert kw_func_udl_z(x=5) == "x=5, y=0" |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 45 | |
| 46 | |
Dean Moldovan | 99dbdc1 | 2016-08-19 13:45:36 +0200 | [diff] [blame] | 47 | def test_arg_and_kwargs(): |
Dean Moldovan | 99dbdc1 | 2016-08-19 13:45:36 +0200 | [diff] [blame] | 48 | args = 'arg1_value', 'arg2_value', 3 |
| 49 | assert args_function(*args) == args |
| 50 | |
| 51 | args = 'a1', 'a2' |
| 52 | kwargs = dict(arg3='a3', arg4=4) |
| 53 | assert args_kwargs_function(*args, **kwargs) == (args, kwargs) |
Jason Rhinelander | 2686da8 | 2017-01-21 23:42:14 -0500 | [diff] [blame] | 54 | |
| 55 | |
| 56 | def test_mixed_args_and_kwargs(msg): |
| 57 | from pybind11_tests import (mixed_plus_args, mixed_plus_kwargs, mixed_plus_args_kwargs, |
| 58 | mixed_plus_args_kwargs_defaults) |
| 59 | mpa = mixed_plus_args |
| 60 | mpk = mixed_plus_kwargs |
| 61 | mpak = mixed_plus_args_kwargs |
| 62 | mpakd = mixed_plus_args_kwargs_defaults |
| 63 | |
| 64 | assert mpa(1, 2.5, 4, 99.5, None) == (1, 2.5, (4, 99.5, None)) |
| 65 | assert mpa(1, 2.5) == (1, 2.5, ()) |
| 66 | with pytest.raises(TypeError) as excinfo: |
| 67 | assert mpa(1) |
| 68 | assert msg(excinfo.value) == """ |
| 69 | mixed_plus_args(): incompatible function arguments. The following argument types are supported: |
| 70 | 1. (arg0: int, arg1: float, *args) -> tuple |
| 71 | |
| 72 | Invoked with: 1 |
Jason Rhinelander | 231e167 | 2017-02-23 21:04:46 -0500 | [diff] [blame] | 73 | """ # noqa: E501 line too long |
Jason Rhinelander | 2686da8 | 2017-01-21 23:42:14 -0500 | [diff] [blame] | 74 | with pytest.raises(TypeError) as excinfo: |
| 75 | assert mpa() |
| 76 | assert msg(excinfo.value) == """ |
| 77 | mixed_plus_args(): incompatible function arguments. The following argument types are supported: |
| 78 | 1. (arg0: int, arg1: float, *args) -> tuple |
| 79 | |
| 80 | Invoked with: |
Jason Rhinelander | 231e167 | 2017-02-23 21:04:46 -0500 | [diff] [blame] | 81 | """ # noqa: E501 line too long |
Jason Rhinelander | 2686da8 | 2017-01-21 23:42:14 -0500 | [diff] [blame] | 82 | |
| 83 | assert mpk(-2, 3.5, pi=3.14159, e=2.71828) == (-2, 3.5, {'e': 2.71828, 'pi': 3.14159}) |
| 84 | assert mpak(7, 7.7, 7.77, 7.777, 7.7777, minusseven=-7) == ( |
| 85 | 7, 7.7, (7.77, 7.777, 7.7777), {'minusseven': -7}) |
| 86 | assert mpakd() == (1, 3.14159, (), {}) |
| 87 | assert mpakd(3) == (3, 3.14159, (), {}) |
| 88 | assert mpakd(j=2.71828) == (1, 2.71828, (), {}) |
| 89 | assert mpakd(k=42) == (1, 3.14159, (), {'k': 42}) |
| 90 | assert mpakd(1, 1, 2, 3, 5, 8, then=13, followedby=21) == ( |
| 91 | 1, 1, (2, 3, 5, 8), {'then': 13, 'followedby': 21}) |
Jason Rhinelander | 231e167 | 2017-02-23 21:04:46 -0500 | [diff] [blame] | 92 | # Arguments specified both positionally and via kwargs should fail: |
Jason Rhinelander | 2686da8 | 2017-01-21 23:42:14 -0500 | [diff] [blame] | 93 | with pytest.raises(TypeError) as excinfo: |
| 94 | assert mpakd(1, i=1) |
| 95 | assert msg(excinfo.value) == """ |
Jason Rhinelander | 231e167 | 2017-02-23 21:04:46 -0500 | [diff] [blame] | 96 | mixed_plus_args_kwargs_defaults(): incompatible function arguments. The following argument types are supported: |
| 97 | 1. (i: int=1, j: float=3.14159, *args, **kwargs) -> tuple |
| 98 | |
| 99 | Invoked with: 1; kwargs: i=1 |
| 100 | """ # noqa: E501 line too long |
Jason Rhinelander | 2686da8 | 2017-01-21 23:42:14 -0500 | [diff] [blame] | 101 | with pytest.raises(TypeError) as excinfo: |
| 102 | assert mpakd(1, 2, j=1) |
| 103 | assert msg(excinfo.value) == """ |
Jason Rhinelander | 231e167 | 2017-02-23 21:04:46 -0500 | [diff] [blame] | 104 | mixed_plus_args_kwargs_defaults(): incompatible function arguments. The following argument types are supported: |
| 105 | 1. (i: int=1, j: float=3.14159, *args, **kwargs) -> tuple |
| 106 | |
| 107 | Invoked with: 1, 2; kwargs: j=1 |
| 108 | """ # noqa: E501 line too long |