pybind11/tests/test_numpy_array.py

7     import numpy as np
27     return np.array([[1, 2, 3], [4, 5, 6]], '=u2')
31     a = np.array(0, 'f8')
47     a = np.array([[1, 2, 3], [4, 5, 6]], 'u2').view()
175         if LooseVersion(np.__version__) >= LooseVersion("1.14.0"):
179         assert np.all(a == b)
186     a1 = np.array([1, 2], dtype=np.int16)
191     a1 = np.array([[1, 2], [3, 4]], dtype=np.float32, order='F')
196     a1 = np.array([[1, 2], [3, 4]], dtype=np.float32, order='C')
201     a1 = np.random.random((4, 4, 4))
223         assert np.all(ac_view_1 == np.array([1, 2], dtype=np.int32))
248     m.function_taking_uint64(np.uint64(123))
252     assert m.isinstance_untyped(np.array([1, 2, 3]), "not an array")
253     assert m.isinstance_typed(np.array([1.0, 2.0, 3.0]))
260     assert defaults["array"].dtype == np.array([]).dtype
261     assert defaults["array_t<int32>"].dtype == np.int32
262     assert defaults["array_t<double>"].dtype == np.float64
266         np.testing.assert_array_equal(a, [1, 2, 3])
267     assert results["array"].dtype == np.int_
268     assert results["array_t<int32>"].dtype == np.int32
269     assert results["array_t<double>"].dtype == np.float64
274     assert m.overloaded(np.array([1], dtype='float64')) == 'double'
275     assert m.overloaded(np.array([1], dtype='float32')) == 'float'
276     assert m.overloaded(np.array([1], dtype='ushort')) == 'unsigned short'
277     assert m.overloaded(np.array([1], dtype='intc')) == 'int'
278     assert m.overloaded(np.array([1], dtype='longlong')) == 'long long'
279     assert m.overloaded(np.array([1], dtype='complex')) == 'double complex'
280     assert m.overloaded(np.array([1], dtype='csingle')) == 'float complex'
283     assert m.overloaded(np.array([1], dtype='uint8')) == 'double'
300     assert m.overloaded2(np.array([1], dtype='float64')) == 'double'
301     assert m.overloaded2(np.array([1], dtype='float32')) == 'float'
302     assert m.overloaded2(np.array([1], dtype='complex64')) == 'float complex'
303     assert m.overloaded2(np.array([1], dtype='complex128')) == 'double complex'
304     assert m.overloaded2(np.array([1], dtype='float32')) == 'float'
306     assert m.overloaded3(np.array([1], dtype='float64')) == 'double'
307     assert m.overloaded3(np.array([1], dtype='intc')) == 'int'
316         m.overloaded3(np.array([1], dtype='uintc'))
317     assert msg(excinfo.value) == expected_exc + repr(np.array([1], dtype='uint32'))
319         m.overloaded3(np.array([1], dtype='float32'))
320     assert msg(excinfo.value) == expected_exc + repr(np.array([1.], dtype='float32'))
322         m.overloaded3(np.array([1], dtype='complex'))
323     assert msg(excinfo.value) == expected_exc + repr(np.array([1. + 0.j]))
326     assert m.overloaded4(np.array([1], dtype='double')) == 'double'
327     assert m.overloaded4(np.array([1], dtype='longlong')) == 'long long'
331     assert m.overloaded4(np.array([1], dtype='float32')) == 'double'
332     assert m.overloaded4(np.array([1], dtype='short')) == 'long long'
334     assert m.overloaded5(np.array([1], dtype='double')) == 'double'
335     assert m.overloaded5(np.array([1], dtype='uintc')) == 'unsigned int'
336     assert m.overloaded5(np.array([1], dtype='float32')) == 'unsigned int'
343     assert m.issue685(np.array([97, 98, 99], dtype='b')) == "array"
348     z1 = np.array([[1, 2], [3, 4]], dtype='float64')
350     assert np.all(z1 == [[11, 12], [13, 14]])
353         m.proxy_add2(np.array([1., 2, 3]), 5.0)
356     expect_c = np.ndarray(shape=(3, 3, 3), buffer=np.array(range(3, 30)), dtype='int')
357     assert np.all(m.proxy_init3(3.0) == expect_c)
358     expect_f = np.transpose(expect_c)
359     assert np.all(m.proxy_init3F(3.0) == expect_f)
361     assert m.proxy_squared_L2_norm(np.array(range(6))) == 55
362     assert m.proxy_squared_L2_norm(np.array(range(6), dtype="float64")) == 55
369     z1 = np.array([[1, 2], [3, 4]], dtype='float64')
371     assert np.all(z1 == [[11, 12], [13, 14]])
373     expect_c = np.ndarray(shape=(3, 3, 3), buffer=np.array(range(3, 30)), dtype='int')
374     assert np.all(m.proxy_init3_dyn(3.0) == expect_c)
402     a = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9], dtype='float64')
405     assert(np.all(a == [[1, 2, 3], [4, 5, 6], [7, 8, 9]]))
430     assert(np.all(a == 42.))
435     a = m.index_using_ellipsis(np.zeros((5, 6, 7)))
442     dtype = np.dtype(np.float_)
443     a = np.array([1], dtype=dtype)