pybind11/tests/test_numpy_array.py

26 def arr():
70 def test_index_offset(arr, args, ret):
71     assert m.index_at(arr, *args) == ret
72     assert m.index_at_t(arr, *args) == ret
73     assert m.offset_at(arr, *args) == ret * arr.dtype.itemsize
74     assert m.offset_at_t(arr, *args) == ret * arr.dtype.itemsize
77 def test_dim_check_fail(arr):
81             func(arr, 1, 2, 3)
90 def test_data(arr, args, ret):
92     assert all(m.data_t(arr, *args) == ret)
93     assert all(m.data(arr, *args)[(0 if byteorder == 'little' else 1)::2] == ret)
94     assert all(m.data(arr, *args)[(1 if byteorder == 'little' else 0)::2] == 0)
98 def test_at_fail(arr, dim):
101             func(arr, *([0] * dim))
105 def test_at(arr):
106     assert m.at_t(arr, 0, 2) == 3
107     assert m.at_t(arr, 1, 0) == 4
109     assert all(m.mutate_at_t(arr, 0, 2).ravel() == [1, 2, 4, 4, 5, 6])
110     assert all(m.mutate_at_t(arr, 1, 0).ravel() == [1, 2, 4, 5, 5, 6])
113 def test_mutate_readonly(arr):
114     arr.flags.writeable = False
117             func(arr, *args)
121 def test_mutate_data(arr):
122     assert all(m.mutate_data(arr).ravel() == [2, 4, 6, 8, 10, 12])
123     assert all(m.mutate_data(arr).ravel() == [4, 8, 12, 16, 20, 24])
124     assert all(m.mutate_data(arr, 1).ravel() == [4, 8, 12, 32, 40, 48])
125     assert all(m.mutate_data(arr, 0, 1).ravel() == [4, 16, 24, 64, 80, 96])
126     assert all(m.mutate_data(arr, 1, 2).ravel() == [4, 16, 24, 64, 80, 192])
128     assert all(m.mutate_data_t(arr).ravel() == [5, 17, 25, 65, 81, 193])
129     assert all(m.mutate_data_t(arr).ravel() == [6, 18, 26, 66, 82, 194])
130     assert all(m.mutate_data_t(arr, 1).ravel() == [6, 18, 26, 67, 83, 195])
131     assert all(m.mutate_data_t(arr, 0, 1).ravel() == [6, 19, 27, 68, 84, 196])
132     assert all(m.mutate_data_t(arr, 1, 2).ravel() == [6, 19, 27, 68, 84, 197])
135 def test_bounds_check(arr):
139             func(arr, 2, 0)
142             func(arr, 0, 4)