mirror of
https://github.com/pybind/pybind11.git
synced 2024-11-25 22:52:01 +00:00
818 lines
29 KiB
Python
818 lines
29 KiB
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from pybind11_tests import ConstructorStats
|
|
|
|
np = pytest.importorskip("numpy")
|
|
m = pytest.importorskip("pybind11_tests.eigen_matrix")
|
|
|
|
|
|
ref = np.array(
|
|
[
|
|
[0.0, 3, 0, 0, 0, 11],
|
|
[22, 0, 0, 0, 17, 11],
|
|
[7, 5, 0, 1, 0, 11],
|
|
[0, 0, 0, 0, 0, 11],
|
|
[0, 0, 14, 0, 8, 11],
|
|
]
|
|
)
|
|
|
|
|
|
def assert_equal_ref(mat):
|
|
np.testing.assert_array_equal(mat, ref)
|
|
|
|
|
|
def assert_sparse_equal_ref(sparse_mat):
|
|
assert_equal_ref(sparse_mat.toarray())
|
|
|
|
|
|
def test_fixed():
|
|
assert_equal_ref(m.fixed_c())
|
|
assert_equal_ref(m.fixed_r())
|
|
assert_equal_ref(m.fixed_copy_r(m.fixed_r()))
|
|
assert_equal_ref(m.fixed_copy_c(m.fixed_c()))
|
|
assert_equal_ref(m.fixed_copy_r(m.fixed_c()))
|
|
assert_equal_ref(m.fixed_copy_c(m.fixed_r()))
|
|
|
|
|
|
def test_dense():
|
|
assert_equal_ref(m.dense_r())
|
|
assert_equal_ref(m.dense_c())
|
|
assert_equal_ref(m.dense_copy_r(m.dense_r()))
|
|
assert_equal_ref(m.dense_copy_c(m.dense_c()))
|
|
assert_equal_ref(m.dense_copy_r(m.dense_c()))
|
|
assert_equal_ref(m.dense_copy_c(m.dense_r()))
|
|
|
|
|
|
def test_partially_fixed():
|
|
ref2 = np.array([[0.0, 1, 2, 3], [4, 5, 6, 7], [8, 9, 10, 11], [12, 13, 14, 15]])
|
|
np.testing.assert_array_equal(m.partial_copy_four_rm_r(ref2), ref2)
|
|
np.testing.assert_array_equal(m.partial_copy_four_rm_c(ref2), ref2)
|
|
np.testing.assert_array_equal(m.partial_copy_four_rm_r(ref2[:, 1]), ref2[:, [1]])
|
|
np.testing.assert_array_equal(m.partial_copy_four_rm_c(ref2[0, :]), ref2[[0], :])
|
|
np.testing.assert_array_equal(
|
|
m.partial_copy_four_rm_r(ref2[:, (0, 2)]), ref2[:, (0, 2)]
|
|
)
|
|
np.testing.assert_array_equal(
|
|
m.partial_copy_four_rm_c(ref2[(3, 1, 2), :]), ref2[(3, 1, 2), :]
|
|
)
|
|
|
|
np.testing.assert_array_equal(m.partial_copy_four_cm_r(ref2), ref2)
|
|
np.testing.assert_array_equal(m.partial_copy_four_cm_c(ref2), ref2)
|
|
np.testing.assert_array_equal(m.partial_copy_four_cm_r(ref2[:, 1]), ref2[:, [1]])
|
|
np.testing.assert_array_equal(m.partial_copy_four_cm_c(ref2[0, :]), ref2[[0], :])
|
|
np.testing.assert_array_equal(
|
|
m.partial_copy_four_cm_r(ref2[:, (0, 2)]), ref2[:, (0, 2)]
|
|
)
|
|
np.testing.assert_array_equal(
|
|
m.partial_copy_four_cm_c(ref2[(3, 1, 2), :]), ref2[(3, 1, 2), :]
|
|
)
|
|
|
|
# TypeError should be raise for a shape mismatch
|
|
functions = [
|
|
m.partial_copy_four_rm_r,
|
|
m.partial_copy_four_rm_c,
|
|
m.partial_copy_four_cm_r,
|
|
m.partial_copy_four_cm_c,
|
|
]
|
|
matrix_with_wrong_shape = [[1, 2], [3, 4]]
|
|
for f in functions:
|
|
with pytest.raises(TypeError) as excinfo:
|
|
f(matrix_with_wrong_shape)
|
|
assert "incompatible function arguments" in str(excinfo.value)
|
|
|
|
|
|
def test_mutator_descriptors():
|
|
zr = np.arange(30, dtype="float32").reshape(5, 6) # row-major
|
|
zc = zr.reshape(6, 5).transpose() # column-major
|
|
|
|
m.fixed_mutator_r(zr)
|
|
m.fixed_mutator_c(zc)
|
|
m.fixed_mutator_a(zr)
|
|
m.fixed_mutator_a(zc)
|
|
with pytest.raises(TypeError) as excinfo:
|
|
m.fixed_mutator_r(zc)
|
|
assert (
|
|
"(arg0: numpy.typing.NDArray[numpy.float32[5, 6],"
|
|
" flags.writeable, flags.c_contiguous]) -> None" in str(excinfo.value)
|
|
)
|
|
with pytest.raises(TypeError) as excinfo:
|
|
m.fixed_mutator_c(zr)
|
|
assert (
|
|
"(arg0: numpy.typing.NDArray[numpy.float32[5, 6],"
|
|
" flags.writeable, flags.f_contiguous]) -> None" in str(excinfo.value)
|
|
)
|
|
with pytest.raises(TypeError) as excinfo:
|
|
m.fixed_mutator_a(np.array([[1, 2], [3, 4]], dtype="float32"))
|
|
assert (
|
|
"(arg0: numpy.typing.NDArray[numpy.float32[5, 6], flags.writeable]) -> None"
|
|
in str(excinfo.value)
|
|
)
|
|
zr.flags.writeable = False
|
|
with pytest.raises(TypeError):
|
|
m.fixed_mutator_r(zr)
|
|
with pytest.raises(TypeError):
|
|
m.fixed_mutator_a(zr)
|
|
|
|
|
|
def test_cpp_casting():
|
|
assert m.cpp_copy(m.fixed_r()) == 22.0
|
|
assert m.cpp_copy(m.fixed_c()) == 22.0
|
|
z = np.array([[5.0, 6], [7, 8]])
|
|
assert m.cpp_copy(z) == 7.0
|
|
assert m.cpp_copy(m.get_cm_ref()) == 21.0
|
|
assert m.cpp_copy(m.get_rm_ref()) == 21.0
|
|
assert m.cpp_ref_c(m.get_cm_ref()) == 21.0
|
|
assert m.cpp_ref_r(m.get_rm_ref()) == 21.0
|
|
with pytest.raises(RuntimeError) as excinfo:
|
|
# Can't reference m.fixed_c: it contains floats, m.cpp_ref_any wants doubles
|
|
m.cpp_ref_any(m.fixed_c())
|
|
assert "Unable to cast Python instance" in str(excinfo.value)
|
|
with pytest.raises(RuntimeError) as excinfo:
|
|
# Can't reference m.fixed_r: it contains floats, m.cpp_ref_any wants doubles
|
|
m.cpp_ref_any(m.fixed_r())
|
|
assert "Unable to cast Python instance" in str(excinfo.value)
|
|
assert m.cpp_ref_any(m.ReturnTester.create()) == 1.0
|
|
|
|
assert m.cpp_ref_any(m.get_cm_ref()) == 21.0
|
|
assert m.cpp_ref_any(m.get_cm_ref()) == 21.0
|
|
|
|
|
|
def test_pass_readonly_array():
|
|
z = np.full((5, 6), 42.0)
|
|
z.flags.writeable = False
|
|
np.testing.assert_array_equal(z, m.fixed_copy_r(z))
|
|
np.testing.assert_array_equal(m.fixed_r_const(), m.fixed_r())
|
|
assert not m.fixed_r_const().flags.writeable
|
|
np.testing.assert_array_equal(m.fixed_copy_r(m.fixed_r_const()), m.fixed_r_const())
|
|
|
|
|
|
def test_nonunit_stride_from_python():
|
|
counting_mat = np.arange(9.0, dtype=np.float32).reshape((3, 3))
|
|
second_row = counting_mat[1, :]
|
|
second_col = counting_mat[:, 1]
|
|
np.testing.assert_array_equal(m.double_row(second_row), 2.0 * second_row)
|
|
np.testing.assert_array_equal(m.double_col(second_row), 2.0 * second_row)
|
|
np.testing.assert_array_equal(m.double_complex(second_row), 2.0 * second_row)
|
|
np.testing.assert_array_equal(m.double_row(second_col), 2.0 * second_col)
|
|
np.testing.assert_array_equal(m.double_col(second_col), 2.0 * second_col)
|
|
np.testing.assert_array_equal(m.double_complex(second_col), 2.0 * second_col)
|
|
|
|
counting_3d = np.arange(27.0, dtype=np.float32).reshape((3, 3, 3))
|
|
slices = [counting_3d[0, :, :], counting_3d[:, 0, :], counting_3d[:, :, 0]]
|
|
for ref_mat in slices:
|
|
np.testing.assert_array_equal(m.double_mat_cm(ref_mat), 2.0 * ref_mat)
|
|
np.testing.assert_array_equal(m.double_mat_rm(ref_mat), 2.0 * ref_mat)
|
|
|
|
# Mutator:
|
|
m.double_threer(second_row)
|
|
m.double_threec(second_col)
|
|
np.testing.assert_array_equal(counting_mat, [[0.0, 2, 2], [6, 16, 10], [6, 14, 8]])
|
|
|
|
|
|
def test_negative_stride_from_python(msg):
|
|
"""Eigen doesn't support (as of yet) negative strides. When a function takes an Eigen matrix by
|
|
copy or const reference, we can pass a numpy array that has negative strides. Otherwise, an
|
|
exception will be thrown as Eigen will not be able to map the numpy array."""
|
|
|
|
counting_mat = np.arange(9.0, dtype=np.float32).reshape((3, 3))
|
|
counting_mat = counting_mat[::-1, ::-1]
|
|
second_row = counting_mat[1, :]
|
|
second_col = counting_mat[:, 1]
|
|
np.testing.assert_array_equal(m.double_row(second_row), 2.0 * second_row)
|
|
np.testing.assert_array_equal(m.double_col(second_row), 2.0 * second_row)
|
|
np.testing.assert_array_equal(m.double_complex(second_row), 2.0 * second_row)
|
|
np.testing.assert_array_equal(m.double_row(second_col), 2.0 * second_col)
|
|
np.testing.assert_array_equal(m.double_col(second_col), 2.0 * second_col)
|
|
np.testing.assert_array_equal(m.double_complex(second_col), 2.0 * second_col)
|
|
|
|
counting_3d = np.arange(27.0, dtype=np.float32).reshape((3, 3, 3))
|
|
counting_3d = counting_3d[::-1, ::-1, ::-1]
|
|
slices = [counting_3d[0, :, :], counting_3d[:, 0, :], counting_3d[:, :, 0]]
|
|
for ref_mat in slices:
|
|
np.testing.assert_array_equal(m.double_mat_cm(ref_mat), 2.0 * ref_mat)
|
|
np.testing.assert_array_equal(m.double_mat_rm(ref_mat), 2.0 * ref_mat)
|
|
|
|
# Mutator:
|
|
with pytest.raises(TypeError) as excinfo:
|
|
m.double_threer(second_row)
|
|
assert (
|
|
msg(excinfo.value)
|
|
== """
|
|
double_threer(): incompatible function arguments. The following argument types are supported:
|
|
1. (arg0: numpy.typing.NDArray[numpy.float32[1, 3], flags.writeable]) -> None
|
|
|
|
Invoked with: """
|
|
+ repr(np.array([5.0, 4.0, 3.0], dtype="float32"))
|
|
)
|
|
|
|
with pytest.raises(TypeError) as excinfo:
|
|
m.double_threec(second_col)
|
|
assert (
|
|
msg(excinfo.value)
|
|
== """
|
|
double_threec(): incompatible function arguments. The following argument types are supported:
|
|
1. (arg0: numpy.typing.NDArray[numpy.float32[3, 1], flags.writeable]) -> None
|
|
|
|
Invoked with: """
|
|
+ repr(np.array([7.0, 4.0, 1.0], dtype="float32"))
|
|
)
|
|
|
|
|
|
def test_block_runtime_error_type_caster_eigen_ref_made_a_copy():
|
|
with pytest.raises(RuntimeError) as excinfo:
|
|
m.block(ref, 0, 0, 0, 0)
|
|
assert str(excinfo.value) == "type_caster for Eigen::Ref made a copy."
|
|
|
|
|
|
def test_nonunit_stride_to_python():
|
|
assert np.all(m.diagonal(ref) == ref.diagonal())
|
|
assert np.all(m.diagonal_1(ref) == ref.diagonal(1))
|
|
for i in range(-5, 7):
|
|
assert np.all(m.diagonal_n(ref, i) == ref.diagonal(i)), f"m.diagonal_n({i})"
|
|
|
|
# Must be order="F", otherwise the type_caster will make a copy and
|
|
# m.block() will return a dangling reference (heap-use-after-free).
|
|
rof = np.asarray(ref, order="F")
|
|
assert np.all(m.block(rof, 2, 1, 3, 3) == rof[2:5, 1:4])
|
|
assert np.all(m.block(rof, 1, 4, 4, 2) == rof[1:, 4:])
|
|
assert np.all(m.block(rof, 1, 4, 3, 2) == rof[1:4, 4:])
|
|
|
|
|
|
def test_eigen_ref_to_python():
|
|
chols = [m.cholesky1, m.cholesky2, m.cholesky3, m.cholesky4]
|
|
for i, chol in enumerate(chols, start=1):
|
|
mymat = chol(np.array([[1.0, 2, 4], [2, 13, 23], [4, 23, 77]]))
|
|
assert np.all(
|
|
mymat == np.array([[1, 0, 0], [2, 3, 0], [4, 5, 6]])
|
|
), f"cholesky{i}"
|
|
|
|
|
|
def assign_both(a1, a2, r, c, v):
|
|
a1[r, c] = v
|
|
a2[r, c] = v
|
|
|
|
|
|
def array_copy_but_one(a, r, c, v):
|
|
z = np.array(a, copy=True)
|
|
z[r, c] = v
|
|
return z
|
|
|
|
|
|
def test_eigen_return_references():
|
|
"""Tests various ways of returning references and non-referencing copies"""
|
|
|
|
primary = np.ones((10, 10))
|
|
a = m.ReturnTester()
|
|
a_get1 = a.get()
|
|
assert not a_get1.flags.owndata
|
|
assert a_get1.flags.writeable
|
|
assign_both(a_get1, primary, 3, 3, 5)
|
|
a_get2 = a.get_ptr()
|
|
assert not a_get2.flags.owndata
|
|
assert a_get2.flags.writeable
|
|
assign_both(a_get1, primary, 2, 3, 6)
|
|
|
|
a_view1 = a.view()
|
|
assert not a_view1.flags.owndata
|
|
assert not a_view1.flags.writeable
|
|
with pytest.raises(ValueError):
|
|
a_view1[2, 3] = 4
|
|
a_view2 = a.view_ptr()
|
|
assert not a_view2.flags.owndata
|
|
assert not a_view2.flags.writeable
|
|
with pytest.raises(ValueError):
|
|
a_view2[2, 3] = 4
|
|
|
|
a_copy1 = a.copy_get()
|
|
assert a_copy1.flags.owndata
|
|
assert a_copy1.flags.writeable
|
|
np.testing.assert_array_equal(a_copy1, primary)
|
|
a_copy1[7, 7] = -44 # Shouldn't affect anything else
|
|
c1want = array_copy_but_one(primary, 7, 7, -44)
|
|
a_copy2 = a.copy_view()
|
|
assert a_copy2.flags.owndata
|
|
assert a_copy2.flags.writeable
|
|
np.testing.assert_array_equal(a_copy2, primary)
|
|
a_copy2[4, 4] = -22 # Shouldn't affect anything else
|
|
c2want = array_copy_but_one(primary, 4, 4, -22)
|
|
|
|
a_ref1 = a.ref()
|
|
assert not a_ref1.flags.owndata
|
|
assert a_ref1.flags.writeable
|
|
assign_both(a_ref1, primary, 1, 1, 15)
|
|
a_ref2 = a.ref_const()
|
|
assert not a_ref2.flags.owndata
|
|
assert not a_ref2.flags.writeable
|
|
with pytest.raises(ValueError):
|
|
a_ref2[5, 5] = 33
|
|
a_ref3 = a.ref_safe()
|
|
assert not a_ref3.flags.owndata
|
|
assert a_ref3.flags.writeable
|
|
assign_both(a_ref3, primary, 0, 7, 99)
|
|
a_ref4 = a.ref_const_safe()
|
|
assert not a_ref4.flags.owndata
|
|
assert not a_ref4.flags.writeable
|
|
with pytest.raises(ValueError):
|
|
a_ref4[7, 0] = 987654321
|
|
|
|
a_copy3 = a.copy_ref()
|
|
assert a_copy3.flags.owndata
|
|
assert a_copy3.flags.writeable
|
|
np.testing.assert_array_equal(a_copy3, primary)
|
|
a_copy3[8, 1] = 11
|
|
c3want = array_copy_but_one(primary, 8, 1, 11)
|
|
a_copy4 = a.copy_ref_const()
|
|
assert a_copy4.flags.owndata
|
|
assert a_copy4.flags.writeable
|
|
np.testing.assert_array_equal(a_copy4, primary)
|
|
a_copy4[8, 4] = 88
|
|
c4want = array_copy_but_one(primary, 8, 4, 88)
|
|
|
|
a_block1 = a.block(3, 3, 2, 2)
|
|
assert not a_block1.flags.owndata
|
|
assert a_block1.flags.writeable
|
|
a_block1[0, 0] = 55
|
|
primary[3, 3] = 55
|
|
a_block2 = a.block_safe(2, 2, 3, 2)
|
|
assert not a_block2.flags.owndata
|
|
assert a_block2.flags.writeable
|
|
a_block2[2, 1] = -123
|
|
primary[4, 3] = -123
|
|
a_block3 = a.block_const(6, 7, 4, 3)
|
|
assert not a_block3.flags.owndata
|
|
assert not a_block3.flags.writeable
|
|
with pytest.raises(ValueError):
|
|
a_block3[2, 2] = -44444
|
|
|
|
a_copy5 = a.copy_block(2, 2, 2, 3)
|
|
assert a_copy5.flags.owndata
|
|
assert a_copy5.flags.writeable
|
|
np.testing.assert_array_equal(a_copy5, primary[2:4, 2:5])
|
|
a_copy5[1, 1] = 777
|
|
c5want = array_copy_but_one(primary[2:4, 2:5], 1, 1, 777)
|
|
|
|
a_corn1 = a.corners()
|
|
assert not a_corn1.flags.owndata
|
|
assert a_corn1.flags.writeable
|
|
a_corn1 *= 50
|
|
a_corn1[1, 1] = 999
|
|
primary[0, 0] = 50
|
|
primary[0, 9] = 50
|
|
primary[9, 0] = 50
|
|
primary[9, 9] = 999
|
|
a_corn2 = a.corners_const()
|
|
assert not a_corn2.flags.owndata
|
|
assert not a_corn2.flags.writeable
|
|
with pytest.raises(ValueError):
|
|
a_corn2[1, 0] = 51
|
|
|
|
# All of the changes made all the way along should be visible everywhere
|
|
# now (except for the copies, of course)
|
|
np.testing.assert_array_equal(a_get1, primary)
|
|
np.testing.assert_array_equal(a_get2, primary)
|
|
np.testing.assert_array_equal(a_view1, primary)
|
|
np.testing.assert_array_equal(a_view2, primary)
|
|
np.testing.assert_array_equal(a_ref1, primary)
|
|
np.testing.assert_array_equal(a_ref2, primary)
|
|
np.testing.assert_array_equal(a_ref3, primary)
|
|
np.testing.assert_array_equal(a_ref4, primary)
|
|
np.testing.assert_array_equal(a_block1, primary[3:5, 3:5])
|
|
np.testing.assert_array_equal(a_block2, primary[2:5, 2:4])
|
|
np.testing.assert_array_equal(a_block3, primary[6:10, 7:10])
|
|
np.testing.assert_array_equal(
|
|
a_corn1, primary[0 :: primary.shape[0] - 1, 0 :: primary.shape[1] - 1]
|
|
)
|
|
np.testing.assert_array_equal(
|
|
a_corn2, primary[0 :: primary.shape[0] - 1, 0 :: primary.shape[1] - 1]
|
|
)
|
|
|
|
np.testing.assert_array_equal(a_copy1, c1want)
|
|
np.testing.assert_array_equal(a_copy2, c2want)
|
|
np.testing.assert_array_equal(a_copy3, c3want)
|
|
np.testing.assert_array_equal(a_copy4, c4want)
|
|
np.testing.assert_array_equal(a_copy5, c5want)
|
|
|
|
|
|
def assert_keeps_alive(cl, method, *args):
|
|
cstats = ConstructorStats.get(cl)
|
|
start_with = cstats.alive()
|
|
a = cl()
|
|
assert cstats.alive() == start_with + 1
|
|
z = method(a, *args)
|
|
assert cstats.alive() == start_with + 1
|
|
del a
|
|
# Here's the keep alive in action:
|
|
assert cstats.alive() == start_with + 1
|
|
del z
|
|
# Keep alive should have expired:
|
|
assert cstats.alive() == start_with
|
|
|
|
|
|
def test_eigen_keepalive():
|
|
a = m.ReturnTester()
|
|
cstats = ConstructorStats.get(m.ReturnTester)
|
|
assert cstats.alive() == 1
|
|
unsafe = [a.ref(), a.ref_const(), a.block(1, 2, 3, 4)]
|
|
copies = [
|
|
a.copy_get(),
|
|
a.copy_view(),
|
|
a.copy_ref(),
|
|
a.copy_ref_const(),
|
|
a.copy_block(4, 3, 2, 1),
|
|
]
|
|
del a
|
|
assert cstats.alive() == 0
|
|
del unsafe
|
|
del copies
|
|
|
|
for meth in [
|
|
m.ReturnTester.get,
|
|
m.ReturnTester.get_ptr,
|
|
m.ReturnTester.view,
|
|
m.ReturnTester.view_ptr,
|
|
m.ReturnTester.ref_safe,
|
|
m.ReturnTester.ref_const_safe,
|
|
m.ReturnTester.corners,
|
|
m.ReturnTester.corners_const,
|
|
]:
|
|
assert_keeps_alive(m.ReturnTester, meth)
|
|
|
|
for meth in [m.ReturnTester.block_safe, m.ReturnTester.block_const]:
|
|
assert_keeps_alive(m.ReturnTester, meth, 4, 3, 2, 1)
|
|
|
|
|
|
def test_eigen_ref_mutators():
|
|
"""Tests Eigen's ability to mutate numpy values"""
|
|
|
|
orig = np.array([[1.0, 2, 3], [4, 5, 6], [7, 8, 9]])
|
|
zr = np.array(orig)
|
|
zc = np.array(orig, order="F")
|
|
m.add_rm(zr, 1, 0, 100)
|
|
assert np.all(zr == np.array([[1.0, 2, 3], [104, 5, 6], [7, 8, 9]]))
|
|
m.add_cm(zc, 1, 0, 200)
|
|
assert np.all(zc == np.array([[1.0, 2, 3], [204, 5, 6], [7, 8, 9]]))
|
|
|
|
m.add_any(zr, 1, 0, 20)
|
|
assert np.all(zr == np.array([[1.0, 2, 3], [124, 5, 6], [7, 8, 9]]))
|
|
m.add_any(zc, 1, 0, 10)
|
|
assert np.all(zc == np.array([[1.0, 2, 3], [214, 5, 6], [7, 8, 9]]))
|
|
|
|
# Can't reference a col-major array with a row-major Ref, and vice versa:
|
|
with pytest.raises(TypeError):
|
|
m.add_rm(zc, 1, 0, 1)
|
|
with pytest.raises(TypeError):
|
|
m.add_cm(zr, 1, 0, 1)
|
|
|
|
# Overloads:
|
|
m.add1(zr, 1, 0, -100)
|
|
m.add2(zr, 1, 0, -20)
|
|
assert np.all(zr == orig)
|
|
m.add1(zc, 1, 0, -200)
|
|
m.add2(zc, 1, 0, -10)
|
|
assert np.all(zc == orig)
|
|
|
|
# a non-contiguous slice (this won't work on either the row- or
|
|
# column-contiguous refs, but should work for the any)
|
|
cornersr = zr[0::2, 0::2]
|
|
cornersc = zc[0::2, 0::2]
|
|
|
|
assert np.all(cornersr == np.array([[1.0, 3], [7, 9]]))
|
|
assert np.all(cornersc == np.array([[1.0, 3], [7, 9]]))
|
|
|
|
with pytest.raises(TypeError):
|
|
m.add_rm(cornersr, 0, 1, 25)
|
|
with pytest.raises(TypeError):
|
|
m.add_cm(cornersr, 0, 1, 25)
|
|
with pytest.raises(TypeError):
|
|
m.add_rm(cornersc, 0, 1, 25)
|
|
with pytest.raises(TypeError):
|
|
m.add_cm(cornersc, 0, 1, 25)
|
|
m.add_any(cornersr, 0, 1, 25)
|
|
m.add_any(cornersc, 0, 1, 44)
|
|
assert np.all(zr == np.array([[1.0, 2, 28], [4, 5, 6], [7, 8, 9]]))
|
|
assert np.all(zc == np.array([[1.0, 2, 47], [4, 5, 6], [7, 8, 9]]))
|
|
|
|
# You shouldn't be allowed to pass a non-writeable array to a mutating Eigen method:
|
|
zro = zr[0:4, 0:4]
|
|
zro.flags.writeable = False
|
|
with pytest.raises(TypeError):
|
|
m.add_rm(zro, 0, 0, 0)
|
|
with pytest.raises(TypeError):
|
|
m.add_any(zro, 0, 0, 0)
|
|
with pytest.raises(TypeError):
|
|
m.add1(zro, 0, 0, 0)
|
|
with pytest.raises(TypeError):
|
|
m.add2(zro, 0, 0, 0)
|
|
|
|
# integer array shouldn't be passable to a double-matrix-accepting mutating func:
|
|
zi = np.array([[1, 2], [3, 4]])
|
|
with pytest.raises(TypeError):
|
|
m.add_rm(zi)
|
|
|
|
|
|
def test_numpy_ref_mutators():
|
|
"""Tests numpy mutating Eigen matrices (for returned Eigen::Ref<...>s)"""
|
|
|
|
m.reset_refs() # In case another test already changed it
|
|
|
|
zc = m.get_cm_ref()
|
|
zcro = m.get_cm_const_ref()
|
|
zr = m.get_rm_ref()
|
|
zrro = m.get_rm_const_ref()
|
|
|
|
assert [zc[1, 2], zcro[1, 2], zr[1, 2], zrro[1, 2]] == [23] * 4
|
|
|
|
assert not zc.flags.owndata
|
|
assert zc.flags.writeable
|
|
assert not zr.flags.owndata
|
|
assert zr.flags.writeable
|
|
assert not zcro.flags.owndata
|
|
assert not zcro.flags.writeable
|
|
assert not zrro.flags.owndata
|
|
assert not zrro.flags.writeable
|
|
|
|
zc[1, 2] = 99
|
|
expect = np.array([[11.0, 12, 13], [21, 22, 99], [31, 32, 33]])
|
|
# We should have just changed zc, of course, but also zcro and the original eigen matrix
|
|
assert np.all(zc == expect)
|
|
assert np.all(zcro == expect)
|
|
assert np.all(m.get_cm_ref() == expect)
|
|
|
|
zr[1, 2] = 99
|
|
assert np.all(zr == expect)
|
|
assert np.all(zrro == expect)
|
|
assert np.all(m.get_rm_ref() == expect)
|
|
|
|
# Make sure the readonly ones are numpy-readonly:
|
|
with pytest.raises(ValueError):
|
|
zcro[1, 2] = 6
|
|
with pytest.raises(ValueError):
|
|
zrro[1, 2] = 6
|
|
|
|
# We should be able to explicitly copy like this (and since we're copying,
|
|
# the const should drop away)
|
|
y1 = np.array(m.get_cm_const_ref())
|
|
|
|
assert y1.flags.owndata
|
|
assert y1.flags.writeable
|
|
# We should get copies of the eigen data, which was modified above:
|
|
assert y1[1, 2] == 99
|
|
y1[1, 2] += 12
|
|
assert y1[1, 2] == 111
|
|
assert zc[1, 2] == 99 # Make sure we aren't referencing the original
|
|
|
|
|
|
def test_both_ref_mutators():
|
|
"""Tests a complex chain of nested eigen/numpy references"""
|
|
|
|
m.reset_refs() # In case another test already changed it
|
|
|
|
z = m.get_cm_ref() # numpy -> eigen
|
|
z[0, 2] -= 3
|
|
z2 = m.incr_matrix(z, 1) # numpy -> eigen -> numpy -> eigen
|
|
z2[1, 1] += 6
|
|
z3 = m.incr_matrix(z, 2) # (numpy -> eigen)^3
|
|
z3[2, 2] += -5
|
|
z4 = m.incr_matrix(z, 3) # (numpy -> eigen)^4
|
|
z4[1, 1] -= 1
|
|
z5 = m.incr_matrix(z, 4) # (numpy -> eigen)^5
|
|
z5[0, 0] = 0
|
|
assert np.all(z == z2)
|
|
assert np.all(z == z3)
|
|
assert np.all(z == z4)
|
|
assert np.all(z == z5)
|
|
expect = np.array([[0.0, 22, 20], [31, 37, 33], [41, 42, 38]])
|
|
assert np.all(z == expect)
|
|
|
|
y = np.array(range(100), dtype="float64").reshape(10, 10)
|
|
y2 = m.incr_matrix_any(y, 10) # np -> eigen -> np
|
|
y3 = m.incr_matrix_any(
|
|
y2[0::2, 0::2], -33
|
|
) # np -> eigen -> np slice -> np -> eigen -> np
|
|
y4 = m.even_rows(y3) # numpy -> eigen slice -> (... y3)
|
|
y5 = m.even_cols(y4) # numpy -> eigen slice -> (... y4)
|
|
y6 = m.incr_matrix_any(y5, 1000) # numpy -> eigen -> (... y5)
|
|
|
|
# Apply same mutations using just numpy:
|
|
yexpect = np.array(range(100), dtype="float64").reshape(10, 10)
|
|
yexpect += 10
|
|
yexpect[0::2, 0::2] -= 33
|
|
yexpect[0::4, 0::4] += 1000
|
|
assert np.all(y6 == yexpect[0::4, 0::4])
|
|
assert np.all(y5 == yexpect[0::4, 0::4])
|
|
assert np.all(y4 == yexpect[0::4, 0::2])
|
|
assert np.all(y3 == yexpect[0::2, 0::2])
|
|
assert np.all(y2 == yexpect)
|
|
assert np.all(y == yexpect)
|
|
|
|
|
|
def test_nocopy_wrapper():
|
|
# get_elem requires a column-contiguous matrix reference, but should be
|
|
# callable with other types of matrix (via copying):
|
|
int_matrix_colmajor = np.array(
|
|
[[1, 2, 3], [4, 5, 6], [7, 8, 9]], dtype="l", order="F"
|
|
)
|
|
dbl_matrix_colmajor = np.array(
|
|
int_matrix_colmajor, dtype="double", order="F", copy=True
|
|
)
|
|
int_matrix_rowmajor = np.array(int_matrix_colmajor, order="C", copy=True)
|
|
dbl_matrix_rowmajor = np.array(
|
|
int_matrix_rowmajor, dtype="double", order="C", copy=True
|
|
)
|
|
|
|
# All should be callable via get_elem:
|
|
assert m.get_elem(int_matrix_colmajor) == 8
|
|
assert m.get_elem(dbl_matrix_colmajor) == 8
|
|
assert m.get_elem(int_matrix_rowmajor) == 8
|
|
assert m.get_elem(dbl_matrix_rowmajor) == 8
|
|
|
|
# All but the second should fail with m.get_elem_nocopy:
|
|
with pytest.raises(TypeError) as excinfo:
|
|
m.get_elem_nocopy(int_matrix_colmajor)
|
|
assert "get_elem_nocopy(): incompatible function arguments." in str(excinfo.value)
|
|
assert ", flags.f_contiguous" in str(excinfo.value)
|
|
assert m.get_elem_nocopy(dbl_matrix_colmajor) == 8
|
|
with pytest.raises(TypeError) as excinfo:
|
|
m.get_elem_nocopy(int_matrix_rowmajor)
|
|
assert "get_elem_nocopy(): incompatible function arguments." in str(excinfo.value)
|
|
assert ", flags.f_contiguous" in str(excinfo.value)
|
|
with pytest.raises(TypeError) as excinfo:
|
|
m.get_elem_nocopy(dbl_matrix_rowmajor)
|
|
assert "get_elem_nocopy(): incompatible function arguments." in str(excinfo.value)
|
|
assert ", flags.f_contiguous" in str(excinfo.value)
|
|
|
|
# For the row-major test, we take a long matrix in row-major, so only the third is allowed:
|
|
with pytest.raises(TypeError) as excinfo:
|
|
m.get_elem_rm_nocopy(int_matrix_colmajor)
|
|
assert "get_elem_rm_nocopy(): incompatible function arguments." in str(
|
|
excinfo.value
|
|
)
|
|
assert ", flags.c_contiguous" in str(excinfo.value)
|
|
with pytest.raises(TypeError) as excinfo:
|
|
m.get_elem_rm_nocopy(dbl_matrix_colmajor)
|
|
assert "get_elem_rm_nocopy(): incompatible function arguments." in str(
|
|
excinfo.value
|
|
)
|
|
assert ", flags.c_contiguous" in str(excinfo.value)
|
|
assert m.get_elem_rm_nocopy(int_matrix_rowmajor) == 8
|
|
with pytest.raises(TypeError) as excinfo:
|
|
m.get_elem_rm_nocopy(dbl_matrix_rowmajor)
|
|
assert "get_elem_rm_nocopy(): incompatible function arguments." in str(
|
|
excinfo.value
|
|
)
|
|
assert ", flags.c_contiguous" in str(excinfo.value)
|
|
|
|
|
|
def test_eigen_ref_life_support():
|
|
"""Ensure the lifetime of temporary arrays created by the `Ref` caster
|
|
|
|
The `Ref` caster sometimes creates a copy which needs to stay alive. This needs to
|
|
happen both for directs casts (just the array) or indirectly (e.g. list of arrays).
|
|
"""
|
|
|
|
a = np.full(shape=10, fill_value=8, dtype=np.int8)
|
|
assert m.get_elem_direct(a) == 8
|
|
|
|
list_of_a = [a]
|
|
assert m.get_elem_indirect(list_of_a) == 8
|
|
|
|
|
|
def test_special_matrix_objects():
|
|
assert np.all(m.incr_diag(7) == np.diag([1.0, 2, 3, 4, 5, 6, 7]))
|
|
|
|
asymm = np.array([[1.0, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12], [13, 14, 15, 16]])
|
|
symm_lower = np.array(asymm)
|
|
symm_upper = np.array(asymm)
|
|
for i in range(4):
|
|
for j in range(i + 1, 4):
|
|
symm_lower[i, j] = symm_lower[j, i]
|
|
symm_upper[j, i] = symm_upper[i, j]
|
|
|
|
assert np.all(m.symmetric_lower(asymm) == symm_lower)
|
|
assert np.all(m.symmetric_upper(asymm) == symm_upper)
|
|
|
|
|
|
def test_dense_signature(doc):
|
|
assert (
|
|
doc(m.double_col)
|
|
== """
|
|
double_col(arg0: numpy.typing.NDArray[numpy.float32[m, 1]]) -> numpy.typing.NDArray[numpy.float32[m, 1]]
|
|
"""
|
|
)
|
|
assert (
|
|
doc(m.double_row)
|
|
== """
|
|
double_row(arg0: numpy.typing.NDArray[numpy.float32[1, n]]) -> numpy.typing.NDArray[numpy.float32[1, n]]
|
|
"""
|
|
)
|
|
assert doc(m.double_complex) == (
|
|
"""
|
|
double_complex(arg0: numpy.typing.NDArray[numpy.complex64[m, 1]])"""
|
|
""" -> numpy.typing.NDArray[numpy.complex64[m, 1]]
|
|
"""
|
|
)
|
|
assert doc(m.double_mat_rm) == (
|
|
"""
|
|
double_mat_rm(arg0: numpy.typing.NDArray[numpy.float32[m, n]])"""
|
|
""" -> numpy.typing.NDArray[numpy.float32[m, n]]
|
|
"""
|
|
)
|
|
|
|
|
|
def test_defaults(doc):
|
|
assert "\n" not in str(doc(m.defaults_mat))
|
|
assert "\n" not in str(doc(m.defaults_vec))
|
|
|
|
|
|
def test_named_arguments():
|
|
a = np.array([[1.0, 2], [3, 4], [5, 6]])
|
|
b = np.ones((2, 1))
|
|
|
|
assert np.all(m.matrix_multiply(a, b) == np.array([[3.0], [7], [11]]))
|
|
assert np.all(m.matrix_multiply(A=a, B=b) == np.array([[3.0], [7], [11]]))
|
|
assert np.all(m.matrix_multiply(B=b, A=a) == np.array([[3.0], [7], [11]]))
|
|
|
|
with pytest.raises(ValueError) as excinfo:
|
|
m.matrix_multiply(b, a)
|
|
assert str(excinfo.value) == "Nonconformable matrices!"
|
|
|
|
with pytest.raises(ValueError) as excinfo:
|
|
m.matrix_multiply(A=b, B=a)
|
|
assert str(excinfo.value) == "Nonconformable matrices!"
|
|
|
|
with pytest.raises(ValueError) as excinfo:
|
|
m.matrix_multiply(B=a, A=b)
|
|
assert str(excinfo.value) == "Nonconformable matrices!"
|
|
|
|
|
|
def test_sparse():
|
|
pytest.importorskip("scipy")
|
|
assert_sparse_equal_ref(m.sparse_r())
|
|
assert_sparse_equal_ref(m.sparse_c())
|
|
assert_sparse_equal_ref(m.sparse_copy_r(m.sparse_r()))
|
|
assert_sparse_equal_ref(m.sparse_copy_c(m.sparse_c()))
|
|
assert_sparse_equal_ref(m.sparse_copy_r(m.sparse_c()))
|
|
assert_sparse_equal_ref(m.sparse_copy_c(m.sparse_r()))
|
|
|
|
|
|
def test_sparse_signature(doc):
|
|
pytest.importorskip("scipy")
|
|
assert (
|
|
doc(m.sparse_copy_r)
|
|
== """
|
|
sparse_copy_r(arg0: scipy.sparse.csr_matrix[numpy.float32]) -> scipy.sparse.csr_matrix[numpy.float32]
|
|
"""
|
|
)
|
|
assert (
|
|
doc(m.sparse_copy_c)
|
|
== """
|
|
sparse_copy_c(arg0: scipy.sparse.csc_matrix[numpy.float32]) -> scipy.sparse.csc_matrix[numpy.float32]
|
|
"""
|
|
)
|
|
|
|
|
|
def test_issue738():
|
|
"""Ignore strides on a length-1 dimension (even if they would be incompatible length > 1)"""
|
|
assert np.all(m.iss738_f1(np.array([[1.0, 2, 3]])) == np.array([[1.0, 102, 203]]))
|
|
assert np.all(
|
|
m.iss738_f1(np.array([[1.0], [2], [3]])) == np.array([[1.0], [12], [23]])
|
|
)
|
|
|
|
assert np.all(m.iss738_f2(np.array([[1.0, 2, 3]])) == np.array([[1.0, 102, 203]]))
|
|
assert np.all(
|
|
m.iss738_f2(np.array([[1.0], [2], [3]])) == np.array([[1.0], [12], [23]])
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize("func", [m.iss738_f1, m.iss738_f2])
|
|
@pytest.mark.parametrize("sizes", [(0, 2), (2, 0)])
|
|
def test_zero_length(func, sizes):
|
|
"""Ignore strides on a length-0 dimension (even if they would be incompatible length > 1)"""
|
|
assert np.all(func(np.zeros(sizes)) == np.zeros(sizes))
|
|
|
|
|
|
def test_issue1105():
|
|
"""Issue 1105: 1xN or Nx1 input arrays weren't accepted for eigen
|
|
compile-time row vectors or column vector"""
|
|
assert m.iss1105_row(np.ones((1, 7)))
|
|
assert m.iss1105_col(np.ones((7, 1)))
|
|
|
|
# These should still fail (incompatible dimensions):
|
|
with pytest.raises(TypeError) as excinfo:
|
|
m.iss1105_row(np.ones((7, 1)))
|
|
assert "incompatible function arguments" in str(excinfo.value)
|
|
with pytest.raises(TypeError) as excinfo:
|
|
m.iss1105_col(np.ones((1, 7)))
|
|
assert "incompatible function arguments" in str(excinfo.value)
|
|
|
|
|
|
def test_custom_operator_new():
|
|
"""Using Eigen types as member variables requires a class-specific
|
|
operator new with proper alignment"""
|
|
|
|
o = m.CustomOperatorNew()
|
|
np.testing.assert_allclose(o.a, 0.0)
|
|
np.testing.assert_allclose(o.b.diagonal(), 1.0)
|