test_buffers.py revision 14299:2fbea9df56d2
110037SARM gem5 Developersimport struct 210844Sandreas.sandberg@arm.comimport pytest 310037SARM gem5 Developersfrom pybind11_tests import buffers as m 410037SARM gem5 Developersfrom pybind11_tests import ConstructorStats 510037SARM gem5 Developers 610037SARM gem5 Developerspytestmark = pytest.requires_numpy 710037SARM gem5 Developers 810037SARM gem5 Developerswith pytest.suppress(ImportError): 910037SARM gem5 Developers import numpy as np 1010037SARM gem5 Developers 1110037SARM gem5 Developers 1210037SARM gem5 Developersdef test_from_python(): 1310037SARM gem5 Developers with pytest.raises(RuntimeError) as excinfo: 1410037SARM gem5 Developers m.Matrix(np.array([1, 2, 3])) # trying to assign a 1D array 1510037SARM gem5 Developers assert str(excinfo.value) == "Incompatible buffer format!" 1610037SARM gem5 Developers 1710037SARM gem5 Developers m3 = np.array([[1, 2, 3], [4, 5, 6]]).astype(np.float32) 1810037SARM gem5 Developers m4 = m.Matrix(m3) 1910037SARM gem5 Developers 2010037SARM gem5 Developers for i in range(m4.rows()): 2110037SARM gem5 Developers for j in range(m4.cols()): 2210037SARM gem5 Developers assert m3[i, j] == m4[i, j] 2310037SARM gem5 Developers 2410037SARM gem5 Developers cstats = ConstructorStats.get(m.Matrix) 2510037SARM gem5 Developers assert cstats.alive() == 1 2610037SARM gem5 Developers del m3, m4 2710037SARM gem5 Developers assert cstats.alive() == 0 2810037SARM gem5 Developers assert cstats.values() == ["2x3 matrix"] 2910037SARM gem5 Developers assert cstats.copy_constructions == 0 3010037SARM gem5 Developers # assert cstats.move_constructions >= 0 # Don't invoke any 3110037SARM gem5 Developers assert cstats.copy_assignments == 0 3210037SARM gem5 Developers assert cstats.move_assignments == 0 3310037SARM gem5 Developers 3410037SARM gem5 Developers 3510037SARM gem5 Developers# PyPy: Memory leak in the "np.array(m, copy=False)" call 3610037SARM gem5 Developers# https://bitbucket.org/pypy/pypy/issues/2444 3710037SARM gem5 Developers@pytest.unsupported_on_pypy 3810844Sandreas.sandberg@arm.comdef test_to_python(): 3910037SARM gem5 Developers mat = m.Matrix(5, 4) 4010037SARM gem5 Developers assert memoryview(mat).shape == (5, 4) 4110037SARM gem5 Developers 4210037SARM gem5 Developers assert mat[2, 3] == 0 4310037SARM gem5 Developers mat[2, 3] = 4.0 4410844Sandreas.sandberg@arm.com mat[3, 2] = 7.0 4510037SARM gem5 Developers assert mat[2, 3] == 4 4610844Sandreas.sandberg@arm.com assert mat[3, 2] == 7 4710037SARM gem5 Developers assert struct.unpack_from('f', mat, (3 * 4 + 2) * 4) == (7, ) 4810037SARM gem5 Developers assert struct.unpack_from('f', mat, (2 * 4 + 3) * 4) == (4, ) 4910037SARM gem5 Developers 5010037SARM gem5 Developers mat2 = np.array(mat, copy=False) 5110037SARM gem5 Developers assert mat2.shape == (5, 4) 5210037SARM gem5 Developers assert abs(mat2).sum() == 11 5310037SARM gem5 Developers assert mat2[2, 3] == 4 and mat2[3, 2] == 7 5410037SARM gem5 Developers mat2[2, 3] = 5 5510037SARM gem5 Developers assert mat2[2, 3] == 5 5610844Sandreas.sandberg@arm.com 5710037SARM gem5 Developers cstats = ConstructorStats.get(m.Matrix) 5810844Sandreas.sandberg@arm.com assert cstats.alive() == 1 5910844Sandreas.sandberg@arm.com del mat 6010844Sandreas.sandberg@arm.com pytest.gc_collect() 6110844Sandreas.sandberg@arm.com assert cstats.alive() == 1 6210844Sandreas.sandberg@arm.com del mat2 # holds a mat reference 6310844Sandreas.sandberg@arm.com pytest.gc_collect() 6410844Sandreas.sandberg@arm.com assert cstats.alive() == 0 6510844Sandreas.sandberg@arm.com assert cstats.values() == ["5x4 matrix"] 6610844Sandreas.sandberg@arm.com assert cstats.copy_constructions == 0 6710844Sandreas.sandberg@arm.com # assert cstats.move_constructions >= 0 # Don't invoke any 6810844Sandreas.sandberg@arm.com assert cstats.copy_assignments == 0 6910844Sandreas.sandberg@arm.com assert cstats.move_assignments == 0 7010844Sandreas.sandberg@arm.com 7110844Sandreas.sandberg@arm.com 7210844Sandreas.sandberg@arm.com@pytest.unsupported_on_pypy 7310844Sandreas.sandberg@arm.comdef test_inherited_protocol(): 7410844Sandreas.sandberg@arm.com """SquareMatrix is derived from Matrix and inherits the buffer protocol""" 7510844Sandreas.sandberg@arm.com 7610844Sandreas.sandberg@arm.com matrix = m.SquareMatrix(5) 7710844Sandreas.sandberg@arm.com assert memoryview(matrix).shape == (5, 5) 7810844Sandreas.sandberg@arm.com assert np.asarray(matrix).shape == (5, 5) 7910844Sandreas.sandberg@arm.com 8010844Sandreas.sandberg@arm.com 8110844Sandreas.sandberg@arm.com@pytest.unsupported_on_pypy 8210844Sandreas.sandberg@arm.comdef test_pointer_to_member_fn(): 8310844Sandreas.sandberg@arm.com for cls in [m.Buffer, m.ConstBuffer, m.DerivedBuffer]: 8410844Sandreas.sandberg@arm.com buf = cls() 8510844Sandreas.sandberg@arm.com buf.value = 0x12345678 8610844Sandreas.sandberg@arm.com value = struct.unpack('i', bytearray(buf))[0] 8710844Sandreas.sandberg@arm.com assert value == 0x12345678 8810844Sandreas.sandberg@arm.com