test_stl_binders.py revision 12391:ceeca8b41e4b 
1import pytest
2import sys
3from pybind11_tests import stl_binders as m
4
5with pytest.suppress(ImportError):
6    import numpy as np
7
8
9def test_vector_int():
10    v_int = m.VectorInt([0, 0])
11    assert len(v_int) == 2
12    assert bool(v_int) is True
13
14    v_int2 = m.VectorInt([0, 0])
15    assert v_int == v_int2
16    v_int2[1] = 1
17    assert v_int != v_int2
18
19    v_int2.append(2)
20    v_int2.insert(0, 1)
21    v_int2.insert(0, 2)
22    v_int2.insert(0, 3)
23    v_int2.insert(6, 3)
24    assert str(v_int2) == "VectorInt[3, 2, 1, 0, 1, 2, 3]"
25    with pytest.raises(IndexError):
26        v_int2.insert(8, 4)
27
28    v_int.append(99)
29    v_int2[2:-2] = v_int
30    assert v_int2 == m.VectorInt([3, 2, 0, 0, 99, 2, 3])
31    del v_int2[1:3]
32    assert v_int2 == m.VectorInt([3, 0, 99, 2, 3])
33    del v_int2[0]
34    assert v_int2 == m.VectorInt([0, 99, 2, 3])
35
36
37# related to the PyPy's buffer protocol.
38@pytest.unsupported_on_pypy
39def test_vector_buffer():
40    b = bytearray([1, 2, 3, 4])
41    v = m.VectorUChar(b)
42    assert v[1] == 2
43    v[2] = 5
44    mv = memoryview(v)  # We expose the buffer interface
45    if sys.version_info.major > 2:
46        assert mv[2] == 5
47        mv[2] = 6
48    else:
49        assert mv[2] == '\x05'
50        mv[2] = '\x06'
51    assert v[2] == 6
52
53    with pytest.raises(RuntimeError) as excinfo:
54        m.create_undeclstruct()  # Undeclared struct contents, no buffer interface
55    assert "NumPy type info missing for " in str(excinfo.value)
56
57
58@pytest.unsupported_on_pypy
59@pytest.requires_numpy
60def test_vector_buffer_numpy():
61    a = np.array([1, 2, 3, 4], dtype=np.int32)
62    with pytest.raises(TypeError):
63        m.VectorInt(a)
64
65    a = np.array([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]], dtype=np.uintc)
66    v = m.VectorInt(a[0, :])
67    assert len(v) == 4
68    assert v[2] == 3
69    ma = np.asarray(v)
70    ma[2] = 5
71    assert v[2] == 5
72
73    v = m.VectorInt(a[:, 1])
74    assert len(v) == 3
75    assert v[2] == 10
76
77    v = m.get_vectorstruct()
78    assert v[0].x == 5
79    ma = np.asarray(v)
80    ma[1]['x'] = 99
81    assert v[1].x == 99
82
83    v = m.VectorStruct(np.zeros(3, dtype=np.dtype([('w', 'bool'), ('x', 'I'),
84                                                   ('y', 'float64'), ('z', 'bool')], align=True)))
85    assert len(v) == 3
86
87
88def test_vector_bool():
89    import pybind11_cross_module_tests as cm
90
91    vv_c = cm.VectorBool()
92    for i in range(10):
93        vv_c.append(i % 2 == 0)
94    for i in range(10):
95        assert vv_c[i] == (i % 2 == 0)
96    assert str(vv_c) == "VectorBool[1, 0, 1, 0, 1, 0, 1, 0, 1, 0]"
97
98
99def test_vector_custom():
100    v_a = m.VectorEl()
101    v_a.append(m.El(1))
102    v_a.append(m.El(2))
103    assert str(v_a) == "VectorEl[El{1}, El{2}]"
104
105    vv_a = m.VectorVectorEl()
106    vv_a.append(v_a)
107    vv_b = vv_a[0]
108    assert str(vv_b) == "VectorEl[El{1}, El{2}]"
109
110
111def test_map_string_double():
112    mm = m.MapStringDouble()
113    mm['a'] = 1
114    mm['b'] = 2.5
115
116    assert list(mm) == ['a', 'b']
117    assert list(mm.items()) == [('a', 1), ('b', 2.5)]
118    assert str(mm) == "MapStringDouble{a: 1, b: 2.5}"
119
120    um = m.UnorderedMapStringDouble()
121    um['ua'] = 1.1
122    um['ub'] = 2.6
123
124    assert sorted(list(um)) == ['ua', 'ub']
125    assert sorted(list(um.items())) == [('ua', 1.1), ('ub', 2.6)]
126    assert "UnorderedMapStringDouble" in str(um)
127
128
129def test_map_string_double_const():
130    mc = m.MapStringDoubleConst()
131    mc['a'] = 10
132    mc['b'] = 20.5
133    assert str(mc) == "MapStringDoubleConst{a: 10, b: 20.5}"
134
135    umc = m.UnorderedMapStringDoubleConst()
136    umc['a'] = 11
137    umc['b'] = 21.5
138
139    str(umc)
140
141
142def test_noncopyable_containers():
143    # std::vector
144    vnc = m.get_vnc(5)
145    for i in range(0, 5):
146        assert vnc[i].value == i + 1
147
148    for i, j in enumerate(vnc, start=1):
149        assert j.value == i
150
151    # std::deque
152    dnc = m.get_dnc(5)
153    for i in range(0, 5):
154        assert dnc[i].value == i + 1
155
156    i = 1
157    for j in dnc:
158        assert(j.value == i)
159        i += 1
160
161    # std::map
162    mnc = m.get_mnc(5)
163    for i in range(1, 6):
164        assert mnc[i].value == 10 * i
165
166    vsum = 0
167    for k, v in mnc.items():
168        assert v.value == 10 * k
169        vsum += v.value
170
171    assert vsum == 150
172
173    # std::unordered_map
174    mnc = m.get_umnc(5)
175    for i in range(1, 6):
176        assert mnc[i].value == 10 * i
177
178    vsum = 0
179    for k, v in mnc.items():
180        assert v.value == 10 * k
181        vsum += v.value
182
183    assert vsum == 150
184