test_eigen.py revision 12037
111986Sandreas.sandberg@arm.comimport pytest 211986Sandreas.sandberg@arm.com 312037Sandreas.sandberg@arm.compytestmark = pytest.requires_eigen_and_numpy 412037Sandreas.sandberg@arm.com 511986Sandreas.sandberg@arm.comwith pytest.suppress(ImportError): 611986Sandreas.sandberg@arm.com import numpy as np 711986Sandreas.sandberg@arm.com 812037Sandreas.sandberg@arm.com ref = np.array([[ 0., 3, 0, 0, 0, 11], 911986Sandreas.sandberg@arm.com [22, 0, 0, 0, 17, 11], 1011986Sandreas.sandberg@arm.com [ 7, 5, 0, 1, 0, 11], 1111986Sandreas.sandberg@arm.com [ 0, 0, 0, 0, 0, 11], 1211986Sandreas.sandberg@arm.com [ 0, 0, 14, 0, 8, 11]]) 1311986Sandreas.sandberg@arm.com 1411986Sandreas.sandberg@arm.com 1511986Sandreas.sandberg@arm.comdef assert_equal_ref(mat): 1611986Sandreas.sandberg@arm.com np.testing.assert_array_equal(mat, ref) 1711986Sandreas.sandberg@arm.com 1811986Sandreas.sandberg@arm.com 1911986Sandreas.sandberg@arm.comdef assert_sparse_equal_ref(sparse_mat): 2011986Sandreas.sandberg@arm.com assert_equal_ref(sparse_mat.todense()) 2111986Sandreas.sandberg@arm.com 2211986Sandreas.sandberg@arm.com 2311986Sandreas.sandberg@arm.comdef test_fixed(): 2412037Sandreas.sandberg@arm.com from pybind11_tests import fixed_r, fixed_c, fixed_copy_r, fixed_copy_c 2511986Sandreas.sandberg@arm.com 2611986Sandreas.sandberg@arm.com assert_equal_ref(fixed_c()) 2711986Sandreas.sandberg@arm.com assert_equal_ref(fixed_r()) 2812037Sandreas.sandberg@arm.com assert_equal_ref(fixed_copy_r(fixed_r())) 2912037Sandreas.sandberg@arm.com assert_equal_ref(fixed_copy_c(fixed_c())) 3012037Sandreas.sandberg@arm.com assert_equal_ref(fixed_copy_r(fixed_c())) 3112037Sandreas.sandberg@arm.com assert_equal_ref(fixed_copy_c(fixed_r())) 3211986Sandreas.sandberg@arm.com 3311986Sandreas.sandberg@arm.com 3411986Sandreas.sandberg@arm.comdef test_dense(): 3512037Sandreas.sandberg@arm.com from pybind11_tests import dense_r, dense_c, dense_copy_r, dense_copy_c 3611986Sandreas.sandberg@arm.com 3711986Sandreas.sandberg@arm.com assert_equal_ref(dense_r()) 3811986Sandreas.sandberg@arm.com assert_equal_ref(dense_c()) 3912037Sandreas.sandberg@arm.com assert_equal_ref(dense_copy_r(dense_r())) 4012037Sandreas.sandberg@arm.com assert_equal_ref(dense_copy_c(dense_c())) 4112037Sandreas.sandberg@arm.com assert_equal_ref(dense_copy_r(dense_c())) 4212037Sandreas.sandberg@arm.com assert_equal_ref(dense_copy_c(dense_r())) 4311986Sandreas.sandberg@arm.com 4411986Sandreas.sandberg@arm.com 4512037Sandreas.sandberg@arm.comdef test_partially_fixed(): 4612037Sandreas.sandberg@arm.com from pybind11_tests import (partial_copy_four_rm_r, partial_copy_four_rm_c, 4712037Sandreas.sandberg@arm.com partial_copy_four_cm_r, partial_copy_four_cm_c) 4812037Sandreas.sandberg@arm.com 4912037Sandreas.sandberg@arm.com ref2 = np.array([[0., 1, 2, 3], [4, 5, 6, 7], [8, 9, 10, 11], [12, 13, 14, 15]]) 5012037Sandreas.sandberg@arm.com np.testing.assert_array_equal(partial_copy_four_rm_r(ref2), ref2) 5112037Sandreas.sandberg@arm.com np.testing.assert_array_equal(partial_copy_four_rm_c(ref2), ref2) 5212037Sandreas.sandberg@arm.com np.testing.assert_array_equal(partial_copy_four_rm_r(ref2[:, 1]), ref2[:, [1]]) 5312037Sandreas.sandberg@arm.com np.testing.assert_array_equal(partial_copy_four_rm_c(ref2[0, :]), ref2[[0], :]) 5412037Sandreas.sandberg@arm.com np.testing.assert_array_equal(partial_copy_four_rm_r(ref2[:, (0, 2)]), ref2[:, (0, 2)]) 5512037Sandreas.sandberg@arm.com np.testing.assert_array_equal( 5612037Sandreas.sandberg@arm.com partial_copy_four_rm_c(ref2[(3, 1, 2), :]), ref2[(3, 1, 2), :]) 5712037Sandreas.sandberg@arm.com 5812037Sandreas.sandberg@arm.com np.testing.assert_array_equal(partial_copy_four_cm_r(ref2), ref2) 5912037Sandreas.sandberg@arm.com np.testing.assert_array_equal(partial_copy_four_cm_c(ref2), ref2) 6012037Sandreas.sandberg@arm.com np.testing.assert_array_equal(partial_copy_four_cm_r(ref2[:, 1]), ref2[:, [1]]) 6112037Sandreas.sandberg@arm.com np.testing.assert_array_equal(partial_copy_four_cm_c(ref2[0, :]), ref2[[0], :]) 6212037Sandreas.sandberg@arm.com np.testing.assert_array_equal(partial_copy_four_cm_r(ref2[:, (0, 2)]), ref2[:, (0, 2)]) 6312037Sandreas.sandberg@arm.com np.testing.assert_array_equal( 6412037Sandreas.sandberg@arm.com partial_copy_four_cm_c(ref2[(3, 1, 2), :]), ref2[(3, 1, 2), :]) 6512037Sandreas.sandberg@arm.com 6612037Sandreas.sandberg@arm.com 6712037Sandreas.sandberg@arm.comdef test_mutator_descriptors(): 6812037Sandreas.sandberg@arm.com from pybind11_tests import fixed_mutator_r, fixed_mutator_c, fixed_mutator_a 6912037Sandreas.sandberg@arm.com zr = np.arange(30, dtype='float32').reshape(5, 6) # row-major 7012037Sandreas.sandberg@arm.com zc = zr.reshape(6, 5).transpose() # column-major 7112037Sandreas.sandberg@arm.com 7212037Sandreas.sandberg@arm.com fixed_mutator_r(zr) 7312037Sandreas.sandberg@arm.com fixed_mutator_c(zc) 7412037Sandreas.sandberg@arm.com fixed_mutator_a(zr) 7512037Sandreas.sandberg@arm.com fixed_mutator_a(zc) 7612037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 7712037Sandreas.sandberg@arm.com fixed_mutator_r(zc) 7812037Sandreas.sandberg@arm.com assert ('(numpy.ndarray[float32[5, 6], flags.writeable, flags.c_contiguous]) -> arg0: None' 7912037Sandreas.sandberg@arm.com in str(excinfo.value)) 8012037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 8112037Sandreas.sandberg@arm.com fixed_mutator_c(zr) 8212037Sandreas.sandberg@arm.com assert ('(numpy.ndarray[float32[5, 6], flags.writeable, flags.f_contiguous]) -> arg0: None' 8312037Sandreas.sandberg@arm.com in str(excinfo.value)) 8412037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 8512037Sandreas.sandberg@arm.com fixed_mutator_a(np.array([[1, 2], [3, 4]], dtype='float32')) 8612037Sandreas.sandberg@arm.com assert ('(numpy.ndarray[float32[5, 6], flags.writeable]) -> arg0: None' 8712037Sandreas.sandberg@arm.com in str(excinfo.value)) 8812037Sandreas.sandberg@arm.com zr.flags.writeable = False 8912037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 9012037Sandreas.sandberg@arm.com fixed_mutator_r(zr) 9112037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 9212037Sandreas.sandberg@arm.com fixed_mutator_a(zr) 9312037Sandreas.sandberg@arm.com 9412037Sandreas.sandberg@arm.com 9512037Sandreas.sandberg@arm.comdef test_cpp_casting(): 9612037Sandreas.sandberg@arm.com from pybind11_tests import (cpp_copy, cpp_ref_c, cpp_ref_r, cpp_ref_any, 9712037Sandreas.sandberg@arm.com fixed_r, fixed_c, get_cm_ref, get_rm_ref, ReturnTester) 9812037Sandreas.sandberg@arm.com assert cpp_copy(fixed_r()) == 22. 9912037Sandreas.sandberg@arm.com assert cpp_copy(fixed_c()) == 22. 10012037Sandreas.sandberg@arm.com z = np.array([[5., 6], [7, 8]]) 10112037Sandreas.sandberg@arm.com assert cpp_copy(z) == 7. 10212037Sandreas.sandberg@arm.com assert cpp_copy(get_cm_ref()) == 21. 10312037Sandreas.sandberg@arm.com assert cpp_copy(get_rm_ref()) == 21. 10412037Sandreas.sandberg@arm.com assert cpp_ref_c(get_cm_ref()) == 21. 10512037Sandreas.sandberg@arm.com assert cpp_ref_r(get_rm_ref()) == 21. 10612037Sandreas.sandberg@arm.com with pytest.raises(RuntimeError) as excinfo: 10712037Sandreas.sandberg@arm.com # Can't reference fixed_c: it contains floats, cpp_ref_any wants doubles 10812037Sandreas.sandberg@arm.com cpp_ref_any(fixed_c()) 10912037Sandreas.sandberg@arm.com assert 'Unable to cast Python instance' in str(excinfo.value) 11012037Sandreas.sandberg@arm.com with pytest.raises(RuntimeError) as excinfo: 11112037Sandreas.sandberg@arm.com # Can't reference fixed_r: it contains floats, cpp_ref_any wants doubles 11212037Sandreas.sandberg@arm.com cpp_ref_any(fixed_r()) 11312037Sandreas.sandberg@arm.com assert 'Unable to cast Python instance' in str(excinfo.value) 11412037Sandreas.sandberg@arm.com assert cpp_ref_any(ReturnTester.create()) == 1. 11512037Sandreas.sandberg@arm.com 11612037Sandreas.sandberg@arm.com assert cpp_ref_any(get_cm_ref()) == 21. 11712037Sandreas.sandberg@arm.com assert cpp_ref_any(get_cm_ref()) == 21. 11812037Sandreas.sandberg@arm.com 11912037Sandreas.sandberg@arm.com 12012037Sandreas.sandberg@arm.comdef test_pass_readonly_array(): 12112037Sandreas.sandberg@arm.com from pybind11_tests import fixed_copy_r, fixed_r, fixed_r_const 12212037Sandreas.sandberg@arm.com z = np.full((5, 6), 42.0) 12312037Sandreas.sandberg@arm.com z.flags.writeable = False 12412037Sandreas.sandberg@arm.com np.testing.assert_array_equal(z, fixed_copy_r(z)) 12512037Sandreas.sandberg@arm.com np.testing.assert_array_equal(fixed_r_const(), fixed_r()) 12612037Sandreas.sandberg@arm.com assert not fixed_r_const().flags.writeable 12712037Sandreas.sandberg@arm.com np.testing.assert_array_equal(fixed_copy_r(fixed_r_const()), fixed_r_const()) 12812037Sandreas.sandberg@arm.com 12912037Sandreas.sandberg@arm.com 13011986Sandreas.sandberg@arm.comdef test_nonunit_stride_from_python(): 13112037Sandreas.sandberg@arm.com from pybind11_tests import ( 13212037Sandreas.sandberg@arm.com double_row, double_col, double_complex, double_mat_cm, double_mat_rm, 13312037Sandreas.sandberg@arm.com double_threec, double_threer) 13411986Sandreas.sandberg@arm.com 13511986Sandreas.sandberg@arm.com counting_mat = np.arange(9.0, dtype=np.float32).reshape((3, 3)) 13612037Sandreas.sandberg@arm.com second_row = counting_mat[1, :] 13712037Sandreas.sandberg@arm.com second_col = counting_mat[:, 1] 13812037Sandreas.sandberg@arm.com np.testing.assert_array_equal(double_row(second_row), 2.0 * second_row) 13912037Sandreas.sandberg@arm.com np.testing.assert_array_equal(double_col(second_row), 2.0 * second_row) 14012037Sandreas.sandberg@arm.com np.testing.assert_array_equal(double_complex(second_row), 2.0 * second_row) 14112037Sandreas.sandberg@arm.com np.testing.assert_array_equal(double_row(second_col), 2.0 * second_col) 14212037Sandreas.sandberg@arm.com np.testing.assert_array_equal(double_col(second_col), 2.0 * second_col) 14312037Sandreas.sandberg@arm.com np.testing.assert_array_equal(double_complex(second_col), 2.0 * second_col) 14411986Sandreas.sandberg@arm.com 14511986Sandreas.sandberg@arm.com counting_3d = np.arange(27.0, dtype=np.float32).reshape((3, 3, 3)) 14611986Sandreas.sandberg@arm.com slices = [counting_3d[0, :, :], counting_3d[:, 0, :], counting_3d[:, :, 0]] 14711986Sandreas.sandberg@arm.com for slice_idx, ref_mat in enumerate(slices): 14812037Sandreas.sandberg@arm.com np.testing.assert_array_equal(double_mat_cm(ref_mat), 2.0 * ref_mat) 14912037Sandreas.sandberg@arm.com np.testing.assert_array_equal(double_mat_rm(ref_mat), 2.0 * ref_mat) 15011986Sandreas.sandberg@arm.com 15112037Sandreas.sandberg@arm.com # Mutator: 15212037Sandreas.sandberg@arm.com double_threer(second_row) 15312037Sandreas.sandberg@arm.com double_threec(second_col) 15412037Sandreas.sandberg@arm.com np.testing.assert_array_equal(counting_mat, [[0., 2, 2], [6, 16, 10], [6, 14, 8]]) 15511986Sandreas.sandberg@arm.com 15612037Sandreas.sandberg@arm.com 15711986Sandreas.sandberg@arm.comdef test_nonunit_stride_to_python(): 15811986Sandreas.sandberg@arm.com from pybind11_tests import diagonal, diagonal_1, diagonal_n, block 15911986Sandreas.sandberg@arm.com 16011986Sandreas.sandberg@arm.com assert np.all(diagonal(ref) == ref.diagonal()) 16111986Sandreas.sandberg@arm.com assert np.all(diagonal_1(ref) == ref.diagonal(1)) 16211986Sandreas.sandberg@arm.com for i in range(-5, 7): 16311986Sandreas.sandberg@arm.com assert np.all(diagonal_n(ref, i) == ref.diagonal(i)), "diagonal_n({})".format(i) 16411986Sandreas.sandberg@arm.com 16511986Sandreas.sandberg@arm.com assert np.all(block(ref, 2, 1, 3, 3) == ref[2:5, 1:4]) 16611986Sandreas.sandberg@arm.com assert np.all(block(ref, 1, 4, 4, 2) == ref[1:, 4:]) 16711986Sandreas.sandberg@arm.com assert np.all(block(ref, 1, 4, 3, 2) == ref[1:4, 4:]) 16811986Sandreas.sandberg@arm.com 16911986Sandreas.sandberg@arm.com 17011986Sandreas.sandberg@arm.comdef test_eigen_ref_to_python(): 17112037Sandreas.sandberg@arm.com from pybind11_tests import cholesky1, cholesky2, cholesky3, cholesky4 17211986Sandreas.sandberg@arm.com 17312037Sandreas.sandberg@arm.com chols = [cholesky1, cholesky2, cholesky3, cholesky4] 17411986Sandreas.sandberg@arm.com for i, chol in enumerate(chols, start=1): 17512037Sandreas.sandberg@arm.com mymat = chol(np.array([[1., 2, 4], [2, 13, 23], [4, 23, 77]])) 17611986Sandreas.sandberg@arm.com assert np.all(mymat == np.array([[1, 0, 0], [2, 3, 0], [4, 5, 6]])), "cholesky{}".format(i) 17711986Sandreas.sandberg@arm.com 17811986Sandreas.sandberg@arm.com 17912037Sandreas.sandberg@arm.comdef assign_both(a1, a2, r, c, v): 18012037Sandreas.sandberg@arm.com a1[r, c] = v 18112037Sandreas.sandberg@arm.com a2[r, c] = v 18212037Sandreas.sandberg@arm.com 18312037Sandreas.sandberg@arm.com 18412037Sandreas.sandberg@arm.comdef array_copy_but_one(a, r, c, v): 18512037Sandreas.sandberg@arm.com z = np.array(a, copy=True) 18612037Sandreas.sandberg@arm.com z[r, c] = v 18712037Sandreas.sandberg@arm.com return z 18812037Sandreas.sandberg@arm.com 18912037Sandreas.sandberg@arm.com 19012037Sandreas.sandberg@arm.comdef test_eigen_return_references(): 19112037Sandreas.sandberg@arm.com """Tests various ways of returning references and non-referencing copies""" 19212037Sandreas.sandberg@arm.com from pybind11_tests import ReturnTester 19312037Sandreas.sandberg@arm.com master = np.ones((10, 10)) 19412037Sandreas.sandberg@arm.com a = ReturnTester() 19512037Sandreas.sandberg@arm.com a_get1 = a.get() 19612037Sandreas.sandberg@arm.com assert not a_get1.flags.owndata and a_get1.flags.writeable 19712037Sandreas.sandberg@arm.com assign_both(a_get1, master, 3, 3, 5) 19812037Sandreas.sandberg@arm.com a_get2 = a.get_ptr() 19912037Sandreas.sandberg@arm.com assert not a_get2.flags.owndata and a_get2.flags.writeable 20012037Sandreas.sandberg@arm.com assign_both(a_get1, master, 2, 3, 6) 20112037Sandreas.sandberg@arm.com 20212037Sandreas.sandberg@arm.com a_view1 = a.view() 20312037Sandreas.sandberg@arm.com assert not a_view1.flags.owndata and not a_view1.flags.writeable 20412037Sandreas.sandberg@arm.com with pytest.raises(ValueError): 20512037Sandreas.sandberg@arm.com a_view1[2, 3] = 4 20612037Sandreas.sandberg@arm.com a_view2 = a.view_ptr() 20712037Sandreas.sandberg@arm.com assert not a_view2.flags.owndata and not a_view2.flags.writeable 20812037Sandreas.sandberg@arm.com with pytest.raises(ValueError): 20912037Sandreas.sandberg@arm.com a_view2[2, 3] = 4 21012037Sandreas.sandberg@arm.com 21112037Sandreas.sandberg@arm.com a_copy1 = a.copy_get() 21212037Sandreas.sandberg@arm.com assert a_copy1.flags.owndata and a_copy1.flags.writeable 21312037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy1, master) 21412037Sandreas.sandberg@arm.com a_copy1[7, 7] = -44 # Shouldn't affect anything else 21512037Sandreas.sandberg@arm.com c1want = array_copy_but_one(master, 7, 7, -44) 21612037Sandreas.sandberg@arm.com a_copy2 = a.copy_view() 21712037Sandreas.sandberg@arm.com assert a_copy2.flags.owndata and a_copy2.flags.writeable 21812037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy2, master) 21912037Sandreas.sandberg@arm.com a_copy2[4, 4] = -22 # Shouldn't affect anything else 22012037Sandreas.sandberg@arm.com c2want = array_copy_but_one(master, 4, 4, -22) 22112037Sandreas.sandberg@arm.com 22212037Sandreas.sandberg@arm.com a_ref1 = a.ref() 22312037Sandreas.sandberg@arm.com assert not a_ref1.flags.owndata and a_ref1.flags.writeable 22412037Sandreas.sandberg@arm.com assign_both(a_ref1, master, 1, 1, 15) 22512037Sandreas.sandberg@arm.com a_ref2 = a.ref_const() 22612037Sandreas.sandberg@arm.com assert not a_ref2.flags.owndata and not a_ref2.flags.writeable 22712037Sandreas.sandberg@arm.com with pytest.raises(ValueError): 22812037Sandreas.sandberg@arm.com a_ref2[5, 5] = 33 22912037Sandreas.sandberg@arm.com a_ref3 = a.ref_safe() 23012037Sandreas.sandberg@arm.com assert not a_ref3.flags.owndata and a_ref3.flags.writeable 23112037Sandreas.sandberg@arm.com assign_both(a_ref3, master, 0, 7, 99) 23212037Sandreas.sandberg@arm.com a_ref4 = a.ref_const_safe() 23312037Sandreas.sandberg@arm.com assert not a_ref4.flags.owndata and not a_ref4.flags.writeable 23412037Sandreas.sandberg@arm.com with pytest.raises(ValueError): 23512037Sandreas.sandberg@arm.com a_ref4[7, 0] = 987654321 23612037Sandreas.sandberg@arm.com 23712037Sandreas.sandberg@arm.com a_copy3 = a.copy_ref() 23812037Sandreas.sandberg@arm.com assert a_copy3.flags.owndata and a_copy3.flags.writeable 23912037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy3, master) 24012037Sandreas.sandberg@arm.com a_copy3[8, 1] = 11 24112037Sandreas.sandberg@arm.com c3want = array_copy_but_one(master, 8, 1, 11) 24212037Sandreas.sandberg@arm.com a_copy4 = a.copy_ref_const() 24312037Sandreas.sandberg@arm.com assert a_copy4.flags.owndata and a_copy4.flags.writeable 24412037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy4, master) 24512037Sandreas.sandberg@arm.com a_copy4[8, 4] = 88 24612037Sandreas.sandberg@arm.com c4want = array_copy_but_one(master, 8, 4, 88) 24712037Sandreas.sandberg@arm.com 24812037Sandreas.sandberg@arm.com a_block1 = a.block(3, 3, 2, 2) 24912037Sandreas.sandberg@arm.com assert not a_block1.flags.owndata and a_block1.flags.writeable 25012037Sandreas.sandberg@arm.com a_block1[0, 0] = 55 25112037Sandreas.sandberg@arm.com master[3, 3] = 55 25212037Sandreas.sandberg@arm.com a_block2 = a.block_safe(2, 2, 3, 2) 25312037Sandreas.sandberg@arm.com assert not a_block2.flags.owndata and a_block2.flags.writeable 25412037Sandreas.sandberg@arm.com a_block2[2, 1] = -123 25512037Sandreas.sandberg@arm.com master[4, 3] = -123 25612037Sandreas.sandberg@arm.com a_block3 = a.block_const(6, 7, 4, 3) 25712037Sandreas.sandberg@arm.com assert not a_block3.flags.owndata and not a_block3.flags.writeable 25812037Sandreas.sandberg@arm.com with pytest.raises(ValueError): 25912037Sandreas.sandberg@arm.com a_block3[2, 2] = -44444 26012037Sandreas.sandberg@arm.com 26112037Sandreas.sandberg@arm.com a_copy5 = a.copy_block(2, 2, 2, 3) 26212037Sandreas.sandberg@arm.com assert a_copy5.flags.owndata and a_copy5.flags.writeable 26312037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy5, master[2:4, 2:5]) 26412037Sandreas.sandberg@arm.com a_copy5[1, 1] = 777 26512037Sandreas.sandberg@arm.com c5want = array_copy_but_one(master[2:4, 2:5], 1, 1, 777) 26612037Sandreas.sandberg@arm.com 26712037Sandreas.sandberg@arm.com a_corn1 = a.corners() 26812037Sandreas.sandberg@arm.com assert not a_corn1.flags.owndata and a_corn1.flags.writeable 26912037Sandreas.sandberg@arm.com a_corn1 *= 50 27012037Sandreas.sandberg@arm.com a_corn1[1, 1] = 999 27112037Sandreas.sandberg@arm.com master[0, 0] = 50 27212037Sandreas.sandberg@arm.com master[0, 9] = 50 27312037Sandreas.sandberg@arm.com master[9, 0] = 50 27412037Sandreas.sandberg@arm.com master[9, 9] = 999 27512037Sandreas.sandberg@arm.com a_corn2 = a.corners_const() 27612037Sandreas.sandberg@arm.com assert not a_corn2.flags.owndata and not a_corn2.flags.writeable 27712037Sandreas.sandberg@arm.com with pytest.raises(ValueError): 27812037Sandreas.sandberg@arm.com a_corn2[1, 0] = 51 27912037Sandreas.sandberg@arm.com 28012037Sandreas.sandberg@arm.com # All of the changes made all the way along should be visible everywhere 28112037Sandreas.sandberg@arm.com # now (except for the copies, of course) 28212037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_get1, master) 28312037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_get2, master) 28412037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_view1, master) 28512037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_view2, master) 28612037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_ref1, master) 28712037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_ref2, master) 28812037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_ref3, master) 28912037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_ref4, master) 29012037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_block1, master[3:5, 3:5]) 29112037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_block2, master[2:5, 2:4]) 29212037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_block3, master[6:10, 7:10]) 29312037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_corn1, master[0::master.shape[0] - 1, 0::master.shape[1] - 1]) 29412037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_corn2, master[0::master.shape[0] - 1, 0::master.shape[1] - 1]) 29512037Sandreas.sandberg@arm.com 29612037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy1, c1want) 29712037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy2, c2want) 29812037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy3, c3want) 29912037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy4, c4want) 30012037Sandreas.sandberg@arm.com np.testing.assert_array_equal(a_copy5, c5want) 30112037Sandreas.sandberg@arm.com 30212037Sandreas.sandberg@arm.com 30312037Sandreas.sandberg@arm.comdef assert_keeps_alive(cl, method, *args): 30412037Sandreas.sandberg@arm.com from pybind11_tests import ConstructorStats 30512037Sandreas.sandberg@arm.com cstats = ConstructorStats.get(cl) 30612037Sandreas.sandberg@arm.com start_with = cstats.alive() 30712037Sandreas.sandberg@arm.com a = cl() 30812037Sandreas.sandberg@arm.com assert cstats.alive() == start_with + 1 30912037Sandreas.sandberg@arm.com z = method(a, *args) 31012037Sandreas.sandberg@arm.com assert cstats.alive() == start_with + 1 31112037Sandreas.sandberg@arm.com del a 31212037Sandreas.sandberg@arm.com # Here's the keep alive in action: 31312037Sandreas.sandberg@arm.com assert cstats.alive() == start_with + 1 31412037Sandreas.sandberg@arm.com del z 31512037Sandreas.sandberg@arm.com # Keep alive should have expired: 31612037Sandreas.sandberg@arm.com assert cstats.alive() == start_with 31712037Sandreas.sandberg@arm.com 31812037Sandreas.sandberg@arm.com 31912037Sandreas.sandberg@arm.comdef test_eigen_keepalive(): 32012037Sandreas.sandberg@arm.com from pybind11_tests import ReturnTester, ConstructorStats 32112037Sandreas.sandberg@arm.com a = ReturnTester() 32212037Sandreas.sandberg@arm.com 32312037Sandreas.sandberg@arm.com cstats = ConstructorStats.get(ReturnTester) 32412037Sandreas.sandberg@arm.com assert cstats.alive() == 1 32512037Sandreas.sandberg@arm.com unsafe = [a.ref(), a.ref_const(), a.block(1, 2, 3, 4)] 32612037Sandreas.sandberg@arm.com copies = [a.copy_get(), a.copy_view(), a.copy_ref(), a.copy_ref_const(), 32712037Sandreas.sandberg@arm.com a.copy_block(4, 3, 2, 1)] 32812037Sandreas.sandberg@arm.com del a 32912037Sandreas.sandberg@arm.com assert cstats.alive() == 0 33012037Sandreas.sandberg@arm.com del unsafe 33112037Sandreas.sandberg@arm.com del copies 33212037Sandreas.sandberg@arm.com 33312037Sandreas.sandberg@arm.com for meth in [ReturnTester.get, ReturnTester.get_ptr, ReturnTester.view, 33412037Sandreas.sandberg@arm.com ReturnTester.view_ptr, ReturnTester.ref_safe, ReturnTester.ref_const_safe, 33512037Sandreas.sandberg@arm.com ReturnTester.corners, ReturnTester.corners_const]: 33612037Sandreas.sandberg@arm.com assert_keeps_alive(ReturnTester, meth) 33712037Sandreas.sandberg@arm.com 33812037Sandreas.sandberg@arm.com for meth in [ReturnTester.block_safe, ReturnTester.block_const]: 33912037Sandreas.sandberg@arm.com assert_keeps_alive(ReturnTester, meth, 4, 3, 2, 1) 34012037Sandreas.sandberg@arm.com 34112037Sandreas.sandberg@arm.com 34212037Sandreas.sandberg@arm.comdef test_eigen_ref_mutators(): 34312037Sandreas.sandberg@arm.com """Tests whether Eigen can mutate numpy values""" 34412037Sandreas.sandberg@arm.com from pybind11_tests import add_rm, add_cm, add_any, add1, add2 34512037Sandreas.sandberg@arm.com orig = np.array([[1., 2, 3], [4, 5, 6], [7, 8, 9]]) 34612037Sandreas.sandberg@arm.com zr = np.array(orig) 34712037Sandreas.sandberg@arm.com zc = np.array(orig, order='F') 34812037Sandreas.sandberg@arm.com add_rm(zr, 1, 0, 100) 34912037Sandreas.sandberg@arm.com assert np.all(zr == np.array([[1., 2, 3], [104, 5, 6], [7, 8, 9]])) 35012037Sandreas.sandberg@arm.com add_cm(zc, 1, 0, 200) 35112037Sandreas.sandberg@arm.com assert np.all(zc == np.array([[1., 2, 3], [204, 5, 6], [7, 8, 9]])) 35212037Sandreas.sandberg@arm.com 35312037Sandreas.sandberg@arm.com add_any(zr, 1, 0, 20) 35412037Sandreas.sandberg@arm.com assert np.all(zr == np.array([[1., 2, 3], [124, 5, 6], [7, 8, 9]])) 35512037Sandreas.sandberg@arm.com add_any(zc, 1, 0, 10) 35612037Sandreas.sandberg@arm.com assert np.all(zc == np.array([[1., 2, 3], [214, 5, 6], [7, 8, 9]])) 35712037Sandreas.sandberg@arm.com 35812037Sandreas.sandberg@arm.com # Can't reference a col-major array with a row-major Ref, and vice versa: 35912037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 36012037Sandreas.sandberg@arm.com add_rm(zc, 1, 0, 1) 36112037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 36212037Sandreas.sandberg@arm.com add_cm(zr, 1, 0, 1) 36312037Sandreas.sandberg@arm.com 36412037Sandreas.sandberg@arm.com # Overloads: 36512037Sandreas.sandberg@arm.com add1(zr, 1, 0, -100) 36612037Sandreas.sandberg@arm.com add2(zr, 1, 0, -20) 36712037Sandreas.sandberg@arm.com assert np.all(zr == orig) 36812037Sandreas.sandberg@arm.com add1(zc, 1, 0, -200) 36912037Sandreas.sandberg@arm.com add2(zc, 1, 0, -10) 37012037Sandreas.sandberg@arm.com assert np.all(zc == orig) 37112037Sandreas.sandberg@arm.com 37212037Sandreas.sandberg@arm.com # a non-contiguous slice (this won't work on either the row- or 37312037Sandreas.sandberg@arm.com # column-contiguous refs, but should work for the any) 37412037Sandreas.sandberg@arm.com cornersr = zr[0::2, 0::2] 37512037Sandreas.sandberg@arm.com cornersc = zc[0::2, 0::2] 37612037Sandreas.sandberg@arm.com 37712037Sandreas.sandberg@arm.com assert np.all(cornersr == np.array([[1., 3], [7, 9]])) 37812037Sandreas.sandberg@arm.com assert np.all(cornersc == np.array([[1., 3], [7, 9]])) 37912037Sandreas.sandberg@arm.com 38012037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 38112037Sandreas.sandberg@arm.com add_rm(cornersr, 0, 1, 25) 38212037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 38312037Sandreas.sandberg@arm.com add_cm(cornersr, 0, 1, 25) 38412037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 38512037Sandreas.sandberg@arm.com add_rm(cornersc, 0, 1, 25) 38612037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 38712037Sandreas.sandberg@arm.com add_cm(cornersc, 0, 1, 25) 38812037Sandreas.sandberg@arm.com add_any(cornersr, 0, 1, 25) 38912037Sandreas.sandberg@arm.com add_any(cornersc, 0, 1, 44) 39012037Sandreas.sandberg@arm.com assert np.all(zr == np.array([[1., 2, 28], [4, 5, 6], [7, 8, 9]])) 39112037Sandreas.sandberg@arm.com assert np.all(zc == np.array([[1., 2, 47], [4, 5, 6], [7, 8, 9]])) 39212037Sandreas.sandberg@arm.com 39312037Sandreas.sandberg@arm.com # You shouldn't be allowed to pass a non-writeable array to a mutating Eigen method: 39412037Sandreas.sandberg@arm.com zro = zr[0:4, 0:4] 39512037Sandreas.sandberg@arm.com zro.flags.writeable = False 39612037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 39712037Sandreas.sandberg@arm.com add_rm(zro, 0, 0, 0) 39812037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 39912037Sandreas.sandberg@arm.com add_any(zro, 0, 0, 0) 40012037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 40112037Sandreas.sandberg@arm.com add1(zro, 0, 0, 0) 40212037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 40312037Sandreas.sandberg@arm.com add2(zro, 0, 0, 0) 40412037Sandreas.sandberg@arm.com 40512037Sandreas.sandberg@arm.com # integer array shouldn't be passable to a double-matrix-accepting mutating func: 40612037Sandreas.sandberg@arm.com zi = np.array([[1, 2], [3, 4]]) 40712037Sandreas.sandberg@arm.com with pytest.raises(TypeError): 40812037Sandreas.sandberg@arm.com add_rm(zi) 40912037Sandreas.sandberg@arm.com 41012037Sandreas.sandberg@arm.com 41112037Sandreas.sandberg@arm.comdef test_numpy_ref_mutators(): 41212037Sandreas.sandberg@arm.com """Tests numpy mutating Eigen matrices (for returned Eigen::Ref<...>s)""" 41312037Sandreas.sandberg@arm.com from pybind11_tests import ( 41412037Sandreas.sandberg@arm.com get_cm_ref, get_cm_const_ref, get_rm_ref, get_rm_const_ref, reset_refs) 41512037Sandreas.sandberg@arm.com reset_refs() # In case another test already changed it 41612037Sandreas.sandberg@arm.com 41712037Sandreas.sandberg@arm.com zc = get_cm_ref() 41812037Sandreas.sandberg@arm.com zcro = get_cm_const_ref() 41912037Sandreas.sandberg@arm.com zr = get_rm_ref() 42012037Sandreas.sandberg@arm.com zrro = get_rm_const_ref() 42112037Sandreas.sandberg@arm.com 42212037Sandreas.sandberg@arm.com assert [zc[1, 2], zcro[1, 2], zr[1, 2], zrro[1, 2]] == [23] * 4 42312037Sandreas.sandberg@arm.com 42412037Sandreas.sandberg@arm.com assert not zc.flags.owndata and zc.flags.writeable 42512037Sandreas.sandberg@arm.com assert not zr.flags.owndata and zr.flags.writeable 42612037Sandreas.sandberg@arm.com assert not zcro.flags.owndata and not zcro.flags.writeable 42712037Sandreas.sandberg@arm.com assert not zrro.flags.owndata and not zrro.flags.writeable 42812037Sandreas.sandberg@arm.com 42912037Sandreas.sandberg@arm.com zc[1, 2] = 99 43012037Sandreas.sandberg@arm.com expect = np.array([[11., 12, 13], [21, 22, 99], [31, 32, 33]]) 43112037Sandreas.sandberg@arm.com # We should have just changed zc, of course, but also zcro and the original eigen matrix 43212037Sandreas.sandberg@arm.com assert np.all(zc == expect) 43312037Sandreas.sandberg@arm.com assert np.all(zcro == expect) 43412037Sandreas.sandberg@arm.com assert np.all(get_cm_ref() == expect) 43512037Sandreas.sandberg@arm.com 43612037Sandreas.sandberg@arm.com zr[1, 2] = 99 43712037Sandreas.sandberg@arm.com assert np.all(zr == expect) 43812037Sandreas.sandberg@arm.com assert np.all(zrro == expect) 43912037Sandreas.sandberg@arm.com assert np.all(get_rm_ref() == expect) 44012037Sandreas.sandberg@arm.com 44112037Sandreas.sandberg@arm.com # Make sure the readonly ones are numpy-readonly: 44212037Sandreas.sandberg@arm.com with pytest.raises(ValueError): 44312037Sandreas.sandberg@arm.com zcro[1, 2] = 6 44412037Sandreas.sandberg@arm.com with pytest.raises(ValueError): 44512037Sandreas.sandberg@arm.com zrro[1, 2] = 6 44612037Sandreas.sandberg@arm.com 44712037Sandreas.sandberg@arm.com # We should be able to explicitly copy like this (and since we're copying, 44812037Sandreas.sandberg@arm.com # the const should drop away) 44912037Sandreas.sandberg@arm.com y1 = np.array(get_cm_const_ref()) 45012037Sandreas.sandberg@arm.com 45112037Sandreas.sandberg@arm.com assert y1.flags.owndata and y1.flags.writeable 45212037Sandreas.sandberg@arm.com # We should get copies of the eigen data, which was modified above: 45312037Sandreas.sandberg@arm.com assert y1[1, 2] == 99 45412037Sandreas.sandberg@arm.com y1[1, 2] += 12 45512037Sandreas.sandberg@arm.com assert y1[1, 2] == 111 45612037Sandreas.sandberg@arm.com assert zc[1, 2] == 99 # Make sure we aren't referencing the original 45712037Sandreas.sandberg@arm.com 45812037Sandreas.sandberg@arm.com 45912037Sandreas.sandberg@arm.comdef test_both_ref_mutators(): 46012037Sandreas.sandberg@arm.com """Tests a complex chain of nested eigen/numpy references""" 46112037Sandreas.sandberg@arm.com from pybind11_tests import ( 46212037Sandreas.sandberg@arm.com incr_matrix, get_cm_ref, incr_matrix_any, even_cols, even_rows, reset_refs) 46312037Sandreas.sandberg@arm.com reset_refs() # In case another test already changed it 46412037Sandreas.sandberg@arm.com 46512037Sandreas.sandberg@arm.com z = get_cm_ref() # numpy -> eigen 46612037Sandreas.sandberg@arm.com z[0, 2] -= 3 46712037Sandreas.sandberg@arm.com z2 = incr_matrix(z, 1) # numpy -> eigen -> numpy -> eigen 46812037Sandreas.sandberg@arm.com z2[1, 1] += 6 46912037Sandreas.sandberg@arm.com z3 = incr_matrix(z, 2) # (numpy -> eigen)^3 47012037Sandreas.sandberg@arm.com z3[2, 2] += -5 47112037Sandreas.sandberg@arm.com z4 = incr_matrix(z, 3) # (numpy -> eigen)^4 47212037Sandreas.sandberg@arm.com z4[1, 1] -= 1 47312037Sandreas.sandberg@arm.com z5 = incr_matrix(z, 4) # (numpy -> eigen)^5 47412037Sandreas.sandberg@arm.com z5[0, 0] = 0 47512037Sandreas.sandberg@arm.com assert np.all(z == z2) 47612037Sandreas.sandberg@arm.com assert np.all(z == z3) 47712037Sandreas.sandberg@arm.com assert np.all(z == z4) 47812037Sandreas.sandberg@arm.com assert np.all(z == z5) 47912037Sandreas.sandberg@arm.com expect = np.array([[0., 22, 20], [31, 37, 33], [41, 42, 38]]) 48012037Sandreas.sandberg@arm.com assert np.all(z == expect) 48112037Sandreas.sandberg@arm.com 48212037Sandreas.sandberg@arm.com y = np.array(range(100), dtype='float64').reshape(10, 10) 48312037Sandreas.sandberg@arm.com y2 = incr_matrix_any(y, 10) # np -> eigen -> np 48412037Sandreas.sandberg@arm.com y3 = incr_matrix_any(y2[0::2, 0::2], -33) # np -> eigen -> np slice -> np -> eigen -> np 48512037Sandreas.sandberg@arm.com y4 = even_rows(y3) # numpy -> eigen slice -> (... y3) 48612037Sandreas.sandberg@arm.com y5 = even_cols(y4) # numpy -> eigen slice -> (... y4) 48712037Sandreas.sandberg@arm.com y6 = incr_matrix_any(y5, 1000) # numpy -> eigen -> (... y5) 48812037Sandreas.sandberg@arm.com 48912037Sandreas.sandberg@arm.com # Apply same mutations using just numpy: 49012037Sandreas.sandberg@arm.com yexpect = np.array(range(100), dtype='float64').reshape(10, 10) 49112037Sandreas.sandberg@arm.com yexpect += 10 49212037Sandreas.sandberg@arm.com yexpect[0::2, 0::2] -= 33 49312037Sandreas.sandberg@arm.com yexpect[0::4, 0::4] += 1000 49412037Sandreas.sandberg@arm.com assert np.all(y6 == yexpect[0::4, 0::4]) 49512037Sandreas.sandberg@arm.com assert np.all(y5 == yexpect[0::4, 0::4]) 49612037Sandreas.sandberg@arm.com assert np.all(y4 == yexpect[0::4, 0::2]) 49712037Sandreas.sandberg@arm.com assert np.all(y3 == yexpect[0::2, 0::2]) 49812037Sandreas.sandberg@arm.com assert np.all(y2 == yexpect) 49912037Sandreas.sandberg@arm.com assert np.all(y == yexpect) 50012037Sandreas.sandberg@arm.com 50112037Sandreas.sandberg@arm.com 50212037Sandreas.sandberg@arm.comdef test_nocopy_wrapper(): 50312037Sandreas.sandberg@arm.com from pybind11_tests import get_elem, get_elem_nocopy, get_elem_rm_nocopy 50412037Sandreas.sandberg@arm.com # get_elem requires a column-contiguous matrix reference, but should be 50512037Sandreas.sandberg@arm.com # callable with other types of matrix (via copying): 50612037Sandreas.sandberg@arm.com int_matrix_colmajor = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]], order='F') 50712037Sandreas.sandberg@arm.com dbl_matrix_colmajor = np.array(int_matrix_colmajor, dtype='double', order='F', copy=True) 50812037Sandreas.sandberg@arm.com int_matrix_rowmajor = np.array(int_matrix_colmajor, order='C', copy=True) 50912037Sandreas.sandberg@arm.com dbl_matrix_rowmajor = np.array(int_matrix_rowmajor, dtype='double', order='C', copy=True) 51012037Sandreas.sandberg@arm.com 51112037Sandreas.sandberg@arm.com # All should be callable via get_elem: 51212037Sandreas.sandberg@arm.com assert get_elem(int_matrix_colmajor) == 8 51312037Sandreas.sandberg@arm.com assert get_elem(dbl_matrix_colmajor) == 8 51412037Sandreas.sandberg@arm.com assert get_elem(int_matrix_rowmajor) == 8 51512037Sandreas.sandberg@arm.com assert get_elem(dbl_matrix_rowmajor) == 8 51612037Sandreas.sandberg@arm.com 51712037Sandreas.sandberg@arm.com # All but the second should fail with get_elem_nocopy: 51812037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 51912037Sandreas.sandberg@arm.com get_elem_nocopy(int_matrix_colmajor) 52012037Sandreas.sandberg@arm.com assert ('get_elem_nocopy(): incompatible function arguments.' in str(excinfo.value) and 52112037Sandreas.sandberg@arm.com ', flags.f_contiguous' in str(excinfo.value)) 52212037Sandreas.sandberg@arm.com assert get_elem_nocopy(dbl_matrix_colmajor) == 8 52312037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 52412037Sandreas.sandberg@arm.com get_elem_nocopy(int_matrix_rowmajor) 52512037Sandreas.sandberg@arm.com assert ('get_elem_nocopy(): incompatible function arguments.' in str(excinfo.value) and 52612037Sandreas.sandberg@arm.com ', flags.f_contiguous' in str(excinfo.value)) 52712037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 52812037Sandreas.sandberg@arm.com get_elem_nocopy(dbl_matrix_rowmajor) 52912037Sandreas.sandberg@arm.com assert ('get_elem_nocopy(): incompatible function arguments.' in str(excinfo.value) and 53012037Sandreas.sandberg@arm.com ', flags.f_contiguous' in str(excinfo.value)) 53112037Sandreas.sandberg@arm.com 53212037Sandreas.sandberg@arm.com # For the row-major test, we take a long matrix in row-major, so only the third is allowed: 53312037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 53412037Sandreas.sandberg@arm.com get_elem_rm_nocopy(int_matrix_colmajor) 53512037Sandreas.sandberg@arm.com assert ('get_elem_rm_nocopy(): incompatible function arguments.' in str(excinfo.value) and 53612037Sandreas.sandberg@arm.com ', flags.c_contiguous' in str(excinfo.value)) 53712037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 53812037Sandreas.sandberg@arm.com get_elem_rm_nocopy(dbl_matrix_colmajor) 53912037Sandreas.sandberg@arm.com assert ('get_elem_rm_nocopy(): incompatible function arguments.' in str(excinfo.value) and 54012037Sandreas.sandberg@arm.com ', flags.c_contiguous' in str(excinfo.value)) 54112037Sandreas.sandberg@arm.com assert get_elem_rm_nocopy(int_matrix_rowmajor) == 8 54212037Sandreas.sandberg@arm.com with pytest.raises(TypeError) as excinfo: 54312037Sandreas.sandberg@arm.com get_elem_rm_nocopy(dbl_matrix_rowmajor) 54412037Sandreas.sandberg@arm.com assert ('get_elem_rm_nocopy(): incompatible function arguments.' in str(excinfo.value) and 54512037Sandreas.sandberg@arm.com ', flags.c_contiguous' in str(excinfo.value)) 54612037Sandreas.sandberg@arm.com 54712037Sandreas.sandberg@arm.com 54811986Sandreas.sandberg@arm.comdef test_special_matrix_objects(): 54911986Sandreas.sandberg@arm.com from pybind11_tests import incr_diag, symmetric_upper, symmetric_lower 55011986Sandreas.sandberg@arm.com 55112037Sandreas.sandberg@arm.com assert np.all(incr_diag(7) == np.diag([1., 2, 3, 4, 5, 6, 7])) 55211986Sandreas.sandberg@arm.com 55312037Sandreas.sandberg@arm.com asymm = np.array([[ 1., 2, 3, 4], 55411986Sandreas.sandberg@arm.com [ 5, 6, 7, 8], 55511986Sandreas.sandberg@arm.com [ 9, 10, 11, 12], 55611986Sandreas.sandberg@arm.com [13, 14, 15, 16]]) 55711986Sandreas.sandberg@arm.com symm_lower = np.array(asymm) 55811986Sandreas.sandberg@arm.com symm_upper = np.array(asymm) 55911986Sandreas.sandberg@arm.com for i in range(4): 56011986Sandreas.sandberg@arm.com for j in range(i + 1, 4): 56111986Sandreas.sandberg@arm.com symm_lower[i, j] = symm_lower[j, i] 56211986Sandreas.sandberg@arm.com symm_upper[j, i] = symm_upper[i, j] 56311986Sandreas.sandberg@arm.com 56411986Sandreas.sandberg@arm.com assert np.all(symmetric_lower(asymm) == symm_lower) 56511986Sandreas.sandberg@arm.com assert np.all(symmetric_upper(asymm) == symm_upper) 56611986Sandreas.sandberg@arm.com 56711986Sandreas.sandberg@arm.com 56811986Sandreas.sandberg@arm.comdef test_dense_signature(doc): 56912037Sandreas.sandberg@arm.com from pybind11_tests import double_col, double_row, double_complex, double_mat_rm 57011986Sandreas.sandberg@arm.com 57111986Sandreas.sandberg@arm.com assert doc(double_col) == """ 57211986Sandreas.sandberg@arm.com double_col(arg0: numpy.ndarray[float32[m, 1]]) -> numpy.ndarray[float32[m, 1]] 57311986Sandreas.sandberg@arm.com """ 57411986Sandreas.sandberg@arm.com assert doc(double_row) == """ 57511986Sandreas.sandberg@arm.com double_row(arg0: numpy.ndarray[float32[1, n]]) -> numpy.ndarray[float32[1, n]] 57611986Sandreas.sandberg@arm.com """ 57712037Sandreas.sandberg@arm.com assert doc(double_complex) == """ 57812037Sandreas.sandberg@arm.com double_complex(arg0: numpy.ndarray[complex64[m, 1]]) -> numpy.ndarray[complex64[m, 1]] 57912037Sandreas.sandberg@arm.com """ 58011986Sandreas.sandberg@arm.com assert doc(double_mat_rm) == """ 58111986Sandreas.sandberg@arm.com double_mat_rm(arg0: numpy.ndarray[float32[m, n]]) -> numpy.ndarray[float32[m, n]] 58211986Sandreas.sandberg@arm.com """ 58311986Sandreas.sandberg@arm.com 58411986Sandreas.sandberg@arm.com 58511986Sandreas.sandberg@arm.com@pytest.requires_eigen_and_scipy 58611986Sandreas.sandberg@arm.comdef test_sparse(): 58712037Sandreas.sandberg@arm.com from pybind11_tests import sparse_r, sparse_c, sparse_copy_r, sparse_copy_c 58811986Sandreas.sandberg@arm.com 58911986Sandreas.sandberg@arm.com assert_sparse_equal_ref(sparse_r()) 59011986Sandreas.sandberg@arm.com assert_sparse_equal_ref(sparse_c()) 59112037Sandreas.sandberg@arm.com assert_sparse_equal_ref(sparse_copy_r(sparse_r())) 59212037Sandreas.sandberg@arm.com assert_sparse_equal_ref(sparse_copy_c(sparse_c())) 59312037Sandreas.sandberg@arm.com assert_sparse_equal_ref(sparse_copy_r(sparse_c())) 59412037Sandreas.sandberg@arm.com assert_sparse_equal_ref(sparse_copy_c(sparse_r())) 59511986Sandreas.sandberg@arm.com 59611986Sandreas.sandberg@arm.com 59711986Sandreas.sandberg@arm.com@pytest.requires_eigen_and_scipy 59811986Sandreas.sandberg@arm.comdef test_sparse_signature(doc): 59912037Sandreas.sandberg@arm.com from pybind11_tests import sparse_copy_r, sparse_copy_c 60011986Sandreas.sandberg@arm.com 60112037Sandreas.sandberg@arm.com assert doc(sparse_copy_r) == """ 60212037Sandreas.sandberg@arm.com sparse_copy_r(arg0: scipy.sparse.csr_matrix[float32]) -> scipy.sparse.csr_matrix[float32] 60311986Sandreas.sandberg@arm.com """ # noqa: E501 line too long 60412037Sandreas.sandberg@arm.com assert doc(sparse_copy_c) == """ 60512037Sandreas.sandberg@arm.com sparse_copy_c(arg0: scipy.sparse.csc_matrix[float32]) -> scipy.sparse.csc_matrix[float32] 60611986Sandreas.sandberg@arm.com """ # noqa: E501 line too long 60712037Sandreas.sandberg@arm.com 60812037Sandreas.sandberg@arm.com 60912037Sandreas.sandberg@arm.comdef test_issue738(): 61012037Sandreas.sandberg@arm.com from pybind11_tests import iss738_f1, iss738_f2 61112037Sandreas.sandberg@arm.com 61212037Sandreas.sandberg@arm.com assert np.all(iss738_f1(np.array([[1., 2, 3]])) == np.array([[1., 102, 203]])) 61312037Sandreas.sandberg@arm.com assert np.all(iss738_f1(np.array([[1.], [2], [3]])) == np.array([[1.], [12], [23]])) 61412037Sandreas.sandberg@arm.com 61512037Sandreas.sandberg@arm.com assert np.all(iss738_f2(np.array([[1., 2, 3]])) == np.array([[1., 102, 203]])) 61612037Sandreas.sandberg@arm.com assert np.all(iss738_f2(np.array([[1.], [2], [3]])) == np.array([[1.], [12], [23]])) 61712037Sandreas.sandberg@arm.com 61812037Sandreas.sandberg@arm.com 61912037Sandreas.sandberg@arm.comdef test_custom_operator_new(): 62012037Sandreas.sandberg@arm.com """Using Eigen types as member variables requires a class-specific 62112037Sandreas.sandberg@arm.com operator new with proper alignment""" 62212037Sandreas.sandberg@arm.com from pybind11_tests import CustomOperatorNew 62312037Sandreas.sandberg@arm.com 62412037Sandreas.sandberg@arm.com o = CustomOperatorNew() 62512037Sandreas.sandberg@arm.com np.testing.assert_allclose(o.a, 0.0) 62612037Sandreas.sandberg@arm.com np.testing.assert_allclose(o.b.diagonal(), 1.0) 627