test_numpy_dtypes.cpp revision 11986
111986Sandreas.sandberg@arm.com/* 211986Sandreas.sandberg@arm.com tests/test_numpy_dtypes.cpp -- Structured and compound NumPy dtypes 311986Sandreas.sandberg@arm.com 411986Sandreas.sandberg@arm.com Copyright (c) 2016 Ivan Smirnov 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#include <pybind11/numpy.h> 1211986Sandreas.sandberg@arm.com 1311986Sandreas.sandberg@arm.com#ifdef __GNUC__ 1411986Sandreas.sandberg@arm.com#define PYBIND11_PACKED(cls) cls __attribute__((__packed__)) 1511986Sandreas.sandberg@arm.com#else 1611986Sandreas.sandberg@arm.com#define PYBIND11_PACKED(cls) __pragma(pack(push, 1)) cls __pragma(pack(pop)) 1711986Sandreas.sandberg@arm.com#endif 1811986Sandreas.sandberg@arm.com 1911986Sandreas.sandberg@arm.comnamespace py = pybind11; 2011986Sandreas.sandberg@arm.com 2111986Sandreas.sandberg@arm.comstruct SimpleStruct { 2211986Sandreas.sandberg@arm.com bool x; 2311986Sandreas.sandberg@arm.com uint32_t y; 2411986Sandreas.sandberg@arm.com float z; 2511986Sandreas.sandberg@arm.com}; 2611986Sandreas.sandberg@arm.com 2711986Sandreas.sandberg@arm.comstd::ostream& operator<<(std::ostream& os, const SimpleStruct& v) { 2811986Sandreas.sandberg@arm.com return os << "s:" << v.x << "," << v.y << "," << v.z; 2911986Sandreas.sandberg@arm.com} 3011986Sandreas.sandberg@arm.com 3111986Sandreas.sandberg@arm.comPYBIND11_PACKED(struct PackedStruct { 3211986Sandreas.sandberg@arm.com bool x; 3311986Sandreas.sandberg@arm.com uint32_t y; 3411986Sandreas.sandberg@arm.com float z; 3511986Sandreas.sandberg@arm.com}); 3611986Sandreas.sandberg@arm.com 3711986Sandreas.sandberg@arm.comstd::ostream& operator<<(std::ostream& os, const PackedStruct& v) { 3811986Sandreas.sandberg@arm.com return os << "p:" << v.x << "," << v.y << "," << v.z; 3911986Sandreas.sandberg@arm.com} 4011986Sandreas.sandberg@arm.com 4111986Sandreas.sandberg@arm.comPYBIND11_PACKED(struct NestedStruct { 4211986Sandreas.sandberg@arm.com SimpleStruct a; 4311986Sandreas.sandberg@arm.com PackedStruct b; 4411986Sandreas.sandberg@arm.com}); 4511986Sandreas.sandberg@arm.com 4611986Sandreas.sandberg@arm.comstd::ostream& operator<<(std::ostream& os, const NestedStruct& v) { 4711986Sandreas.sandberg@arm.com return os << "n:a=" << v.a << ";b=" << v.b; 4811986Sandreas.sandberg@arm.com} 4911986Sandreas.sandberg@arm.com 5011986Sandreas.sandberg@arm.comstruct PartialStruct { 5111986Sandreas.sandberg@arm.com bool x; 5211986Sandreas.sandberg@arm.com uint32_t y; 5311986Sandreas.sandberg@arm.com float z; 5411986Sandreas.sandberg@arm.com uint64_t dummy2; 5511986Sandreas.sandberg@arm.com}; 5611986Sandreas.sandberg@arm.com 5711986Sandreas.sandberg@arm.comstruct PartialNestedStruct { 5811986Sandreas.sandberg@arm.com uint64_t dummy1; 5911986Sandreas.sandberg@arm.com PartialStruct a; 6011986Sandreas.sandberg@arm.com uint64_t dummy2; 6111986Sandreas.sandberg@arm.com}; 6211986Sandreas.sandberg@arm.com 6311986Sandreas.sandberg@arm.comstruct UnboundStruct { }; 6411986Sandreas.sandberg@arm.com 6511986Sandreas.sandberg@arm.comstruct StringStruct { 6611986Sandreas.sandberg@arm.com char a[3]; 6711986Sandreas.sandberg@arm.com std::array<char, 3> b; 6811986Sandreas.sandberg@arm.com}; 6911986Sandreas.sandberg@arm.com 7011986Sandreas.sandberg@arm.comPYBIND11_PACKED(struct StructWithUglyNames { 7111986Sandreas.sandberg@arm.com int8_t __x__; 7211986Sandreas.sandberg@arm.com uint64_t __y__; 7311986Sandreas.sandberg@arm.com}); 7411986Sandreas.sandberg@arm.com 7511986Sandreas.sandberg@arm.comenum class E1 : int64_t { A = -1, B = 1 }; 7611986Sandreas.sandberg@arm.comenum E2 : uint8_t { X = 1, Y = 2 }; 7711986Sandreas.sandberg@arm.com 7811986Sandreas.sandberg@arm.comPYBIND11_PACKED(struct EnumStruct { 7911986Sandreas.sandberg@arm.com E1 e1; 8011986Sandreas.sandberg@arm.com E2 e2; 8111986Sandreas.sandberg@arm.com}); 8211986Sandreas.sandberg@arm.com 8311986Sandreas.sandberg@arm.comstd::ostream& operator<<(std::ostream& os, const StringStruct& v) { 8411986Sandreas.sandberg@arm.com os << "a='"; 8511986Sandreas.sandberg@arm.com for (size_t i = 0; i < 3 && v.a[i]; i++) os << v.a[i]; 8611986Sandreas.sandberg@arm.com os << "',b='"; 8711986Sandreas.sandberg@arm.com for (size_t i = 0; i < 3 && v.b[i]; i++) os << v.b[i]; 8811986Sandreas.sandberg@arm.com return os << "'"; 8911986Sandreas.sandberg@arm.com} 9011986Sandreas.sandberg@arm.com 9111986Sandreas.sandberg@arm.comstd::ostream& operator<<(std::ostream& os, const EnumStruct& v) { 9211986Sandreas.sandberg@arm.com return os << "e1=" << (v.e1 == E1::A ? "A" : "B") << ",e2=" << (v.e2 == E2::X ? "X" : "Y"); 9311986Sandreas.sandberg@arm.com} 9411986Sandreas.sandberg@arm.com 9511986Sandreas.sandberg@arm.comtemplate <typename T> 9611986Sandreas.sandberg@arm.compy::array mkarray_via_buffer(size_t n) { 9711986Sandreas.sandberg@arm.com return py::array(py::buffer_info(nullptr, sizeof(T), 9811986Sandreas.sandberg@arm.com py::format_descriptor<T>::format(), 9911986Sandreas.sandberg@arm.com 1, { n }, { sizeof(T) })); 10011986Sandreas.sandberg@arm.com} 10111986Sandreas.sandberg@arm.com 10211986Sandreas.sandberg@arm.comtemplate <typename S> 10311986Sandreas.sandberg@arm.compy::array_t<S, 0> create_recarray(size_t n) { 10411986Sandreas.sandberg@arm.com auto arr = mkarray_via_buffer<S>(n); 10511986Sandreas.sandberg@arm.com auto req = arr.request(); 10611986Sandreas.sandberg@arm.com auto ptr = static_cast<S*>(req.ptr); 10711986Sandreas.sandberg@arm.com for (size_t i = 0; i < n; i++) { 10811986Sandreas.sandberg@arm.com ptr[i].x = i % 2 != 0; ptr[i].y = (uint32_t) i; ptr[i].z = (float) i * 1.5f; 10911986Sandreas.sandberg@arm.com } 11011986Sandreas.sandberg@arm.com return arr; 11111986Sandreas.sandberg@arm.com} 11211986Sandreas.sandberg@arm.com 11311986Sandreas.sandberg@arm.comstd::string get_format_unbound() { 11411986Sandreas.sandberg@arm.com return py::format_descriptor<UnboundStruct>::format(); 11511986Sandreas.sandberg@arm.com} 11611986Sandreas.sandberg@arm.com 11711986Sandreas.sandberg@arm.compy::array_t<NestedStruct, 0> create_nested(size_t n) { 11811986Sandreas.sandberg@arm.com auto arr = mkarray_via_buffer<NestedStruct>(n); 11911986Sandreas.sandberg@arm.com auto req = arr.request(); 12011986Sandreas.sandberg@arm.com auto ptr = static_cast<NestedStruct*>(req.ptr); 12111986Sandreas.sandberg@arm.com for (size_t i = 0; i < n; i++) { 12211986Sandreas.sandberg@arm.com ptr[i].a.x = i % 2 != 0; ptr[i].a.y = (uint32_t) i; ptr[i].a.z = (float) i * 1.5f; 12311986Sandreas.sandberg@arm.com ptr[i].b.x = (i + 1) % 2 != 0; ptr[i].b.y = (uint32_t) (i + 1); ptr[i].b.z = (float) (i + 1) * 1.5f; 12411986Sandreas.sandberg@arm.com } 12511986Sandreas.sandberg@arm.com return arr; 12611986Sandreas.sandberg@arm.com} 12711986Sandreas.sandberg@arm.com 12811986Sandreas.sandberg@arm.compy::array_t<PartialNestedStruct, 0> create_partial_nested(size_t n) { 12911986Sandreas.sandberg@arm.com auto arr = mkarray_via_buffer<PartialNestedStruct>(n); 13011986Sandreas.sandberg@arm.com auto req = arr.request(); 13111986Sandreas.sandberg@arm.com auto ptr = static_cast<PartialNestedStruct*>(req.ptr); 13211986Sandreas.sandberg@arm.com for (size_t i = 0; i < n; i++) { 13311986Sandreas.sandberg@arm.com ptr[i].a.x = i % 2 != 0; ptr[i].a.y = (uint32_t) i; ptr[i].a.z = (float) i * 1.5f; 13411986Sandreas.sandberg@arm.com } 13511986Sandreas.sandberg@arm.com return arr; 13611986Sandreas.sandberg@arm.com} 13711986Sandreas.sandberg@arm.com 13811986Sandreas.sandberg@arm.compy::array_t<StringStruct, 0> create_string_array(bool non_empty) { 13911986Sandreas.sandberg@arm.com auto arr = mkarray_via_buffer<StringStruct>(non_empty ? 4 : 0); 14011986Sandreas.sandberg@arm.com if (non_empty) { 14111986Sandreas.sandberg@arm.com auto req = arr.request(); 14211986Sandreas.sandberg@arm.com auto ptr = static_cast<StringStruct*>(req.ptr); 14311986Sandreas.sandberg@arm.com for (size_t i = 0; i < req.size * req.itemsize; i++) 14411986Sandreas.sandberg@arm.com static_cast<char*>(req.ptr)[i] = 0; 14511986Sandreas.sandberg@arm.com ptr[1].a[0] = 'a'; ptr[1].b[0] = 'a'; 14611986Sandreas.sandberg@arm.com ptr[2].a[0] = 'a'; ptr[2].b[0] = 'a'; 14711986Sandreas.sandberg@arm.com ptr[3].a[0] = 'a'; ptr[3].b[0] = 'a'; 14811986Sandreas.sandberg@arm.com 14911986Sandreas.sandberg@arm.com ptr[2].a[1] = 'b'; ptr[2].b[1] = 'b'; 15011986Sandreas.sandberg@arm.com ptr[3].a[1] = 'b'; ptr[3].b[1] = 'b'; 15111986Sandreas.sandberg@arm.com 15211986Sandreas.sandberg@arm.com ptr[3].a[2] = 'c'; ptr[3].b[2] = 'c'; 15311986Sandreas.sandberg@arm.com } 15411986Sandreas.sandberg@arm.com return arr; 15511986Sandreas.sandberg@arm.com} 15611986Sandreas.sandberg@arm.com 15711986Sandreas.sandberg@arm.compy::array_t<EnumStruct, 0> create_enum_array(size_t n) { 15811986Sandreas.sandberg@arm.com auto arr = mkarray_via_buffer<EnumStruct>(n); 15911986Sandreas.sandberg@arm.com auto ptr = (EnumStruct *) arr.mutable_data(); 16011986Sandreas.sandberg@arm.com for (size_t i = 0; i < n; i++) { 16111986Sandreas.sandberg@arm.com ptr[i].e1 = static_cast<E1>(-1 + ((int) i % 2) * 2); 16211986Sandreas.sandberg@arm.com ptr[i].e2 = static_cast<E2>(1 + (i % 2)); 16311986Sandreas.sandberg@arm.com } 16411986Sandreas.sandberg@arm.com return arr; 16511986Sandreas.sandberg@arm.com} 16611986Sandreas.sandberg@arm.com 16711986Sandreas.sandberg@arm.comtemplate <typename S> 16811986Sandreas.sandberg@arm.compy::list print_recarray(py::array_t<S, 0> arr) { 16911986Sandreas.sandberg@arm.com const auto req = arr.request(); 17011986Sandreas.sandberg@arm.com const auto ptr = static_cast<S*>(req.ptr); 17111986Sandreas.sandberg@arm.com auto l = py::list(); 17211986Sandreas.sandberg@arm.com for (size_t i = 0; i < req.size; i++) { 17311986Sandreas.sandberg@arm.com std::stringstream ss; 17411986Sandreas.sandberg@arm.com ss << ptr[i]; 17511986Sandreas.sandberg@arm.com l.append(py::str(ss.str())); 17611986Sandreas.sandberg@arm.com } 17711986Sandreas.sandberg@arm.com return l; 17811986Sandreas.sandberg@arm.com} 17911986Sandreas.sandberg@arm.com 18011986Sandreas.sandberg@arm.compy::list print_format_descriptors() { 18111986Sandreas.sandberg@arm.com const auto fmts = { 18211986Sandreas.sandberg@arm.com py::format_descriptor<SimpleStruct>::format(), 18311986Sandreas.sandberg@arm.com py::format_descriptor<PackedStruct>::format(), 18411986Sandreas.sandberg@arm.com py::format_descriptor<NestedStruct>::format(), 18511986Sandreas.sandberg@arm.com py::format_descriptor<PartialStruct>::format(), 18611986Sandreas.sandberg@arm.com py::format_descriptor<PartialNestedStruct>::format(), 18711986Sandreas.sandberg@arm.com py::format_descriptor<StringStruct>::format(), 18811986Sandreas.sandberg@arm.com py::format_descriptor<EnumStruct>::format() 18911986Sandreas.sandberg@arm.com }; 19011986Sandreas.sandberg@arm.com auto l = py::list(); 19111986Sandreas.sandberg@arm.com for (const auto &fmt : fmts) { 19211986Sandreas.sandberg@arm.com l.append(py::cast(fmt)); 19311986Sandreas.sandberg@arm.com } 19411986Sandreas.sandberg@arm.com return l; 19511986Sandreas.sandberg@arm.com} 19611986Sandreas.sandberg@arm.com 19711986Sandreas.sandberg@arm.compy::list print_dtypes() { 19811986Sandreas.sandberg@arm.com const auto dtypes = { 19911986Sandreas.sandberg@arm.com py::str(py::dtype::of<SimpleStruct>()), 20011986Sandreas.sandberg@arm.com py::str(py::dtype::of<PackedStruct>()), 20111986Sandreas.sandberg@arm.com py::str(py::dtype::of<NestedStruct>()), 20211986Sandreas.sandberg@arm.com py::str(py::dtype::of<PartialStruct>()), 20311986Sandreas.sandberg@arm.com py::str(py::dtype::of<PartialNestedStruct>()), 20411986Sandreas.sandberg@arm.com py::str(py::dtype::of<StringStruct>()), 20511986Sandreas.sandberg@arm.com py::str(py::dtype::of<EnumStruct>()), 20611986Sandreas.sandberg@arm.com py::str(py::dtype::of<StructWithUglyNames>()) 20711986Sandreas.sandberg@arm.com }; 20811986Sandreas.sandberg@arm.com auto l = py::list(); 20911986Sandreas.sandberg@arm.com for (const auto &s : dtypes) { 21011986Sandreas.sandberg@arm.com l.append(s); 21111986Sandreas.sandberg@arm.com } 21211986Sandreas.sandberg@arm.com return l; 21311986Sandreas.sandberg@arm.com} 21411986Sandreas.sandberg@arm.com 21511986Sandreas.sandberg@arm.compy::array_t<int32_t, 0> test_array_ctors(int i) { 21611986Sandreas.sandberg@arm.com using arr_t = py::array_t<int32_t, 0>; 21711986Sandreas.sandberg@arm.com 21811986Sandreas.sandberg@arm.com std::vector<int32_t> data { 1, 2, 3, 4, 5, 6 }; 21911986Sandreas.sandberg@arm.com std::vector<size_t> shape { 3, 2 }; 22011986Sandreas.sandberg@arm.com std::vector<size_t> strides { 8, 4 }; 22111986Sandreas.sandberg@arm.com 22211986Sandreas.sandberg@arm.com auto ptr = data.data(); 22311986Sandreas.sandberg@arm.com auto vptr = (void *) ptr; 22411986Sandreas.sandberg@arm.com auto dtype = py::dtype("int32"); 22511986Sandreas.sandberg@arm.com 22611986Sandreas.sandberg@arm.com py::buffer_info buf_ndim1(vptr, 4, "i", 6); 22711986Sandreas.sandberg@arm.com py::buffer_info buf_ndim1_null(nullptr, 4, "i", 6); 22811986Sandreas.sandberg@arm.com py::buffer_info buf_ndim2(vptr, 4, "i", 2, shape, strides); 22911986Sandreas.sandberg@arm.com py::buffer_info buf_ndim2_null(nullptr, 4, "i", 2, shape, strides); 23011986Sandreas.sandberg@arm.com 23111986Sandreas.sandberg@arm.com auto fill = [](py::array arr) { 23211986Sandreas.sandberg@arm.com auto req = arr.request(); 23311986Sandreas.sandberg@arm.com for (int i = 0; i < 6; i++) ((int32_t *) req.ptr)[i] = i + 1; 23411986Sandreas.sandberg@arm.com return arr; 23511986Sandreas.sandberg@arm.com }; 23611986Sandreas.sandberg@arm.com 23711986Sandreas.sandberg@arm.com switch (i) { 23811986Sandreas.sandberg@arm.com // shape: (3, 2) 23911986Sandreas.sandberg@arm.com case 10: return arr_t(shape, strides, ptr); 24011986Sandreas.sandberg@arm.com case 11: return py::array(shape, strides, ptr); 24111986Sandreas.sandberg@arm.com case 12: return py::array(dtype, shape, strides, vptr); 24211986Sandreas.sandberg@arm.com case 13: return arr_t(shape, ptr); 24311986Sandreas.sandberg@arm.com case 14: return py::array(shape, ptr); 24411986Sandreas.sandberg@arm.com case 15: return py::array(dtype, shape, vptr); 24511986Sandreas.sandberg@arm.com case 16: return arr_t(buf_ndim2); 24611986Sandreas.sandberg@arm.com case 17: return py::array(buf_ndim2); 24711986Sandreas.sandberg@arm.com // shape: (3, 2) - post-fill 24811986Sandreas.sandberg@arm.com case 20: return fill(arr_t(shape, strides)); 24911986Sandreas.sandberg@arm.com case 21: return py::array(shape, strides, ptr); // can't have nullptr due to templated ctor 25011986Sandreas.sandberg@arm.com case 22: return fill(py::array(dtype, shape, strides)); 25111986Sandreas.sandberg@arm.com case 23: return fill(arr_t(shape)); 25211986Sandreas.sandberg@arm.com case 24: return py::array(shape, ptr); // can't have nullptr due to templated ctor 25311986Sandreas.sandberg@arm.com case 25: return fill(py::array(dtype, shape)); 25411986Sandreas.sandberg@arm.com case 26: return fill(arr_t(buf_ndim2_null)); 25511986Sandreas.sandberg@arm.com case 27: return fill(py::array(buf_ndim2_null)); 25611986Sandreas.sandberg@arm.com // shape: (6, ) 25711986Sandreas.sandberg@arm.com case 30: return arr_t(6, ptr); 25811986Sandreas.sandberg@arm.com case 31: return py::array(6, ptr); 25911986Sandreas.sandberg@arm.com case 32: return py::array(dtype, 6, vptr); 26011986Sandreas.sandberg@arm.com case 33: return arr_t(buf_ndim1); 26111986Sandreas.sandberg@arm.com case 34: return py::array(buf_ndim1); 26211986Sandreas.sandberg@arm.com // shape: (6, ) 26311986Sandreas.sandberg@arm.com case 40: return fill(arr_t(6)); 26411986Sandreas.sandberg@arm.com case 41: return py::array(6, ptr); // can't have nullptr due to templated ctor 26511986Sandreas.sandberg@arm.com case 42: return fill(py::array(dtype, 6)); 26611986Sandreas.sandberg@arm.com case 43: return fill(arr_t(buf_ndim1_null)); 26711986Sandreas.sandberg@arm.com case 44: return fill(py::array(buf_ndim1_null)); 26811986Sandreas.sandberg@arm.com } 26911986Sandreas.sandberg@arm.com return arr_t(); 27011986Sandreas.sandberg@arm.com} 27111986Sandreas.sandberg@arm.com 27211986Sandreas.sandberg@arm.compy::list test_dtype_ctors() { 27311986Sandreas.sandberg@arm.com py::list list; 27411986Sandreas.sandberg@arm.com list.append(py::dtype("int32")); 27511986Sandreas.sandberg@arm.com list.append(py::dtype(std::string("float64"))); 27611986Sandreas.sandberg@arm.com list.append(py::dtype::from_args(py::str("bool"))); 27711986Sandreas.sandberg@arm.com py::list names, offsets, formats; 27811986Sandreas.sandberg@arm.com py::dict dict; 27911986Sandreas.sandberg@arm.com names.append(py::str("a")); names.append(py::str("b")); dict["names"] = names; 28011986Sandreas.sandberg@arm.com offsets.append(py::int_(1)); offsets.append(py::int_(10)); dict["offsets"] = offsets; 28111986Sandreas.sandberg@arm.com formats.append(py::dtype("int32")); formats.append(py::dtype("float64")); dict["formats"] = formats; 28211986Sandreas.sandberg@arm.com dict["itemsize"] = py::int_(20); 28311986Sandreas.sandberg@arm.com list.append(py::dtype::from_args(dict)); 28411986Sandreas.sandberg@arm.com list.append(py::dtype(names, formats, offsets, 20)); 28511986Sandreas.sandberg@arm.com list.append(py::dtype(py::buffer_info((void *) 0, sizeof(unsigned int), "I", 1))); 28611986Sandreas.sandberg@arm.com list.append(py::dtype(py::buffer_info((void *) 0, 0, "T{i:a:f:b:}", 1))); 28711986Sandreas.sandberg@arm.com return list; 28811986Sandreas.sandberg@arm.com} 28911986Sandreas.sandberg@arm.com 29011986Sandreas.sandberg@arm.comstruct TrailingPaddingStruct { 29111986Sandreas.sandberg@arm.com int32_t a; 29211986Sandreas.sandberg@arm.com char b; 29311986Sandreas.sandberg@arm.com}; 29411986Sandreas.sandberg@arm.com 29511986Sandreas.sandberg@arm.compy::dtype trailing_padding_dtype() { 29611986Sandreas.sandberg@arm.com return py::dtype::of<TrailingPaddingStruct>(); 29711986Sandreas.sandberg@arm.com} 29811986Sandreas.sandberg@arm.com 29911986Sandreas.sandberg@arm.compy::dtype buffer_to_dtype(py::buffer& buf) { 30011986Sandreas.sandberg@arm.com return py::dtype(buf.request()); 30111986Sandreas.sandberg@arm.com} 30211986Sandreas.sandberg@arm.com 30311986Sandreas.sandberg@arm.compy::list test_dtype_methods() { 30411986Sandreas.sandberg@arm.com py::list list; 30511986Sandreas.sandberg@arm.com auto dt1 = py::dtype::of<int32_t>(); 30611986Sandreas.sandberg@arm.com auto dt2 = py::dtype::of<SimpleStruct>(); 30711986Sandreas.sandberg@arm.com list.append(dt1); list.append(dt2); 30811986Sandreas.sandberg@arm.com list.append(py::bool_(dt1.has_fields())); list.append(py::bool_(dt2.has_fields())); 30911986Sandreas.sandberg@arm.com list.append(py::int_(dt1.itemsize())); list.append(py::int_(dt2.itemsize())); 31011986Sandreas.sandberg@arm.com return list; 31111986Sandreas.sandberg@arm.com} 31211986Sandreas.sandberg@arm.com 31311986Sandreas.sandberg@arm.comtest_initializer numpy_dtypes([](py::module &m) { 31411986Sandreas.sandberg@arm.com try { 31511986Sandreas.sandberg@arm.com py::module::import("numpy"); 31611986Sandreas.sandberg@arm.com } catch (...) { 31711986Sandreas.sandberg@arm.com return; 31811986Sandreas.sandberg@arm.com } 31911986Sandreas.sandberg@arm.com 32011986Sandreas.sandberg@arm.com // typeinfo may be registered before the dtype descriptor for scalar casts to work... 32111986Sandreas.sandberg@arm.com py::class_<SimpleStruct>(m, "SimpleStruct"); 32211986Sandreas.sandberg@arm.com 32311986Sandreas.sandberg@arm.com PYBIND11_NUMPY_DTYPE(SimpleStruct, x, y, z); 32411986Sandreas.sandberg@arm.com PYBIND11_NUMPY_DTYPE(PackedStruct, x, y, z); 32511986Sandreas.sandberg@arm.com PYBIND11_NUMPY_DTYPE(NestedStruct, a, b); 32611986Sandreas.sandberg@arm.com PYBIND11_NUMPY_DTYPE(PartialStruct, x, y, z); 32711986Sandreas.sandberg@arm.com PYBIND11_NUMPY_DTYPE(PartialNestedStruct, a); 32811986Sandreas.sandberg@arm.com PYBIND11_NUMPY_DTYPE(StringStruct, a, b); 32911986Sandreas.sandberg@arm.com PYBIND11_NUMPY_DTYPE(EnumStruct, e1, e2); 33011986Sandreas.sandberg@arm.com PYBIND11_NUMPY_DTYPE(TrailingPaddingStruct, a, b); 33111986Sandreas.sandberg@arm.com 33211986Sandreas.sandberg@arm.com // ... or after 33311986Sandreas.sandberg@arm.com py::class_<PackedStruct>(m, "PackedStruct"); 33411986Sandreas.sandberg@arm.com 33511986Sandreas.sandberg@arm.com PYBIND11_NUMPY_DTYPE_EX(StructWithUglyNames, __x__, "x", __y__, "y"); 33611986Sandreas.sandberg@arm.com 33711986Sandreas.sandberg@arm.com m.def("create_rec_simple", &create_recarray<SimpleStruct>); 33811986Sandreas.sandberg@arm.com m.def("create_rec_packed", &create_recarray<PackedStruct>); 33911986Sandreas.sandberg@arm.com m.def("create_rec_nested", &create_nested); 34011986Sandreas.sandberg@arm.com m.def("create_rec_partial", &create_recarray<PartialStruct>); 34111986Sandreas.sandberg@arm.com m.def("create_rec_partial_nested", &create_partial_nested); 34211986Sandreas.sandberg@arm.com m.def("print_format_descriptors", &print_format_descriptors); 34311986Sandreas.sandberg@arm.com m.def("print_rec_simple", &print_recarray<SimpleStruct>); 34411986Sandreas.sandberg@arm.com m.def("print_rec_packed", &print_recarray<PackedStruct>); 34511986Sandreas.sandberg@arm.com m.def("print_rec_nested", &print_recarray<NestedStruct>); 34611986Sandreas.sandberg@arm.com m.def("print_dtypes", &print_dtypes); 34711986Sandreas.sandberg@arm.com m.def("get_format_unbound", &get_format_unbound); 34811986Sandreas.sandberg@arm.com m.def("create_string_array", &create_string_array); 34911986Sandreas.sandberg@arm.com m.def("print_string_array", &print_recarray<StringStruct>); 35011986Sandreas.sandberg@arm.com m.def("create_enum_array", &create_enum_array); 35111986Sandreas.sandberg@arm.com m.def("print_enum_array", &print_recarray<EnumStruct>); 35211986Sandreas.sandberg@arm.com m.def("test_array_ctors", &test_array_ctors); 35311986Sandreas.sandberg@arm.com m.def("test_dtype_ctors", &test_dtype_ctors); 35411986Sandreas.sandberg@arm.com m.def("test_dtype_methods", &test_dtype_methods); 35511986Sandreas.sandberg@arm.com m.def("trailing_padding_dtype", &trailing_padding_dtype); 35611986Sandreas.sandberg@arm.com m.def("buffer_to_dtype", &buffer_to_dtype); 35711986Sandreas.sandberg@arm.com m.def("f_simple", [](SimpleStruct s) { return s.y * 10; }); 35811986Sandreas.sandberg@arm.com m.def("f_packed", [](PackedStruct s) { return s.y * 10; }); 35911986Sandreas.sandberg@arm.com m.def("f_nested", [](NestedStruct s) { return s.a.y * 10; }); 36011986Sandreas.sandberg@arm.com m.def("register_dtype", []() { PYBIND11_NUMPY_DTYPE(SimpleStruct, x, y, z); }); 36111986Sandreas.sandberg@arm.com}); 36211986Sandreas.sandberg@arm.com 36311986Sandreas.sandberg@arm.com#undef PYBIND11_PACKED 364