mirror of
https://github.com/pybind/pybind11.git
synced 2024-11-23 05:35:13 +00:00
52f4be8946
Adding or removing tests is a little bit cumbersome currently: the test needs to be added to CMakeLists.txt, the init function needs to be predeclared in pybind11_tests.cpp, then called in the plugin initialization. While this isn't a big deal for tests that are being committed, it's more of a hassle when working on some new feature or test code for which I temporarily only care about building and linking the test being worked on rather than the entire test suite. This commit changes tests to self-register their initialization by having each test initialize a local object (which stores the initialization function in a static variable). This makes changing the set of tests being build easy: one only needs to add or comment out test names in tests/CMakeLists.txt. A couple other minor changes that go along with this: - test_eigen.cpp is now included in the test list, then removed if eigen isn't available. This lets you disable the eigen tests by commenting it out, just like all the other tests, but keeps the build working without eigen eigen isn't available. (Also, if it's commented out, we don't even bother looking for and reporting the building with/without eigen status message). - pytest is now invoked with all the built test names (with .cpp changed to .py) so that it doesn't try to run tests that weren't built.
99 lines
2.9 KiB
C++
99 lines
2.9 KiB
C++
/*
|
|
tests/test_callbacks.cpp -- callbacks
|
|
|
|
Copyright (c) 2016 Wenzel Jakob <wenzel.jakob@epfl.ch>
|
|
|
|
All rights reserved. Use of this source code is governed by a
|
|
BSD-style license that can be found in the LICENSE file.
|
|
*/
|
|
|
|
#include "pybind11_tests.h"
|
|
#include "constructor_stats.h"
|
|
#include <pybind11/functional.h>
|
|
|
|
|
|
py::object test_callback1(py::object func) {
|
|
return func();
|
|
}
|
|
|
|
py::tuple test_callback2(py::object func) {
|
|
return func("Hello", 'x', true, 5);
|
|
}
|
|
|
|
std::string test_callback3(const std::function<int(int)> &func) {
|
|
return "func(43) = " + std::to_string(func(43));
|
|
}
|
|
|
|
std::function<int(int)> test_callback4() {
|
|
return [](int i) { return i+1; };
|
|
}
|
|
|
|
py::cpp_function test_callback5() {
|
|
return py::cpp_function([](int i) { return i+1; },
|
|
py::arg("number"));
|
|
}
|
|
|
|
int dummy_function(int i) { return i + 1; }
|
|
int dummy_function2(int i, int j) { return i + j; }
|
|
std::function<int(int)> roundtrip(std::function<int(int)> f, bool expect_none = false) {
|
|
if (expect_none && f) {
|
|
throw std::runtime_error("Expected None to be converted to empty std::function");
|
|
}
|
|
return f;
|
|
}
|
|
|
|
std::string test_dummy_function(const std::function<int(int)> &f) {
|
|
using fn_type = int (*)(int);
|
|
auto result = f.target<fn_type>();
|
|
if (!result) {
|
|
auto r = f(1);
|
|
return "can't convert to function pointer: eval(1) = " + std::to_string(r);
|
|
} else if (*result == dummy_function) {
|
|
auto r = (*result)(1);
|
|
return "matches dummy_function: eval(1) = " + std::to_string(r);
|
|
} else {
|
|
return "argument does NOT match dummy_function. This should never happen!";
|
|
}
|
|
}
|
|
|
|
struct Payload {
|
|
Payload() {
|
|
print_default_created(this);
|
|
}
|
|
~Payload() {
|
|
print_destroyed(this);
|
|
}
|
|
Payload(const Payload &) {
|
|
print_copy_created(this);
|
|
}
|
|
Payload(Payload &&) {
|
|
print_move_created(this);
|
|
}
|
|
};
|
|
|
|
test_initializer callbacks([](py::module &m) {
|
|
m.def("test_callback1", &test_callback1);
|
|
m.def("test_callback2", &test_callback2);
|
|
m.def("test_callback3", &test_callback3);
|
|
m.def("test_callback4", &test_callback4);
|
|
m.def("test_callback5", &test_callback5);
|
|
|
|
/* Test cleanup of lambda closure */
|
|
|
|
m.def("test_cleanup", []() -> std::function<void(void)> {
|
|
Payload p;
|
|
|
|
return [p]() {
|
|
/* p should be cleaned up when the returned function is garbage collected */
|
|
};
|
|
});
|
|
|
|
/* Test if passing a function pointer from C++ -> Python -> C++ yields the original pointer */
|
|
m.def("dummy_function", &dummy_function);
|
|
m.def("dummy_function2", &dummy_function2);
|
|
m.def("roundtrip", &roundtrip, py::arg("f"), py::arg("expect_none")=false);
|
|
m.def("test_dummy_function", &test_dummy_function);
|
|
// Export the payload constructor statistics for testing purposes:
|
|
m.def("payload_cstats", &ConstructorStats::get<Payload>);
|
|
});
|