test_numpy_array.py revision 14299
111986Sandreas.sandberg@arm.comimport pytest 212391Sjason@lowepower.comfrom pybind11_tests import numpy_array as m 312037Sandreas.sandberg@arm.com 412037Sandreas.sandberg@arm.compytestmark = pytest.requires_numpy 511986Sandreas.sandberg@arm.com 611986Sandreas.sandberg@arm.comwith pytest.suppress(ImportError): 711986Sandreas.sandberg@arm.com import numpy as np 811986Sandreas.sandberg@arm.com 911986Sandreas.sandberg@arm.com 1014299Sbbruce@ucdavis.edudef test_dtypes(): 1114299Sbbruce@ucdavis.edu # See issue #1328. 1214299Sbbruce@ucdavis.edu # - Platform-dependent sizes. 1314299Sbbruce@ucdavis.edu for size_check in m.get_platform_dtype_size_checks(): 1414299Sbbruce@ucdavis.edu print(size_check) 1514299Sbbruce@ucdavis.edu assert size_check.size_cpp == size_check.size_numpy, size_check 1614299Sbbruce@ucdavis.edu # - Concrete sizes. 1714299Sbbruce@ucdavis.edu for check in m.get_concrete_dtype_checks(): 1814299Sbbruce@ucdavis.edu print(check) 1914299Sbbruce@ucdavis.edu assert check.numpy == check.pybind11, check 2014299Sbbruce@ucdavis.edu if check.numpy.num != check.pybind11.num: 2114299Sbbruce@ucdavis.edu print("NOTE: typenum mismatch for {}: {} != {}".format( 2214299Sbbruce@ucdavis.edu check, check.numpy.num, check.pybind11.num)) 2314299Sbbruce@ucdavis.edu 2414299Sbbruce@ucdavis.edu 2511986Sandreas.sandberg@arm.com@pytest.fixture(scope='function') 2611986Sandreas.sandberg@arm.comdef arr(): 2712037Sandreas.sandberg@arm.com return np.array([[1, 2, 3], [4, 5, 6]], '=u2') 2811986Sandreas.sandberg@arm.com 2911986Sandreas.sandberg@arm.com 3011986Sandreas.sandberg@arm.comdef test_array_attributes(): 3111986Sandreas.sandberg@arm.com a = np.array(0, 'f8') 3212391Sjason@lowepower.com assert m.ndim(a) == 0 3312391Sjason@lowepower.com assert all(m.shape(a) == []) 3412391Sjason@lowepower.com assert all(m.strides(a) == []) 3511986Sandreas.sandberg@arm.com with pytest.raises(IndexError) as excinfo: 3612391Sjason@lowepower.com m.shape(a, 0) 3711986Sandreas.sandberg@arm.com assert str(excinfo.value) == 'invalid axis: 0 (ndim = 0)' 3811986Sandreas.sandberg@arm.com with pytest.raises(IndexError) as excinfo: 3912391Sjason@lowepower.com m.strides(a, 0) 4011986Sandreas.sandberg@arm.com assert str(excinfo.value) == 'invalid axis: 0 (ndim = 0)' 4112391Sjason@lowepower.com assert m.writeable(a) 4212391Sjason@lowepower.com assert m.size(a) == 1 4312391Sjason@lowepower.com assert m.itemsize(a) == 8 4412391Sjason@lowepower.com assert m.nbytes(a) == 8 4512391Sjason@lowepower.com assert m.owndata(a) 4611986Sandreas.sandberg@arm.com 4711986Sandreas.sandberg@arm.com a = np.array([[1, 2, 3], [4, 5, 6]], 'u2').view() 4811986Sandreas.sandberg@arm.com a.flags.writeable = False 4912391Sjason@lowepower.com assert m.ndim(a) == 2 5012391Sjason@lowepower.com assert all(m.shape(a) == [2, 3]) 5112391Sjason@lowepower.com assert m.shape(a, 0) == 2 5212391Sjason@lowepower.com assert m.shape(a, 1) == 3 5312391Sjason@lowepower.com assert all(m.strides(a) == [6, 2]) 5412391Sjason@lowepower.com assert m.strides(a, 0) == 6 5512391Sjason@lowepower.com assert m.strides(a, 1) == 2 5611986Sandreas.sandberg@arm.com with pytest.raises(IndexError) as excinfo: 5712391Sjason@lowepower.com m.shape(a, 2) 5811986Sandreas.sandberg@arm.com assert str(excinfo.value) == 'invalid axis: 2 (ndim = 2)' 5911986Sandreas.sandberg@arm.com with pytest.raises(IndexError) as excinfo: 6012391Sjason@lowepower.com m.strides(a, 2) 6111986Sandreas.sandberg@arm.com assert str(excinfo.value) == 'invalid axis: 2 (ndim = 2)' 6212391Sjason@lowepower.com assert not m.writeable(a) 6312391Sjason@lowepower.com assert m.size(a) == 6 6412391Sjason@lowepower.com assert m.itemsize(a) == 2 6512391Sjason@lowepower.com assert m.nbytes(a) == 12 6612391Sjason@lowepower.com assert not m.owndata(a) 6711986Sandreas.sandberg@arm.com 6811986Sandreas.sandberg@arm.com 6911986Sandreas.sandberg@arm.com@pytest.mark.parametrize('args, ret', [([], 0), ([0], 0), ([1], 3), ([0, 1], 1), ([1, 2], 5)]) 7011986Sandreas.sandberg@arm.comdef test_index_offset(arr, args, ret): 7112391Sjason@lowepower.com assert m.index_at(arr, *args) == ret 7212391Sjason@lowepower.com assert m.index_at_t(arr, *args) == ret 7312391Sjason@lowepower.com assert m.offset_at(arr, *args) == ret * arr.dtype.itemsize 7412391Sjason@lowepower.com assert m.offset_at_t(arr, *args) == ret * arr.dtype.itemsize 7511986Sandreas.sandberg@arm.com 7611986Sandreas.sandberg@arm.com 7711986Sandreas.sandberg@arm.comdef test_dim_check_fail(arr): 7812391Sjason@lowepower.com for func in (m.index_at, m.index_at_t, m.offset_at, m.offset_at_t, m.data, m.data_t, 7912391Sjason@lowepower.com m.mutate_data, m.mutate_data_t): 8011986Sandreas.sandberg@arm.com with pytest.raises(IndexError) as excinfo: 8111986Sandreas.sandberg@arm.com func(arr, 1, 2, 3) 8211986Sandreas.sandberg@arm.com assert str(excinfo.value) == 'too many indices for an array: 3 (ndim = 2)' 8311986Sandreas.sandberg@arm.com 8411986Sandreas.sandberg@arm.com 8511986Sandreas.sandberg@arm.com@pytest.mark.parametrize('args, ret', 8611986Sandreas.sandberg@arm.com [([], [1, 2, 3, 4, 5, 6]), 8711986Sandreas.sandberg@arm.com ([1], [4, 5, 6]), 8811986Sandreas.sandberg@arm.com ([0, 1], [2, 3, 4, 5, 6]), 8911986Sandreas.sandberg@arm.com ([1, 2], [6])]) 9011986Sandreas.sandberg@arm.comdef test_data(arr, args, ret): 9112037Sandreas.sandberg@arm.com from sys import byteorder 9212391Sjason@lowepower.com assert all(m.data_t(arr, *args) == ret) 9312391Sjason@lowepower.com assert all(m.data(arr, *args)[(0 if byteorder == 'little' else 1)::2] == ret) 9412391Sjason@lowepower.com assert all(m.data(arr, *args)[(1 if byteorder == 'little' else 0)::2] == 0) 9512391Sjason@lowepower.com 9612391Sjason@lowepower.com 9712391Sjason@lowepower.com@pytest.mark.parametrize('dim', [0, 1, 3]) 9812391Sjason@lowepower.comdef test_at_fail(arr, dim): 9912391Sjason@lowepower.com for func in m.at_t, m.mutate_at_t: 10012391Sjason@lowepower.com with pytest.raises(IndexError) as excinfo: 10112391Sjason@lowepower.com func(arr, *([0] * dim)) 10212391Sjason@lowepower.com assert str(excinfo.value) == 'index dimension mismatch: {} (ndim = 2)'.format(dim) 10312391Sjason@lowepower.com 10412391Sjason@lowepower.com 10512391Sjason@lowepower.comdef test_at(arr): 10612391Sjason@lowepower.com assert m.at_t(arr, 0, 2) == 3 10712391Sjason@lowepower.com assert m.at_t(arr, 1, 0) == 4 10812391Sjason@lowepower.com 10912391Sjason@lowepower.com assert all(m.mutate_at_t(arr, 0, 2).ravel() == [1, 2, 4, 4, 5, 6]) 11012391Sjason@lowepower.com assert all(m.mutate_at_t(arr, 1, 0).ravel() == [1, 2, 4, 5, 5, 6]) 11111986Sandreas.sandberg@arm.com 11211986Sandreas.sandberg@arm.com 11311986Sandreas.sandberg@arm.comdef test_mutate_readonly(arr): 11411986Sandreas.sandberg@arm.com arr.flags.writeable = False 11512391Sjason@lowepower.com for func, args in (m.mutate_data, ()), (m.mutate_data_t, ()), (m.mutate_at_t, (0, 0)): 11612037Sandreas.sandberg@arm.com with pytest.raises(ValueError) as excinfo: 11711986Sandreas.sandberg@arm.com func(arr, *args) 11811986Sandreas.sandberg@arm.com assert str(excinfo.value) == 'array is not writeable' 11911986Sandreas.sandberg@arm.com 12011986Sandreas.sandberg@arm.com 12112391Sjason@lowepower.comdef test_mutate_data(arr): 12212391Sjason@lowepower.com assert all(m.mutate_data(arr).ravel() == [2, 4, 6, 8, 10, 12]) 12312391Sjason@lowepower.com assert all(m.mutate_data(arr).ravel() == [4, 8, 12, 16, 20, 24]) 12412391Sjason@lowepower.com assert all(m.mutate_data(arr, 1).ravel() == [4, 8, 12, 32, 40, 48]) 12512391Sjason@lowepower.com assert all(m.mutate_data(arr, 0, 1).ravel() == [4, 16, 24, 64, 80, 96]) 12612391Sjason@lowepower.com assert all(m.mutate_data(arr, 1, 2).ravel() == [4, 16, 24, 64, 80, 192]) 12711986Sandreas.sandberg@arm.com 12812391Sjason@lowepower.com assert all(m.mutate_data_t(arr).ravel() == [5, 17, 25, 65, 81, 193]) 12912391Sjason@lowepower.com assert all(m.mutate_data_t(arr).ravel() == [6, 18, 26, 66, 82, 194]) 13012391Sjason@lowepower.com assert all(m.mutate_data_t(arr, 1).ravel() == [6, 18, 26, 67, 83, 195]) 13112391Sjason@lowepower.com assert all(m.mutate_data_t(arr, 0, 1).ravel() == [6, 19, 27, 68, 84, 196]) 13212391Sjason@lowepower.com assert all(m.mutate_data_t(arr, 1, 2).ravel() == [6, 19, 27, 68, 84, 197]) 13311986Sandreas.sandberg@arm.com 13411986Sandreas.sandberg@arm.com 13511986Sandreas.sandberg@arm.comdef test_bounds_check(arr): 13612391Sjason@lowepower.com for func in (m.index_at, m.index_at_t, m.data, m.data_t, 13712391Sjason@lowepower.com m.mutate_data, m.mutate_data_t, m.at_t, m.mutate_at_t): 13811986Sandreas.sandberg@arm.com with pytest.raises(IndexError) as excinfo: 13911986Sandreas.sandberg@arm.com func(arr, 2, 0) 14011986Sandreas.sandberg@arm.com assert str(excinfo.value) == 'index 2 is out of bounds for axis 0 with size 2' 14111986Sandreas.sandberg@arm.com with pytest.raises(IndexError) as excinfo: 14211986Sandreas.sandberg@arm.com func(arr, 0, 4) 14311986Sandreas.sandberg@arm.com assert str(excinfo.value) == 'index 4 is out of bounds for axis 1 with size 3' 14411986Sandreas.sandberg@arm.com 14511986Sandreas.sandberg@arm.com 14611986Sandreas.sandberg@arm.comdef test_make_c_f_array(): 14712391Sjason@lowepower.com assert m.make_c_array().flags.c_contiguous 14812391Sjason@lowepower.com assert not m.make_c_array().flags.f_contiguous 14912391Sjason@lowepower.com assert m.make_f_array().flags.f_contiguous 15012391Sjason@lowepower.com assert not m.make_f_array().flags.c_contiguous 15111986Sandreas.sandberg@arm.com 15211986Sandreas.sandberg@arm.com 15314299Sbbruce@ucdavis.edudef test_make_empty_shaped_array(): 15414299Sbbruce@ucdavis.edu m.make_empty_shaped_array() 15514299Sbbruce@ucdavis.edu 15614299Sbbruce@ucdavis.edu # empty shape means numpy scalar, PEP 3118 15714299Sbbruce@ucdavis.edu assert m.scalar_int().ndim == 0 15814299Sbbruce@ucdavis.edu assert m.scalar_int().shape == () 15914299Sbbruce@ucdavis.edu assert m.scalar_int() == 42 16014299Sbbruce@ucdavis.edu 16114299Sbbruce@ucdavis.edu 16211986Sandreas.sandberg@arm.comdef test_wrap(): 16312037Sandreas.sandberg@arm.com def assert_references(a, b, base=None): 16414299Sbbruce@ucdavis.edu from distutils.version import LooseVersion 16512037Sandreas.sandberg@arm.com if base is None: 16612037Sandreas.sandberg@arm.com base = a 16711986Sandreas.sandberg@arm.com assert a is not b 16811986Sandreas.sandberg@arm.com assert a.__array_interface__['data'][0] == b.__array_interface__['data'][0] 16911986Sandreas.sandberg@arm.com assert a.shape == b.shape 17011986Sandreas.sandberg@arm.com assert a.strides == b.strides 17111986Sandreas.sandberg@arm.com assert a.flags.c_contiguous == b.flags.c_contiguous 17211986Sandreas.sandberg@arm.com assert a.flags.f_contiguous == b.flags.f_contiguous 17311986Sandreas.sandberg@arm.com assert a.flags.writeable == b.flags.writeable 17411986Sandreas.sandberg@arm.com assert a.flags.aligned == b.flags.aligned 17514299Sbbruce@ucdavis.edu if LooseVersion(np.__version__) >= LooseVersion("1.14.0"): 17614299Sbbruce@ucdavis.edu assert a.flags.writebackifcopy == b.flags.writebackifcopy 17714299Sbbruce@ucdavis.edu else: 17814299Sbbruce@ucdavis.edu assert a.flags.updateifcopy == b.flags.updateifcopy 17911986Sandreas.sandberg@arm.com assert np.all(a == b) 18011986Sandreas.sandberg@arm.com assert not b.flags.owndata 18112037Sandreas.sandberg@arm.com assert b.base is base 18211986Sandreas.sandberg@arm.com if a.flags.writeable and a.ndim == 2: 18311986Sandreas.sandberg@arm.com a[0, 0] = 1234 18411986Sandreas.sandberg@arm.com assert b[0, 0] == 1234 18511986Sandreas.sandberg@arm.com 18611986Sandreas.sandberg@arm.com a1 = np.array([1, 2], dtype=np.int16) 18711986Sandreas.sandberg@arm.com assert a1.flags.owndata and a1.base is None 18812391Sjason@lowepower.com a2 = m.wrap(a1) 18911986Sandreas.sandberg@arm.com assert_references(a1, a2) 19011986Sandreas.sandberg@arm.com 19111986Sandreas.sandberg@arm.com a1 = np.array([[1, 2], [3, 4]], dtype=np.float32, order='F') 19211986Sandreas.sandberg@arm.com assert a1.flags.owndata and a1.base is None 19312391Sjason@lowepower.com a2 = m.wrap(a1) 19411986Sandreas.sandberg@arm.com assert_references(a1, a2) 19511986Sandreas.sandberg@arm.com 19611986Sandreas.sandberg@arm.com a1 = np.array([[1, 2], [3, 4]], dtype=np.float32, order='C') 19711986Sandreas.sandberg@arm.com a1.flags.writeable = False 19812391Sjason@lowepower.com a2 = m.wrap(a1) 19911986Sandreas.sandberg@arm.com assert_references(a1, a2) 20011986Sandreas.sandberg@arm.com 20111986Sandreas.sandberg@arm.com a1 = np.random.random((4, 4, 4)) 20212391Sjason@lowepower.com a2 = m.wrap(a1) 20311986Sandreas.sandberg@arm.com assert_references(a1, a2) 20411986Sandreas.sandberg@arm.com 20512037Sandreas.sandberg@arm.com a1t = a1.transpose() 20612391Sjason@lowepower.com a2 = m.wrap(a1t) 20712037Sandreas.sandberg@arm.com assert_references(a1t, a2, a1) 20811986Sandreas.sandberg@arm.com 20912037Sandreas.sandberg@arm.com a1d = a1.diagonal() 21012391Sjason@lowepower.com a2 = m.wrap(a1d) 21112037Sandreas.sandberg@arm.com assert_references(a1d, a2, a1) 21211986Sandreas.sandberg@arm.com 21312391Sjason@lowepower.com a1m = a1[::-1, ::-1, ::-1] 21412391Sjason@lowepower.com a2 = m.wrap(a1m) 21512391Sjason@lowepower.com assert_references(a1m, a2, a1) 21612391Sjason@lowepower.com 21711986Sandreas.sandberg@arm.com 21811986Sandreas.sandberg@arm.comdef test_numpy_view(capture): 21911986Sandreas.sandberg@arm.com with capture: 22012391Sjason@lowepower.com ac = m.ArrayClass() 22111986Sandreas.sandberg@arm.com ac_view_1 = ac.numpy_view() 22211986Sandreas.sandberg@arm.com ac_view_2 = ac.numpy_view() 22311986Sandreas.sandberg@arm.com assert np.all(ac_view_1 == np.array([1, 2], dtype=np.int32)) 22411986Sandreas.sandberg@arm.com del ac 22512037Sandreas.sandberg@arm.com pytest.gc_collect() 22611986Sandreas.sandberg@arm.com assert capture == """ 22711986Sandreas.sandberg@arm.com ArrayClass() 22811986Sandreas.sandberg@arm.com ArrayClass::numpy_view() 22911986Sandreas.sandberg@arm.com ArrayClass::numpy_view() 23011986Sandreas.sandberg@arm.com """ 23111986Sandreas.sandberg@arm.com ac_view_1[0] = 4 23211986Sandreas.sandberg@arm.com ac_view_1[1] = 3 23311986Sandreas.sandberg@arm.com assert ac_view_2[0] == 4 23411986Sandreas.sandberg@arm.com assert ac_view_2[1] == 3 23511986Sandreas.sandberg@arm.com with capture: 23611986Sandreas.sandberg@arm.com del ac_view_1 23711986Sandreas.sandberg@arm.com del ac_view_2 23812037Sandreas.sandberg@arm.com pytest.gc_collect() 23912037Sandreas.sandberg@arm.com pytest.gc_collect() 24011986Sandreas.sandberg@arm.com assert capture == """ 24111986Sandreas.sandberg@arm.com ~ArrayClass() 24211986Sandreas.sandberg@arm.com """ 24311986Sandreas.sandberg@arm.com 24411986Sandreas.sandberg@arm.com 24512037Sandreas.sandberg@arm.com@pytest.unsupported_on_pypy 24611986Sandreas.sandberg@arm.comdef test_cast_numpy_int64_to_uint64(): 24712391Sjason@lowepower.com m.function_taking_uint64(123) 24812391Sjason@lowepower.com m.function_taking_uint64(np.uint64(123)) 24911986Sandreas.sandberg@arm.com 25011986Sandreas.sandberg@arm.com 25111986Sandreas.sandberg@arm.comdef test_isinstance(): 25212391Sjason@lowepower.com assert m.isinstance_untyped(np.array([1, 2, 3]), "not an array") 25312391Sjason@lowepower.com assert m.isinstance_typed(np.array([1.0, 2.0, 3.0])) 25411986Sandreas.sandberg@arm.com 25511986Sandreas.sandberg@arm.com 25611986Sandreas.sandberg@arm.comdef test_constructors(): 25712391Sjason@lowepower.com defaults = m.default_constructors() 25811986Sandreas.sandberg@arm.com for a in defaults.values(): 25911986Sandreas.sandberg@arm.com assert a.size == 0 26011986Sandreas.sandberg@arm.com assert defaults["array"].dtype == np.array([]).dtype 26111986Sandreas.sandberg@arm.com assert defaults["array_t<int32>"].dtype == np.int32 26211986Sandreas.sandberg@arm.com assert defaults["array_t<double>"].dtype == np.float64 26311986Sandreas.sandberg@arm.com 26412391Sjason@lowepower.com results = m.converting_constructors([1, 2, 3]) 26511986Sandreas.sandberg@arm.com for a in results.values(): 26611986Sandreas.sandberg@arm.com np.testing.assert_array_equal(a, [1, 2, 3]) 26711986Sandreas.sandberg@arm.com assert results["array"].dtype == np.int_ 26811986Sandreas.sandberg@arm.com assert results["array_t<int32>"].dtype == np.int32 26911986Sandreas.sandberg@arm.com assert results["array_t<double>"].dtype == np.float64 27012037Sandreas.sandberg@arm.com 27112037Sandreas.sandberg@arm.com 27212037Sandreas.sandberg@arm.comdef test_overload_resolution(msg): 27312037Sandreas.sandberg@arm.com # Exact overload matches: 27412391Sjason@lowepower.com assert m.overloaded(np.array([1], dtype='float64')) == 'double' 27512391Sjason@lowepower.com assert m.overloaded(np.array([1], dtype='float32')) == 'float' 27612391Sjason@lowepower.com assert m.overloaded(np.array([1], dtype='ushort')) == 'unsigned short' 27712391Sjason@lowepower.com assert m.overloaded(np.array([1], dtype='intc')) == 'int' 27812391Sjason@lowepower.com assert m.overloaded(np.array([1], dtype='longlong')) == 'long long' 27912391Sjason@lowepower.com assert m.overloaded(np.array([1], dtype='complex')) == 'double complex' 28012391Sjason@lowepower.com assert m.overloaded(np.array([1], dtype='csingle')) == 'float complex' 28112037Sandreas.sandberg@arm.com 28212037Sandreas.sandberg@arm.com # No exact match, should call first convertible version: 28312391Sjason@lowepower.com assert m.overloaded(np.array([1], dtype='uint8')) == 'double' 28412037Sandreas.sandberg@arm.com 28512037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 28612391Sjason@lowepower.com m.overloaded("not an array") 28712037Sandreas.sandberg@arm.com assert msg(excinfo.value) == """ 28812037Sandreas.sandberg@arm.com overloaded(): incompatible function arguments. The following argument types are supported: 28912037Sandreas.sandberg@arm.com 1. (arg0: numpy.ndarray[float64]) -> str 29012037Sandreas.sandberg@arm.com 2. (arg0: numpy.ndarray[float32]) -> str 29112037Sandreas.sandberg@arm.com 3. (arg0: numpy.ndarray[int32]) -> str 29212037Sandreas.sandberg@arm.com 4. (arg0: numpy.ndarray[uint16]) -> str 29312037Sandreas.sandberg@arm.com 5. (arg0: numpy.ndarray[int64]) -> str 29412037Sandreas.sandberg@arm.com 6. (arg0: numpy.ndarray[complex128]) -> str 29512037Sandreas.sandberg@arm.com 7. (arg0: numpy.ndarray[complex64]) -> str 29612037Sandreas.sandberg@arm.com 29712037Sandreas.sandberg@arm.com Invoked with: 'not an array' 29812037Sandreas.sandberg@arm.com """ 29912037Sandreas.sandberg@arm.com 30012391Sjason@lowepower.com assert m.overloaded2(np.array([1], dtype='float64')) == 'double' 30112391Sjason@lowepower.com assert m.overloaded2(np.array([1], dtype='float32')) == 'float' 30212391Sjason@lowepower.com assert m.overloaded2(np.array([1], dtype='complex64')) == 'float complex' 30312391Sjason@lowepower.com assert m.overloaded2(np.array([1], dtype='complex128')) == 'double complex' 30412391Sjason@lowepower.com assert m.overloaded2(np.array([1], dtype='float32')) == 'float' 30512037Sandreas.sandberg@arm.com 30612391Sjason@lowepower.com assert m.overloaded3(np.array([1], dtype='float64')) == 'double' 30712391Sjason@lowepower.com assert m.overloaded3(np.array([1], dtype='intc')) == 'int' 30812037Sandreas.sandberg@arm.com expected_exc = """ 30912037Sandreas.sandberg@arm.com overloaded3(): incompatible function arguments. The following argument types are supported: 31012037Sandreas.sandberg@arm.com 1. (arg0: numpy.ndarray[int32]) -> str 31112037Sandreas.sandberg@arm.com 2. (arg0: numpy.ndarray[float64]) -> str 31212037Sandreas.sandberg@arm.com 31314299Sbbruce@ucdavis.edu Invoked with: """ 31412037Sandreas.sandberg@arm.com 31512037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 31612391Sjason@lowepower.com m.overloaded3(np.array([1], dtype='uintc')) 31714299Sbbruce@ucdavis.edu assert msg(excinfo.value) == expected_exc + repr(np.array([1], dtype='uint32')) 31812037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 31912391Sjason@lowepower.com m.overloaded3(np.array([1], dtype='float32')) 32014299Sbbruce@ucdavis.edu assert msg(excinfo.value) == expected_exc + repr(np.array([1.], dtype='float32')) 32112037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 32212391Sjason@lowepower.com m.overloaded3(np.array([1], dtype='complex')) 32314299Sbbruce@ucdavis.edu assert msg(excinfo.value) == expected_exc + repr(np.array([1. + 0.j])) 32412037Sandreas.sandberg@arm.com 32512037Sandreas.sandberg@arm.com # Exact matches: 32612391Sjason@lowepower.com assert m.overloaded4(np.array([1], dtype='double')) == 'double' 32712391Sjason@lowepower.com assert m.overloaded4(np.array([1], dtype='longlong')) == 'long long' 32812037Sandreas.sandberg@arm.com # Non-exact matches requiring conversion. Since float to integer isn't a 32912037Sandreas.sandberg@arm.com # save conversion, it should go to the double overload, but short can go to 33012037Sandreas.sandberg@arm.com # either (and so should end up on the first-registered, the long long). 33112391Sjason@lowepower.com assert m.overloaded4(np.array([1], dtype='float32')) == 'double' 33212391Sjason@lowepower.com assert m.overloaded4(np.array([1], dtype='short')) == 'long long' 33312037Sandreas.sandberg@arm.com 33412391Sjason@lowepower.com assert m.overloaded5(np.array([1], dtype='double')) == 'double' 33512391Sjason@lowepower.com assert m.overloaded5(np.array([1], dtype='uintc')) == 'unsigned int' 33612391Sjason@lowepower.com assert m.overloaded5(np.array([1], dtype='float32')) == 'unsigned int' 33712037Sandreas.sandberg@arm.com 33812037Sandreas.sandberg@arm.com 33912391Sjason@lowepower.comdef test_greedy_string_overload(): 34012391Sjason@lowepower.com """Tests fix for #685 - ndarray shouldn't go to std::string overload""" 34112037Sandreas.sandberg@arm.com 34212391Sjason@lowepower.com assert m.issue685("abc") == "string" 34312391Sjason@lowepower.com assert m.issue685(np.array([97, 98, 99], dtype='b')) == "array" 34412391Sjason@lowepower.com assert m.issue685(123) == "other" 34512037Sandreas.sandberg@arm.com 34612037Sandreas.sandberg@arm.com 34712037Sandreas.sandberg@arm.comdef test_array_unchecked_fixed_dims(msg): 34812037Sandreas.sandberg@arm.com z1 = np.array([[1, 2], [3, 4]], dtype='float64') 34912391Sjason@lowepower.com m.proxy_add2(z1, 10) 35012037Sandreas.sandberg@arm.com assert np.all(z1 == [[11, 12], [13, 14]]) 35112037Sandreas.sandberg@arm.com 35212037Sandreas.sandberg@arm.com with pytest.raises(ValueError) as excinfo: 35312391Sjason@lowepower.com m.proxy_add2(np.array([1., 2, 3]), 5.0) 35412037Sandreas.sandberg@arm.com assert msg(excinfo.value) == "array has incorrect number of dimensions: 1; expected 2" 35512037Sandreas.sandberg@arm.com 35612037Sandreas.sandberg@arm.com expect_c = np.ndarray(shape=(3, 3, 3), buffer=np.array(range(3, 30)), dtype='int') 35712391Sjason@lowepower.com assert np.all(m.proxy_init3(3.0) == expect_c) 35812037Sandreas.sandberg@arm.com expect_f = np.transpose(expect_c) 35912391Sjason@lowepower.com assert np.all(m.proxy_init3F(3.0) == expect_f) 36012037Sandreas.sandberg@arm.com 36112391Sjason@lowepower.com assert m.proxy_squared_L2_norm(np.array(range(6))) == 55 36212391Sjason@lowepower.com assert m.proxy_squared_L2_norm(np.array(range(6), dtype="float64")) == 55 36312037Sandreas.sandberg@arm.com 36412391Sjason@lowepower.com assert m.proxy_auxiliaries2(z1) == [11, 11, True, 2, 8, 2, 2, 4, 32] 36512391Sjason@lowepower.com assert m.proxy_auxiliaries2(z1) == m.array_auxiliaries2(z1) 36612037Sandreas.sandberg@arm.com 36712037Sandreas.sandberg@arm.com 36812037Sandreas.sandberg@arm.comdef test_array_unchecked_dyn_dims(msg): 36912037Sandreas.sandberg@arm.com z1 = np.array([[1, 2], [3, 4]], dtype='float64') 37012391Sjason@lowepower.com m.proxy_add2_dyn(z1, 10) 37112037Sandreas.sandberg@arm.com assert np.all(z1 == [[11, 12], [13, 14]]) 37212037Sandreas.sandberg@arm.com 37312037Sandreas.sandberg@arm.com expect_c = np.ndarray(shape=(3, 3, 3), buffer=np.array(range(3, 30)), dtype='int') 37412391Sjason@lowepower.com assert np.all(m.proxy_init3_dyn(3.0) == expect_c) 37512037Sandreas.sandberg@arm.com 37612391Sjason@lowepower.com assert m.proxy_auxiliaries2_dyn(z1) == [11, 11, True, 2, 8, 2, 2, 4, 32] 37712391Sjason@lowepower.com assert m.proxy_auxiliaries2_dyn(z1) == m.array_auxiliaries2(z1) 37812391Sjason@lowepower.com 37912391Sjason@lowepower.com 38012391Sjason@lowepower.comdef test_array_failure(): 38112391Sjason@lowepower.com with pytest.raises(ValueError) as excinfo: 38212391Sjason@lowepower.com m.array_fail_test() 38312391Sjason@lowepower.com assert str(excinfo.value) == 'cannot create a pybind11::array from a nullptr' 38412391Sjason@lowepower.com 38512391Sjason@lowepower.com with pytest.raises(ValueError) as excinfo: 38612391Sjason@lowepower.com m.array_t_fail_test() 38712391Sjason@lowepower.com assert str(excinfo.value) == 'cannot create a pybind11::array_t from a nullptr' 38812391Sjason@lowepower.com 38912391Sjason@lowepower.com with pytest.raises(ValueError) as excinfo: 39012391Sjason@lowepower.com m.array_fail_test_negative_size() 39112391Sjason@lowepower.com assert str(excinfo.value) == 'negative dimensions are not allowed' 39212391Sjason@lowepower.com 39312391Sjason@lowepower.com 39412391Sjason@lowepower.comdef test_initializer_list(): 39512391Sjason@lowepower.com assert m.array_initializer_list1().shape == (1,) 39612391Sjason@lowepower.com assert m.array_initializer_list2().shape == (1, 2) 39712391Sjason@lowepower.com assert m.array_initializer_list3().shape == (1, 2, 3) 39812391Sjason@lowepower.com assert m.array_initializer_list4().shape == (1, 2, 3, 4) 39912391Sjason@lowepower.com 40012391Sjason@lowepower.com 40112391Sjason@lowepower.comdef test_array_resize(msg): 40212391Sjason@lowepower.com a = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9], dtype='float64') 40312391Sjason@lowepower.com m.array_reshape2(a) 40412391Sjason@lowepower.com assert(a.size == 9) 40512391Sjason@lowepower.com assert(np.all(a == [[1, 2, 3], [4, 5, 6], [7, 8, 9]])) 40612391Sjason@lowepower.com 40712391Sjason@lowepower.com # total size change should succced with refcheck off 40812391Sjason@lowepower.com m.array_resize3(a, 4, False) 40912391Sjason@lowepower.com assert(a.size == 64) 41012391Sjason@lowepower.com # ... and fail with refcheck on 41112391Sjason@lowepower.com try: 41212391Sjason@lowepower.com m.array_resize3(a, 3, True) 41312391Sjason@lowepower.com except ValueError as e: 41412391Sjason@lowepower.com assert(str(e).startswith("cannot resize an array")) 41512391Sjason@lowepower.com # transposed array doesn't own data 41612391Sjason@lowepower.com b = a.transpose() 41712391Sjason@lowepower.com try: 41812391Sjason@lowepower.com m.array_resize3(b, 3, False) 41912391Sjason@lowepower.com except ValueError as e: 42012391Sjason@lowepower.com assert(str(e).startswith("cannot resize this array: it does not own its data")) 42112391Sjason@lowepower.com # ... but reshape should be fine 42212391Sjason@lowepower.com m.array_reshape2(b) 42312391Sjason@lowepower.com assert(b.shape == (8, 8)) 42412391Sjason@lowepower.com 42512391Sjason@lowepower.com 42612391Sjason@lowepower.com@pytest.unsupported_on_pypy 42712391Sjason@lowepower.comdef test_array_create_and_resize(msg): 42812391Sjason@lowepower.com a = m.create_and_resize(2) 42912391Sjason@lowepower.com assert(a.size == 4) 43012391Sjason@lowepower.com assert(np.all(a == 42.)) 43114299Sbbruce@ucdavis.edu 43214299Sbbruce@ucdavis.edu 43314299Sbbruce@ucdavis.edu@pytest.unsupported_on_py2 43414299Sbbruce@ucdavis.edudef test_index_using_ellipsis(): 43514299Sbbruce@ucdavis.edu a = m.index_using_ellipsis(np.zeros((5, 6, 7))) 43614299Sbbruce@ucdavis.edu assert a.shape == (6,) 43714299Sbbruce@ucdavis.edu 43814299Sbbruce@ucdavis.edu 43914299Sbbruce@ucdavis.edu@pytest.unsupported_on_pypy 44014299Sbbruce@ucdavis.edudef test_dtype_refcount_leak(): 44114299Sbbruce@ucdavis.edu from sys import getrefcount 44214299Sbbruce@ucdavis.edu dtype = np.dtype(np.float_) 44314299Sbbruce@ucdavis.edu a = np.array([1], dtype=dtype) 44414299Sbbruce@ucdavis.edu before = getrefcount(dtype) 44514299Sbbruce@ucdavis.edu m.ndim(a) 44614299Sbbruce@ucdavis.edu after = getrefcount(dtype) 44714299Sbbruce@ucdavis.edu assert after == before 448