Blame external/pybind11/tests/test_cmake_build/embed.cpp

Packit 534379
#include <pybind11/embed.h>
Packit 534379
namespace py = pybind11;
Packit 534379
Packit 534379
PYBIND11_EMBEDDED_MODULE(test_cmake_build, m) {
Packit 534379
    m.def("add", [](int i, int j) { return i + j; });
Packit 534379
}
Packit 534379
Packit 534379
int main(int argc, char *argv[]) {
Packit 534379
    if (argc != 2)
Packit 534379
        throw std::runtime_error("Expected test.py file as the first argument");
Packit 534379
    auto test_py_file = argv[1];
Packit 534379
Packit 534379
    py::scoped_interpreter guard{};
Packit 534379
Packit 534379
    auto m = py::module::import("test_cmake_build");
Packit 534379
    if (m.attr("add")(1, 2).cast<int>() != 3)
Packit 534379
        throw std::runtime_error("embed.cpp failed");
Packit 534379
Packit 534379
    py::module::import("sys").attr("argv") = py::make_tuple("test.py", "embed.cpp");
Packit 534379
    py::eval_file(test_py_file, py::globals());
Packit 534379
}