test_buffers.py revision 11986:c12e4625ab56
1import pytest 2from pybind11_tests import Matrix, ConstructorStats 3 4with pytest.suppress(ImportError): 5 import numpy as np 6 7 8@pytest.requires_numpy 9def test_to_python(): 10 m = Matrix(5, 5) 11 12 assert m[2, 3] == 0 13 m[2, 3] = 4 14 assert m[2, 3] == 4 15 16 m2 = np.array(m, copy=False) 17 assert m2.shape == (5, 5) 18 assert abs(m2).sum() == 4 19 assert m2[2, 3] == 4 20 m2[2, 3] = 5 21 assert m2[2, 3] == 5 22 23 cstats = ConstructorStats.get(Matrix) 24 assert cstats.alive() == 1 25 del m 26 assert cstats.alive() == 1 27 del m2 # holds an m reference 28 assert cstats.alive() == 0 29 assert cstats.values() == ["5x5 matrix"] 30 assert cstats.copy_constructions == 0 31 # assert cstats.move_constructions >= 0 # Don't invoke any 32 assert cstats.copy_assignments == 0 33 assert cstats.move_assignments == 0 34 35 36@pytest.requires_numpy 37def test_from_python(): 38 with pytest.raises(RuntimeError) as excinfo: 39 Matrix(np.array([1, 2, 3])) # trying to assign a 1D array 40 assert str(excinfo.value) == "Incompatible buffer format!" 41 42 m3 = np.array([[1, 2, 3], [4, 5, 6]]).astype(np.float32) 43 m4 = Matrix(m3) 44 45 for i in range(m4.rows()): 46 for j in range(m4.cols()): 47 assert m3[i, j] == m4[i, j] 48 49 cstats = ConstructorStats.get(Matrix) 50 assert cstats.alive() == 1 51 del m3, m4 52 assert cstats.alive() == 0 53 assert cstats.values() == ["2x3 matrix"] 54 assert cstats.copy_constructions == 0 55 # assert cstats.move_constructions >= 0 # Don't invoke any 56 assert cstats.copy_assignments == 0 57 assert cstats.move_assignments == 0 58