test_buffers.py revision 14299
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(): 3914299Sbbruce@ucdavis.edu mat = m.Matrix(5, 4) 4014299Sbbruce@ucdavis.edu assert memoryview(mat).shape == (5, 4) 4112037Sandreas.sandberg@arm.com 4212391Sjason@lowepower.com assert mat[2, 3] == 0 4314299Sbbruce@ucdavis.edu mat[2, 3] = 4.0 4414299Sbbruce@ucdavis.edu mat[3, 2] = 7.0 4512391Sjason@lowepower.com assert mat[2, 3] == 4 4614299Sbbruce@ucdavis.edu assert mat[3, 2] == 7 4714299Sbbruce@ucdavis.edu assert struct.unpack_from('f', mat, (3 * 4 + 2) * 4) == (7, ) 4814299Sbbruce@ucdavis.edu assert struct.unpack_from('f', mat, (2 * 4 + 3) * 4) == (4, ) 4912037Sandreas.sandberg@arm.com 5012391Sjason@lowepower.com mat2 = np.array(mat, copy=False) 5114299Sbbruce@ucdavis.edu assert mat2.shape == (5, 4) 5214299Sbbruce@ucdavis.edu assert abs(mat2).sum() == 11 5314299Sbbruce@ucdavis.edu assert mat2[2, 3] == 4 and mat2[3, 2] == 7 5412391Sjason@lowepower.com mat2[2, 3] = 5 5512391Sjason@lowepower.com assert mat2[2, 3] == 5 5612037Sandreas.sandberg@arm.com 5712391Sjason@lowepower.com cstats = ConstructorStats.get(m.Matrix) 5812037Sandreas.sandberg@arm.com assert cstats.alive() == 1 5912391Sjason@lowepower.com del mat 6012037Sandreas.sandberg@arm.com pytest.gc_collect() 6112037Sandreas.sandberg@arm.com assert cstats.alive() == 1 6212391Sjason@lowepower.com del mat2 # holds a mat reference 6312037Sandreas.sandberg@arm.com pytest.gc_collect() 6412037Sandreas.sandberg@arm.com assert cstats.alive() == 0 6514299Sbbruce@ucdavis.edu assert cstats.values() == ["5x4 matrix"] 6612037Sandreas.sandberg@arm.com assert cstats.copy_constructions == 0 6712037Sandreas.sandberg@arm.com # assert cstats.move_constructions >= 0 # Don't invoke any 6812037Sandreas.sandberg@arm.com assert cstats.copy_assignments == 0 6912037Sandreas.sandberg@arm.com assert cstats.move_assignments == 0 7012391Sjason@lowepower.com 7112391Sjason@lowepower.com 7212391Sjason@lowepower.com@pytest.unsupported_on_pypy 7312391Sjason@lowepower.comdef test_inherited_protocol(): 7412391Sjason@lowepower.com """SquareMatrix is derived from Matrix and inherits the buffer protocol""" 7512391Sjason@lowepower.com 7612391Sjason@lowepower.com matrix = m.SquareMatrix(5) 7712391Sjason@lowepower.com assert memoryview(matrix).shape == (5, 5) 7812391Sjason@lowepower.com assert np.asarray(matrix).shape == (5, 5) 7912391Sjason@lowepower.com 8012391Sjason@lowepower.com 8112391Sjason@lowepower.com@pytest.unsupported_on_pypy 8212391Sjason@lowepower.comdef test_pointer_to_member_fn(): 8312391Sjason@lowepower.com for cls in [m.Buffer, m.ConstBuffer, m.DerivedBuffer]: 8412391Sjason@lowepower.com buf = cls() 8512391Sjason@lowepower.com buf.value = 0x12345678 8612391Sjason@lowepower.com value = struct.unpack('i', bytearray(buf))[0] 8712391Sjason@lowepower.com assert value == 0x12345678 88