test_buffers.py revision 12391
112391Sjason@lowepower.comimport struct 211986Sandreas.sandberg@arm.comimport pytest 312391Sjason@lowepower.comfrom pybind11_tests import buffers as m 412391Sjason@lowepower.comfrom pybind11_tests import ConstructorStats 511986Sandreas.sandberg@arm.com 612037Sandreas.sandberg@arm.compytestmark = pytest.requires_numpy 712037Sandreas.sandberg@arm.com 811986Sandreas.sandberg@arm.comwith pytest.suppress(ImportError): 911986Sandreas.sandberg@arm.com import numpy as np 1011986Sandreas.sandberg@arm.com 1111986Sandreas.sandberg@arm.com 1211986Sandreas.sandberg@arm.comdef test_from_python(): 1311986Sandreas.sandberg@arm.com with pytest.raises(RuntimeError) as excinfo: 1412391Sjason@lowepower.com m.Matrix(np.array([1, 2, 3])) # trying to assign a 1D array 1511986Sandreas.sandberg@arm.com assert str(excinfo.value) == "Incompatible buffer format!" 1611986Sandreas.sandberg@arm.com 1711986Sandreas.sandberg@arm.com m3 = np.array([[1, 2, 3], [4, 5, 6]]).astype(np.float32) 1812391Sjason@lowepower.com m4 = m.Matrix(m3) 1911986Sandreas.sandberg@arm.com 2011986Sandreas.sandberg@arm.com for i in range(m4.rows()): 2111986Sandreas.sandberg@arm.com for j in range(m4.cols()): 2211986Sandreas.sandberg@arm.com assert m3[i, j] == m4[i, j] 2311986Sandreas.sandberg@arm.com 2412391Sjason@lowepower.com cstats = ConstructorStats.get(m.Matrix) 2511986Sandreas.sandberg@arm.com assert cstats.alive() == 1 2611986Sandreas.sandberg@arm.com del m3, m4 2711986Sandreas.sandberg@arm.com assert cstats.alive() == 0 2811986Sandreas.sandberg@arm.com assert cstats.values() == ["2x3 matrix"] 2911986Sandreas.sandberg@arm.com assert cstats.copy_constructions == 0 3011986Sandreas.sandberg@arm.com # assert cstats.move_constructions >= 0 # Don't invoke any 3111986Sandreas.sandberg@arm.com assert cstats.copy_assignments == 0 3211986Sandreas.sandberg@arm.com assert cstats.move_assignments == 0 3312037Sandreas.sandberg@arm.com 3412037Sandreas.sandberg@arm.com 3512037Sandreas.sandberg@arm.com# PyPy: Memory leak in the "np.array(m, copy=False)" call 3612037Sandreas.sandberg@arm.com# https://bitbucket.org/pypy/pypy/issues/2444 3712037Sandreas.sandberg@arm.com@pytest.unsupported_on_pypy 3812037Sandreas.sandberg@arm.comdef test_to_python(): 3912391Sjason@lowepower.com mat = m.Matrix(5, 5) 4012391Sjason@lowepower.com assert memoryview(mat).shape == (5, 5) 4112037Sandreas.sandberg@arm.com 4212391Sjason@lowepower.com assert mat[2, 3] == 0 4312391Sjason@lowepower.com mat[2, 3] = 4 4412391Sjason@lowepower.com assert mat[2, 3] == 4 4512037Sandreas.sandberg@arm.com 4612391Sjason@lowepower.com mat2 = np.array(mat, copy=False) 4712391Sjason@lowepower.com assert mat2.shape == (5, 5) 4812391Sjason@lowepower.com assert abs(mat2).sum() == 4 4912391Sjason@lowepower.com assert mat2[2, 3] == 4 5012391Sjason@lowepower.com mat2[2, 3] = 5 5112391Sjason@lowepower.com assert mat2[2, 3] == 5 5212037Sandreas.sandberg@arm.com 5312391Sjason@lowepower.com cstats = ConstructorStats.get(m.Matrix) 5412037Sandreas.sandberg@arm.com assert cstats.alive() == 1 5512391Sjason@lowepower.com del mat 5612037Sandreas.sandberg@arm.com pytest.gc_collect() 5712037Sandreas.sandberg@arm.com assert cstats.alive() == 1 5812391Sjason@lowepower.com del mat2 # holds a mat reference 5912037Sandreas.sandberg@arm.com pytest.gc_collect() 6012037Sandreas.sandberg@arm.com assert cstats.alive() == 0 6112037Sandreas.sandberg@arm.com assert cstats.values() == ["5x5 matrix"] 6212037Sandreas.sandberg@arm.com assert cstats.copy_constructions == 0 6312037Sandreas.sandberg@arm.com # assert cstats.move_constructions >= 0 # Don't invoke any 6412037Sandreas.sandberg@arm.com assert cstats.copy_assignments == 0 6512037Sandreas.sandberg@arm.com assert cstats.move_assignments == 0 6612391Sjason@lowepower.com 6712391Sjason@lowepower.com 6812391Sjason@lowepower.com@pytest.unsupported_on_pypy 6912391Sjason@lowepower.comdef test_inherited_protocol(): 7012391Sjason@lowepower.com """SquareMatrix is derived from Matrix and inherits the buffer protocol""" 7112391Sjason@lowepower.com 7212391Sjason@lowepower.com matrix = m.SquareMatrix(5) 7312391Sjason@lowepower.com assert memoryview(matrix).shape == (5, 5) 7412391Sjason@lowepower.com assert np.asarray(matrix).shape == (5, 5) 7512391Sjason@lowepower.com 7612391Sjason@lowepower.com 7712391Sjason@lowepower.com@pytest.unsupported_on_pypy 7812391Sjason@lowepower.comdef test_pointer_to_member_fn(): 7912391Sjason@lowepower.com for cls in [m.Buffer, m.ConstBuffer, m.DerivedBuffer]: 8012391Sjason@lowepower.com buf = cls() 8112391Sjason@lowepower.com buf.value = 0x12345678 8212391Sjason@lowepower.com value = struct.unpack('i', bytearray(buf))[0] 8312391Sjason@lowepower.com assert value == 0x12345678 84