test_stl_binders.cpp revision 12037
111986Sandreas.sandberg@arm.com/* 211986Sandreas.sandberg@arm.com tests/test_stl_binders.cpp -- Usage of stl_binders functions 311986Sandreas.sandberg@arm.com 411986Sandreas.sandberg@arm.com Copyright (c) 2016 Sergey Lyskov 511986Sandreas.sandberg@arm.com 611986Sandreas.sandberg@arm.com All rights reserved. Use of this source code is governed by a 711986Sandreas.sandberg@arm.com BSD-style license that can be found in the LICENSE file. 811986Sandreas.sandberg@arm.com*/ 911986Sandreas.sandberg@arm.com 1011986Sandreas.sandberg@arm.com#include "pybind11_tests.h" 1111986Sandreas.sandberg@arm.com 1211986Sandreas.sandberg@arm.com#include <pybind11/stl_bind.h> 1312037Sandreas.sandberg@arm.com#include <pybind11/numpy.h> 1411986Sandreas.sandberg@arm.com#include <map> 1511986Sandreas.sandberg@arm.com#include <deque> 1611986Sandreas.sandberg@arm.com#include <unordered_map> 1711986Sandreas.sandberg@arm.com 1812037Sandreas.sandberg@arm.com#ifdef _MSC_VER 1912037Sandreas.sandberg@arm.com// We get some really long type names here which causes MSVC to emit warnings 2012037Sandreas.sandberg@arm.com# pragma warning(disable: 4503) // warning C4503: decorated name length exceeded, name was truncated 2112037Sandreas.sandberg@arm.com#endif 2212037Sandreas.sandberg@arm.com 2311986Sandreas.sandberg@arm.comclass El { 2411986Sandreas.sandberg@arm.compublic: 2511986Sandreas.sandberg@arm.com El() = delete; 2611986Sandreas.sandberg@arm.com El(int v) : a(v) { } 2711986Sandreas.sandberg@arm.com 2811986Sandreas.sandberg@arm.com int a; 2911986Sandreas.sandberg@arm.com}; 3011986Sandreas.sandberg@arm.com 3111986Sandreas.sandberg@arm.comstd::ostream & operator<<(std::ostream &s, El const&v) { 3211986Sandreas.sandberg@arm.com s << "El{" << v.a << '}'; 3311986Sandreas.sandberg@arm.com return s; 3411986Sandreas.sandberg@arm.com} 3511986Sandreas.sandberg@arm.com 3611986Sandreas.sandberg@arm.com/// Issue #487: binding std::vector<E> with E non-copyable 3711986Sandreas.sandberg@arm.comclass E_nc { 3811986Sandreas.sandberg@arm.compublic: 3911986Sandreas.sandberg@arm.com explicit E_nc(int i) : value{i} {} 4011986Sandreas.sandberg@arm.com E_nc(const E_nc &) = delete; 4111986Sandreas.sandberg@arm.com E_nc &operator=(const E_nc &) = delete; 4211986Sandreas.sandberg@arm.com E_nc(E_nc &&) = default; 4311986Sandreas.sandberg@arm.com E_nc &operator=(E_nc &&) = default; 4411986Sandreas.sandberg@arm.com 4511986Sandreas.sandberg@arm.com int value; 4611986Sandreas.sandberg@arm.com}; 4711986Sandreas.sandberg@arm.com 4811986Sandreas.sandberg@arm.comtemplate <class Container> Container *one_to_n(int n) { 4911986Sandreas.sandberg@arm.com auto v = new Container(); 5011986Sandreas.sandberg@arm.com for (int i = 1; i <= n; i++) 5111986Sandreas.sandberg@arm.com v->emplace_back(i); 5211986Sandreas.sandberg@arm.com return v; 5311986Sandreas.sandberg@arm.com} 5411986Sandreas.sandberg@arm.com 5511986Sandreas.sandberg@arm.comtemplate <class Map> Map *times_ten(int n) { 5611986Sandreas.sandberg@arm.com auto m = new Map(); 5711986Sandreas.sandberg@arm.com for (int i = 1; i <= n; i++) 5811986Sandreas.sandberg@arm.com m->emplace(int(i), E_nc(10*i)); 5911986Sandreas.sandberg@arm.com return m; 6011986Sandreas.sandberg@arm.com} 6111986Sandreas.sandberg@arm.com 6212037Sandreas.sandberg@arm.comstruct VStruct { 6312037Sandreas.sandberg@arm.com bool w; 6412037Sandreas.sandberg@arm.com uint32_t x; 6512037Sandreas.sandberg@arm.com double y; 6612037Sandreas.sandberg@arm.com bool z; 6712037Sandreas.sandberg@arm.com}; 6812037Sandreas.sandberg@arm.com 6912037Sandreas.sandberg@arm.comstruct VUndeclStruct { //dtype not declared for this version 7012037Sandreas.sandberg@arm.com bool w; 7112037Sandreas.sandberg@arm.com uint32_t x; 7212037Sandreas.sandberg@arm.com double y; 7312037Sandreas.sandberg@arm.com bool z; 7412037Sandreas.sandberg@arm.com}; 7512037Sandreas.sandberg@arm.com 7611986Sandreas.sandberg@arm.comtest_initializer stl_binder_vector([](py::module &m) { 7711986Sandreas.sandberg@arm.com py::class_<El>(m, "El") 7811986Sandreas.sandberg@arm.com .def(py::init<int>()); 7911986Sandreas.sandberg@arm.com 8012037Sandreas.sandberg@arm.com py::bind_vector<std::vector<unsigned char>>(m, "VectorUChar", py::buffer_protocol()); 8112037Sandreas.sandberg@arm.com py::bind_vector<std::vector<unsigned int>>(m, "VectorInt", py::buffer_protocol()); 8211986Sandreas.sandberg@arm.com py::bind_vector<std::vector<bool>>(m, "VectorBool"); 8311986Sandreas.sandberg@arm.com 8411986Sandreas.sandberg@arm.com py::bind_vector<std::vector<El>>(m, "VectorEl"); 8511986Sandreas.sandberg@arm.com 8611986Sandreas.sandberg@arm.com py::bind_vector<std::vector<std::vector<El>>>(m, "VectorVectorEl"); 8711986Sandreas.sandberg@arm.com 8812037Sandreas.sandberg@arm.com m.def("create_undeclstruct", [m] () mutable { 8912037Sandreas.sandberg@arm.com py::bind_vector<std::vector<VUndeclStruct>>(m, "VectorUndeclStruct", py::buffer_protocol()); 9012037Sandreas.sandberg@arm.com }); 9112037Sandreas.sandberg@arm.com 9212037Sandreas.sandberg@arm.com try { 9312037Sandreas.sandberg@arm.com py::module::import("numpy"); 9412037Sandreas.sandberg@arm.com } catch (...) { 9512037Sandreas.sandberg@arm.com return; 9612037Sandreas.sandberg@arm.com } 9712037Sandreas.sandberg@arm.com PYBIND11_NUMPY_DTYPE(VStruct, w, x, y, z); 9812037Sandreas.sandberg@arm.com py::class_<VStruct>(m, "VStruct").def_readwrite("x", &VStruct::x); 9912037Sandreas.sandberg@arm.com py::bind_vector<std::vector<VStruct>>(m, "VectorStruct", py::buffer_protocol()); 10012037Sandreas.sandberg@arm.com m.def("get_vectorstruct", [] {return std::vector<VStruct> {{0, 5, 3.0, 1}, {1, 30, -1e4, 0}};}); 10111986Sandreas.sandberg@arm.com}); 10211986Sandreas.sandberg@arm.com 10311986Sandreas.sandberg@arm.comtest_initializer stl_binder_map([](py::module &m) { 10411986Sandreas.sandberg@arm.com py::bind_map<std::map<std::string, double>>(m, "MapStringDouble"); 10511986Sandreas.sandberg@arm.com py::bind_map<std::unordered_map<std::string, double>>(m, "UnorderedMapStringDouble"); 10611986Sandreas.sandberg@arm.com 10711986Sandreas.sandberg@arm.com py::bind_map<std::map<std::string, double const>>(m, "MapStringDoubleConst"); 10811986Sandreas.sandberg@arm.com py::bind_map<std::unordered_map<std::string, double const>>(m, "UnorderedMapStringDoubleConst"); 10911986Sandreas.sandberg@arm.com 11011986Sandreas.sandberg@arm.com}); 11111986Sandreas.sandberg@arm.com 11211986Sandreas.sandberg@arm.comtest_initializer stl_binder_noncopyable([](py::module &m) { 11311986Sandreas.sandberg@arm.com py::class_<E_nc>(m, "ENC") 11411986Sandreas.sandberg@arm.com .def(py::init<int>()) 11511986Sandreas.sandberg@arm.com .def_readwrite("value", &E_nc::value); 11611986Sandreas.sandberg@arm.com 11711986Sandreas.sandberg@arm.com py::bind_vector<std::vector<E_nc>>(m, "VectorENC"); 11811986Sandreas.sandberg@arm.com m.def("get_vnc", &one_to_n<std::vector<E_nc>>, py::return_value_policy::reference); 11911986Sandreas.sandberg@arm.com 12011986Sandreas.sandberg@arm.com py::bind_vector<std::deque<E_nc>>(m, "DequeENC"); 12111986Sandreas.sandberg@arm.com m.def("get_dnc", &one_to_n<std::deque<E_nc>>, py::return_value_policy::reference); 12211986Sandreas.sandberg@arm.com 12311986Sandreas.sandberg@arm.com py::bind_map<std::map<int, E_nc>>(m, "MapENC"); 12411986Sandreas.sandberg@arm.com m.def("get_mnc", ×_ten<std::map<int, E_nc>>, py::return_value_policy::reference); 12511986Sandreas.sandberg@arm.com 12611986Sandreas.sandberg@arm.com py::bind_map<std::unordered_map<int, E_nc>>(m, "UmapENC"); 12711986Sandreas.sandberg@arm.com m.def("get_umnc", ×_ten<std::unordered_map<int, E_nc>>, py::return_value_policy::reference); 12811986Sandreas.sandberg@arm.com}); 129