test_buffers.py 2.99 KB
Newer Older
1
# -*- coding: utf-8 -*-
2
import io
3
import struct
4

Dean Moldovan's avatar
Dean Moldovan committed
5
import pytest
6

7
8
import env  # noqa: F401

9
10
from pybind11_tests import buffers as m
from pybind11_tests import ConstructorStats
Dean Moldovan's avatar
Dean Moldovan committed
11

12
np = pytest.importorskip("numpy")
Dean Moldovan's avatar
Dean Moldovan committed
13
14


Wenzel Jakob's avatar
Wenzel Jakob committed
15
16
def test_from_python():
    with pytest.raises(RuntimeError) as excinfo:
17
        m.Matrix(np.array([1, 2, 3]))  # trying to assign a 1D array
Wenzel Jakob's avatar
Wenzel Jakob committed
18
19
20
    assert str(excinfo.value) == "Incompatible buffer format!"

    m3 = np.array([[1, 2, 3], [4, 5, 6]]).astype(np.float32)
21
    m4 = m.Matrix(m3)
Wenzel Jakob's avatar
Wenzel Jakob committed
22
23
24
25
26

    for i in range(m4.rows()):
        for j in range(m4.cols()):
            assert m3[i, j] == m4[i, j]

27
    cstats = ConstructorStats.get(m.Matrix)
Wenzel Jakob's avatar
Wenzel Jakob committed
28
29
30
31
32
33
34
35
36
37
    assert cstats.alive() == 1
    del m3, m4
    assert cstats.alive() == 0
    assert cstats.values() == ["2x3 matrix"]
    assert cstats.copy_constructions == 0
    # assert cstats.move_constructions >= 0  # Don't invoke any
    assert cstats.copy_assignments == 0
    assert cstats.move_assignments == 0


38
# https://foss.heptapod.net/pypy/pypy/-/issues/2444
Dean Moldovan's avatar
Dean Moldovan committed
39
def test_to_python():
40
41
    mat = m.Matrix(5, 4)
    assert memoryview(mat).shape == (5, 4)
Dean Moldovan's avatar
Dean Moldovan committed
42

43
    assert mat[2, 3] == 0
44
45
    mat[2, 3] = 4.0
    mat[3, 2] = 7.0
46
    assert mat[2, 3] == 4
47
48
49
    assert mat[3, 2] == 7
    assert struct.unpack_from('f', mat, (3 * 4 + 2) * 4) == (7, )
    assert struct.unpack_from('f', mat, (2 * 4 + 3) * 4) == (4, )
Dean Moldovan's avatar
Dean Moldovan committed
50

51
    mat2 = np.array(mat, copy=False)
52
53
54
    assert mat2.shape == (5, 4)
    assert abs(mat2).sum() == 11
    assert mat2[2, 3] == 4 and mat2[3, 2] == 7
55
56
    mat2[2, 3] = 5
    assert mat2[2, 3] == 5
Dean Moldovan's avatar
Dean Moldovan committed
57

58
    cstats = ConstructorStats.get(m.Matrix)
Dean Moldovan's avatar
Dean Moldovan committed
59
    assert cstats.alive() == 1
60
    del mat
Wenzel Jakob's avatar
Wenzel Jakob committed
61
    pytest.gc_collect()
Dean Moldovan's avatar
Dean Moldovan committed
62
    assert cstats.alive() == 1
63
    del mat2  # holds a mat reference
Wenzel Jakob's avatar
Wenzel Jakob committed
64
    pytest.gc_collect()
Dean Moldovan's avatar
Dean Moldovan committed
65
    assert cstats.alive() == 0
66
    assert cstats.values() == ["5x4 matrix"]
Dean Moldovan's avatar
Dean Moldovan committed
67
68
69
70
    assert cstats.copy_constructions == 0
    # assert cstats.move_constructions >= 0  # Don't invoke any
    assert cstats.copy_assignments == 0
    assert cstats.move_assignments == 0
71
72


73
74
75
def test_inherited_protocol():
    """SquareMatrix is derived from Matrix and inherits the buffer protocol"""

76
    matrix = m.SquareMatrix(5)
77
78
79
80
    assert memoryview(matrix).shape == (5, 5)
    assert np.asarray(matrix).shape == (5, 5)


81
82
def test_pointer_to_member_fn():
    for cls in [m.Buffer, m.ConstBuffer, m.DerivedBuffer]:
83
84
85
86
        buf = cls()
        buf.value = 0x12345678
        value = struct.unpack('i', bytearray(buf))[0]
        assert value == 0x12345678
87
88
89
90
91


def test_readonly_buffer():
    buf = m.BufferReadOnly(0x64)
    view = memoryview(buf)
92
    assert view[0] == b'd' if env.PY2 else 0x64
93
94
95
96
97
98
    assert view.readonly


def test_selective_readonly_buffer():
    buf = m.BufferReadOnlySelect()

99
    memoryview(buf)[0] = b'd' if env.PY2 else 0x64
100
101
102
103
104
105
106
    assert buf.value == 0x64

    io.BytesIO(b'A').readinto(buf)
    assert buf.value == ord(b'A')

    buf.readonly = True
    with pytest.raises(TypeError):
107
        memoryview(buf)[0] = b'\0' if env.PY2 else 0
108
109
    with pytest.raises(TypeError):
        io.BytesIO(b'1').readinto(buf)