diff options
Diffstat (limited to '3rdparty/pybind11/tests/test_stl_binders.cpp')
| -rw-r--r-- | 3rdparty/pybind11/tests/test_stl_binders.cpp | 111 |
1 files changed, 67 insertions, 44 deletions
diff --git a/3rdparty/pybind11/tests/test_stl_binders.cpp b/3rdparty/pybind11/tests/test_stl_binders.cpp index 1c0df984..ca9630bd 100644 --- a/3rdparty/pybind11/tests/test_stl_binders.cpp +++ b/3rdparty/pybind11/tests/test_stl_binders.cpp @@ -7,23 +7,24 @@ BSD-style license that can be found in the LICENSE file. */ +#include <pybind11/numpy.h> +#include <pybind11/stl_bind.h> + #include "pybind11_tests.h" -#include <pybind11/stl_bind.h> -#include <pybind11/numpy.h> -#include <map> #include <deque> +#include <map> #include <unordered_map> class El { public: El() = delete; - El(int v) : a(v) { } + explicit El(int v) : a(v) {} int a; }; -std::ostream & operator<<(std::ostream &s, El const&v) { +std::ostream &operator<<(std::ostream &s, El const &v) { s << "El{" << v.a << '}'; return s; } @@ -40,25 +41,32 @@ public: int value; }; -template <class Container> Container *one_to_n(int n) { - auto v = new Container(); - for (int i = 1; i <= n; i++) +template <class Container> +Container *one_to_n(int n) { + auto *v = new Container(); + for (int i = 1; i <= n; i++) { v->emplace_back(i); + } return v; } -template <class Map> Map *times_ten(int n) { - auto m = new Map(); - for (int i = 1; i <= n; i++) - m->emplace(int(i), E_nc(10*i)); +template <class Map> +Map *times_ten(int n) { + auto *m = new Map(); + for (int i = 1; i <= n; i++) { + m->emplace(int(i), E_nc(10 * i)); + } return m; } -template <class NestMap> NestMap *times_hundred(int n) { - auto m = new NestMap(); - for (int i = 1; i <= n; i++) - for (int j = 1; j <= n; j++) - (*m)[i].emplace(int(j*10), E_nc(100*j)); +template <class NestMap> +NestMap *times_hundred(int n) { + auto *m = new NestMap(); + for (int i = 1; i <= n; i++) { + for (int j = 1; j <= n; j++) { + (*m)[i].emplace(int(j * 10), E_nc(100 * j)); + } + } return m; } @@ -67,8 +75,7 @@ TEST_SUBMODULE(stl_binders, m) { py::bind_vector<std::vector<unsigned int>>(m, "VectorInt", py::buffer_protocol()); // test_vector_custom - py::class_<El>(m, "El") - .def(py::init<int>()); + py::class_<El>(m, "El").def(py::init<int>()); py::bind_vector<std::vector<El>>(m, "VectorEl"); py::bind_vector<std::vector<std::vector<El>>>(m, "VectorVectorEl"); @@ -78,52 +85,68 @@ TEST_SUBMODULE(stl_binders, m) { // test_map_string_double_const py::bind_map<std::map<std::string, double const>>(m, "MapStringDoubleConst"); - py::bind_map<std::unordered_map<std::string, double const>>(m, "UnorderedMapStringDoubleConst"); + py::bind_map<std::unordered_map<std::string, double const>>(m, + "UnorderedMapStringDoubleConst"); - py::class_<E_nc>(m, "ENC") - .def(py::init<int>()) - .def_readwrite("value", &E_nc::value); + py::class_<E_nc>(m, "ENC").def(py::init<int>()).def_readwrite("value", &E_nc::value); // test_noncopyable_containers py::bind_vector<std::vector<E_nc>>(m, "VectorENC"); - m.def("get_vnc", &one_to_n<std::vector<E_nc>>, py::return_value_policy::reference); + m.def("get_vnc", &one_to_n<std::vector<E_nc>>); py::bind_vector<std::deque<E_nc>>(m, "DequeENC"); - m.def("get_dnc", &one_to_n<std::deque<E_nc>>, py::return_value_policy::reference); + m.def("get_dnc", &one_to_n<std::deque<E_nc>>); py::bind_map<std::map<int, E_nc>>(m, "MapENC"); - m.def("get_mnc", ×_ten<std::map<int, E_nc>>, py::return_value_policy::reference); + m.def("get_mnc", ×_ten<std::map<int, E_nc>>); py::bind_map<std::unordered_map<int, E_nc>>(m, "UmapENC"); - m.def("get_umnc", ×_ten<std::unordered_map<int, E_nc>>, py::return_value_policy::reference); + m.def("get_umnc", ×_ten<std::unordered_map<int, E_nc>>); // Issue #1885: binding nested std::map<X, Container<E>> with E non-copyable py::bind_map<std::map<int, std::vector<E_nc>>>(m, "MapVecENC"); - m.def("get_nvnc", [](int n) - { - auto m = new std::map<int, std::vector<E_nc>>(); - for (int i = 1; i <= n; i++) - for (int j = 1; j <= n; j++) - (*m)[i].emplace_back(j); - return m; - }, py::return_value_policy::reference); + m.def("get_nvnc", [](int n) { + auto *m = new std::map<int, std::vector<E_nc>>(); + for (int i = 1; i <= n; i++) { + for (int j = 1; j <= n; j++) { + (*m)[i].emplace_back(j); + } + } + return m; + }); py::bind_map<std::map<int, std::map<int, E_nc>>>(m, "MapMapENC"); - m.def("get_nmnc", ×_hundred<std::map<int, std::map<int, E_nc>>>, py::return_value_policy::reference); + m.def("get_nmnc", ×_hundred<std::map<int, std::map<int, E_nc>>>); py::bind_map<std::unordered_map<int, std::unordered_map<int, E_nc>>>(m, "UmapUmapENC"); - m.def("get_numnc", ×_hundred<std::unordered_map<int, std::unordered_map<int, E_nc>>>, py::return_value_policy::reference); + m.def("get_numnc", ×_hundred<std::unordered_map<int, std::unordered_map<int, E_nc>>>); // test_vector_buffer py::bind_vector<std::vector<unsigned char>>(m, "VectorUChar", py::buffer_protocol()); // no dtype declared for this version: - struct VUndeclStruct { bool w; uint32_t x; double y; bool z; }; - m.def("create_undeclstruct", [m] () mutable { - py::bind_vector<std::vector<VUndeclStruct>>(m, "VectorUndeclStruct", py::buffer_protocol()); + struct VUndeclStruct { + bool w; + uint32_t x; + double y; + bool z; + }; + m.def("create_undeclstruct", [m]() mutable { + py::bind_vector<std::vector<VUndeclStruct>>( + m, "VectorUndeclStruct", py::buffer_protocol()); }); // The rest depends on numpy: - try { py::module_::import("numpy"); } - catch (...) { return; } + try { + py::module_::import("numpy"); + } catch (...) { + return; + } // test_vector_buffer_numpy - struct VStruct { bool w; uint32_t x; double y; bool z; }; + struct VStruct { + bool w; + uint32_t x; + double y; + bool z; + }; PYBIND11_NUMPY_DTYPE(VStruct, w, x, y, z); py::class_<VStruct>(m, "VStruct").def_readwrite("x", &VStruct::x); py::bind_vector<std::vector<VStruct>>(m, "VectorStruct", py::buffer_protocol()); - m.def("get_vectorstruct", [] {return std::vector<VStruct> {{0, 5, 3.0, 1}, {1, 30, -1e4, 0}};}); + m.def("get_vectorstruct", [] { + return std::vector<VStruct>{{false, 5, 3.0, true}, {true, 30, -1e4, false}}; + }); } |