test_callbacks.py revision 14299
111986Sandreas.sandberg@arm.comimport pytest 212391Sjason@lowepower.comfrom pybind11_tests import callbacks as m 314299Sbbruce@ucdavis.edufrom threading import Thread 411986Sandreas.sandberg@arm.com 511986Sandreas.sandberg@arm.com 611986Sandreas.sandberg@arm.comdef test_callbacks(): 711986Sandreas.sandberg@arm.com from functools import partial 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 1812391Sjason@lowepower.com assert m.test_callback1(func1) == "func1" 1912391Sjason@lowepower.com assert m.test_callback2(func2) == ("func2", "Hello", "x", True, 5) 2012391Sjason@lowepower.com assert m.test_callback1(partial(func2, 1, 2, 3, 4)) == ("func2", 1, 2, 3, 4) 2112391Sjason@lowepower.com assert m.test_callback1(partial(func3, "partial")) == "func3(partial)" 2212391Sjason@lowepower.com assert m.test_callback3(lambda i: i + 1) == "func(43) = 44" 2311986Sandreas.sandberg@arm.com 2412391Sjason@lowepower.com f = m.test_callback4() 2511986Sandreas.sandberg@arm.com assert f(43) == 44 2612391Sjason@lowepower.com f = m.test_callback5() 2711986Sandreas.sandberg@arm.com assert f(number=43) == 44 2811986Sandreas.sandberg@arm.com 2911986Sandreas.sandberg@arm.com 3012391Sjason@lowepower.comdef test_bound_method_callback(): 3112391Sjason@lowepower.com # Bound Python method: 3212391Sjason@lowepower.com class MyClass: 3312391Sjason@lowepower.com def double(self, val): 3412391Sjason@lowepower.com return 2 * val 3512391Sjason@lowepower.com 3612391Sjason@lowepower.com z = MyClass() 3712391Sjason@lowepower.com assert m.test_callback3(z.double) == "func(43) = 86" 3812391Sjason@lowepower.com 3912391Sjason@lowepower.com z = m.CppBoundMethodTest() 4012391Sjason@lowepower.com assert m.test_callback3(z.triple) == "func(43) = 129" 4112391Sjason@lowepower.com 4212391Sjason@lowepower.com 4311986Sandreas.sandberg@arm.comdef test_keyword_args_and_generalized_unpacking(): 4411986Sandreas.sandberg@arm.com 4511986Sandreas.sandberg@arm.com def f(*args, **kwargs): 4611986Sandreas.sandberg@arm.com return args, kwargs 4711986Sandreas.sandberg@arm.com 4812391Sjason@lowepower.com assert m.test_tuple_unpacking(f) == (("positional", 1, 2, 3, 4, 5, 6), {}) 4912391Sjason@lowepower.com assert m.test_dict_unpacking(f) == (("positional", 1), {"key": "value", "a": 1, "b": 2}) 5012391Sjason@lowepower.com assert m.test_keyword_args(f) == ((), {"x": 10, "y": 20}) 5112391Sjason@lowepower.com assert m.test_unpacking_and_keywords1(f) == ((1, 2), {"c": 3, "d": 4}) 5212391Sjason@lowepower.com assert m.test_unpacking_and_keywords2(f) == ( 5311986Sandreas.sandberg@arm.com ("positional", 1, 2, 3, 4, 5), 5411986Sandreas.sandberg@arm.com {"key": "value", "a": 1, "b": 2, "c": 3, "d": 4, "e": 5} 5511986Sandreas.sandberg@arm.com ) 5611986Sandreas.sandberg@arm.com 5711986Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 5812391Sjason@lowepower.com m.test_unpacking_error1(f) 5911986Sandreas.sandberg@arm.com assert "Got multiple values for keyword argument" in str(excinfo.value) 6011986Sandreas.sandberg@arm.com 6111986Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 6212391Sjason@lowepower.com m.test_unpacking_error2(f) 6311986Sandreas.sandberg@arm.com assert "Got multiple values for keyword argument" in str(excinfo.value) 6411986Sandreas.sandberg@arm.com 6511986Sandreas.sandberg@arm.com with pytest.raises(RuntimeError) as excinfo: 6612391Sjason@lowepower.com m.test_arg_conversion_error1(f) 6711986Sandreas.sandberg@arm.com assert "Unable to convert call argument" in str(excinfo.value) 6811986Sandreas.sandberg@arm.com 6911986Sandreas.sandberg@arm.com with pytest.raises(RuntimeError) as excinfo: 7012391Sjason@lowepower.com m.test_arg_conversion_error2(f) 7111986Sandreas.sandberg@arm.com assert "Unable to convert call argument" in str(excinfo.value) 7211986Sandreas.sandberg@arm.com 7311986Sandreas.sandberg@arm.com 7411986Sandreas.sandberg@arm.comdef test_lambda_closure_cleanup(): 7512391Sjason@lowepower.com m.test_cleanup() 7612391Sjason@lowepower.com cstats = m.payload_cstats() 7711986Sandreas.sandberg@arm.com assert cstats.alive() == 0 7811986Sandreas.sandberg@arm.com assert cstats.copy_constructions == 1 7911986Sandreas.sandberg@arm.com assert cstats.move_constructions >= 1 8011986Sandreas.sandberg@arm.com 8111986Sandreas.sandberg@arm.com 8211986Sandreas.sandberg@arm.comdef test_cpp_function_roundtrip(): 8311986Sandreas.sandberg@arm.com """Test if passing a function pointer from C++ -> Python -> C++ yields the original pointer""" 8411986Sandreas.sandberg@arm.com 8512391Sjason@lowepower.com assert m.test_dummy_function(m.dummy_function) == "matches dummy_function: eval(1) = 2" 8612391Sjason@lowepower.com assert (m.test_dummy_function(m.roundtrip(m.dummy_function)) == 8712391Sjason@lowepower.com "matches dummy_function: eval(1) = 2") 8812391Sjason@lowepower.com assert m.roundtrip(None, expect_none=True) is None 8912391Sjason@lowepower.com assert (m.test_dummy_function(lambda x: x + 2) == 9012391Sjason@lowepower.com "can't convert to function pointer: eval(1) = 3") 9111986Sandreas.sandberg@arm.com 9211986Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 9312391Sjason@lowepower.com m.test_dummy_function(m.dummy_function2) 9411986Sandreas.sandberg@arm.com assert "incompatible function arguments" in str(excinfo.value) 9511986Sandreas.sandberg@arm.com 9611986Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 9712391Sjason@lowepower.com m.test_dummy_function(lambda x, y: x + y) 9811986Sandreas.sandberg@arm.com assert any(s in str(excinfo.value) for s in ("missing 1 required positional argument", 9911986Sandreas.sandberg@arm.com "takes exactly 2 arguments")) 10011986Sandreas.sandberg@arm.com 10111986Sandreas.sandberg@arm.com 10211986Sandreas.sandberg@arm.comdef test_function_signatures(doc): 10312391Sjason@lowepower.com assert doc(m.test_callback3) == "test_callback3(arg0: Callable[[int], int]) -> str" 10412391Sjason@lowepower.com assert doc(m.test_callback4) == "test_callback4() -> Callable[[int], int]" 10512037Sandreas.sandberg@arm.com 10612037Sandreas.sandberg@arm.com 10712037Sandreas.sandberg@arm.comdef test_movable_object(): 10812391Sjason@lowepower.com assert m.callback_with_movable(lambda _: None) is True 10914299Sbbruce@ucdavis.edu 11014299Sbbruce@ucdavis.edu 11114299Sbbruce@ucdavis.edudef test_async_callbacks(): 11214299Sbbruce@ucdavis.edu # serves as state for async callback 11314299Sbbruce@ucdavis.edu class Item: 11414299Sbbruce@ucdavis.edu def __init__(self, value): 11514299Sbbruce@ucdavis.edu self.value = value 11614299Sbbruce@ucdavis.edu 11714299Sbbruce@ucdavis.edu res = [] 11814299Sbbruce@ucdavis.edu 11914299Sbbruce@ucdavis.edu # generate stateful lambda that will store result in `res` 12014299Sbbruce@ucdavis.edu def gen_f(): 12114299Sbbruce@ucdavis.edu s = Item(3) 12214299Sbbruce@ucdavis.edu return lambda j: res.append(s.value + j) 12314299Sbbruce@ucdavis.edu 12414299Sbbruce@ucdavis.edu # do some work async 12514299Sbbruce@ucdavis.edu work = [1, 2, 3, 4] 12614299Sbbruce@ucdavis.edu m.test_async_callback(gen_f(), work) 12714299Sbbruce@ucdavis.edu # wait until work is done 12814299Sbbruce@ucdavis.edu from time import sleep 12914299Sbbruce@ucdavis.edu sleep(0.5) 13014299Sbbruce@ucdavis.edu assert sum(res) == sum([x + 3 for x in work]) 13114299Sbbruce@ucdavis.edu 13214299Sbbruce@ucdavis.edu 13314299Sbbruce@ucdavis.edudef test_async_async_callbacks(): 13414299Sbbruce@ucdavis.edu t = Thread(target=test_async_callbacks) 13514299Sbbruce@ucdavis.edu t.start() 13614299Sbbruce@ucdavis.edu t.join() 137