test_callbacks.py
Go to the documentation of this file.
1 # -*- coding: utf-8 -*-
2 import pytest
3 from pybind11_tests import callbacks as m
4 from threading import Thread
5 
6 
8  from functools import partial
9 
10  def func1():
11  return "func1"
12 
13  def func2(a, b, c, d):
14  return "func2", a, b, c, d
15 
16  def func3(a):
17  return "func3({})".format(a)
18 
19  assert m.test_callback1(func1) == "func1"
20  assert m.test_callback2(func2) == ("func2", "Hello", "x", True, 5)
21  assert m.test_callback1(partial(func2, 1, 2, 3, 4)) == ("func2", 1, 2, 3, 4)
22  assert m.test_callback1(partial(func3, "partial")) == "func3(partial)"
23  assert m.test_callback3(lambda i: i + 1) == "func(43) = 44"
24 
25  f = m.test_callback4()
26  assert f(43) == 44
27  f = m.test_callback5()
28  assert f(number=43) == 44
29 
30 
32  # Bound Python method:
33  class MyClass:
34  def double(self, val):
35  return 2 * val
36 
37  z = MyClass()
38  assert m.test_callback3(z.double) == "func(43) = 86"
39 
40  z = m.CppBoundMethodTest()
41  assert m.test_callback3(z.triple) == "func(43) = 129"
42 
43 
45 
46  def f(*args, **kwargs):
47  return args, kwargs
48 
49  assert m.test_tuple_unpacking(f) == (("positional", 1, 2, 3, 4, 5, 6), {})
50  assert m.test_dict_unpacking(f) == (("positional", 1), {"key": "value", "a": 1, "b": 2})
51  assert m.test_keyword_args(f) == ((), {"x": 10, "y": 20})
52  assert m.test_unpacking_and_keywords1(f) == ((1, 2), {"c": 3, "d": 4})
53  assert m.test_unpacking_and_keywords2(f) == (
54  ("positional", 1, 2, 3, 4, 5),
55  {"key": "value", "a": 1, "b": 2, "c": 3, "d": 4, "e": 5}
56  )
57 
58  with pytest.raises(TypeError) as excinfo:
59  m.test_unpacking_error1(f)
60  assert "Got multiple values for keyword argument" in str(excinfo.value)
61 
62  with pytest.raises(TypeError) as excinfo:
63  m.test_unpacking_error2(f)
64  assert "Got multiple values for keyword argument" in str(excinfo.value)
65 
66  with pytest.raises(RuntimeError) as excinfo:
67  m.test_arg_conversion_error1(f)
68  assert "Unable to convert call argument" in str(excinfo.value)
69 
70  with pytest.raises(RuntimeError) as excinfo:
71  m.test_arg_conversion_error2(f)
72  assert "Unable to convert call argument" in str(excinfo.value)
73 
74 
76  m.test_cleanup()
77  cstats = m.payload_cstats()
78  assert cstats.alive() == 0
79  assert cstats.copy_constructions == 1
80  assert cstats.move_constructions >= 1
81 
82 
84  """Test if passing a function pointer from C++ -> Python -> C++ yields the original pointer"""
85 
86  assert m.test_dummy_function(m.dummy_function) == "matches dummy_function: eval(1) = 2"
87  assert (m.test_dummy_function(m.roundtrip(m.dummy_function)) ==
88  "matches dummy_function: eval(1) = 2")
89  assert m.roundtrip(None, expect_none=True) is None
90  assert (m.test_dummy_function(lambda x: x + 2) ==
91  "can't convert to function pointer: eval(1) = 3")
92 
93  with pytest.raises(TypeError) as excinfo:
94  m.test_dummy_function(m.dummy_function2)
95  assert "incompatible function arguments" in str(excinfo.value)
96 
97  with pytest.raises(TypeError) as excinfo:
98  m.test_dummy_function(lambda x, y: x + y)
99  assert any(s in str(excinfo.value) for s in ("missing 1 required positional argument",
100  "takes exactly 2 arguments"))
101 
102 
104  assert doc(m.test_callback3) == "test_callback3(arg0: Callable[[int], int]) -> str"
105  assert doc(m.test_callback4) == "test_callback4() -> Callable[[int], int]"
106 
107 
109  assert m.callback_with_movable(lambda _: None) is True
110 
111 
113  # serves as state for async callback
114  class Item:
115  def __init__(self, value):
116  self.value = value
117 
118  res = []
119 
120  # generate stateful lambda that will store result in `res`
121  def gen_f():
122  s = Item(3)
123  return lambda j: res.append(s.value + j)
124 
125  # do some work async
126  work = [1, 2, 3, 4]
127  m.test_async_callback(gen_f(), work)
128  # wait until work is done
129  from time import sleep
130  sleep(0.5)
131  assert sum(res) == sum([x + 3 for x in work])
132 
133 
135  t = Thread(target=test_async_callbacks)
136  t.start()
137  t.join()
Annotation for documentation.
Definition: attr.h:33
def test_bound_method_callback()
def test_async_async_callbacks()
def test_lambda_closure_cleanup()
def test_keyword_args_and_generalized_unpacking()
Definition: pytypes.h:928
def test_callbacks()
Point2(* f)(const Point3 &, OptionalJacobian< 2, 3 >)
const mpreal sum(const mpreal tab[], const unsigned long int n, int &status, mp_rnd_t mode=mpreal::get_default_rnd())
Definition: mpreal.h:2381
def test_function_signatures(doc)
def test_movable_object()
def test_cpp_function_roundtrip()
def test_async_callbacks()


gtsam
Author(s):
autogenerated on Sat May 8 2021 02:46:03