test_numpy_dtypes.cpp (12037:d28054ac6ec9) test_numpy_dtypes.cpp (12391:ceeca8b41e4b)
1/*
2 tests/test_numpy_dtypes.cpp -- Structured and compound NumPy dtypes
3
4 Copyright (c) 2016 Ivan Smirnov
5
6 All rights reserved. Use of this source code is governed by a
7 BSD-style license that can be found in the LICENSE file.
8*/

--- 56 unchanged lines hidden (view full) ---

65
66struct UnboundStruct { };
67
68struct StringStruct {
69 char a[3];
70 std::array<char, 3> b;
71};
72
1/*
2 tests/test_numpy_dtypes.cpp -- Structured and compound NumPy dtypes
3
4 Copyright (c) 2016 Ivan Smirnov
5
6 All rights reserved. Use of this source code is governed by a
7 BSD-style license that can be found in the LICENSE file.
8*/

--- 56 unchanged lines hidden (view full) ---

65
66struct UnboundStruct { };
67
68struct StringStruct {
69 char a[3];
70 std::array<char, 3> b;
71};
72
73struct ComplexStruct {
74 std::complex<float> cflt;
75 std::complex<double> cdbl;
76};
77
78std::ostream& operator<<(std::ostream& os, const ComplexStruct& v) {
79 return os << "c:" << v.cflt << "," << v.cdbl;
80}
81
82struct ArrayStruct {
83 char a[3][4];
84 int32_t b[2];
85 std::array<uint8_t, 3> c;
86 std::array<float, 2> d[4];
87};
88
73PYBIND11_PACKED(struct StructWithUglyNames {
74 int8_t __x__;
75 uint64_t __y__;
76});
77
78enum class E1 : int64_t { A = -1, B = 1 };
79enum E2 : uint8_t { X = 1, Y = 2 };
80

--- 5 unchanged lines hidden (view full) ---

86std::ostream& operator<<(std::ostream& os, const StringStruct& v) {
87 os << "a='";
88 for (size_t i = 0; i < 3 && v.a[i]; i++) os << v.a[i];
89 os << "',b='";
90 for (size_t i = 0; i < 3 && v.b[i]; i++) os << v.b[i];
91 return os << "'";
92}
93
89PYBIND11_PACKED(struct StructWithUglyNames {
90 int8_t __x__;
91 uint64_t __y__;
92});
93
94enum class E1 : int64_t { A = -1, B = 1 };
95enum E2 : uint8_t { X = 1, Y = 2 };
96

--- 5 unchanged lines hidden (view full) ---

102std::ostream& operator<<(std::ostream& os, const StringStruct& v) {
103 os << "a='";
104 for (size_t i = 0; i < 3 && v.a[i]; i++) os << v.a[i];
105 os << "',b='";
106 for (size_t i = 0; i < 3 && v.b[i]; i++) os << v.b[i];
107 return os << "'";
108}
109
110std::ostream& operator<<(std::ostream& os, const ArrayStruct& v) {
111 os << "a={";
112 for (int i = 0; i < 3; i++) {
113 if (i > 0)
114 os << ',';
115 os << '{';
116 for (int j = 0; j < 3; j++)
117 os << v.a[i][j] << ',';
118 os << v.a[i][3] << '}';
119 }
120 os << "},b={" << v.b[0] << ',' << v.b[1];
121 os << "},c={" << int(v.c[0]) << ',' << int(v.c[1]) << ',' << int(v.c[2]);
122 os << "},d={";
123 for (int i = 0; i < 4; i++) {
124 if (i > 0)
125 os << ',';
126 os << '{' << v.d[i][0] << ',' << v.d[i][1] << '}';
127 }
128 return os << '}';
129}
130
94std::ostream& operator<<(std::ostream& os, const EnumStruct& v) {
95 return os << "e1=" << (v.e1 == E1::A ? "A" : "B") << ",e2=" << (v.e2 == E2::X ? "X" : "Y");
96}
97
98template <typename T>
99py::array mkarray_via_buffer(size_t n) {
100 return py::array(py::buffer_info(nullptr, sizeof(T),
101 py::format_descriptor<T>::format(),

--- 12 unchanged lines hidden (view full) ---

114 auto req = arr.request();
115 auto ptr = static_cast<S*>(req.ptr);
116 for (size_t i = 0; i < n; i++) {
117 SET_TEST_VALS(ptr[i], i);
118 }
119 return arr;
120}
121
131std::ostream& operator<<(std::ostream& os, const EnumStruct& v) {
132 return os << "e1=" << (v.e1 == E1::A ? "A" : "B") << ",e2=" << (v.e2 == E2::X ? "X" : "Y");
133}
134
135template <typename T>
136py::array mkarray_via_buffer(size_t n) {
137 return py::array(py::buffer_info(nullptr, sizeof(T),
138 py::format_descriptor<T>::format(),

--- 12 unchanged lines hidden (view full) ---

151 auto req = arr.request();
152 auto ptr = static_cast<S*>(req.ptr);
153 for (size_t i = 0; i < n; i++) {
154 SET_TEST_VALS(ptr[i], i);
155 }
156 return arr;
157}
158
122std::string get_format_unbound() {
123 return py::format_descriptor<UnboundStruct>::format();
124}
125
126py::array_t<NestedStruct, 0> create_nested(size_t n) {
127 auto arr = mkarray_via_buffer<NestedStruct>(n);
128 auto req = arr.request();
129 auto ptr = static_cast<NestedStruct*>(req.ptr);
130 for (size_t i = 0; i < n; i++) {
131 SET_TEST_VALS(ptr[i].a, i);
132 SET_TEST_VALS(ptr[i].b, i + 1);
133 }
134 return arr;
135}
136
137py::array_t<PartialNestedStruct, 0> create_partial_nested(size_t n) {
138 auto arr = mkarray_via_buffer<PartialNestedStruct>(n);
139 auto req = arr.request();
140 auto ptr = static_cast<PartialNestedStruct*>(req.ptr);
141 for (size_t i = 0; i < n; i++) {
142 SET_TEST_VALS(ptr[i].a, i);
143 }
144 return arr;
145}
146
147py::array_t<StringStruct, 0> create_string_array(bool non_empty) {
148 auto arr = mkarray_via_buffer<StringStruct>(non_empty ? 4 : 0);
149 if (non_empty) {
150 auto req = arr.request();
151 auto ptr = static_cast<StringStruct*>(req.ptr);
152 for (size_t i = 0; i < req.size * req.itemsize; i++)
153 static_cast<char*>(req.ptr)[i] = 0;
154 ptr[1].a[0] = 'a'; ptr[1].b[0] = 'a';
155 ptr[2].a[0] = 'a'; ptr[2].b[0] = 'a';
156 ptr[3].a[0] = 'a'; ptr[3].b[0] = 'a';
157
158 ptr[2].a[1] = 'b'; ptr[2].b[1] = 'b';
159 ptr[3].a[1] = 'b'; ptr[3].b[1] = 'b';
160
161 ptr[3].a[2] = 'c'; ptr[3].b[2] = 'c';
162 }
163 return arr;
164}
165
166py::array_t<EnumStruct, 0> create_enum_array(size_t n) {
167 auto arr = mkarray_via_buffer<EnumStruct>(n);
168 auto ptr = (EnumStruct *) arr.mutable_data();
169 for (size_t i = 0; i < n; i++) {
170 ptr[i].e1 = static_cast<E1>(-1 + ((int) i % 2) * 2);
171 ptr[i].e2 = static_cast<E2>(1 + (i % 2));
172 }
173 return arr;
174}
175
176template <typename S>
177py::list print_recarray(py::array_t<S, 0> arr) {
178 const auto req = arr.request();
179 const auto ptr = static_cast<S*>(req.ptr);
180 auto l = py::list();
159template <typename S>
160py::list print_recarray(py::array_t<S, 0> arr) {
161 const auto req = arr.request();
162 const auto ptr = static_cast<S*>(req.ptr);
163 auto l = py::list();
181 for (size_t i = 0; i < req.size; i++) {
164 for (ssize_t i = 0; i < req.size; i++) {
182 std::stringstream ss;
183 ss << ptr[i];
184 l.append(py::str(ss.str()));
185 }
186 return l;
187}
188
165 std::stringstream ss;
166 ss << ptr[i];
167 l.append(py::str(ss.str()));
168 }
169 return l;
170}
171
189py::list print_format_descriptors() {
190 const auto fmts = {
191 py::format_descriptor<SimpleStruct>::format(),
192 py::format_descriptor<PackedStruct>::format(),
193 py::format_descriptor<NestedStruct>::format(),
194 py::format_descriptor<PartialStruct>::format(),
195 py::format_descriptor<PartialNestedStruct>::format(),
196 py::format_descriptor<StringStruct>::format(),
197 py::format_descriptor<EnumStruct>::format()
198 };
199 auto l = py::list();
200 for (const auto &fmt : fmts) {
201 l.append(py::cast(fmt));
202 }
203 return l;
204}
205
206py::list print_dtypes() {
207 const auto dtypes = {
208 py::str(py::dtype::of<SimpleStruct>()),
209 py::str(py::dtype::of<PackedStruct>()),
210 py::str(py::dtype::of<NestedStruct>()),
211 py::str(py::dtype::of<PartialStruct>()),
212 py::str(py::dtype::of<PartialNestedStruct>()),
213 py::str(py::dtype::of<StringStruct>()),
214 py::str(py::dtype::of<EnumStruct>()),
215 py::str(py::dtype::of<StructWithUglyNames>())
216 };
217 auto l = py::list();
218 for (const auto &s : dtypes) {
219 l.append(s);
220 }
221 return l;
222}
223
224py::array_t<int32_t, 0> test_array_ctors(int i) {
225 using arr_t = py::array_t<int32_t, 0>;
226
227 std::vector<int32_t> data { 1, 2, 3, 4, 5, 6 };
172py::array_t<int32_t, 0> test_array_ctors(int i) {
173 using arr_t = py::array_t<int32_t, 0>;
174
175 std::vector<int32_t> data { 1, 2, 3, 4, 5, 6 };
228 std::vector shape { 3, 2 };
229 std::vector strides { 8, 4 };
176 std::vector<ssize_t> shape { 3, 2 };
177 std::vector<ssize_t> strides { 8, 4 };
230
231 auto ptr = data.data();
232 auto vptr = (void *) ptr;
233 auto dtype = py::dtype("int32");
234
235 py::buffer_info buf_ndim1(vptr, 4, "i", 6);
236 py::buffer_info buf_ndim1_null(nullptr, 4, "i", 6);
237 py::buffer_info buf_ndim2(vptr, 4, "i", 2, shape, strides);

--- 53 unchanged lines hidden (view full) ---

291 dict["itemsize"] = py::int_(20);
292 list.append(py::dtype::from_args(dict));
293 list.append(py::dtype(names, formats, offsets, 20));
294 list.append(py::dtype(py::buffer_info((void *) 0, sizeof(unsigned int), "I", 1)));
295 list.append(py::dtype(py::buffer_info((void *) 0, 0, "T{i:a:f:b:}", 1)));
296 return list;
297}
298
178
179 auto ptr = data.data();
180 auto vptr = (void *) ptr;
181 auto dtype = py::dtype("int32");
182
183 py::buffer_info buf_ndim1(vptr, 4, "i", 6);
184 py::buffer_info buf_ndim1_null(nullptr, 4, "i", 6);
185 py::buffer_info buf_ndim2(vptr, 4, "i", 2, shape, strides);

--- 53 unchanged lines hidden (view full) ---

239 dict["itemsize"] = py::int_(20);
240 list.append(py::dtype::from_args(dict));
241 list.append(py::dtype(names, formats, offsets, 20));
242 list.append(py::dtype(py::buffer_info((void *) 0, sizeof(unsigned int), "I", 1)));
243 list.append(py::dtype(py::buffer_info((void *) 0, 0, "T{i:a:f:b:}", 1)));
244 return list;
245}
246
299struct TrailingPaddingStruct {
300 int32_t a;
301 char b;
302};
247TEST_SUBMODULE(numpy_dtypes, m) {
248 try { py::module::import("numpy"); }
249 catch (...) { return; }
303
250
304py::dtype trailing_padding_dtype() {
305 return py::dtype::of<TrailingPaddingStruct>();
306}
307
308py::dtype buffer_to_dtype(py::buffer& buf) {
309 return py::dtype(buf.request());
310}
311
312py::list test_dtype_methods() {
313 py::list list;
314 auto dt1 = py::dtype::of<int32_t>();
315 auto dt2 = py::dtype::of<SimpleStruct>();
316 list.append(dt1); list.append(dt2);
317 list.append(py::bool_(dt1.has_fields())); list.append(py::bool_(dt2.has_fields()));
318 list.append(py::int_(dt1.itemsize())); list.append(py::int_(dt2.itemsize()));
319 return list;
320}
321
322struct CompareStruct {
323 bool x;
324 uint32_t y;
325 float z;
326};
327
328py::list test_compare_buffer_info() {
329 py::list list;
330 list.append(py::bool_(py::detail::compare_buffer_info<float>::compare(py::buffer_info(nullptr, sizeof(float), "f", 1))));
331 list.append(py::bool_(py::detail::compare_buffer_info<unsigned>::compare(py::buffer_info(nullptr, sizeof(int), "I", 1))));
332 list.append(py::bool_(py::detail::compare_buffer_info<long>::compare(py::buffer_info(nullptr, sizeof(long), "l", 1))));
333 list.append(py::bool_(py::detail::compare_buffer_info<long>::compare(py::buffer_info(nullptr, sizeof(long), sizeof(long) == sizeof(int) ? "i" : "q", 1))));
334 list.append(py::bool_(py::detail::compare_buffer_info<CompareStruct>::compare(py::buffer_info(nullptr, sizeof(CompareStruct), "T{?:x:3xI:y:f:z:}", 1))));
335 return list;
336}
337
338test_initializer numpy_dtypes([](py::module &m) {
339 try {
340 py::module::import("numpy");
341 } catch (...) {
342 return;
343 }
344
345 // typeinfo may be registered before the dtype descriptor for scalar casts to work...
346 py::class_<SimpleStruct>(m, "SimpleStruct");
347
348 PYBIND11_NUMPY_DTYPE(SimpleStruct, bool_, uint_, float_, ldbl_);
349 PYBIND11_NUMPY_DTYPE(PackedStruct, bool_, uint_, float_, ldbl_);
350 PYBIND11_NUMPY_DTYPE(NestedStruct, a, b);
351 PYBIND11_NUMPY_DTYPE(PartialStruct, bool_, uint_, float_, ldbl_);
352 PYBIND11_NUMPY_DTYPE(PartialNestedStruct, a);
353 PYBIND11_NUMPY_DTYPE(StringStruct, a, b);
251 // typeinfo may be registered before the dtype descriptor for scalar casts to work...
252 py::class_<SimpleStruct>(m, "SimpleStruct");
253
254 PYBIND11_NUMPY_DTYPE(SimpleStruct, bool_, uint_, float_, ldbl_);
255 PYBIND11_NUMPY_DTYPE(PackedStruct, bool_, uint_, float_, ldbl_);
256 PYBIND11_NUMPY_DTYPE(NestedStruct, a, b);
257 PYBIND11_NUMPY_DTYPE(PartialStruct, bool_, uint_, float_, ldbl_);
258 PYBIND11_NUMPY_DTYPE(PartialNestedStruct, a);
259 PYBIND11_NUMPY_DTYPE(StringStruct, a, b);
260 PYBIND11_NUMPY_DTYPE(ArrayStruct, a, b, c, d);
354 PYBIND11_NUMPY_DTYPE(EnumStruct, e1, e2);
261 PYBIND11_NUMPY_DTYPE(EnumStruct, e1, e2);
355 PYBIND11_NUMPY_DTYPE(TrailingPaddingStruct, a, b);
356 PYBIND11_NUMPY_DTYPE(CompareStruct, x, y, z);
262 PYBIND11_NUMPY_DTYPE(ComplexStruct, cflt, cdbl);
357
358 // ... or after
359 py::class_<PackedStruct>(m, "PackedStruct");
360
361 PYBIND11_NUMPY_DTYPE_EX(StructWithUglyNames, __x__, "x", __y__, "y");
362
363 // If uncommented, this should produce a static_assert failure telling the user that the struct
364 // is not a POD type
365// struct NotPOD { std::string v; NotPOD() : v("hi") {}; };
366// PYBIND11_NUMPY_DTYPE(NotPOD, v);
367
263
264 // ... or after
265 py::class_<PackedStruct>(m, "PackedStruct");
266
267 PYBIND11_NUMPY_DTYPE_EX(StructWithUglyNames, __x__, "x", __y__, "y");
268
269 // If uncommented, this should produce a static_assert failure telling the user that the struct
270 // is not a POD type
271// struct NotPOD { std::string v; NotPOD() : v("hi") {}; };
272// PYBIND11_NUMPY_DTYPE(NotPOD, v);
273
274 // test_recarray, test_scalar_conversion
368 m.def("create_rec_simple", &create_recarray<SimpleStruct>);
369 m.def("create_rec_packed", &create_recarray<PackedStruct>);
275 m.def("create_rec_simple", &create_recarray<SimpleStruct>);
276 m.def("create_rec_packed", &create_recarray<PackedStruct>);
370 m.def("create_rec_nested", &create_nested);
277 m.def("create_rec_nested", [](size_t n) { // test_signature
278 py::array_t<NestedStruct, 0> arr = mkarray_via_buffer<NestedStruct>(n);
279 auto req = arr.request();
280 auto ptr = static_cast<NestedStruct*>(req.ptr);
281 for (size_t i = 0; i < n; i++) {
282 SET_TEST_VALS(ptr[i].a, i);
283 SET_TEST_VALS(ptr[i].b, i + 1);
284 }
285 return arr;
286 });
371 m.def("create_rec_partial", &create_recarray<PartialStruct>);
287 m.def("create_rec_partial", &create_recarray<PartialStruct>);
372 m.def("create_rec_partial_nested", &create_partial_nested);
373 m.def("print_format_descriptors", &print_format_descriptors);
288 m.def("create_rec_partial_nested", [](size_t n) {
289 py::array_t<PartialNestedStruct, 0> arr = mkarray_via_buffer<PartialNestedStruct>(n);
290 auto req = arr.request();
291 auto ptr = static_cast<PartialNestedStruct*>(req.ptr);
292 for (size_t i = 0; i < n; i++) {
293 SET_TEST_VALS(ptr[i].a, i);
294 }
295 return arr;
296 });
374 m.def("print_rec_simple", &print_recarray<SimpleStruct>);
375 m.def("print_rec_packed", &print_recarray<PackedStruct>);
376 m.def("print_rec_nested", &print_recarray<NestedStruct>);
297 m.def("print_rec_simple", &print_recarray<SimpleStruct>);
298 m.def("print_rec_packed", &print_recarray<PackedStruct>);
299 m.def("print_rec_nested", &print_recarray<NestedStruct>);
377 m.def("print_dtypes", &print_dtypes);
378 m.def("get_format_unbound", &get_format_unbound);
379 m.def("create_string_array", &create_string_array);
300
301 // test_format_descriptors
302 m.def("get_format_unbound", []() { return py::format_descriptor<UnboundStruct>::format(); });
303 m.def("print_format_descriptors", []() {
304 py::list l;
305 for (const auto &fmt : {
306 py::format_descriptor<SimpleStruct>::format(),
307 py::format_descriptor<PackedStruct>::format(),
308 py::format_descriptor<NestedStruct>::format(),
309 py::format_descriptor<PartialStruct>::format(),
310 py::format_descriptor<PartialNestedStruct>::format(),
311 py::format_descriptor<StringStruct>::format(),
312 py::format_descriptor<ArrayStruct>::format(),
313 py::format_descriptor<EnumStruct>::format(),
314 py::format_descriptor<ComplexStruct>::format()
315 }) {
316 l.append(py::cast(fmt));
317 }
318 return l;
319 });
320
321 // test_dtype
322 m.def("print_dtypes", []() {
323 py::list l;
324 for (const py::handle &d : {
325 py::dtype::of<SimpleStruct>(),
326 py::dtype::of<PackedStruct>(),
327 py::dtype::of<NestedStruct>(),
328 py::dtype::of<PartialStruct>(),
329 py::dtype::of<PartialNestedStruct>(),
330 py::dtype::of<StringStruct>(),
331 py::dtype::of<ArrayStruct>(),
332 py::dtype::of<EnumStruct>(),
333 py::dtype::of<StructWithUglyNames>(),
334 py::dtype::of<ComplexStruct>()
335 })
336 l.append(py::str(d));
337 return l;
338 });
339 m.def("test_dtype_ctors", &test_dtype_ctors);
340 m.def("test_dtype_methods", []() {
341 py::list list;
342 auto dt1 = py::dtype::of<int32_t>();
343 auto dt2 = py::dtype::of<SimpleStruct>();
344 list.append(dt1); list.append(dt2);
345 list.append(py::bool_(dt1.has_fields())); list.append(py::bool_(dt2.has_fields()));
346 list.append(py::int_(dt1.itemsize())); list.append(py::int_(dt2.itemsize()));
347 return list;
348 });
349 struct TrailingPaddingStruct {
350 int32_t a;
351 char b;
352 };
353 PYBIND11_NUMPY_DTYPE(TrailingPaddingStruct, a, b);
354 m.def("trailing_padding_dtype", []() { return py::dtype::of<TrailingPaddingStruct>(); });
355
356 // test_string_array
357 m.def("create_string_array", [](bool non_empty) {
358 py::array_t<StringStruct, 0> arr = mkarray_via_buffer<StringStruct>(non_empty ? 4 : 0);
359 if (non_empty) {
360 auto req = arr.request();
361 auto ptr = static_cast<StringStruct*>(req.ptr);
362 for (ssize_t i = 0; i < req.size * req.itemsize; i++)
363 static_cast<char*>(req.ptr)[i] = 0;
364 ptr[1].a[0] = 'a'; ptr[1].b[0] = 'a';
365 ptr[2].a[0] = 'a'; ptr[2].b[0] = 'a';
366 ptr[3].a[0] = 'a'; ptr[3].b[0] = 'a';
367
368 ptr[2].a[1] = 'b'; ptr[2].b[1] = 'b';
369 ptr[3].a[1] = 'b'; ptr[3].b[1] = 'b';
370
371 ptr[3].a[2] = 'c'; ptr[3].b[2] = 'c';
372 }
373 return arr;
374 });
380 m.def("print_string_array", &print_recarray<StringStruct>);
375 m.def("print_string_array", &print_recarray<StringStruct>);
381 m.def("create_enum_array", &create_enum_array);
376
377 // test_array_array
378 m.def("create_array_array", [](size_t n) {
379 py::array_t<ArrayStruct, 0> arr = mkarray_via_buffer<ArrayStruct>(n);
380 auto ptr = (ArrayStruct *) arr.mutable_data();
381 for (size_t i = 0; i < n; i++) {
382 for (size_t j = 0; j < 3; j++)
383 for (size_t k = 0; k < 4; k++)
384 ptr[i].a[j][k] = char('A' + (i * 100 + j * 10 + k) % 26);
385 for (size_t j = 0; j < 2; j++)
386 ptr[i].b[j] = int32_t(i * 1000 + j);
387 for (size_t j = 0; j < 3; j++)
388 ptr[i].c[j] = uint8_t(i * 10 + j);
389 for (size_t j = 0; j < 4; j++)
390 for (size_t k = 0; k < 2; k++)
391 ptr[i].d[j][k] = float(i) * 100.0f + float(j) * 10.0f + float(k);
392 }
393 return arr;
394 });
395 m.def("print_array_array", &print_recarray<ArrayStruct>);
396
397 // test_enum_array
398 m.def("create_enum_array", [](size_t n) {
399 py::array_t<EnumStruct, 0> arr = mkarray_via_buffer<EnumStruct>(n);
400 auto ptr = (EnumStruct *) arr.mutable_data();
401 for (size_t i = 0; i < n; i++) {
402 ptr[i].e1 = static_cast<E1>(-1 + ((int) i % 2) * 2);
403 ptr[i].e2 = static_cast<E2>(1 + (i % 2));
404 }
405 return arr;
406 });
382 m.def("print_enum_array", &print_recarray<EnumStruct>);
407 m.def("print_enum_array", &print_recarray<EnumStruct>);
408
409 // test_complex_array
410 m.def("create_complex_array", [](size_t n) {
411 py::array_t<ComplexStruct, 0> arr = mkarray_via_buffer<ComplexStruct>(n);
412 auto ptr = (ComplexStruct *) arr.mutable_data();
413 for (size_t i = 0; i < n; i++) {
414 ptr[i].cflt.real(float(i));
415 ptr[i].cflt.imag(float(i) + 0.25f);
416 ptr[i].cdbl.real(double(i) + 0.5);
417 ptr[i].cdbl.imag(double(i) + 0.75);
418 }
419 return arr;
420 });
421 m.def("print_complex_array", &print_recarray<ComplexStruct>);
422
423 // test_array_constructors
383 m.def("test_array_ctors", &test_array_ctors);
424 m.def("test_array_ctors", &test_array_ctors);
384 m.def("test_dtype_ctors", &test_dtype_ctors);
385 m.def("test_dtype_methods", &test_dtype_methods);
386 m.def("compare_buffer_info", &test_compare_buffer_info);
387 m.def("trailing_padding_dtype", &trailing_padding_dtype);
388 m.def("buffer_to_dtype", &buffer_to_dtype);
425
426 // test_compare_buffer_info
427 struct CompareStruct {
428 bool x;
429 uint32_t y;
430 float z;
431 };
432 PYBIND11_NUMPY_DTYPE(CompareStruct, x, y, z);
433 m.def("compare_buffer_info", []() {
434 py::list list;
435 list.append(py::bool_(py::detail::compare_buffer_info<float>::compare(py::buffer_info(nullptr, sizeof(float), "f", 1))));
436 list.append(py::bool_(py::detail::compare_buffer_info<unsigned>::compare(py::buffer_info(nullptr, sizeof(int), "I", 1))));
437 list.append(py::bool_(py::detail::compare_buffer_info<long>::compare(py::buffer_info(nullptr, sizeof(long), "l", 1))));
438 list.append(py::bool_(py::detail::compare_buffer_info<long>::compare(py::buffer_info(nullptr, sizeof(long), sizeof(long) == sizeof(int) ? "i" : "q", 1))));
439 list.append(py::bool_(py::detail::compare_buffer_info<CompareStruct>::compare(py::buffer_info(nullptr, sizeof(CompareStruct), "T{?:x:3xI:y:f:z:}", 1))));
440 return list;
441 });
442 m.def("buffer_to_dtype", [](py::buffer& buf) { return py::dtype(buf.request()); });
443
444 // test_scalar_conversion
389 m.def("f_simple", [](SimpleStruct s) { return s.uint_ * 10; });
390 m.def("f_packed", [](PackedStruct s) { return s.uint_ * 10; });
391 m.def("f_nested", [](NestedStruct s) { return s.a.uint_ * 10; });
445 m.def("f_simple", [](SimpleStruct s) { return s.uint_ * 10; });
446 m.def("f_packed", [](PackedStruct s) { return s.uint_ * 10; });
447 m.def("f_nested", [](NestedStruct s) { return s.a.uint_ * 10; });
392 m.def("register_dtype", []() { PYBIND11_NUMPY_DTYPE(SimpleStruct, bool_, uint_, float_, ldbl_); });
393});
394
448
395#undef PYBIND11_PACKED
449 // test_register_dtype
450 m.def("register_dtype", []() { PYBIND11_NUMPY_DTYPE(SimpleStruct, bool_, uint_, float_, ldbl_); });
451}