mirror of https://github.com/pybind/pybind11.git
65 lines
1.9 KiB
C++
65 lines
1.9 KiB
C++
|
#include <pybind11/pybind11.h>
|
||
|
|
#include <pybind11/eval.h>
|
||
|
|
|
||
|
|
#include <catch.hpp>
|
||
|
|
|
||
|
|
namespace py = pybind11;
|
||
|
|
using namespace py::literals;
|
||
|
|
|
||
|
|
class Widget {
|
||
|
|
public:
|
||
|
|
Widget(std::string message) : message(message) { }
|
||
|
|
virtual ~Widget() = default;
|
||
|
|
|
||
|
|
std::string the_message() const { return message; }
|
||
|
|
virtual int the_answer() const = 0;
|
||
|
|
|
||
|
|
private:
|
||
|
|
std::string message;
|
||
|
|
};
|
||
|
|
|
||
|
|
class PyWidget final : public Widget {
|
||
|
|
using Widget::Widget;
|
||
|
|
|
||
|
|
int the_answer() const override { PYBIND11_OVERLOAD_PURE(int, Widget, the_answer); }
|
||
|
|
};
|
||
|
|
|
||
|
|
PyObject *make_embedded_module() {
|
||
|
|
py::module m("widget_module");
|
||
|
|
|
||
|
|
py::class_<Widget, PyWidget>(m, "Widget")
|
||
|
|
.def(py::init<std::string>())
|
||
|
|
.def_property_readonly("the_message", &Widget::the_message);
|
||
|
|
|
||
|
|
return m.ptr();
|
||
|
|
}
|
||
|
|
|
||
|
|
py::object import_file(const std::string &module, const std::string &path, py::object globals) {
|
||
|
|
auto locals = py::dict("module_name"_a=module, "path"_a=path);
|
||
|
|
py::eval<py::eval_statements>(
|
||
|
|
"import imp\n"
|
||
|
|
"with open(path) as file:\n"
|
||
|
|
" new_module = imp.load_module(module_name, file, path, ('py', 'U', imp.PY_SOURCE))",
|
||
|
|
globals, locals
|
||
|
|
);
|
||
|
|
return locals["new_module"];
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_CASE("Pass classes and data between modules defined in C++ and Python") {
|
||
|
|
PyImport_AppendInittab("widget_module", &make_embedded_module);
|
||
|
|
Py_Initialize();
|
||
|
|
{
|
||
|
|
auto globals = py::module::import("__main__").attr("__dict__");
|
||
|
|
auto module = import_file("widget", "test_interpreter.py", globals);
|
||
|
|
REQUIRE(py::hasattr(module, "DerivedWidget"));
|
||
|
|
|
||
|
|
auto py_widget = module.attr("DerivedWidget")("Hello, World!");
|
||
|
|
auto message = py_widget.attr("the_message");
|
||
|
|
REQUIRE(message.cast<std::string>() == "Hello, World!");
|
||
|
|
|
||
|
|
const auto &cpp_widget = py_widget.cast<const Widget &>();
|
||
|
|
REQUIRE(cpp_widget.the_answer() == 42);
|
||
|
|
}
|
||
|
|
Py_Finalize();
|
||
|
|
}
|