| Ivan Smirnov | bb4015d | 2016-06-19 15:50:31 +0100 | [diff] [blame] | 1 | #!/usr/bin/env python |
| 2 | from __future__ import print_function |
| 3 | |
| 4 | import numpy as np |
| Ivan Smirnov | 8502f54 | 2016-06-19 16:09:44 +0100 | [diff] [blame^] | 5 | from example import create_rec_simple, create_rec_packed |
| Ivan Smirnov | bb4015d | 2016-06-19 15:50:31 +0100 | [diff] [blame] | 6 | |
| 7 | |
| 8 | def check_eq(arr, data, dtype): |
| 9 | np.testing.assert_equal(arr, np.array(data, dtype=dtype)) |
| 10 | |
| Ivan Smirnov | 8502f54 | 2016-06-19 16:09:44 +0100 | [diff] [blame^] | 11 | simple_dtype = np.dtype({'names': ['x', 'y', 'z'], |
| 12 | 'formats': ['?', 'u4', 'f4'], |
| 13 | 'offsets': [0, 4, 8]}) |
| 14 | packed_dtype = np.dtype([('x', '?'), ('y', 'u4'), ('z', 'f4')]) |
| Ivan Smirnov | bb4015d | 2016-06-19 15:50:31 +0100 | [diff] [blame] | 15 | |
| Ivan Smirnov | 8502f54 | 2016-06-19 16:09:44 +0100 | [diff] [blame^] | 16 | for func, dtype in [(create_rec_simple, simple_dtype), (create_rec_packed, packed_dtype)]: |
| 17 | arr = func(3) |
| 18 | assert arr.dtype == dtype |
| 19 | check_eq(arr, [(False, 0, 0.0), (True, 1, 1.5), (False, 2, 3.0)], simple_dtype) |
| 20 | check_eq(arr, [(False, 0, 0.0), (True, 1, 1.5), (False, 2, 3.0)], packed_dtype) |
| Ivan Smirnov | 2e1565e | 2016-06-19 16:05:23 +0100 | [diff] [blame] | 21 | |
| Ivan Smirnov | 8502f54 | 2016-06-19 16:09:44 +0100 | [diff] [blame^] | 22 | arr = func(0) |
| 23 | assert arr.dtype == dtype |
| 24 | check_eq(arr, [], simple_dtype) |
| 25 | check_eq(arr, [], packed_dtype) |