Zengtudor/pybind11
1
0
Fork 0
mirror of https://github.com/pybind/pybind11.git synced 2024-11-21 20:55:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

std::array<> caster: support arbitrary sequences

Browse Source 
This PR brings the std::array<> caster in sync with the other STL type
casters: to accept an arbitrary sequence as input (rather than a list,
which is too restrictive).
This commit is contained in:
Wenzel Jakob 2018-11-09 10:15:17 +01:00
parent 978d439e92
commit 619fb23056
3 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
include/pybind11/stl.h
View File

@ -199,9 +199,9 @@ private:
public: public:
bool load(handle src, bool convert) { bool load(handle src, bool convert) {
if (!isinstance<list>(src)) if (!isinstance<sequence>(src))
return false; return false;
auto l = reinterpret_borrow<list>(src); auto l = reinterpret_borrow<sequence>(src);
if (!require_size(l.size())) if (!require_size(l.size()))
return false; return false;
size_t ctr = 0; size_t ctr = 0;

2
tests/test_stl.cpp
View File

@ -263,4 +263,6 @@ TEST_SUBMODULE(stl, m) {
return result; return result;
}, },
py::return_value_policy::take_ownership); py::return_value_policy::take_ownership);
m.def("array_cast_sequence", [](std::array<int, 3> x) { return x; });
} }

4
tests/test_stl.py
View File

@ -216,3 +216,7 @@ def test_stl_ownership():
assert len(r) == 1 assert len(r) == 1
del r del r
assert cstats.alive() == 0 assert cstats.alive() == 0
def test_array_cast_sequence():
assert m.array_cast_sequence((1, 2, 3)) == [1, 2, 3]