test_stl_binders.py 7.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291
  1. # -*- coding: utf-8 -*-
  2. import pytest
  3. import env # noqa: F401
  4. from pybind11_tests import stl_binders as m
  5. def test_vector_int():
  6. v_int = m.VectorInt([0, 0])
  7. assert len(v_int) == 2
  8. assert bool(v_int) is True
  9. # test construction from a generator
  10. v_int1 = m.VectorInt(x for x in range(5))
  11. assert v_int1 == m.VectorInt([0, 1, 2, 3, 4])
  12. v_int2 = m.VectorInt([0, 0])
  13. assert v_int == v_int2
  14. v_int2[1] = 1
  15. assert v_int != v_int2
  16. v_int2.append(2)
  17. v_int2.insert(0, 1)
  18. v_int2.insert(0, 2)
  19. v_int2.insert(0, 3)
  20. v_int2.insert(6, 3)
  21. assert str(v_int2) == "VectorInt[3, 2, 1, 0, 1, 2, 3]"
  22. with pytest.raises(IndexError):
  23. v_int2.insert(8, 4)
  24. v_int.append(99)
  25. v_int2[2:-2] = v_int
  26. assert v_int2 == m.VectorInt([3, 2, 0, 0, 99, 2, 3])
  27. del v_int2[1:3]
  28. assert v_int2 == m.VectorInt([3, 0, 99, 2, 3])
  29. del v_int2[0]
  30. assert v_int2 == m.VectorInt([0, 99, 2, 3])
  31. v_int2.extend(m.VectorInt([4, 5]))
  32. assert v_int2 == m.VectorInt([0, 99, 2, 3, 4, 5])
  33. v_int2.extend([6, 7])
  34. assert v_int2 == m.VectorInt([0, 99, 2, 3, 4, 5, 6, 7])
  35. # test error handling, and that the vector is unchanged
  36. with pytest.raises(RuntimeError):
  37. v_int2.extend([8, "a"])
  38. assert v_int2 == m.VectorInt([0, 99, 2, 3, 4, 5, 6, 7])
  39. # test extending from a generator
  40. v_int2.extend(x for x in range(5))
  41. assert v_int2 == m.VectorInt([0, 99, 2, 3, 4, 5, 6, 7, 0, 1, 2, 3, 4])
  42. # test negative indexing
  43. assert v_int2[-1] == 4
  44. # insert with negative index
  45. v_int2.insert(-1, 88)
  46. assert v_int2 == m.VectorInt([0, 99, 2, 3, 4, 5, 6, 7, 0, 1, 2, 3, 88, 4])
  47. # delete negative index
  48. del v_int2[-1]
  49. assert v_int2 == m.VectorInt([0, 99, 2, 3, 4, 5, 6, 7, 0, 1, 2, 3, 88])
  50. v_int2.clear()
  51. assert len(v_int2) == 0
  52. # Older PyPy's failed here, related to the PyPy's buffer protocol.
  53. def test_vector_buffer():
  54. b = bytearray([1, 2, 3, 4])
  55. v = m.VectorUChar(b)
  56. assert v[1] == 2
  57. v[2] = 5
  58. mv = memoryview(v) # We expose the buffer interface
  59. if not env.PY2:
  60. assert mv[2] == 5
  61. mv[2] = 6
  62. else:
  63. assert mv[2] == "\x05"
  64. mv[2] = "\x06"
  65. assert v[2] == 6
  66. if not env.PY2:
  67. mv = memoryview(b)
  68. v = m.VectorUChar(mv[::2])
  69. assert v[1] == 3
  70. with pytest.raises(RuntimeError) as excinfo:
  71. m.create_undeclstruct() # Undeclared struct contents, no buffer interface
  72. assert "NumPy type info missing for " in str(excinfo.value)
  73. def test_vector_buffer_numpy():
  74. np = pytest.importorskip("numpy")
  75. a = np.array([1, 2, 3, 4], dtype=np.int32)
  76. with pytest.raises(TypeError):
  77. m.VectorInt(a)
  78. a = np.array([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]], dtype=np.uintc)
  79. v = m.VectorInt(a[0, :])
  80. assert len(v) == 4
  81. assert v[2] == 3
  82. ma = np.asarray(v)
  83. ma[2] = 5
  84. assert v[2] == 5
  85. v = m.VectorInt(a[:, 1])
  86. assert len(v) == 3
  87. assert v[2] == 10
  88. v = m.get_vectorstruct()
  89. assert v[0].x == 5
  90. ma = np.asarray(v)
  91. ma[1]["x"] = 99
  92. assert v[1].x == 99
  93. v = m.VectorStruct(
  94. np.zeros(
  95. 3,
  96. dtype=np.dtype(
  97. [("w", "bool"), ("x", "I"), ("y", "float64"), ("z", "bool")], align=True
  98. ),
  99. )
  100. )
  101. assert len(v) == 3
  102. b = np.array([1, 2, 3, 4], dtype=np.uint8)
  103. v = m.VectorUChar(b[::2])
  104. assert v[1] == 3
  105. def test_vector_bool():
  106. import pybind11_cross_module_tests as cm
  107. vv_c = cm.VectorBool()
  108. for i in range(10):
  109. vv_c.append(i % 2 == 0)
  110. for i in range(10):
  111. assert vv_c[i] == (i % 2 == 0)
  112. assert str(vv_c) == "VectorBool[1, 0, 1, 0, 1, 0, 1, 0, 1, 0]"
  113. def test_vector_custom():
  114. v_a = m.VectorEl()
  115. v_a.append(m.El(1))
  116. v_a.append(m.El(2))
  117. assert str(v_a) == "VectorEl[El{1}, El{2}]"
  118. vv_a = m.VectorVectorEl()
  119. vv_a.append(v_a)
  120. vv_b = vv_a[0]
  121. assert str(vv_b) == "VectorEl[El{1}, El{2}]"
  122. def test_map_string_double():
  123. mm = m.MapStringDouble()
  124. mm["a"] = 1
  125. mm["b"] = 2.5
  126. assert list(mm) == ["a", "b"]
  127. assert list(mm.items()) == [("a", 1), ("b", 2.5)]
  128. assert str(mm) == "MapStringDouble{a: 1, b: 2.5}"
  129. um = m.UnorderedMapStringDouble()
  130. um["ua"] = 1.1
  131. um["ub"] = 2.6
  132. assert sorted(list(um)) == ["ua", "ub"]
  133. assert sorted(list(um.items())) == [("ua", 1.1), ("ub", 2.6)]
  134. assert "UnorderedMapStringDouble" in str(um)
  135. def test_map_string_double_const():
  136. mc = m.MapStringDoubleConst()
  137. mc["a"] = 10
  138. mc["b"] = 20.5
  139. assert str(mc) == "MapStringDoubleConst{a: 10, b: 20.5}"
  140. umc = m.UnorderedMapStringDoubleConst()
  141. umc["a"] = 11
  142. umc["b"] = 21.5
  143. str(umc)
  144. def test_noncopyable_containers():
  145. # std::vector
  146. vnc = m.get_vnc(5)
  147. for i in range(0, 5):
  148. assert vnc[i].value == i + 1
  149. for i, j in enumerate(vnc, start=1):
  150. assert j.value == i
  151. # std::deque
  152. dnc = m.get_dnc(5)
  153. for i in range(0, 5):
  154. assert dnc[i].value == i + 1
  155. i = 1
  156. for j in dnc:
  157. assert j.value == i
  158. i += 1
  159. # std::map
  160. mnc = m.get_mnc(5)
  161. for i in range(1, 6):
  162. assert mnc[i].value == 10 * i
  163. vsum = 0
  164. for k, v in mnc.items():
  165. assert v.value == 10 * k
  166. vsum += v.value
  167. assert vsum == 150
  168. # std::unordered_map
  169. mnc = m.get_umnc(5)
  170. for i in range(1, 6):
  171. assert mnc[i].value == 10 * i
  172. vsum = 0
  173. for k, v in mnc.items():
  174. assert v.value == 10 * k
  175. vsum += v.value
  176. assert vsum == 150
  177. # nested std::map<std::vector>
  178. nvnc = m.get_nvnc(5)
  179. for i in range(1, 6):
  180. for j in range(0, 5):
  181. assert nvnc[i][j].value == j + 1
  182. # Note: maps do not have .values()
  183. for _, v in nvnc.items():
  184. for i, j in enumerate(v, start=1):
  185. assert j.value == i
  186. # nested std::map<std::map>
  187. nmnc = m.get_nmnc(5)
  188. for i in range(1, 6):
  189. for j in range(10, 60, 10):
  190. assert nmnc[i][j].value == 10 * j
  191. vsum = 0
  192. for _, v_o in nmnc.items():
  193. for k_i, v_i in v_o.items():
  194. assert v_i.value == 10 * k_i
  195. vsum += v_i.value
  196. assert vsum == 7500
  197. # nested std::unordered_map<std::unordered_map>
  198. numnc = m.get_numnc(5)
  199. for i in range(1, 6):
  200. for j in range(10, 60, 10):
  201. assert numnc[i][j].value == 10 * j
  202. vsum = 0
  203. for _, v_o in numnc.items():
  204. for k_i, v_i in v_o.items():
  205. assert v_i.value == 10 * k_i
  206. vsum += v_i.value
  207. assert vsum == 7500
  208. def test_map_delitem():
  209. mm = m.MapStringDouble()
  210. mm["a"] = 1
  211. mm["b"] = 2.5
  212. assert list(mm) == ["a", "b"]
  213. assert list(mm.items()) == [("a", 1), ("b", 2.5)]
  214. del mm["a"]
  215. assert list(mm) == ["b"]
  216. assert list(mm.items()) == [("b", 2.5)]
  217. um = m.UnorderedMapStringDouble()
  218. um["ua"] = 1.1
  219. um["ub"] = 2.6
  220. assert sorted(list(um)) == ["ua", "ub"]
  221. assert sorted(list(um.items())) == [("ua", 1.1), ("ub", 2.6)]
  222. del um["ua"]
  223. assert sorted(list(um)) == ["ub"]
  224. assert sorted(list(um.items())) == [("ub", 2.6)]