Skip to content

Commit

Permalink
Add async Python callbacks test that runs in separate Python thread
Browse files Browse the repository at this point in the history
  • Loading branch information
uentity committed Dec 2, 2018
1 parent e2b884c commit 83113a0
Show file tree
Hide file tree
Showing 2 changed files with 48 additions and 0 deletions.
19 changes: 19 additions & 0 deletions tests/test_callbacks.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@
#include "pybind11_tests.h"
#include "constructor_stats.h"
#include <pybind11/functional.h>
#include <thread>


int dummy_function(int i) { return i + 1; }
Expand Down Expand Up @@ -146,4 +147,22 @@ TEST_SUBMODULE(callbacks, m) {
py::class_<CppBoundMethodTest>(m, "CppBoundMethodTest")
.def(py::init<>())
.def("triple", [](CppBoundMethodTest &, int val) { return 3 * val; });

// test async Python callbacks
using callback_f = std::function<void(int)>;
m.def("test_async_callback", [](callback_f f, py::list work) {
// make detached thread that calls `f` with piece of work after a little delay
auto start_f = [f](int j) {
auto invoke_f = [f, j] {
std::this_thread::sleep_for(std::chrono::milliseconds(50));
f(j);
};
auto t = std::thread(std::move(invoke_f));
t.detach();
};

// spawn worker threads
for (auto i : work)
start_f(py::cast<int>(i));
});
}
29 changes: 29 additions & 0 deletions tests/test_callbacks.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
import pytest
from pybind11_tests import callbacks as m
from threading import Thread


def test_callbacks():
Expand Down Expand Up @@ -105,3 +106,31 @@ def test_function_signatures(doc):

def test_movable_object():
assert m.callback_with_movable(lambda _: None) is True


def test_async_callbacks():
# serves as state for async callback
class Item:
def __init__(self, value):
self.value = value

res = []

# generate stateful lambda that will store result in `res`
def gen_f():
s = Item(3)
return lambda j: res.append(s.value + j)

# do some work async
work = [1, 2, 3, 4]
m.test_async_callback(gen_f(), work)
# wait until work is done
from time import sleep
sleep(0.5)
assert sum(res) == sum([x + 3 for x in work])


def test_async_async_callbacks():
t = Thread(target=test_async_callbacks)
t.start()
t.join()

0 comments on commit 83113a0

Please sign in to comment.