test_eigen.py revision 12391
111986Sandreas.sandberg@arm.comimport pytest 212391Sjason@lowepower.comfrom pybind11_tests import ConstructorStats 311986Sandreas.sandberg@arm.com 412037Sandreas.sandberg@arm.compytestmark = pytest.requires_eigen_and_numpy 512037Sandreas.sandberg@arm.com 611986Sandreas.sandberg@arm.comwith pytest.suppress(ImportError): 712391Sjason@lowepower.com from pybind11_tests import eigen as m 811986Sandreas.sandberg@arm.com import numpy as np 911986Sandreas.sandberg@arm.com 1012037Sandreas.sandberg@arm.com ref = np.array([[ 0., 3, 0, 0, 0, 11], 1111986Sandreas.sandberg@arm.com [22, 0, 0, 0, 17, 11], 1211986Sandreas.sandberg@arm.com [ 7, 5, 0, 1, 0, 11], 1311986Sandreas.sandberg@arm.com [ 0, 0, 0, 0, 0, 11], 1411986Sandreas.sandberg@arm.com [ 0, 0, 14, 0, 8, 11]]) 1511986Sandreas.sandberg@arm.com 1611986Sandreas.sandberg@arm.com 1711986Sandreas.sandberg@arm.comdef assert_equal_ref(mat): 1811986Sandreas.sandberg@arm.com np.testing.assert_array_equal(mat, ref) 1911986Sandreas.sandberg@arm.com 2011986Sandreas.sandberg@arm.com 2111986Sandreas.sandberg@arm.comdef assert_sparse_equal_ref(sparse_mat): 2211986Sandreas.sandberg@arm.com assert_equal_ref(sparse_mat.todense()) 2311986Sandreas.sandberg@arm.com 2411986Sandreas.sandberg@arm.com 2511986Sandreas.sandberg@arm.comdef test_fixed(): 2612391Sjason@lowepower.com assert_equal_ref(m.fixed_c()) 2712391Sjason@lowepower.com assert_equal_ref(m.fixed_r()) 2812391Sjason@lowepower.com assert_equal_ref(m.fixed_copy_r(m.fixed_r())) 2912391Sjason@lowepower.com assert_equal_ref(m.fixed_copy_c(m.fixed_c())) 3012391Sjason@lowepower.com assert_equal_ref(m.fixed_copy_r(m.fixed_c())) 3112391Sjason@lowepower.com assert_equal_ref(m.fixed_copy_c(m.fixed_r())) 3211986Sandreas.sandberg@arm.com 3311986Sandreas.sandberg@arm.com 3411986Sandreas.sandberg@arm.comdef test_dense(): 3512391Sjason@lowepower.com assert_equal_ref(m.dense_r()) 3612391Sjason@lowepower.com assert_equal_ref(m.dense_c()) 3712391Sjason@lowepower.com assert_equal_ref(m.dense_copy_r(m.dense_r())) 3812391Sjason@lowepower.com assert_equal_ref(m.dense_copy_c(m.dense_c())) 3912391Sjason@lowepower.com assert_equal_ref(m.dense_copy_r(m.dense_c())) 4012391Sjason@lowepower.com assert_equal_ref(m.dense_copy_c(m.dense_r())) 4111986Sandreas.sandberg@arm.com 4211986Sandreas.sandberg@arm.com 4312037Sandreas.sandberg@arm.comdef test_partially_fixed(): 4412391Sjason@lowepower.com ref2 = np.array([[0., 1, 2, 3], [4, 5, 6, 7], [8, 9, 10, 11], [12, 13, 14, 15]]) 4512391Sjason@lowepower.com np.testing.assert_array_equal(m.partial_copy_four_rm_r(ref2), ref2) 4612391Sjason@lowepower.com np.testing.assert_array_equal(m.partial_copy_four_rm_c(ref2), ref2) 4712391Sjason@lowepower.com np.testing.assert_array_equal(m.partial_copy_four_rm_r(ref2[:, 1]), ref2[:, [1]]) 4812391Sjason@lowepower.com np.testing.assert_array_equal(m.partial_copy_four_rm_c(ref2[0, :]), ref2[[0], :]) 4912391Sjason@lowepower.com np.testing.assert_array_equal(m.partial_copy_four_rm_r(ref2[:, (0, 2)]), ref2[:, (0, 2)]) 5012391Sjason@lowepower.com np.testing.assert_array_equal( 5112391Sjason@lowepower.com m.partial_copy_four_rm_c(ref2[(3, 1, 2), :]), ref2[(3, 1, 2), :]) 5212037Sandreas.sandberg@arm.com 5312391Sjason@lowepower.com np.testing.assert_array_equal(m.partial_copy_four_cm_r(ref2), ref2) 5412391Sjason@lowepower.com np.testing.assert_array_equal(m.partial_copy_four_cm_c(ref2), ref2) 5512391Sjason@lowepower.com np.testing.assert_array_equal(m.partial_copy_four_cm_r(ref2[:, 1]), ref2[:, [1]]) 5612391Sjason@lowepower.com np.testing.assert_array_equal(m.partial_copy_four_cm_c(ref2[0, :]), ref2[[0], :]) 5712391Sjason@lowepower.com np.testing.assert_array_equal(m.partial_copy_four_cm_r(ref2[:, (0, 2)]), ref2[:, (0, 2)]) 5812037Sandreas.sandberg@arm.com np.testing.assert_array_equal( 5912391Sjason@lowepower.com m.partial_copy_four_cm_c(ref2[(3, 1, 2), :]), ref2[(3, 1, 2), :]) 6012037Sandreas.sandberg@arm.com 6112391Sjason@lowepower.com # TypeError should be raise for a shape mismatch 6212391Sjason@lowepower.com functions = [m.partial_copy_four_rm_r, m.partial_copy_four_rm_c, 6312391Sjason@lowepower.com m.partial_copy_four_cm_r, m.partial_copy_four_cm_c] 6412391Sjason@lowepower.com matrix_with_wrong_shape = [[1, 2], 6512391Sjason@lowepower.com [3, 4]] 6612391Sjason@lowepower.com for f in functions: 6712391Sjason@lowepower.com with pytest.raises(TypeError) as excinfo: 6812391Sjason@lowepower.com f(matrix_with_wrong_shape) 6912391Sjason@lowepower.com assert "incompatible function arguments" in str(excinfo.value) 7012037Sandreas.sandberg@arm.com 7112037Sandreas.sandberg@arm.com 7212037Sandreas.sandberg@arm.comdef test_mutator_descriptors(): 7312037Sandreas.sandberg@arm.com zr = np.arange(30, dtype='float32').reshape(5, 6) # row-major 7412037Sandreas.sandberg@arm.com zc = zr.reshape(6, 5).transpose() # column-major 7512037Sandreas.sandberg@arm.com 7612391Sjason@lowepower.com m.fixed_mutator_r(zr) 7712391Sjason@lowepower.com m.fixed_mutator_c(zc) 7812391Sjason@lowepower.com m.fixed_mutator_a(zr) 7912391Sjason@lowepower.com m.fixed_mutator_a(zc) 8012037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 8112391Sjason@lowepower.com m.fixed_mutator_r(zc) 8212391Sjason@lowepower.com assert ('(arg0: numpy.ndarray[float32[5, 6], flags.writeable, flags.c_contiguous]) -> None' 8312037Sandreas.sandberg@arm.com in str(excinfo.value)) 8412037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 8512391Sjason@lowepower.com m.fixed_mutator_c(zr) 8612391Sjason@lowepower.com assert ('(arg0: numpy.ndarray[float32[5, 6], flags.writeable, flags.f_contiguous]) -> None' 8712037Sandreas.sandberg@arm.com in str(excinfo.value)) 8812037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 8912391Sjason@lowepower.com m.fixed_mutator_a(np.array([[1, 2], [3, 4]], dtype='float32')) 9012391Sjason@lowepower.com assert ('(arg0: numpy.ndarray[float32[5, 6], flags.writeable]) -> None' 9112037Sandreas.sandberg@arm.com in str(excinfo.value)) 9212037Sandreas.sandberg@arm.com zr.flags.writeable = False 9312037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 9412391Sjason@lowepower.com m.fixed_mutator_r(zr) 9512037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 9612391Sjason@lowepower.com m.fixed_mutator_a(zr) 9712037Sandreas.sandberg@arm.com 9812037Sandreas.sandberg@arm.com 9912037Sandreas.sandberg@arm.comdef test_cpp_casting(): 10012391Sjason@lowepower.com assert m.cpp_copy(m.fixed_r()) == 22. 10112391Sjason@lowepower.com assert m.cpp_copy(m.fixed_c()) == 22. 10212037Sandreas.sandberg@arm.com z = np.array([[5., 6], [7, 8]]) 10312391Sjason@lowepower.com assert m.cpp_copy(z) == 7. 10412391Sjason@lowepower.com assert m.cpp_copy(m.get_cm_ref()) == 21. 10512391Sjason@lowepower.com assert m.cpp_copy(m.get_rm_ref()) == 21. 10612391Sjason@lowepower.com assert m.cpp_ref_c(m.get_cm_ref()) == 21. 10712391Sjason@lowepower.com assert m.cpp_ref_r(m.get_rm_ref()) == 21. 10812037Sandreas.sandberg@arm.com with pytest.raises(RuntimeError) as excinfo: 10912391Sjason@lowepower.com # Can't reference m.fixed_c: it contains floats, m.cpp_ref_any wants doubles 11012391Sjason@lowepower.com m.cpp_ref_any(m.fixed_c()) 11112037Sandreas.sandberg@arm.com assert 'Unable to cast Python instance' in str(excinfo.value) 11212037Sandreas.sandberg@arm.com with pytest.raises(RuntimeError) as excinfo: 11312391Sjason@lowepower.com # Can't reference m.fixed_r: it contains floats, m.cpp_ref_any wants doubles 11412391Sjason@lowepower.com m.cpp_ref_any(m.fixed_r()) 11512037Sandreas.sandberg@arm.com assert 'Unable to cast Python instance' in str(excinfo.value) 11612391Sjason@lowepower.com assert m.cpp_ref_any(m.ReturnTester.create()) == 1. 11712037Sandreas.sandberg@arm.com 11812391Sjason@lowepower.com assert m.cpp_ref_any(m.get_cm_ref()) == 21. 11912391Sjason@lowepower.com assert m.cpp_ref_any(m.get_cm_ref()) == 21. 12012037Sandreas.sandberg@arm.com 12112037Sandreas.sandberg@arm.com 12212037Sandreas.sandberg@arm.comdef test_pass_readonly_array(): 12312037Sandreas.sandberg@arm.com z = np.full((5, 6), 42.0) 12412037Sandreas.sandberg@arm.com z.flags.writeable = False 12512391Sjason@lowepower.com np.testing.assert_array_equal(z, m.fixed_copy_r(z)) 12612391Sjason@lowepower.com np.testing.assert_array_equal(m.fixed_r_const(), m.fixed_r()) 12712391Sjason@lowepower.com assert not m.fixed_r_const().flags.writeable 12812391Sjason@lowepower.com np.testing.assert_array_equal(m.fixed_copy_r(m.fixed_r_const()), m.fixed_r_const()) 12912037Sandreas.sandberg@arm.com 13012037Sandreas.sandberg@arm.com 13111986Sandreas.sandberg@arm.comdef test_nonunit_stride_from_python(): 13211986Sandreas.sandberg@arm.com counting_mat = np.arange(9.0, dtype=np.float32).reshape((3, 3)) 13312037Sandreas.sandberg@arm.com second_row = counting_mat[1, :] 13412037Sandreas.sandberg@arm.com second_col = counting_mat[:, 1] 13512391Sjason@lowepower.com np.testing.assert_array_equal(m.double_row(second_row), 2.0 * second_row) 13612391Sjason@lowepower.com np.testing.assert_array_equal(m.double_col(second_row), 2.0 * second_row) 13712391Sjason@lowepower.com np.testing.assert_array_equal(m.double_complex(second_row), 2.0 * second_row) 13812391Sjason@lowepower.com np.testing.assert_array_equal(m.double_row(second_col), 2.0 * second_col) 13912391Sjason@lowepower.com np.testing.assert_array_equal(m.double_col(second_col), 2.0 * second_col) 14012391Sjason@lowepower.com np.testing.assert_array_equal(m.double_complex(second_col), 2.0 * second_col) 14111986Sandreas.sandberg@arm.com 14211986Sandreas.sandberg@arm.com counting_3d = np.arange(27.0, dtype=np.float32).reshape((3, 3, 3)) 14311986Sandreas.sandberg@arm.com slices = [counting_3d[0, :, :], counting_3d[:, 0, :], counting_3d[:, :, 0]] 14411986Sandreas.sandberg@arm.com for slice_idx, ref_mat in enumerate(slices): 14512391Sjason@lowepower.com np.testing.assert_array_equal(m.double_mat_cm(ref_mat), 2.0 * ref_mat) 14612391Sjason@lowepower.com np.testing.assert_array_equal(m.double_mat_rm(ref_mat), 2.0 * ref_mat) 14711986Sandreas.sandberg@arm.com 14812037Sandreas.sandberg@arm.com # Mutator: 14912391Sjason@lowepower.com m.double_threer(second_row) 15012391Sjason@lowepower.com m.double_threec(second_col) 15112037Sandreas.sandberg@arm.com np.testing.assert_array_equal(counting_mat, [[0., 2, 2], [6, 16, 10], [6, 14, 8]]) 15211986Sandreas.sandberg@arm.com 15312037Sandreas.sandberg@arm.com 15412391Sjason@lowepower.comdef test_negative_stride_from_python(msg): 15512391Sjason@lowepower.com """Eigen doesn't support (as of yet) negative strides. When a function takes an Eigen matrix by 15612391Sjason@lowepower.com copy or const reference, we can pass a numpy array that has negative strides. Otherwise, an 15712391Sjason@lowepower.com exception will be thrown as Eigen will not be able to map the numpy array.""" 15812391Sjason@lowepower.com 15912391Sjason@lowepower.com counting_mat = np.arange(9.0, dtype=np.float32).reshape((3, 3)) 16012391Sjason@lowepower.com counting_mat = counting_mat[::-1, ::-1] 16112391Sjason@lowepower.com second_row = counting_mat[1, :] 16212391Sjason@lowepower.com second_col = counting_mat[:, 1] 16312391Sjason@lowepower.com np.testing.assert_array_equal(m.double_row(second_row), 2.0 * second_row) 16412391Sjason@lowepower.com np.testing.assert_array_equal(m.double_col(second_row), 2.0 * second_row) 16512391Sjason@lowepower.com np.testing.assert_array_equal(m.double_complex(second_row), 2.0 * second_row) 16612391Sjason@lowepower.com np.testing.assert_array_equal(m.double_row(second_col), 2.0 * second_col) 16712391Sjason@lowepower.com np.testing.assert_array_equal(m.double_col(second_col), 2.0 * second_col) 16812391Sjason@lowepower.com np.testing.assert_array_equal(m.double_complex(second_col), 2.0 * second_col) 16912391Sjason@lowepower.com 17012391Sjason@lowepower.com counting_3d = np.arange(27.0, dtype=np.float32).reshape((3, 3, 3)) 17112391Sjason@lowepower.com counting_3d = counting_3d[::-1, ::-1, ::-1] 17212391Sjason@lowepower.com slices = [counting_3d[0, :, :], counting_3d[:, 0, :], counting_3d[:, :, 0]] 17312391Sjason@lowepower.com for slice_idx, ref_mat in enumerate(slices): 17412391Sjason@lowepower.com np.testing.assert_array_equal(m.double_mat_cm(ref_mat), 2.0 * ref_mat) 17512391Sjason@lowepower.com np.testing.assert_array_equal(m.double_mat_rm(ref_mat), 2.0 * ref_mat) 17612391Sjason@lowepower.com 17712391Sjason@lowepower.com # Mutator: 17812391Sjason@lowepower.com with pytest.raises(TypeError) as excinfo: 17912391Sjason@lowepower.com m.double_threer(second_row) 18012391Sjason@lowepower.com assert msg(excinfo.value) == """ 18112391Sjason@lowepower.com double_threer(): incompatible function arguments. The following argument types are supported: 18212391Sjason@lowepower.com 1. (arg0: numpy.ndarray[float32[1, 3], flags.writeable]) -> None 18312391Sjason@lowepower.com 18412391Sjason@lowepower.com Invoked with: array([ 5., 4., 3.], dtype=float32) 18512391Sjason@lowepower.com """ # noqa: E501 line too long 18612391Sjason@lowepower.com 18712391Sjason@lowepower.com with pytest.raises(TypeError) as excinfo: 18812391Sjason@lowepower.com m.double_threec(second_col) 18912391Sjason@lowepower.com assert msg(excinfo.value) == """ 19012391Sjason@lowepower.com double_threec(): incompatible function arguments. The following argument types are supported: 19112391Sjason@lowepower.com 1. (arg0: numpy.ndarray[float32[3, 1], flags.writeable]) -> None 19212391Sjason@lowepower.com 19312391Sjason@lowepower.com Invoked with: array([ 7., 4., 1.], dtype=float32) 19412391Sjason@lowepower.com """ # noqa: E501 line too long 19512391Sjason@lowepower.com 19612391Sjason@lowepower.com 19711986Sandreas.sandberg@arm.comdef test_nonunit_stride_to_python(): 19812391Sjason@lowepower.com assert np.all(m.diagonal(ref) == ref.diagonal()) 19912391Sjason@lowepower.com assert np.all(m.diagonal_1(ref) == ref.diagonal(1)) 20012391Sjason@lowepower.com for i in range(-5, 7): 20112391Sjason@lowepower.com assert np.all(m.diagonal_n(ref, i) == ref.diagonal(i)), "m.diagonal_n({})".format(i) 20211986Sandreas.sandberg@arm.com 20312391Sjason@lowepower.com assert np.all(m.block(ref, 2, 1, 3, 3) == ref[2:5, 1:4]) 20412391Sjason@lowepower.com assert np.all(m.block(ref, 1, 4, 4, 2) == ref[1:, 4:]) 20512391Sjason@lowepower.com assert np.all(m.block(ref, 1, 4, 3, 2) == ref[1:4, 4:]) 20611986Sandreas.sandberg@arm.com 20711986Sandreas.sandberg@arm.com 20811986Sandreas.sandberg@arm.comdef test_eigen_ref_to_python(): 20912391Sjason@lowepower.com chols = [m.cholesky1, m.cholesky2, m.cholesky3, m.cholesky4] 21011986Sandreas.sandberg@arm.com for i, chol in enumerate(chols, start=1): 21112037Sandreas.sandberg@arm.com mymat = chol(np.array([[1., 2, 4], [2, 13, 23], [4, 23, 77]])) 21211986Sandreas.sandberg@arm.com assert np.all(mymat == np.array([[1, 0, 0], [2, 3, 0], [4, 5, 6]])), "cholesky{}".format(i) 21311986Sandreas.sandberg@arm.com 21411986Sandreas.sandberg@arm.com 21512037Sandreas.sandberg@arm.comdef assign_both(a1, a2, r, c, v): 21612037Sandreas.sandberg@arm.com a1[r, c] = v 21712037Sandreas.sandberg@arm.com a2[r, c] = v 21812037Sandreas.sandberg@arm.com 21912037Sandreas.sandberg@arm.com 22012037Sandreas.sandberg@arm.comdef array_copy_but_one(a, r, c, v): 22112037Sandreas.sandberg@arm.com z = np.array(a, copy=True) 22212037Sandreas.sandberg@arm.com z[r, c] = v 22312037Sandreas.sandberg@arm.com return z 22412037Sandreas.sandberg@arm.com 22512037Sandreas.sandberg@arm.com 22612037Sandreas.sandberg@arm.comdef test_eigen_return_references(): 22712037Sandreas.sandberg@arm.com """Tests various ways of returning references and non-referencing copies""" 22812391Sjason@lowepower.com 22912037Sandreas.sandberg@arm.com master = np.ones((10, 10)) 23012391Sjason@lowepower.com a = m.ReturnTester() 23112037Sandreas.sandberg@arm.com a_get1 = a.get() 23212037Sandreas.sandberg@arm.com assert not a_get1.flags.owndata and a_get1.flags.writeable 23312037Sandreas.sandberg@arm.com assign_both(a_get1, master, 3, 3, 5) 23412037Sandreas.sandberg@arm.com a_get2 = a.get_ptr() 23512037Sandreas.sandberg@arm.com assert not a_get2.flags.owndata and a_get2.flags.writeable 23612037Sandreas.sandberg@arm.com assign_both(a_get1, master, 2, 3, 6) 23712037Sandreas.sandberg@arm.com 23812037Sandreas.sandberg@arm.com a_view1 = a.view() 23912037Sandreas.sandberg@arm.com assert not a_view1.flags.owndata and not a_view1.flags.writeable 24012037Sandreas.sandberg@arm.com with pytest.raises(ValueError): 24112037Sandreas.sandberg@arm.com a_view1[2, 3] = 4 24212037Sandreas.sandberg@arm.com a_view2 = a.view_ptr() 24312037Sandreas.sandberg@arm.com assert not a_view2.flags.owndata and not a_view2.flags.writeable 24412037Sandreas.sandberg@arm.com with pytest.raises(ValueError): 24512037Sandreas.sandberg@arm.com a_view2[2, 3] = 4 24612037Sandreas.sandberg@arm.com 24712037Sandreas.sandberg@arm.com a_copy1 = a.copy_get() 24812037Sandreas.sandberg@arm.com assert a_copy1.flags.owndata and a_copy1.flags.writeable 24912037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy1, master) 25012037Sandreas.sandberg@arm.com a_copy1[7, 7] = -44 # Shouldn't affect anything else 25112037Sandreas.sandberg@arm.com c1want = array_copy_but_one(master, 7, 7, -44) 25212037Sandreas.sandberg@arm.com a_copy2 = a.copy_view() 25312037Sandreas.sandberg@arm.com assert a_copy2.flags.owndata and a_copy2.flags.writeable 25412037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy2, master) 25512037Sandreas.sandberg@arm.com a_copy2[4, 4] = -22 # Shouldn't affect anything else 25612037Sandreas.sandberg@arm.com c2want = array_copy_but_one(master, 4, 4, -22) 25712037Sandreas.sandberg@arm.com 25812037Sandreas.sandberg@arm.com a_ref1 = a.ref() 25912037Sandreas.sandberg@arm.com assert not a_ref1.flags.owndata and a_ref1.flags.writeable 26012037Sandreas.sandberg@arm.com assign_both(a_ref1, master, 1, 1, 15) 26112037Sandreas.sandberg@arm.com a_ref2 = a.ref_const() 26212037Sandreas.sandberg@arm.com assert not a_ref2.flags.owndata and not a_ref2.flags.writeable 26312037Sandreas.sandberg@arm.com with pytest.raises(ValueError): 26412037Sandreas.sandberg@arm.com a_ref2[5, 5] = 33 26512037Sandreas.sandberg@arm.com a_ref3 = a.ref_safe() 26612037Sandreas.sandberg@arm.com assert not a_ref3.flags.owndata and a_ref3.flags.writeable 26712037Sandreas.sandberg@arm.com assign_both(a_ref3, master, 0, 7, 99) 26812037Sandreas.sandberg@arm.com a_ref4 = a.ref_const_safe() 26912037Sandreas.sandberg@arm.com assert not a_ref4.flags.owndata and not a_ref4.flags.writeable 27012037Sandreas.sandberg@arm.com with pytest.raises(ValueError): 27112037Sandreas.sandberg@arm.com a_ref4[7, 0] = 987654321 27212037Sandreas.sandberg@arm.com 27312037Sandreas.sandberg@arm.com a_copy3 = a.copy_ref() 27412037Sandreas.sandberg@arm.com assert a_copy3.flags.owndata and a_copy3.flags.writeable 27512037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy3, master) 27612037Sandreas.sandberg@arm.com a_copy3[8, 1] = 11 27712037Sandreas.sandberg@arm.com c3want = array_copy_but_one(master, 8, 1, 11) 27812037Sandreas.sandberg@arm.com a_copy4 = a.copy_ref_const() 27912037Sandreas.sandberg@arm.com assert a_copy4.flags.owndata and a_copy4.flags.writeable 28012037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy4, master) 28112037Sandreas.sandberg@arm.com a_copy4[8, 4] = 88 28212037Sandreas.sandberg@arm.com c4want = array_copy_but_one(master, 8, 4, 88) 28312037Sandreas.sandberg@arm.com 28412037Sandreas.sandberg@arm.com a_block1 = a.block(3, 3, 2, 2) 28512037Sandreas.sandberg@arm.com assert not a_block1.flags.owndata and a_block1.flags.writeable 28612037Sandreas.sandberg@arm.com a_block1[0, 0] = 55 28712037Sandreas.sandberg@arm.com master[3, 3] = 55 28812037Sandreas.sandberg@arm.com a_block2 = a.block_safe(2, 2, 3, 2) 28912037Sandreas.sandberg@arm.com assert not a_block2.flags.owndata and a_block2.flags.writeable 29012037Sandreas.sandberg@arm.com a_block2[2, 1] = -123 29112037Sandreas.sandberg@arm.com master[4, 3] = -123 29212037Sandreas.sandberg@arm.com a_block3 = a.block_const(6, 7, 4, 3) 29312037Sandreas.sandberg@arm.com assert not a_block3.flags.owndata and not a_block3.flags.writeable 29412037Sandreas.sandberg@arm.com with pytest.raises(ValueError): 29512037Sandreas.sandberg@arm.com a_block3[2, 2] = -44444 29612037Sandreas.sandberg@arm.com 29712037Sandreas.sandberg@arm.com a_copy5 = a.copy_block(2, 2, 2, 3) 29812037Sandreas.sandberg@arm.com assert a_copy5.flags.owndata and a_copy5.flags.writeable 29912037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy5, master[2:4, 2:5]) 30012037Sandreas.sandberg@arm.com a_copy5[1, 1] = 777 30112037Sandreas.sandberg@arm.com c5want = array_copy_but_one(master[2:4, 2:5], 1, 1, 777) 30212037Sandreas.sandberg@arm.com 30312037Sandreas.sandberg@arm.com a_corn1 = a.corners() 30412037Sandreas.sandberg@arm.com assert not a_corn1.flags.owndata and a_corn1.flags.writeable 30512037Sandreas.sandberg@arm.com a_corn1 *= 50 30612037Sandreas.sandberg@arm.com a_corn1[1, 1] = 999 30712037Sandreas.sandberg@arm.com master[0, 0] = 50 30812037Sandreas.sandberg@arm.com master[0, 9] = 50 30912037Sandreas.sandberg@arm.com master[9, 0] = 50 31012037Sandreas.sandberg@arm.com master[9, 9] = 999 31112037Sandreas.sandberg@arm.com a_corn2 = a.corners_const() 31212037Sandreas.sandberg@arm.com assert not a_corn2.flags.owndata and not a_corn2.flags.writeable 31312037Sandreas.sandberg@arm.com with pytest.raises(ValueError): 31412037Sandreas.sandberg@arm.com a_corn2[1, 0] = 51 31512037Sandreas.sandberg@arm.com 31612037Sandreas.sandberg@arm.com # All of the changes made all the way along should be visible everywhere 31712037Sandreas.sandberg@arm.com # now (except for the copies, of course) 31812037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_get1, master) 31912037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_get2, master) 32012037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_view1, master) 32112037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_view2, master) 32212037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_ref1, master) 32312037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_ref2, master) 32412037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_ref3, master) 32512037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_ref4, master) 32612037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_block1, master[3:5, 3:5]) 32712037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_block2, master[2:5, 2:4]) 32812037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_block3, master[6:10, 7:10]) 32912037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_corn1, master[0::master.shape[0] - 1, 0::master.shape[1] - 1]) 33012037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_corn2, master[0::master.shape[0] - 1, 0::master.shape[1] - 1]) 33112037Sandreas.sandberg@arm.com 33212037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy1, c1want) 33312037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy2, c2want) 33412037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy3, c3want) 33512037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy4, c4want) 33612037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy5, c5want) 33712037Sandreas.sandberg@arm.com 33812037Sandreas.sandberg@arm.com 33912037Sandreas.sandberg@arm.comdef assert_keeps_alive(cl, method, *args): 34012037Sandreas.sandberg@arm.com cstats = ConstructorStats.get(cl) 34112037Sandreas.sandberg@arm.com start_with = cstats.alive() 34212037Sandreas.sandberg@arm.com a = cl() 34312037Sandreas.sandberg@arm.com assert cstats.alive() == start_with + 1 34412037Sandreas.sandberg@arm.com z = method(a, *args) 34512037Sandreas.sandberg@arm.com assert cstats.alive() == start_with + 1 34612037Sandreas.sandberg@arm.com del a 34712037Sandreas.sandberg@arm.com # Here's the keep alive in action: 34812037Sandreas.sandberg@arm.com assert cstats.alive() == start_with + 1 34912037Sandreas.sandberg@arm.com del z 35012037Sandreas.sandberg@arm.com # Keep alive should have expired: 35112037Sandreas.sandberg@arm.com assert cstats.alive() == start_with 35212037Sandreas.sandberg@arm.com 35312037Sandreas.sandberg@arm.com 35412037Sandreas.sandberg@arm.comdef test_eigen_keepalive(): 35512391Sjason@lowepower.com a = m.ReturnTester() 35612391Sjason@lowepower.com cstats = ConstructorStats.get(m.ReturnTester) 35712037Sandreas.sandberg@arm.com assert cstats.alive() == 1 35812037Sandreas.sandberg@arm.com unsafe = [a.ref(), a.ref_const(), a.block(1, 2, 3, 4)] 35912037Sandreas.sandberg@arm.com copies = [a.copy_get(), a.copy_view(), a.copy_ref(), a.copy_ref_const(), 36012037Sandreas.sandberg@arm.com a.copy_block(4, 3, 2, 1)] 36112037Sandreas.sandberg@arm.com del a 36212037Sandreas.sandberg@arm.com assert cstats.alive() == 0 36312037Sandreas.sandberg@arm.com del unsafe 36412037Sandreas.sandberg@arm.com del copies 36512037Sandreas.sandberg@arm.com 36612391Sjason@lowepower.com for meth in [m.ReturnTester.get, m.ReturnTester.get_ptr, m.ReturnTester.view, 36712391Sjason@lowepower.com m.ReturnTester.view_ptr, m.ReturnTester.ref_safe, m.ReturnTester.ref_const_safe, 36812391Sjason@lowepower.com m.ReturnTester.corners, m.ReturnTester.corners_const]: 36912391Sjason@lowepower.com assert_keeps_alive(m.ReturnTester, meth) 37012037Sandreas.sandberg@arm.com 37112391Sjason@lowepower.com for meth in [m.ReturnTester.block_safe, m.ReturnTester.block_const]: 37212391Sjason@lowepower.com assert_keeps_alive(m.ReturnTester, meth, 4, 3, 2, 1) 37312037Sandreas.sandberg@arm.com 37412037Sandreas.sandberg@arm.com 37512037Sandreas.sandberg@arm.comdef test_eigen_ref_mutators(): 37612391Sjason@lowepower.com """Tests Eigen's ability to mutate numpy values""" 37712391Sjason@lowepower.com 37812037Sandreas.sandberg@arm.com orig = np.array([[1., 2, 3], [4, 5, 6], [7, 8, 9]]) 37912037Sandreas.sandberg@arm.com zr = np.array(orig) 38012037Sandreas.sandberg@arm.com zc = np.array(orig, order='F') 38112391Sjason@lowepower.com m.add_rm(zr, 1, 0, 100) 38212037Sandreas.sandberg@arm.com assert np.all(zr == np.array([[1., 2, 3], [104, 5, 6], [7, 8, 9]])) 38312391Sjason@lowepower.com m.add_cm(zc, 1, 0, 200) 38412037Sandreas.sandberg@arm.com assert np.all(zc == np.array([[1., 2, 3], [204, 5, 6], [7, 8, 9]])) 38512037Sandreas.sandberg@arm.com 38612391Sjason@lowepower.com m.add_any(zr, 1, 0, 20) 38712037Sandreas.sandberg@arm.com assert np.all(zr == np.array([[1., 2, 3], [124, 5, 6], [7, 8, 9]])) 38812391Sjason@lowepower.com m.add_any(zc, 1, 0, 10) 38912037Sandreas.sandberg@arm.com assert np.all(zc == np.array([[1., 2, 3], [214, 5, 6], [7, 8, 9]])) 39012037Sandreas.sandberg@arm.com 39112037Sandreas.sandberg@arm.com # Can't reference a col-major array with a row-major Ref, and vice versa: 39212037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 39312391Sjason@lowepower.com m.add_rm(zc, 1, 0, 1) 39412037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 39512391Sjason@lowepower.com m.add_cm(zr, 1, 0, 1) 39612037Sandreas.sandberg@arm.com 39712037Sandreas.sandberg@arm.com # Overloads: 39812391Sjason@lowepower.com m.add1(zr, 1, 0, -100) 39912391Sjason@lowepower.com m.add2(zr, 1, 0, -20) 40012037Sandreas.sandberg@arm.com assert np.all(zr == orig) 40112391Sjason@lowepower.com m.add1(zc, 1, 0, -200) 40212391Sjason@lowepower.com m.add2(zc, 1, 0, -10) 40312037Sandreas.sandberg@arm.com assert np.all(zc == orig) 40412037Sandreas.sandberg@arm.com 40512037Sandreas.sandberg@arm.com # a non-contiguous slice (this won't work on either the row- or 40612037Sandreas.sandberg@arm.com # column-contiguous refs, but should work for the any) 40712037Sandreas.sandberg@arm.com cornersr = zr[0::2, 0::2] 40812037Sandreas.sandberg@arm.com cornersc = zc[0::2, 0::2] 40912037Sandreas.sandberg@arm.com 41012037Sandreas.sandberg@arm.com assert np.all(cornersr == np.array([[1., 3], [7, 9]])) 41112037Sandreas.sandberg@arm.com assert np.all(cornersc == np.array([[1., 3], [7, 9]])) 41212037Sandreas.sandberg@arm.com 41312037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 41412391Sjason@lowepower.com m.add_rm(cornersr, 0, 1, 25) 41512037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 41612391Sjason@lowepower.com m.add_cm(cornersr, 0, 1, 25) 41712037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 41812391Sjason@lowepower.com m.add_rm(cornersc, 0, 1, 25) 41912037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 42012391Sjason@lowepower.com m.add_cm(cornersc, 0, 1, 25) 42112391Sjason@lowepower.com m.add_any(cornersr, 0, 1, 25) 42212391Sjason@lowepower.com m.add_any(cornersc, 0, 1, 44) 42312037Sandreas.sandberg@arm.com assert np.all(zr == np.array([[1., 2, 28], [4, 5, 6], [7, 8, 9]])) 42412037Sandreas.sandberg@arm.com assert np.all(zc == np.array([[1., 2, 47], [4, 5, 6], [7, 8, 9]])) 42512037Sandreas.sandberg@arm.com 42612037Sandreas.sandberg@arm.com # You shouldn't be allowed to pass a non-writeable array to a mutating Eigen method: 42712037Sandreas.sandberg@arm.com zro = zr[0:4, 0:4] 42812037Sandreas.sandberg@arm.com zro.flags.writeable = False 42912037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 43012391Sjason@lowepower.com m.add_rm(zro, 0, 0, 0) 43112037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 43212391Sjason@lowepower.com m.add_any(zro, 0, 0, 0) 43312037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 43412391Sjason@lowepower.com m.add1(zro, 0, 0, 0) 43512037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 43612391Sjason@lowepower.com m.add2(zro, 0, 0, 0) 43712037Sandreas.sandberg@arm.com 43812037Sandreas.sandberg@arm.com # integer array shouldn't be passable to a double-matrix-accepting mutating func: 43912037Sandreas.sandberg@arm.com zi = np.array([[1, 2], [3, 4]]) 44012037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 44112391Sjason@lowepower.com m.add_rm(zi) 44212037Sandreas.sandberg@arm.com 44312037Sandreas.sandberg@arm.com 44412037Sandreas.sandberg@arm.comdef test_numpy_ref_mutators(): 44512037Sandreas.sandberg@arm.com """Tests numpy mutating Eigen matrices (for returned Eigen::Ref<...>s)""" 44612037Sandreas.sandberg@arm.com 44712391Sjason@lowepower.com m.reset_refs() # In case another test already changed it 44812391Sjason@lowepower.com 44912391Sjason@lowepower.com zc = m.get_cm_ref() 45012391Sjason@lowepower.com zcro = m.get_cm_const_ref() 45112391Sjason@lowepower.com zr = m.get_rm_ref() 45212391Sjason@lowepower.com zrro = m.get_rm_const_ref() 45312037Sandreas.sandberg@arm.com 45412037Sandreas.sandberg@arm.com assert [zc[1, 2], zcro[1, 2], zr[1, 2], zrro[1, 2]] == [23] * 4 45512037Sandreas.sandberg@arm.com 45612037Sandreas.sandberg@arm.com assert not zc.flags.owndata and zc.flags.writeable 45712037Sandreas.sandberg@arm.com assert not zr.flags.owndata and zr.flags.writeable 45812037Sandreas.sandberg@arm.com assert not zcro.flags.owndata and not zcro.flags.writeable 45912037Sandreas.sandberg@arm.com assert not zrro.flags.owndata and not zrro.flags.writeable 46012037Sandreas.sandberg@arm.com 46112037Sandreas.sandberg@arm.com zc[1, 2] = 99 46212037Sandreas.sandberg@arm.com expect = np.array([[11., 12, 13], [21, 22, 99], [31, 32, 33]]) 46312037Sandreas.sandberg@arm.com # We should have just changed zc, of course, but also zcro and the original eigen matrix 46412037Sandreas.sandberg@arm.com assert np.all(zc == expect) 46512037Sandreas.sandberg@arm.com assert np.all(zcro == expect) 46612391Sjason@lowepower.com assert np.all(m.get_cm_ref() == expect) 46712037Sandreas.sandberg@arm.com 46812037Sandreas.sandberg@arm.com zr[1, 2] = 99 46912037Sandreas.sandberg@arm.com assert np.all(zr == expect) 47012037Sandreas.sandberg@arm.com assert np.all(zrro == expect) 47112391Sjason@lowepower.com assert np.all(m.get_rm_ref() == expect) 47212037Sandreas.sandberg@arm.com 47312037Sandreas.sandberg@arm.com # Make sure the readonly ones are numpy-readonly: 47412037Sandreas.sandberg@arm.com with pytest.raises(ValueError): 47512037Sandreas.sandberg@arm.com zcro[1, 2] = 6 47612037Sandreas.sandberg@arm.com with pytest.raises(ValueError): 47712037Sandreas.sandberg@arm.com zrro[1, 2] = 6 47812037Sandreas.sandberg@arm.com 47912037Sandreas.sandberg@arm.com # We should be able to explicitly copy like this (and since we're copying, 48012037Sandreas.sandberg@arm.com # the const should drop away) 48112391Sjason@lowepower.com y1 = np.array(m.get_cm_const_ref()) 48212037Sandreas.sandberg@arm.com 48312037Sandreas.sandberg@arm.com assert y1.flags.owndata and y1.flags.writeable 48412037Sandreas.sandberg@arm.com # We should get copies of the eigen data, which was modified above: 48512037Sandreas.sandberg@arm.com assert y1[1, 2] == 99 48612037Sandreas.sandberg@arm.com y1[1, 2] += 12 48712037Sandreas.sandberg@arm.com assert y1[1, 2] == 111 48812037Sandreas.sandberg@arm.com assert zc[1, 2] == 99 # Make sure we aren't referencing the original 48912037Sandreas.sandberg@arm.com 49012037Sandreas.sandberg@arm.com 49112037Sandreas.sandberg@arm.comdef test_both_ref_mutators(): 49212037Sandreas.sandberg@arm.com """Tests a complex chain of nested eigen/numpy references""" 49312037Sandreas.sandberg@arm.com 49412391Sjason@lowepower.com m.reset_refs() # In case another test already changed it 49512391Sjason@lowepower.com 49612391Sjason@lowepower.com z = m.get_cm_ref() # numpy -> eigen 49712037Sandreas.sandberg@arm.com z[0, 2] -= 3 49812391Sjason@lowepower.com z2 = m.incr_matrix(z, 1) # numpy -> eigen -> numpy -> eigen 49912037Sandreas.sandberg@arm.com z2[1, 1] += 6 50012391Sjason@lowepower.com z3 = m.incr_matrix(z, 2) # (numpy -> eigen)^3 50112037Sandreas.sandberg@arm.com z3[2, 2] += -5 50212391Sjason@lowepower.com z4 = m.incr_matrix(z, 3) # (numpy -> eigen)^4 50312037Sandreas.sandberg@arm.com z4[1, 1] -= 1 50412391Sjason@lowepower.com z5 = m.incr_matrix(z, 4) # (numpy -> eigen)^5 50512037Sandreas.sandberg@arm.com z5[0, 0] = 0 50612037Sandreas.sandberg@arm.com assert np.all(z == z2) 50712037Sandreas.sandberg@arm.com assert np.all(z == z3) 50812037Sandreas.sandberg@arm.com assert np.all(z == z4) 50912037Sandreas.sandberg@arm.com assert np.all(z == z5) 51012037Sandreas.sandberg@arm.com expect = np.array([[0., 22, 20], [31, 37, 33], [41, 42, 38]]) 51112037Sandreas.sandberg@arm.com assert np.all(z == expect) 51212037Sandreas.sandberg@arm.com 51312037Sandreas.sandberg@arm.com y = np.array(range(100), dtype='float64').reshape(10, 10) 51412391Sjason@lowepower.com y2 = m.incr_matrix_any(y, 10) # np -> eigen -> np 51512391Sjason@lowepower.com y3 = m.incr_matrix_any(y2[0::2, 0::2], -33) # np -> eigen -> np slice -> np -> eigen -> np 51612391Sjason@lowepower.com y4 = m.even_rows(y3) # numpy -> eigen slice -> (... y3) 51712391Sjason@lowepower.com y5 = m.even_cols(y4) # numpy -> eigen slice -> (... y4) 51812391Sjason@lowepower.com y6 = m.incr_matrix_any(y5, 1000) # numpy -> eigen -> (... y5) 51912037Sandreas.sandberg@arm.com 52012037Sandreas.sandberg@arm.com # Apply same mutations using just numpy: 52112037Sandreas.sandberg@arm.com yexpect = np.array(range(100), dtype='float64').reshape(10, 10) 52212037Sandreas.sandberg@arm.com yexpect += 10 52312037Sandreas.sandberg@arm.com yexpect[0::2, 0::2] -= 33 52412037Sandreas.sandberg@arm.com yexpect[0::4, 0::4] += 1000 52512037Sandreas.sandberg@arm.com assert np.all(y6 == yexpect[0::4, 0::4]) 52612037Sandreas.sandberg@arm.com assert np.all(y5 == yexpect[0::4, 0::4]) 52712037Sandreas.sandberg@arm.com assert np.all(y4 == yexpect[0::4, 0::2]) 52812037Sandreas.sandberg@arm.com assert np.all(y3 == yexpect[0::2, 0::2]) 52912037Sandreas.sandberg@arm.com assert np.all(y2 == yexpect) 53012037Sandreas.sandberg@arm.com assert np.all(y == yexpect) 53112037Sandreas.sandberg@arm.com 53212037Sandreas.sandberg@arm.com 53312037Sandreas.sandberg@arm.comdef test_nocopy_wrapper(): 53412037Sandreas.sandberg@arm.com # get_elem requires a column-contiguous matrix reference, but should be 53512037Sandreas.sandberg@arm.com # callable with other types of matrix (via copying): 53612037Sandreas.sandberg@arm.com int_matrix_colmajor = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]], order='F') 53712037Sandreas.sandberg@arm.com dbl_matrix_colmajor = np.array(int_matrix_colmajor, dtype='double', order='F', copy=True) 53812037Sandreas.sandberg@arm.com int_matrix_rowmajor = np.array(int_matrix_colmajor, order='C', copy=True) 53912037Sandreas.sandberg@arm.com dbl_matrix_rowmajor = np.array(int_matrix_rowmajor, dtype='double', order='C', copy=True) 54012037Sandreas.sandberg@arm.com 54112037Sandreas.sandberg@arm.com # All should be callable via get_elem: 54212391Sjason@lowepower.com assert m.get_elem(int_matrix_colmajor) == 8 54312391Sjason@lowepower.com assert m.get_elem(dbl_matrix_colmajor) == 8 54412391Sjason@lowepower.com assert m.get_elem(int_matrix_rowmajor) == 8 54512391Sjason@lowepower.com assert m.get_elem(dbl_matrix_rowmajor) == 8 54612037Sandreas.sandberg@arm.com 54712391Sjason@lowepower.com # All but the second should fail with m.get_elem_nocopy: 54812037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 54912391Sjason@lowepower.com m.get_elem_nocopy(int_matrix_colmajor) 55012037Sandreas.sandberg@arm.com assert ('get_elem_nocopy(): incompatible function arguments.' in str(excinfo.value) and 55112037Sandreas.sandberg@arm.com ', flags.f_contiguous' in str(excinfo.value)) 55212391Sjason@lowepower.com assert m.get_elem_nocopy(dbl_matrix_colmajor) == 8 55312037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 55412391Sjason@lowepower.com m.get_elem_nocopy(int_matrix_rowmajor) 55512037Sandreas.sandberg@arm.com assert ('get_elem_nocopy(): incompatible function arguments.' in str(excinfo.value) and 55612037Sandreas.sandberg@arm.com ', flags.f_contiguous' in str(excinfo.value)) 55712037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 55812391Sjason@lowepower.com m.get_elem_nocopy(dbl_matrix_rowmajor) 55912037Sandreas.sandberg@arm.com assert ('get_elem_nocopy(): incompatible function arguments.' in str(excinfo.value) and 56012037Sandreas.sandberg@arm.com ', flags.f_contiguous' in str(excinfo.value)) 56112037Sandreas.sandberg@arm.com 56212037Sandreas.sandberg@arm.com # For the row-major test, we take a long matrix in row-major, so only the third is allowed: 56312037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 56412391Sjason@lowepower.com m.get_elem_rm_nocopy(int_matrix_colmajor) 56512037Sandreas.sandberg@arm.com assert ('get_elem_rm_nocopy(): incompatible function arguments.' in str(excinfo.value) and 56612037Sandreas.sandberg@arm.com ', flags.c_contiguous' in str(excinfo.value)) 56712037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 56812391Sjason@lowepower.com m.get_elem_rm_nocopy(dbl_matrix_colmajor) 56912037Sandreas.sandberg@arm.com assert ('get_elem_rm_nocopy(): incompatible function arguments.' in str(excinfo.value) and 57012037Sandreas.sandberg@arm.com ', flags.c_contiguous' in str(excinfo.value)) 57112391Sjason@lowepower.com assert m.get_elem_rm_nocopy(int_matrix_rowmajor) == 8 57212037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 57312391Sjason@lowepower.com m.get_elem_rm_nocopy(dbl_matrix_rowmajor) 57412037Sandreas.sandberg@arm.com assert ('get_elem_rm_nocopy(): incompatible function arguments.' in str(excinfo.value) and 57512037Sandreas.sandberg@arm.com ', flags.c_contiguous' in str(excinfo.value)) 57612037Sandreas.sandberg@arm.com 57712037Sandreas.sandberg@arm.com 57812391Sjason@lowepower.comdef test_eigen_ref_life_support(): 57912391Sjason@lowepower.com """Ensure the lifetime of temporary arrays created by the `Ref` caster 58012391Sjason@lowepower.com 58112391Sjason@lowepower.com The `Ref` caster sometimes creates a copy which needs to stay alive. This needs to 58212391Sjason@lowepower.com happen both for directs casts (just the array) or indirectly (e.g. list of arrays). 58312391Sjason@lowepower.com """ 58412391Sjason@lowepower.com 58512391Sjason@lowepower.com a = np.full(shape=10, fill_value=8, dtype=np.int8) 58612391Sjason@lowepower.com assert m.get_elem_direct(a) == 8 58712391Sjason@lowepower.com 58812391Sjason@lowepower.com list_of_a = [a] 58912391Sjason@lowepower.com assert m.get_elem_indirect(list_of_a) == 8 59012391Sjason@lowepower.com 59112391Sjason@lowepower.com 59211986Sandreas.sandberg@arm.comdef test_special_matrix_objects(): 59312391Sjason@lowepower.com assert np.all(m.incr_diag(7) == np.diag([1., 2, 3, 4, 5, 6, 7])) 59411986Sandreas.sandberg@arm.com 59512037Sandreas.sandberg@arm.com asymm = np.array([[ 1., 2, 3, 4], 59611986Sandreas.sandberg@arm.com [ 5, 6, 7, 8], 59711986Sandreas.sandberg@arm.com [ 9, 10, 11, 12], 59811986Sandreas.sandberg@arm.com [13, 14, 15, 16]]) 59911986Sandreas.sandberg@arm.com symm_lower = np.array(asymm) 60011986Sandreas.sandberg@arm.com symm_upper = np.array(asymm) 60111986Sandreas.sandberg@arm.com for i in range(4): 60211986Sandreas.sandberg@arm.com for j in range(i + 1, 4): 60311986Sandreas.sandberg@arm.com symm_lower[i, j] = symm_lower[j, i] 60411986Sandreas.sandberg@arm.com symm_upper[j, i] = symm_upper[i, j] 60511986Sandreas.sandberg@arm.com 60612391Sjason@lowepower.com assert np.all(m.symmetric_lower(asymm) == symm_lower) 60712391Sjason@lowepower.com assert np.all(m.symmetric_upper(asymm) == symm_upper) 60811986Sandreas.sandberg@arm.com 60911986Sandreas.sandberg@arm.com 61011986Sandreas.sandberg@arm.comdef test_dense_signature(doc): 61112391Sjason@lowepower.com assert doc(m.double_col) == """ 61211986Sandreas.sandberg@arm.com double_col(arg0: numpy.ndarray[float32[m, 1]]) -> numpy.ndarray[float32[m, 1]] 61311986Sandreas.sandberg@arm.com """ 61412391Sjason@lowepower.com assert doc(m.double_row) == """ 61511986Sandreas.sandberg@arm.com double_row(arg0: numpy.ndarray[float32[1, n]]) -> numpy.ndarray[float32[1, n]] 61611986Sandreas.sandberg@arm.com """ 61712391Sjason@lowepower.com assert doc(m.double_complex) == """ 61812037Sandreas.sandberg@arm.com double_complex(arg0: numpy.ndarray[complex64[m, 1]]) -> numpy.ndarray[complex64[m, 1]] 61912037Sandreas.sandberg@arm.com """ 62012391Sjason@lowepower.com assert doc(m.double_mat_rm) == """ 62111986Sandreas.sandberg@arm.com double_mat_rm(arg0: numpy.ndarray[float32[m, n]]) -> numpy.ndarray[float32[m, n]] 62211986Sandreas.sandberg@arm.com """ 62311986Sandreas.sandberg@arm.com 62411986Sandreas.sandberg@arm.com 62512391Sjason@lowepower.comdef test_named_arguments(): 62612391Sjason@lowepower.com a = np.array([[1.0, 2], [3, 4], [5, 6]]) 62712391Sjason@lowepower.com b = np.ones((2, 1)) 62812391Sjason@lowepower.com 62912391Sjason@lowepower.com assert np.all(m.matrix_multiply(a, b) == np.array([[3.], [7], [11]])) 63012391Sjason@lowepower.com assert np.all(m.matrix_multiply(A=a, B=b) == np.array([[3.], [7], [11]])) 63112391Sjason@lowepower.com assert np.all(m.matrix_multiply(B=b, A=a) == np.array([[3.], [7], [11]])) 63212391Sjason@lowepower.com 63312391Sjason@lowepower.com with pytest.raises(ValueError) as excinfo: 63412391Sjason@lowepower.com m.matrix_multiply(b, a) 63512391Sjason@lowepower.com assert str(excinfo.value) == 'Nonconformable matrices!' 63612391Sjason@lowepower.com 63712391Sjason@lowepower.com with pytest.raises(ValueError) as excinfo: 63812391Sjason@lowepower.com m.matrix_multiply(A=b, B=a) 63912391Sjason@lowepower.com assert str(excinfo.value) == 'Nonconformable matrices!' 64012391Sjason@lowepower.com 64112391Sjason@lowepower.com with pytest.raises(ValueError) as excinfo: 64212391Sjason@lowepower.com m.matrix_multiply(B=a, A=b) 64312391Sjason@lowepower.com assert str(excinfo.value) == 'Nonconformable matrices!' 64412391Sjason@lowepower.com 64512391Sjason@lowepower.com 64611986Sandreas.sandberg@arm.com@pytest.requires_eigen_and_scipy 64711986Sandreas.sandberg@arm.comdef test_sparse(): 64812391Sjason@lowepower.com assert_sparse_equal_ref(m.sparse_r()) 64912391Sjason@lowepower.com assert_sparse_equal_ref(m.sparse_c()) 65012391Sjason@lowepower.com assert_sparse_equal_ref(m.sparse_copy_r(m.sparse_r())) 65112391Sjason@lowepower.com assert_sparse_equal_ref(m.sparse_copy_c(m.sparse_c())) 65212391Sjason@lowepower.com assert_sparse_equal_ref(m.sparse_copy_r(m.sparse_c())) 65312391Sjason@lowepower.com assert_sparse_equal_ref(m.sparse_copy_c(m.sparse_r())) 65411986Sandreas.sandberg@arm.com 65511986Sandreas.sandberg@arm.com 65611986Sandreas.sandberg@arm.com@pytest.requires_eigen_and_scipy 65711986Sandreas.sandberg@arm.comdef test_sparse_signature(doc): 65812391Sjason@lowepower.com assert doc(m.sparse_copy_r) == """ 65912037Sandreas.sandberg@arm.com sparse_copy_r(arg0: scipy.sparse.csr_matrix[float32]) -> scipy.sparse.csr_matrix[float32] 66011986Sandreas.sandberg@arm.com """ # noqa: E501 line too long 66112391Sjason@lowepower.com assert doc(m.sparse_copy_c) == """ 66212037Sandreas.sandberg@arm.com sparse_copy_c(arg0: scipy.sparse.csc_matrix[float32]) -> scipy.sparse.csc_matrix[float32] 66311986Sandreas.sandberg@arm.com """ # noqa: E501 line too long 66412037Sandreas.sandberg@arm.com 66512037Sandreas.sandberg@arm.com 66612037Sandreas.sandberg@arm.comdef test_issue738(): 66712391Sjason@lowepower.com """Ignore strides on a length-1 dimension (even if they would be incompatible length > 1)""" 66812391Sjason@lowepower.com assert np.all(m.iss738_f1(np.array([[1., 2, 3]])) == np.array([[1., 102, 203]])) 66912391Sjason@lowepower.com assert np.all(m.iss738_f1(np.array([[1.], [2], [3]])) == np.array([[1.], [12], [23]])) 67012037Sandreas.sandberg@arm.com 67112391Sjason@lowepower.com assert np.all(m.iss738_f2(np.array([[1., 2, 3]])) == np.array([[1., 102, 203]])) 67212391Sjason@lowepower.com assert np.all(m.iss738_f2(np.array([[1.], [2], [3]])) == np.array([[1.], [12], [23]])) 67312037Sandreas.sandberg@arm.com 67412037Sandreas.sandberg@arm.com 67512037Sandreas.sandberg@arm.comdef test_custom_operator_new(): 67612037Sandreas.sandberg@arm.com """Using Eigen types as member variables requires a class-specific 67712037Sandreas.sandberg@arm.com operator new with proper alignment""" 67812037Sandreas.sandberg@arm.com 67912391Sjason@lowepower.com o = m.CustomOperatorNew() 68012037Sandreas.sandberg@arm.com np.testing.assert_allclose(o.a, 0.0) 68112037Sandreas.sandberg@arm.com np.testing.assert_allclose(o.b.diagonal(), 1.0) 682