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