test_callbacks.py revision 11986
111986Sandreas.sandberg@arm.comimport pytest 211986Sandreas.sandberg@arm.com 311986Sandreas.sandberg@arm.com 411986Sandreas.sandberg@arm.comdef test_callbacks(): 511986Sandreas.sandberg@arm.com from functools import partial 611986Sandreas.sandberg@arm.com from pybind11_tests import (test_callback1, test_callback2, test_callback3, 711986Sandreas.sandberg@arm.com test_callback4, test_callback5) 811986Sandreas.sandberg@arm.com 911986Sandreas.sandberg@arm.com def func1(): 1011986Sandreas.sandberg@arm.com return "func1" 1111986Sandreas.sandberg@arm.com 1211986Sandreas.sandberg@arm.com def func2(a, b, c, d): 1311986Sandreas.sandberg@arm.com return "func2", a, b, c, d 1411986Sandreas.sandberg@arm.com 1511986Sandreas.sandberg@arm.com def func3(a): 1611986Sandreas.sandberg@arm.com return "func3({})".format(a) 1711986Sandreas.sandberg@arm.com 1811986Sandreas.sandberg@arm.com assert test_callback1(func1) == "func1" 1911986Sandreas.sandberg@arm.com assert test_callback2(func2) == ("func2", "Hello", "x", True, 5) 2011986Sandreas.sandberg@arm.com assert test_callback1(partial(func2, 1, 2, 3, 4)) == ("func2", 1, 2, 3, 4) 2111986Sandreas.sandberg@arm.com assert test_callback1(partial(func3, "partial")) == "func3(partial)" 2211986Sandreas.sandberg@arm.com assert test_callback3(lambda i: i + 1) == "func(43) = 44" 2311986Sandreas.sandberg@arm.com 2411986Sandreas.sandberg@arm.com f = test_callback4() 2511986Sandreas.sandberg@arm.com assert f(43) == 44 2611986Sandreas.sandberg@arm.com f = test_callback5() 2711986Sandreas.sandberg@arm.com assert f(number=43) == 44 2811986Sandreas.sandberg@arm.com 2911986Sandreas.sandberg@arm.com 3011986Sandreas.sandberg@arm.comdef test_keyword_args_and_generalized_unpacking(): 3111986Sandreas.sandberg@arm.com from pybind11_tests import (test_tuple_unpacking, test_dict_unpacking, test_keyword_args, 3211986Sandreas.sandberg@arm.com test_unpacking_and_keywords1, test_unpacking_and_keywords2, 3311986Sandreas.sandberg@arm.com test_unpacking_error1, test_unpacking_error2, 3411986Sandreas.sandberg@arm.com test_arg_conversion_error1, test_arg_conversion_error2) 3511986Sandreas.sandberg@arm.com 3611986Sandreas.sandberg@arm.com def f(*args, **kwargs): 3711986Sandreas.sandberg@arm.com return args, kwargs 3811986Sandreas.sandberg@arm.com 3911986Sandreas.sandberg@arm.com assert test_tuple_unpacking(f) == (("positional", 1, 2, 3, 4, 5, 6), {}) 4011986Sandreas.sandberg@arm.com assert test_dict_unpacking(f) == (("positional", 1), {"key": "value", "a": 1, "b": 2}) 4111986Sandreas.sandberg@arm.com assert test_keyword_args(f) == ((), {"x": 10, "y": 20}) 4211986Sandreas.sandberg@arm.com assert test_unpacking_and_keywords1(f) == ((1, 2), {"c": 3, "d": 4}) 4311986Sandreas.sandberg@arm.com assert test_unpacking_and_keywords2(f) == ( 4411986Sandreas.sandberg@arm.com ("positional", 1, 2, 3, 4, 5), 4511986Sandreas.sandberg@arm.com {"key": "value", "a": 1, "b": 2, "c": 3, "d": 4, "e": 5} 4611986Sandreas.sandberg@arm.com ) 4711986Sandreas.sandberg@arm.com 4811986Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 4911986Sandreas.sandberg@arm.com test_unpacking_error1(f) 5011986Sandreas.sandberg@arm.com assert "Got multiple values for keyword argument" in str(excinfo.value) 5111986Sandreas.sandberg@arm.com 5211986Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 5311986Sandreas.sandberg@arm.com test_unpacking_error2(f) 5411986Sandreas.sandberg@arm.com assert "Got multiple values for keyword argument" in str(excinfo.value) 5511986Sandreas.sandberg@arm.com 5611986Sandreas.sandberg@arm.com with pytest.raises(RuntimeError) as excinfo: 5711986Sandreas.sandberg@arm.com test_arg_conversion_error1(f) 5811986Sandreas.sandberg@arm.com assert "Unable to convert call argument" in str(excinfo.value) 5911986Sandreas.sandberg@arm.com 6011986Sandreas.sandberg@arm.com with pytest.raises(RuntimeError) as excinfo: 6111986Sandreas.sandberg@arm.com test_arg_conversion_error2(f) 6211986Sandreas.sandberg@arm.com assert "Unable to convert call argument" in str(excinfo.value) 6311986Sandreas.sandberg@arm.com 6411986Sandreas.sandberg@arm.com 6511986Sandreas.sandberg@arm.comdef test_lambda_closure_cleanup(): 6611986Sandreas.sandberg@arm.com from pybind11_tests import test_cleanup, payload_cstats 6711986Sandreas.sandberg@arm.com 6811986Sandreas.sandberg@arm.com test_cleanup() 6911986Sandreas.sandberg@arm.com cstats = payload_cstats() 7011986Sandreas.sandberg@arm.com assert cstats.alive() == 0 7111986Sandreas.sandberg@arm.com assert cstats.copy_constructions == 1 7211986Sandreas.sandberg@arm.com assert cstats.move_constructions >= 1 7311986Sandreas.sandberg@arm.com 7411986Sandreas.sandberg@arm.com 7511986Sandreas.sandberg@arm.comdef test_cpp_function_roundtrip(): 7611986Sandreas.sandberg@arm.com """Test if passing a function pointer from C++ -> Python -> C++ yields the original pointer""" 7711986Sandreas.sandberg@arm.com from pybind11_tests import dummy_function, dummy_function2, test_dummy_function, roundtrip 7811986Sandreas.sandberg@arm.com 7911986Sandreas.sandberg@arm.com assert test_dummy_function(dummy_function) == "matches dummy_function: eval(1) = 2" 8011986Sandreas.sandberg@arm.com assert test_dummy_function(roundtrip(dummy_function)) == "matches dummy_function: eval(1) = 2" 8111986Sandreas.sandberg@arm.com assert roundtrip(None, expect_none=True) is None 8211986Sandreas.sandberg@arm.com assert test_dummy_function(lambda x: x + 2) == "can't convert to function pointer: eval(1) = 3" 8311986Sandreas.sandberg@arm.com 8411986Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 8511986Sandreas.sandberg@arm.com test_dummy_function(dummy_function2) 8611986Sandreas.sandberg@arm.com assert "incompatible function arguments" in str(excinfo.value) 8711986Sandreas.sandberg@arm.com 8811986Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 8911986Sandreas.sandberg@arm.com test_dummy_function(lambda x, y: x + y) 9011986Sandreas.sandberg@arm.com assert any(s in str(excinfo.value) for s in ("missing 1 required positional argument", 9111986Sandreas.sandberg@arm.com "takes exactly 2 arguments")) 9211986Sandreas.sandberg@arm.com 9311986Sandreas.sandberg@arm.com 9411986Sandreas.sandberg@arm.comdef test_function_signatures(doc): 9511986Sandreas.sandberg@arm.com from pybind11_tests import test_callback3, test_callback4 9611986Sandreas.sandberg@arm.com 9711986Sandreas.sandberg@arm.com assert doc(test_callback3) == "test_callback3(arg0: Callable[[int], int]) -> str" 9811986Sandreas.sandberg@arm.com assert doc(test_callback4) == "test_callback4() -> Callable[[int], int]" 99