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): 2214299Sbbruce@ucdavis.edu assert_equal_ref(sparse_mat.toarray()) 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 18414299Sbbruce@ucdavis.edu Invoked with: """ + repr(np.array([ 5., 4., 3.], dtype='float32')) # noqa: E501 line too long 18512391Sjason@lowepower.com 18612391Sjason@lowepower.com with pytest.raises(TypeError) as excinfo: 18712391Sjason@lowepower.com m.double_threec(second_col) 18812391Sjason@lowepower.com assert msg(excinfo.value) == """ 18912391Sjason@lowepower.com double_threec(): incompatible function arguments. The following argument types are supported: 19012391Sjason@lowepower.com 1. (arg0: numpy.ndarray[float32[3, 1], flags.writeable]) -> None 19112391Sjason@lowepower.com 19214299Sbbruce@ucdavis.edu Invoked with: """ + repr(np.array([ 7., 4., 1.], dtype='float32')) # noqa: E501 line too long 19312391Sjason@lowepower.com 19412391Sjason@lowepower.com 19511986Sandreas.sandberg@arm.comdef test_nonunit_stride_to_python(): 19612391Sjason@lowepower.com assert np.all(m.diagonal(ref) == ref.diagonal()) 19712391Sjason@lowepower.com assert np.all(m.diagonal_1(ref) == ref.diagonal(1)) 19812391Sjason@lowepower.com for i in range(-5, 7): 19912391Sjason@lowepower.com assert np.all(m.diagonal_n(ref, i) == ref.diagonal(i)), "m.diagonal_n({})".format(i) 20011986Sandreas.sandberg@arm.com 20112391Sjason@lowepower.com assert np.all(m.block(ref, 2, 1, 3, 3) == ref[2:5, 1:4]) 20212391Sjason@lowepower.com assert np.all(m.block(ref, 1, 4, 4, 2) == ref[1:, 4:]) 20312391Sjason@lowepower.com assert np.all(m.block(ref, 1, 4, 3, 2) == ref[1:4, 4:]) 20411986Sandreas.sandberg@arm.com 20511986Sandreas.sandberg@arm.com 20611986Sandreas.sandberg@arm.comdef test_eigen_ref_to_python(): 20712391Sjason@lowepower.com chols = [m.cholesky1, m.cholesky2, m.cholesky3, m.cholesky4] 20811986Sandreas.sandberg@arm.com for i, chol in enumerate(chols, start=1): 20912037Sandreas.sandberg@arm.com mymat = chol(np.array([[1., 2, 4], [2, 13, 23], [4, 23, 77]])) 21011986Sandreas.sandberg@arm.com assert np.all(mymat == np.array([[1, 0, 0], [2, 3, 0], [4, 5, 6]])), "cholesky{}".format(i) 21111986Sandreas.sandberg@arm.com 21211986Sandreas.sandberg@arm.com 21312037Sandreas.sandberg@arm.comdef assign_both(a1, a2, r, c, v): 21412037Sandreas.sandberg@arm.com a1[r, c] = v 21512037Sandreas.sandberg@arm.com a2[r, c] = v 21612037Sandreas.sandberg@arm.com 21712037Sandreas.sandberg@arm.com 21812037Sandreas.sandberg@arm.comdef array_copy_but_one(a, r, c, v): 21912037Sandreas.sandberg@arm.com z = np.array(a, copy=True) 22012037Sandreas.sandberg@arm.com z[r, c] = v 22112037Sandreas.sandberg@arm.com return z 22212037Sandreas.sandberg@arm.com 22312037Sandreas.sandberg@arm.com 22412037Sandreas.sandberg@arm.comdef test_eigen_return_references(): 22512037Sandreas.sandberg@arm.com """Tests various ways of returning references and non-referencing copies""" 22612391Sjason@lowepower.com 22712037Sandreas.sandberg@arm.com master = np.ones((10, 10)) 22812391Sjason@lowepower.com a = m.ReturnTester() 22912037Sandreas.sandberg@arm.com a_get1 = a.get() 23012037Sandreas.sandberg@arm.com assert not a_get1.flags.owndata and a_get1.flags.writeable 23112037Sandreas.sandberg@arm.com assign_both(a_get1, master, 3, 3, 5) 23212037Sandreas.sandberg@arm.com a_get2 = a.get_ptr() 23312037Sandreas.sandberg@arm.com assert not a_get2.flags.owndata and a_get2.flags.writeable 23412037Sandreas.sandberg@arm.com assign_both(a_get1, master, 2, 3, 6) 23512037Sandreas.sandberg@arm.com 23612037Sandreas.sandberg@arm.com a_view1 = a.view() 23712037Sandreas.sandberg@arm.com assert not a_view1.flags.owndata and not a_view1.flags.writeable 23812037Sandreas.sandberg@arm.com with pytest.raises(ValueError): 23912037Sandreas.sandberg@arm.com a_view1[2, 3] = 4 24012037Sandreas.sandberg@arm.com a_view2 = a.view_ptr() 24112037Sandreas.sandberg@arm.com assert not a_view2.flags.owndata and not a_view2.flags.writeable 24212037Sandreas.sandberg@arm.com with pytest.raises(ValueError): 24312037Sandreas.sandberg@arm.com a_view2[2, 3] = 4 24412037Sandreas.sandberg@arm.com 24512037Sandreas.sandberg@arm.com a_copy1 = a.copy_get() 24612037Sandreas.sandberg@arm.com assert a_copy1.flags.owndata and a_copy1.flags.writeable 24712037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy1, master) 24812037Sandreas.sandberg@arm.com a_copy1[7, 7] = -44 # Shouldn't affect anything else 24912037Sandreas.sandberg@arm.com c1want = array_copy_but_one(master, 7, 7, -44) 25012037Sandreas.sandberg@arm.com a_copy2 = a.copy_view() 25112037Sandreas.sandberg@arm.com assert a_copy2.flags.owndata and a_copy2.flags.writeable 25212037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy2, master) 25312037Sandreas.sandberg@arm.com a_copy2[4, 4] = -22 # Shouldn't affect anything else 25412037Sandreas.sandberg@arm.com c2want = array_copy_but_one(master, 4, 4, -22) 25512037Sandreas.sandberg@arm.com 25612037Sandreas.sandberg@arm.com a_ref1 = a.ref() 25712037Sandreas.sandberg@arm.com assert not a_ref1.flags.owndata and a_ref1.flags.writeable 25812037Sandreas.sandberg@arm.com assign_both(a_ref1, master, 1, 1, 15) 25912037Sandreas.sandberg@arm.com a_ref2 = a.ref_const() 26012037Sandreas.sandberg@arm.com assert not a_ref2.flags.owndata and not a_ref2.flags.writeable 26112037Sandreas.sandberg@arm.com with pytest.raises(ValueError): 26212037Sandreas.sandberg@arm.com a_ref2[5, 5] = 33 26312037Sandreas.sandberg@arm.com a_ref3 = a.ref_safe() 26412037Sandreas.sandberg@arm.com assert not a_ref3.flags.owndata and a_ref3.flags.writeable 26512037Sandreas.sandberg@arm.com assign_both(a_ref3, master, 0, 7, 99) 26612037Sandreas.sandberg@arm.com a_ref4 = a.ref_const_safe() 26712037Sandreas.sandberg@arm.com assert not a_ref4.flags.owndata and not a_ref4.flags.writeable 26812037Sandreas.sandberg@arm.com with pytest.raises(ValueError): 26912037Sandreas.sandberg@arm.com a_ref4[7, 0] = 987654321 27012037Sandreas.sandberg@arm.com 27112037Sandreas.sandberg@arm.com a_copy3 = a.copy_ref() 27212037Sandreas.sandberg@arm.com assert a_copy3.flags.owndata and a_copy3.flags.writeable 27312037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy3, master) 27412037Sandreas.sandberg@arm.com a_copy3[8, 1] = 11 27512037Sandreas.sandberg@arm.com c3want = array_copy_but_one(master, 8, 1, 11) 27612037Sandreas.sandberg@arm.com a_copy4 = a.copy_ref_const() 27712037Sandreas.sandberg@arm.com assert a_copy4.flags.owndata and a_copy4.flags.writeable 27812037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy4, master) 27912037Sandreas.sandberg@arm.com a_copy4[8, 4] = 88 28012037Sandreas.sandberg@arm.com c4want = array_copy_but_one(master, 8, 4, 88) 28112037Sandreas.sandberg@arm.com 28212037Sandreas.sandberg@arm.com a_block1 = a.block(3, 3, 2, 2) 28312037Sandreas.sandberg@arm.com assert not a_block1.flags.owndata and a_block1.flags.writeable 28412037Sandreas.sandberg@arm.com a_block1[0, 0] = 55 28512037Sandreas.sandberg@arm.com master[3, 3] = 55 28612037Sandreas.sandberg@arm.com a_block2 = a.block_safe(2, 2, 3, 2) 28712037Sandreas.sandberg@arm.com assert not a_block2.flags.owndata and a_block2.flags.writeable 28812037Sandreas.sandberg@arm.com a_block2[2, 1] = -123 28912037Sandreas.sandberg@arm.com master[4, 3] = -123 29012037Sandreas.sandberg@arm.com a_block3 = a.block_const(6, 7, 4, 3) 29112037Sandreas.sandberg@arm.com assert not a_block3.flags.owndata and not a_block3.flags.writeable 29212037Sandreas.sandberg@arm.com with pytest.raises(ValueError): 29312037Sandreas.sandberg@arm.com a_block3[2, 2] = -44444 29412037Sandreas.sandberg@arm.com 29512037Sandreas.sandberg@arm.com a_copy5 = a.copy_block(2, 2, 2, 3) 29612037Sandreas.sandberg@arm.com assert a_copy5.flags.owndata and a_copy5.flags.writeable 29712037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy5, master[2:4, 2:5]) 29812037Sandreas.sandberg@arm.com a_copy5[1, 1] = 777 29912037Sandreas.sandberg@arm.com c5want = array_copy_but_one(master[2:4, 2:5], 1, 1, 777) 30012037Sandreas.sandberg@arm.com 30112037Sandreas.sandberg@arm.com a_corn1 = a.corners() 30212037Sandreas.sandberg@arm.com assert not a_corn1.flags.owndata and a_corn1.flags.writeable 30312037Sandreas.sandberg@arm.com a_corn1 *= 50 30412037Sandreas.sandberg@arm.com a_corn1[1, 1] = 999 30512037Sandreas.sandberg@arm.com master[0, 0] = 50 30612037Sandreas.sandberg@arm.com master[0, 9] = 50 30712037Sandreas.sandberg@arm.com master[9, 0] = 50 30812037Sandreas.sandberg@arm.com master[9, 9] = 999 30912037Sandreas.sandberg@arm.com a_corn2 = a.corners_const() 31012037Sandreas.sandberg@arm.com assert not a_corn2.flags.owndata and not a_corn2.flags.writeable 31112037Sandreas.sandberg@arm.com with pytest.raises(ValueError): 31212037Sandreas.sandberg@arm.com a_corn2[1, 0] = 51 31312037Sandreas.sandberg@arm.com 31412037Sandreas.sandberg@arm.com # All of the changes made all the way along should be visible everywhere 31512037Sandreas.sandberg@arm.com # now (except for the copies, of course) 31612037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_get1, master) 31712037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_get2, master) 31812037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_view1, master) 31912037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_view2, master) 32012037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_ref1, master) 32112037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_ref2, master) 32212037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_ref3, master) 32312037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_ref4, master) 32412037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_block1, master[3:5, 3:5]) 32512037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_block2, master[2:5, 2:4]) 32612037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_block3, master[6:10, 7:10]) 32712037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_corn1, master[0::master.shape[0] - 1, 0::master.shape[1] - 1]) 32812037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_corn2, master[0::master.shape[0] - 1, 0::master.shape[1] - 1]) 32912037Sandreas.sandberg@arm.com 33012037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy1, c1want) 33112037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy2, c2want) 33212037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy3, c3want) 33312037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy4, c4want) 33412037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy5, c5want) 33512037Sandreas.sandberg@arm.com 33612037Sandreas.sandberg@arm.com 33712037Sandreas.sandberg@arm.comdef assert_keeps_alive(cl, method, *args): 33812037Sandreas.sandberg@arm.com cstats = ConstructorStats.get(cl) 33912037Sandreas.sandberg@arm.com start_with = cstats.alive() 34012037Sandreas.sandberg@arm.com a = cl() 34112037Sandreas.sandberg@arm.com assert cstats.alive() == start_with + 1 34212037Sandreas.sandberg@arm.com z = method(a, *args) 34312037Sandreas.sandberg@arm.com assert cstats.alive() == start_with + 1 34412037Sandreas.sandberg@arm.com del a 34512037Sandreas.sandberg@arm.com # Here's the keep alive in action: 34612037Sandreas.sandberg@arm.com assert cstats.alive() == start_with + 1 34712037Sandreas.sandberg@arm.com del z 34812037Sandreas.sandberg@arm.com # Keep alive should have expired: 34912037Sandreas.sandberg@arm.com assert cstats.alive() == start_with 35012037Sandreas.sandberg@arm.com 35112037Sandreas.sandberg@arm.com 35212037Sandreas.sandberg@arm.comdef test_eigen_keepalive(): 35312391Sjason@lowepower.com a = m.ReturnTester() 35412391Sjason@lowepower.com cstats = ConstructorStats.get(m.ReturnTester) 35512037Sandreas.sandberg@arm.com assert cstats.alive() == 1 35612037Sandreas.sandberg@arm.com unsafe = [a.ref(), a.ref_const(), a.block(1, 2, 3, 4)] 35712037Sandreas.sandberg@arm.com copies = [a.copy_get(), a.copy_view(), a.copy_ref(), a.copy_ref_const(), 35812037Sandreas.sandberg@arm.com a.copy_block(4, 3, 2, 1)] 35912037Sandreas.sandberg@arm.com del a 36012037Sandreas.sandberg@arm.com assert cstats.alive() == 0 36112037Sandreas.sandberg@arm.com del unsafe 36212037Sandreas.sandberg@arm.com del copies 36312037Sandreas.sandberg@arm.com 36412391Sjason@lowepower.com for meth in [m.ReturnTester.get, m.ReturnTester.get_ptr, m.ReturnTester.view, 36512391Sjason@lowepower.com m.ReturnTester.view_ptr, m.ReturnTester.ref_safe, m.ReturnTester.ref_const_safe, 36612391Sjason@lowepower.com m.ReturnTester.corners, m.ReturnTester.corners_const]: 36712391Sjason@lowepower.com assert_keeps_alive(m.ReturnTester, meth) 36812037Sandreas.sandberg@arm.com 36912391Sjason@lowepower.com for meth in [m.ReturnTester.block_safe, m.ReturnTester.block_const]: 37012391Sjason@lowepower.com assert_keeps_alive(m.ReturnTester, meth, 4, 3, 2, 1) 37112037Sandreas.sandberg@arm.com 37212037Sandreas.sandberg@arm.com 37312037Sandreas.sandberg@arm.comdef test_eigen_ref_mutators(): 37412391Sjason@lowepower.com """Tests Eigen's ability to mutate numpy values""" 37512391Sjason@lowepower.com 37612037Sandreas.sandberg@arm.com orig = np.array([[1., 2, 3], [4, 5, 6], [7, 8, 9]]) 37712037Sandreas.sandberg@arm.com zr = np.array(orig) 37812037Sandreas.sandberg@arm.com zc = np.array(orig, order='F') 37912391Sjason@lowepower.com m.add_rm(zr, 1, 0, 100) 38012037Sandreas.sandberg@arm.com assert np.all(zr == np.array([[1., 2, 3], [104, 5, 6], [7, 8, 9]])) 38112391Sjason@lowepower.com m.add_cm(zc, 1, 0, 200) 38212037Sandreas.sandberg@arm.com assert np.all(zc == np.array([[1., 2, 3], [204, 5, 6], [7, 8, 9]])) 38312037Sandreas.sandberg@arm.com 38412391Sjason@lowepower.com m.add_any(zr, 1, 0, 20) 38512037Sandreas.sandberg@arm.com assert np.all(zr == np.array([[1., 2, 3], [124, 5, 6], [7, 8, 9]])) 38612391Sjason@lowepower.com m.add_any(zc, 1, 0, 10) 38712037Sandreas.sandberg@arm.com assert np.all(zc == np.array([[1., 2, 3], [214, 5, 6], [7, 8, 9]])) 38812037Sandreas.sandberg@arm.com 38912037Sandreas.sandberg@arm.com # Can't reference a col-major array with a row-major Ref, and vice versa: 39012037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 39112391Sjason@lowepower.com m.add_rm(zc, 1, 0, 1) 39212037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 39312391Sjason@lowepower.com m.add_cm(zr, 1, 0, 1) 39412037Sandreas.sandberg@arm.com 39512037Sandreas.sandberg@arm.com # Overloads: 39612391Sjason@lowepower.com m.add1(zr, 1, 0, -100) 39712391Sjason@lowepower.com m.add2(zr, 1, 0, -20) 39812037Sandreas.sandberg@arm.com assert np.all(zr == orig) 39912391Sjason@lowepower.com m.add1(zc, 1, 0, -200) 40012391Sjason@lowepower.com m.add2(zc, 1, 0, -10) 40112037Sandreas.sandberg@arm.com assert np.all(zc == orig) 40212037Sandreas.sandberg@arm.com 40312037Sandreas.sandberg@arm.com # a non-contiguous slice (this won't work on either the row- or 40412037Sandreas.sandberg@arm.com # column-contiguous refs, but should work for the any) 40512037Sandreas.sandberg@arm.com cornersr = zr[0::2, 0::2] 40612037Sandreas.sandberg@arm.com cornersc = zc[0::2, 0::2] 40712037Sandreas.sandberg@arm.com 40812037Sandreas.sandberg@arm.com assert np.all(cornersr == np.array([[1., 3], [7, 9]])) 40912037Sandreas.sandberg@arm.com assert np.all(cornersc == np.array([[1., 3], [7, 9]])) 41012037Sandreas.sandberg@arm.com 41112037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 41212391Sjason@lowepower.com m.add_rm(cornersr, 0, 1, 25) 41312037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 41412391Sjason@lowepower.com m.add_cm(cornersr, 0, 1, 25) 41512037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 41612391Sjason@lowepower.com m.add_rm(cornersc, 0, 1, 25) 41712037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 41812391Sjason@lowepower.com m.add_cm(cornersc, 0, 1, 25) 41912391Sjason@lowepower.com m.add_any(cornersr, 0, 1, 25) 42012391Sjason@lowepower.com m.add_any(cornersc, 0, 1, 44) 42112037Sandreas.sandberg@arm.com assert np.all(zr == np.array([[1., 2, 28], [4, 5, 6], [7, 8, 9]])) 42212037Sandreas.sandberg@arm.com assert np.all(zc == np.array([[1., 2, 47], [4, 5, 6], [7, 8, 9]])) 42312037Sandreas.sandberg@arm.com 42412037Sandreas.sandberg@arm.com # You shouldn't be allowed to pass a non-writeable array to a mutating Eigen method: 42512037Sandreas.sandberg@arm.com zro = zr[0:4, 0:4] 42612037Sandreas.sandberg@arm.com zro.flags.writeable = False 42712037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 42812391Sjason@lowepower.com m.add_rm(zro, 0, 0, 0) 42912037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 43012391Sjason@lowepower.com m.add_any(zro, 0, 0, 0) 43112037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 43212391Sjason@lowepower.com m.add1(zro, 0, 0, 0) 43312037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 43412391Sjason@lowepower.com m.add2(zro, 0, 0, 0) 43512037Sandreas.sandberg@arm.com 43612037Sandreas.sandberg@arm.com # integer array shouldn't be passable to a double-matrix-accepting mutating func: 43712037Sandreas.sandberg@arm.com zi = np.array([[1, 2], [3, 4]]) 43812037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 43912391Sjason@lowepower.com m.add_rm(zi) 44012037Sandreas.sandberg@arm.com 44112037Sandreas.sandberg@arm.com 44212037Sandreas.sandberg@arm.comdef test_numpy_ref_mutators(): 44312037Sandreas.sandberg@arm.com """Tests numpy mutating Eigen matrices (for returned Eigen::Ref<...>s)""" 44412037Sandreas.sandberg@arm.com 44512391Sjason@lowepower.com m.reset_refs() # In case another test already changed it 44612391Sjason@lowepower.com 44712391Sjason@lowepower.com zc = m.get_cm_ref() 44812391Sjason@lowepower.com zcro = m.get_cm_const_ref() 44912391Sjason@lowepower.com zr = m.get_rm_ref() 45012391Sjason@lowepower.com zrro = m.get_rm_const_ref() 45112037Sandreas.sandberg@arm.com 45212037Sandreas.sandberg@arm.com assert [zc[1, 2], zcro[1, 2], zr[1, 2], zrro[1, 2]] == [23] * 4 45312037Sandreas.sandberg@arm.com 45412037Sandreas.sandberg@arm.com assert not zc.flags.owndata and zc.flags.writeable 45512037Sandreas.sandberg@arm.com assert not zr.flags.owndata and zr.flags.writeable 45612037Sandreas.sandberg@arm.com assert not zcro.flags.owndata and not zcro.flags.writeable 45712037Sandreas.sandberg@arm.com assert not zrro.flags.owndata and not zrro.flags.writeable 45812037Sandreas.sandberg@arm.com 45912037Sandreas.sandberg@arm.com zc[1, 2] = 99 46012037Sandreas.sandberg@arm.com expect = np.array([[11., 12, 13], [21, 22, 99], [31, 32, 33]]) 46112037Sandreas.sandberg@arm.com # We should have just changed zc, of course, but also zcro and the original eigen matrix 46212037Sandreas.sandberg@arm.com assert np.all(zc == expect) 46312037Sandreas.sandberg@arm.com assert np.all(zcro == expect) 46412391Sjason@lowepower.com assert np.all(m.get_cm_ref() == expect) 46512037Sandreas.sandberg@arm.com 46612037Sandreas.sandberg@arm.com zr[1, 2] = 99 46712037Sandreas.sandberg@arm.com assert np.all(zr == expect) 46812037Sandreas.sandberg@arm.com assert np.all(zrro == expect) 46912391Sjason@lowepower.com assert np.all(m.get_rm_ref() == expect) 47012037Sandreas.sandberg@arm.com 47112037Sandreas.sandberg@arm.com # Make sure the readonly ones are numpy-readonly: 47212037Sandreas.sandberg@arm.com with pytest.raises(ValueError): 47312037Sandreas.sandberg@arm.com zcro[1, 2] = 6 47412037Sandreas.sandberg@arm.com with pytest.raises(ValueError): 47512037Sandreas.sandberg@arm.com zrro[1, 2] = 6 47612037Sandreas.sandberg@arm.com 47712037Sandreas.sandberg@arm.com # We should be able to explicitly copy like this (and since we're copying, 47812037Sandreas.sandberg@arm.com # the const should drop away) 47912391Sjason@lowepower.com y1 = np.array(m.get_cm_const_ref()) 48012037Sandreas.sandberg@arm.com 48112037Sandreas.sandberg@arm.com assert y1.flags.owndata and y1.flags.writeable 48212037Sandreas.sandberg@arm.com # We should get copies of the eigen data, which was modified above: 48312037Sandreas.sandberg@arm.com assert y1[1, 2] == 99 48412037Sandreas.sandberg@arm.com y1[1, 2] += 12 48512037Sandreas.sandberg@arm.com assert y1[1, 2] == 111 48612037Sandreas.sandberg@arm.com assert zc[1, 2] == 99 # Make sure we aren't referencing the original 48712037Sandreas.sandberg@arm.com 48812037Sandreas.sandberg@arm.com 48912037Sandreas.sandberg@arm.comdef test_both_ref_mutators(): 49012037Sandreas.sandberg@arm.com """Tests a complex chain of nested eigen/numpy references""" 49112037Sandreas.sandberg@arm.com 49212391Sjason@lowepower.com m.reset_refs() # In case another test already changed it 49312391Sjason@lowepower.com 49412391Sjason@lowepower.com z = m.get_cm_ref() # numpy -> eigen 49512037Sandreas.sandberg@arm.com z[0, 2] -= 3 49612391Sjason@lowepower.com z2 = m.incr_matrix(z, 1) # numpy -> eigen -> numpy -> eigen 49712037Sandreas.sandberg@arm.com z2[1, 1] += 6 49812391Sjason@lowepower.com z3 = m.incr_matrix(z, 2) # (numpy -> eigen)^3 49912037Sandreas.sandberg@arm.com z3[2, 2] += -5 50012391Sjason@lowepower.com z4 = m.incr_matrix(z, 3) # (numpy -> eigen)^4 50112037Sandreas.sandberg@arm.com z4[1, 1] -= 1 50212391Sjason@lowepower.com z5 = m.incr_matrix(z, 4) # (numpy -> eigen)^5 50312037Sandreas.sandberg@arm.com z5[0, 0] = 0 50412037Sandreas.sandberg@arm.com assert np.all(z == z2) 50512037Sandreas.sandberg@arm.com assert np.all(z == z3) 50612037Sandreas.sandberg@arm.com assert np.all(z == z4) 50712037Sandreas.sandberg@arm.com assert np.all(z == z5) 50812037Sandreas.sandberg@arm.com expect = np.array([[0., 22, 20], [31, 37, 33], [41, 42, 38]]) 50912037Sandreas.sandberg@arm.com assert np.all(z == expect) 51012037Sandreas.sandberg@arm.com 51112037Sandreas.sandberg@arm.com y = np.array(range(100), dtype='float64').reshape(10, 10) 51212391Sjason@lowepower.com y2 = m.incr_matrix_any(y, 10) # np -> eigen -> np 51312391Sjason@lowepower.com y3 = m.incr_matrix_any(y2[0::2, 0::2], -33) # np -> eigen -> np slice -> np -> eigen -> np 51412391Sjason@lowepower.com y4 = m.even_rows(y3) # numpy -> eigen slice -> (... y3) 51512391Sjason@lowepower.com y5 = m.even_cols(y4) # numpy -> eigen slice -> (... y4) 51612391Sjason@lowepower.com y6 = m.incr_matrix_any(y5, 1000) # numpy -> eigen -> (... y5) 51712037Sandreas.sandberg@arm.com 51812037Sandreas.sandberg@arm.com # Apply same mutations using just numpy: 51912037Sandreas.sandberg@arm.com yexpect = np.array(range(100), dtype='float64').reshape(10, 10) 52012037Sandreas.sandberg@arm.com yexpect += 10 52112037Sandreas.sandberg@arm.com yexpect[0::2, 0::2] -= 33 52212037Sandreas.sandberg@arm.com yexpect[0::4, 0::4] += 1000 52312037Sandreas.sandberg@arm.com assert np.all(y6 == yexpect[0::4, 0::4]) 52412037Sandreas.sandberg@arm.com assert np.all(y5 == yexpect[0::4, 0::4]) 52512037Sandreas.sandberg@arm.com assert np.all(y4 == yexpect[0::4, 0::2]) 52612037Sandreas.sandberg@arm.com assert np.all(y3 == yexpect[0::2, 0::2]) 52712037Sandreas.sandberg@arm.com assert np.all(y2 == yexpect) 52812037Sandreas.sandberg@arm.com assert np.all(y == yexpect) 52912037Sandreas.sandberg@arm.com 53012037Sandreas.sandberg@arm.com 53112037Sandreas.sandberg@arm.comdef test_nocopy_wrapper(): 53212037Sandreas.sandberg@arm.com # get_elem requires a column-contiguous matrix reference, but should be 53312037Sandreas.sandberg@arm.com # callable with other types of matrix (via copying): 53412037Sandreas.sandberg@arm.com int_matrix_colmajor = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]], order='F') 53512037Sandreas.sandberg@arm.com dbl_matrix_colmajor = np.array(int_matrix_colmajor, dtype='double', order='F', copy=True) 53612037Sandreas.sandberg@arm.com int_matrix_rowmajor = np.array(int_matrix_colmajor, order='C', copy=True) 53712037Sandreas.sandberg@arm.com dbl_matrix_rowmajor = np.array(int_matrix_rowmajor, dtype='double', order='C', copy=True) 53812037Sandreas.sandberg@arm.com 53912037Sandreas.sandberg@arm.com # All should be callable via get_elem: 54012391Sjason@lowepower.com assert m.get_elem(int_matrix_colmajor) == 8 54112391Sjason@lowepower.com assert m.get_elem(dbl_matrix_colmajor) == 8 54212391Sjason@lowepower.com assert m.get_elem(int_matrix_rowmajor) == 8 54312391Sjason@lowepower.com assert m.get_elem(dbl_matrix_rowmajor) == 8 54412037Sandreas.sandberg@arm.com 54512391Sjason@lowepower.com # All but the second should fail with m.get_elem_nocopy: 54612037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 54712391Sjason@lowepower.com m.get_elem_nocopy(int_matrix_colmajor) 54812037Sandreas.sandberg@arm.com assert ('get_elem_nocopy(): incompatible function arguments.' in str(excinfo.value) and 54912037Sandreas.sandberg@arm.com ', flags.f_contiguous' in str(excinfo.value)) 55012391Sjason@lowepower.com assert m.get_elem_nocopy(dbl_matrix_colmajor) == 8 55112037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 55212391Sjason@lowepower.com m.get_elem_nocopy(int_matrix_rowmajor) 55312037Sandreas.sandberg@arm.com assert ('get_elem_nocopy(): incompatible function arguments.' in str(excinfo.value) and 55412037Sandreas.sandberg@arm.com ', flags.f_contiguous' in str(excinfo.value)) 55512037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 55612391Sjason@lowepower.com m.get_elem_nocopy(dbl_matrix_rowmajor) 55712037Sandreas.sandberg@arm.com assert ('get_elem_nocopy(): incompatible function arguments.' in str(excinfo.value) and 55812037Sandreas.sandberg@arm.com ', flags.f_contiguous' in str(excinfo.value)) 55912037Sandreas.sandberg@arm.com 56012037Sandreas.sandberg@arm.com # For the row-major test, we take a long matrix in row-major, so only the third is allowed: 56112037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 56212391Sjason@lowepower.com m.get_elem_rm_nocopy(int_matrix_colmajor) 56312037Sandreas.sandberg@arm.com assert ('get_elem_rm_nocopy(): incompatible function arguments.' in str(excinfo.value) and 56412037Sandreas.sandberg@arm.com ', flags.c_contiguous' in str(excinfo.value)) 56512037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 56612391Sjason@lowepower.com m.get_elem_rm_nocopy(dbl_matrix_colmajor) 56712037Sandreas.sandberg@arm.com assert ('get_elem_rm_nocopy(): incompatible function arguments.' in str(excinfo.value) and 56812037Sandreas.sandberg@arm.com ', flags.c_contiguous' in str(excinfo.value)) 56912391Sjason@lowepower.com assert m.get_elem_rm_nocopy(int_matrix_rowmajor) == 8 57012037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 57112391Sjason@lowepower.com m.get_elem_rm_nocopy(dbl_matrix_rowmajor) 57212037Sandreas.sandberg@arm.com assert ('get_elem_rm_nocopy(): incompatible function arguments.' in str(excinfo.value) and 57312037Sandreas.sandberg@arm.com ', flags.c_contiguous' in str(excinfo.value)) 57412037Sandreas.sandberg@arm.com 57512037Sandreas.sandberg@arm.com 57612391Sjason@lowepower.comdef test_eigen_ref_life_support(): 57712391Sjason@lowepower.com """Ensure the lifetime of temporary arrays created by the `Ref` caster 57812391Sjason@lowepower.com 57912391Sjason@lowepower.com The `Ref` caster sometimes creates a copy which needs to stay alive. This needs to 58012391Sjason@lowepower.com happen both for directs casts (just the array) or indirectly (e.g. list of arrays). 58112391Sjason@lowepower.com """ 58212391Sjason@lowepower.com 58312391Sjason@lowepower.com a = np.full(shape=10, fill_value=8, dtype=np.int8) 58412391Sjason@lowepower.com assert m.get_elem_direct(a) == 8 58512391Sjason@lowepower.com 58612391Sjason@lowepower.com list_of_a = [a] 58712391Sjason@lowepower.com assert m.get_elem_indirect(list_of_a) == 8 58812391Sjason@lowepower.com 58912391Sjason@lowepower.com 59011986Sandreas.sandberg@arm.comdef test_special_matrix_objects(): 59112391Sjason@lowepower.com assert np.all(m.incr_diag(7) == np.diag([1., 2, 3, 4, 5, 6, 7])) 59211986Sandreas.sandberg@arm.com 59312037Sandreas.sandberg@arm.com asymm = np.array([[ 1., 2, 3, 4], 59411986Sandreas.sandberg@arm.com [ 5, 6, 7, 8], 59511986Sandreas.sandberg@arm.com [ 9, 10, 11, 12], 59611986Sandreas.sandberg@arm.com [13, 14, 15, 16]]) 59711986Sandreas.sandberg@arm.com symm_lower = np.array(asymm) 59811986Sandreas.sandberg@arm.com symm_upper = np.array(asymm) 59911986Sandreas.sandberg@arm.com for i in range(4): 60011986Sandreas.sandberg@arm.com for j in range(i + 1, 4): 60111986Sandreas.sandberg@arm.com symm_lower[i, j] = symm_lower[j, i] 60211986Sandreas.sandberg@arm.com symm_upper[j, i] = symm_upper[i, j] 60311986Sandreas.sandberg@arm.com 60412391Sjason@lowepower.com assert np.all(m.symmetric_lower(asymm) == symm_lower) 60512391Sjason@lowepower.com assert np.all(m.symmetric_upper(asymm) == symm_upper) 60611986Sandreas.sandberg@arm.com 60711986Sandreas.sandberg@arm.com 60811986Sandreas.sandberg@arm.comdef test_dense_signature(doc): 60912391Sjason@lowepower.com assert doc(m.double_col) == """ 61011986Sandreas.sandberg@arm.com double_col(arg0: numpy.ndarray[float32[m, 1]]) -> numpy.ndarray[float32[m, 1]] 61111986Sandreas.sandberg@arm.com """ 61212391Sjason@lowepower.com assert doc(m.double_row) == """ 61311986Sandreas.sandberg@arm.com double_row(arg0: numpy.ndarray[float32[1, n]]) -> numpy.ndarray[float32[1, n]] 61411986Sandreas.sandberg@arm.com """ 61512391Sjason@lowepower.com assert doc(m.double_complex) == """ 61612037Sandreas.sandberg@arm.com double_complex(arg0: numpy.ndarray[complex64[m, 1]]) -> numpy.ndarray[complex64[m, 1]] 61712037Sandreas.sandberg@arm.com """ 61812391Sjason@lowepower.com assert doc(m.double_mat_rm) == """ 61911986Sandreas.sandberg@arm.com double_mat_rm(arg0: numpy.ndarray[float32[m, n]]) -> numpy.ndarray[float32[m, n]] 62011986Sandreas.sandberg@arm.com """ 62111986Sandreas.sandberg@arm.com 62211986Sandreas.sandberg@arm.com 62312391Sjason@lowepower.comdef test_named_arguments(): 62412391Sjason@lowepower.com a = np.array([[1.0, 2], [3, 4], [5, 6]]) 62512391Sjason@lowepower.com b = np.ones((2, 1)) 62612391Sjason@lowepower.com 62712391Sjason@lowepower.com assert np.all(m.matrix_multiply(a, b) == np.array([[3.], [7], [11]])) 62812391Sjason@lowepower.com assert np.all(m.matrix_multiply(A=a, B=b) == np.array([[3.], [7], [11]])) 62912391Sjason@lowepower.com assert np.all(m.matrix_multiply(B=b, A=a) == np.array([[3.], [7], [11]])) 63012391Sjason@lowepower.com 63112391Sjason@lowepower.com with pytest.raises(ValueError) as excinfo: 63212391Sjason@lowepower.com m.matrix_multiply(b, a) 63312391Sjason@lowepower.com assert str(excinfo.value) == 'Nonconformable matrices!' 63412391Sjason@lowepower.com 63512391Sjason@lowepower.com with pytest.raises(ValueError) as excinfo: 63612391Sjason@lowepower.com m.matrix_multiply(A=b, B=a) 63712391Sjason@lowepower.com assert str(excinfo.value) == 'Nonconformable matrices!' 63812391Sjason@lowepower.com 63912391Sjason@lowepower.com with pytest.raises(ValueError) as excinfo: 64012391Sjason@lowepower.com m.matrix_multiply(B=a, A=b) 64112391Sjason@lowepower.com assert str(excinfo.value) == 'Nonconformable matrices!' 64212391Sjason@lowepower.com 64312391Sjason@lowepower.com 64411986Sandreas.sandberg@arm.com@pytest.requires_eigen_and_scipy 64511986Sandreas.sandberg@arm.comdef test_sparse(): 64612391Sjason@lowepower.com assert_sparse_equal_ref(m.sparse_r()) 64712391Sjason@lowepower.com assert_sparse_equal_ref(m.sparse_c()) 64812391Sjason@lowepower.com assert_sparse_equal_ref(m.sparse_copy_r(m.sparse_r())) 64912391Sjason@lowepower.com assert_sparse_equal_ref(m.sparse_copy_c(m.sparse_c())) 65012391Sjason@lowepower.com assert_sparse_equal_ref(m.sparse_copy_r(m.sparse_c())) 65112391Sjason@lowepower.com assert_sparse_equal_ref(m.sparse_copy_c(m.sparse_r())) 65211986Sandreas.sandberg@arm.com 65311986Sandreas.sandberg@arm.com 65411986Sandreas.sandberg@arm.com@pytest.requires_eigen_and_scipy 65511986Sandreas.sandberg@arm.comdef test_sparse_signature(doc): 65612391Sjason@lowepower.com assert doc(m.sparse_copy_r) == """ 65712037Sandreas.sandberg@arm.com sparse_copy_r(arg0: scipy.sparse.csr_matrix[float32]) -> scipy.sparse.csr_matrix[float32] 65811986Sandreas.sandberg@arm.com """ # noqa: E501 line too long 65912391Sjason@lowepower.com assert doc(m.sparse_copy_c) == """ 66012037Sandreas.sandberg@arm.com sparse_copy_c(arg0: scipy.sparse.csc_matrix[float32]) -> scipy.sparse.csc_matrix[float32] 66111986Sandreas.sandberg@arm.com """ # noqa: E501 line too long 66212037Sandreas.sandberg@arm.com 66312037Sandreas.sandberg@arm.com 66412037Sandreas.sandberg@arm.comdef test_issue738(): 66512391Sjason@lowepower.com """Ignore strides on a length-1 dimension (even if they would be incompatible length > 1)""" 66612391Sjason@lowepower.com assert np.all(m.iss738_f1(np.array([[1., 2, 3]])) == np.array([[1., 102, 203]])) 66712391Sjason@lowepower.com assert np.all(m.iss738_f1(np.array([[1.], [2], [3]])) == np.array([[1.], [12], [23]])) 66812037Sandreas.sandberg@arm.com 66912391Sjason@lowepower.com assert np.all(m.iss738_f2(np.array([[1., 2, 3]])) == np.array([[1., 102, 203]])) 67012391Sjason@lowepower.com assert np.all(m.iss738_f2(np.array([[1.], [2], [3]])) == np.array([[1.], [12], [23]])) 67112037Sandreas.sandberg@arm.com 67212037Sandreas.sandberg@arm.com 67314299Sbbruce@ucdavis.edudef test_issue1105(): 67414299Sbbruce@ucdavis.edu """Issue 1105: 1xN or Nx1 input arrays weren't accepted for eigen 67514299Sbbruce@ucdavis.edu compile-time row vectors or column vector""" 67614299Sbbruce@ucdavis.edu assert m.iss1105_row(np.ones((1, 7))) 67714299Sbbruce@ucdavis.edu assert m.iss1105_col(np.ones((7, 1))) 67814299Sbbruce@ucdavis.edu 67914299Sbbruce@ucdavis.edu # These should still fail (incompatible dimensions): 68014299Sbbruce@ucdavis.edu with pytest.raises(TypeError) as excinfo: 68114299Sbbruce@ucdavis.edu m.iss1105_row(np.ones((7, 1))) 68214299Sbbruce@ucdavis.edu assert "incompatible function arguments" in str(excinfo.value) 68314299Sbbruce@ucdavis.edu with pytest.raises(TypeError) as excinfo: 68414299Sbbruce@ucdavis.edu m.iss1105_col(np.ones((1, 7))) 68514299Sbbruce@ucdavis.edu assert "incompatible function arguments" in str(excinfo.value) 68614299Sbbruce@ucdavis.edu 68714299Sbbruce@ucdavis.edu 68812037Sandreas.sandberg@arm.comdef test_custom_operator_new(): 68912037Sandreas.sandberg@arm.com """Using Eigen types as member variables requires a class-specific 69012037Sandreas.sandberg@arm.com operator new with proper alignment""" 69112037Sandreas.sandberg@arm.com 69212391Sjason@lowepower.com o = m.CustomOperatorNew() 69312037Sandreas.sandberg@arm.com np.testing.assert_allclose(o.a, 0.0) 69412037Sandreas.sandberg@arm.com np.testing.assert_allclose(o.b.diagonal(), 1.0) 695