pybind/pybind11
1
0
Fork 0
mirror of https://github.com/pybind/pybind11.git synced 2026-05-11 17:00:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add a basic test for recarrays and complex dtypes

Browse Source
This commit is contained in:
Ivan Smirnov
2016-06-19 15:50:31 +01:00
parent 2488b32066
commit bb4015ded3
5 changed files with 79 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
example/example20.py Normal file
View File

@@ -0,0 +1,19 @@
#!/usr/bin/env python
from __future__ import print_function
import numpy as np
from example import create_rec_simple
def check_eq(arr, data, dtype):
np.testing.assert_equal(arr, np.array(data, dtype=dtype))
dtype = np.dtype({'names': ['x', 'y', 'z'],
'formats': ['?', 'u4', 'f4'],
'offsets': [0, 4, 8]})
base_dtype = np.dtype([('x', '?'), ('y', 'u4'), ('z', 'f4')])
arr = create_rec_simple(3)
assert arr.dtype == dtype
check_eq(arr, [(False, 0, 0.0), (True, 1, 1.5), (False, 2, 3.0)], dtype)
check_eq(arr, [(False, 0, 0.0), (True, 1, 1.5), (False, 2, 3.0)], base_dtype)