123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291 |
- # -*- coding: utf-8 -*-
- import pytest
- import env # noqa: F401
- from pybind11_tests import stl_binders as m
- def test_vector_int():
- v_int = m.VectorInt([0, 0])
- assert len(v_int) == 2
- assert bool(v_int) is True
- # test construction from a generator
- v_int1 = m.VectorInt(x for x in range(5))
- assert v_int1 == m.VectorInt([0, 1, 2, 3, 4])
- v_int2 = m.VectorInt([0, 0])
- assert v_int == v_int2
- v_int2[1] = 1
- assert v_int != v_int2
- v_int2.append(2)
- v_int2.insert(0, 1)
- v_int2.insert(0, 2)
- v_int2.insert(0, 3)
- v_int2.insert(6, 3)
- assert str(v_int2) == "VectorInt[3, 2, 1, 0, 1, 2, 3]"
- with pytest.raises(IndexError):
- v_int2.insert(8, 4)
- v_int.append(99)
- v_int2[2:-2] = v_int
- assert v_int2 == m.VectorInt([3, 2, 0, 0, 99, 2, 3])
- del v_int2[1:3]
- assert v_int2 == m.VectorInt([3, 0, 99, 2, 3])
- del v_int2[0]
- assert v_int2 == m.VectorInt([0, 99, 2, 3])
- v_int2.extend(m.VectorInt([4, 5]))
- assert v_int2 == m.VectorInt([0, 99, 2, 3, 4, 5])
- v_int2.extend([6, 7])
- assert v_int2 == m.VectorInt([0, 99, 2, 3, 4, 5, 6, 7])
- # test error handling, and that the vector is unchanged
- with pytest.raises(RuntimeError):
- v_int2.extend([8, "a"])
- assert v_int2 == m.VectorInt([0, 99, 2, 3, 4, 5, 6, 7])
- # test extending from a generator
- v_int2.extend(x for x in range(5))
- assert v_int2 == m.VectorInt([0, 99, 2, 3, 4, 5, 6, 7, 0, 1, 2, 3, 4])
- # test negative indexing
- assert v_int2[-1] == 4
- # insert with negative index
- v_int2.insert(-1, 88)
- assert v_int2 == m.VectorInt([0, 99, 2, 3, 4, 5, 6, 7, 0, 1, 2, 3, 88, 4])
- # delete negative index
- del v_int2[-1]
- assert v_int2 == m.VectorInt([0, 99, 2, 3, 4, 5, 6, 7, 0, 1, 2, 3, 88])
- v_int2.clear()
- assert len(v_int2) == 0
- # Older PyPy's failed here, related to the PyPy's buffer protocol.
- def test_vector_buffer():
- b = bytearray([1, 2, 3, 4])
- v = m.VectorUChar(b)
- assert v[1] == 2
- v[2] = 5
- mv = memoryview(v) # We expose the buffer interface
- if not env.PY2:
- assert mv[2] == 5
- mv[2] = 6
- else:
- assert mv[2] == "\x05"
- mv[2] = "\x06"
- assert v[2] == 6
- if not env.PY2:
- mv = memoryview(b)
- v = m.VectorUChar(mv[::2])
- assert v[1] == 3
- with pytest.raises(RuntimeError) as excinfo:
- m.create_undeclstruct() # Undeclared struct contents, no buffer interface
- assert "NumPy type info missing for " in str(excinfo.value)
- def test_vector_buffer_numpy():
- np = pytest.importorskip("numpy")
- a = np.array([1, 2, 3, 4], dtype=np.int32)
- with pytest.raises(TypeError):
- m.VectorInt(a)
- a = np.array([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]], dtype=np.uintc)
- v = m.VectorInt(a[0, :])
- assert len(v) == 4
- assert v[2] == 3
- ma = np.asarray(v)
- ma[2] = 5
- assert v[2] == 5
- v = m.VectorInt(a[:, 1])
- assert len(v) == 3
- assert v[2] == 10
- v = m.get_vectorstruct()
- assert v[0].x == 5
- ma = np.asarray(v)
- ma[1]["x"] = 99
- assert v[1].x == 99
- v = m.VectorStruct(
- np.zeros(
- 3,
- dtype=np.dtype(
- [("w", "bool"), ("x", "I"), ("y", "float64"), ("z", "bool")], align=True
- ),
- )
- )
- assert len(v) == 3
- b = np.array([1, 2, 3, 4], dtype=np.uint8)
- v = m.VectorUChar(b[::2])
- assert v[1] == 3
- def test_vector_bool():
- import pybind11_cross_module_tests as cm
- vv_c = cm.VectorBool()
- for i in range(10):
- vv_c.append(i % 2 == 0)
- for i in range(10):
- assert vv_c[i] == (i % 2 == 0)
- assert str(vv_c) == "VectorBool[1, 0, 1, 0, 1, 0, 1, 0, 1, 0]"
- def test_vector_custom():
- v_a = m.VectorEl()
- v_a.append(m.El(1))
- v_a.append(m.El(2))
- assert str(v_a) == "VectorEl[El{1}, El{2}]"
- vv_a = m.VectorVectorEl()
- vv_a.append(v_a)
- vv_b = vv_a[0]
- assert str(vv_b) == "VectorEl[El{1}, El{2}]"
- def test_map_string_double():
- mm = m.MapStringDouble()
- mm["a"] = 1
- mm["b"] = 2.5
- assert list(mm) == ["a", "b"]
- assert list(mm.items()) == [("a", 1), ("b", 2.5)]
- assert str(mm) == "MapStringDouble{a: 1, b: 2.5}"
- um = m.UnorderedMapStringDouble()
- um["ua"] = 1.1
- um["ub"] = 2.6
- assert sorted(list(um)) == ["ua", "ub"]
- assert sorted(list(um.items())) == [("ua", 1.1), ("ub", 2.6)]
- assert "UnorderedMapStringDouble" in str(um)
- def test_map_string_double_const():
- mc = m.MapStringDoubleConst()
- mc["a"] = 10
- mc["b"] = 20.5
- assert str(mc) == "MapStringDoubleConst{a: 10, b: 20.5}"
- umc = m.UnorderedMapStringDoubleConst()
- umc["a"] = 11
- umc["b"] = 21.5
- str(umc)
- def test_noncopyable_containers():
- # std::vector
- vnc = m.get_vnc(5)
- for i in range(0, 5):
- assert vnc[i].value == i + 1
- for i, j in enumerate(vnc, start=1):
- assert j.value == i
- # std::deque
- dnc = m.get_dnc(5)
- for i in range(0, 5):
- assert dnc[i].value == i + 1
- i = 1
- for j in dnc:
- assert j.value == i
- i += 1
- # std::map
- mnc = m.get_mnc(5)
- for i in range(1, 6):
- assert mnc[i].value == 10 * i
- vsum = 0
- for k, v in mnc.items():
- assert v.value == 10 * k
- vsum += v.value
- assert vsum == 150
- # std::unordered_map
- mnc = m.get_umnc(5)
- for i in range(1, 6):
- assert mnc[i].value == 10 * i
- vsum = 0
- for k, v in mnc.items():
- assert v.value == 10 * k
- vsum += v.value
- assert vsum == 150
- # nested std::map<std::vector>
- nvnc = m.get_nvnc(5)
- for i in range(1, 6):
- for j in range(0, 5):
- assert nvnc[i][j].value == j + 1
- # Note: maps do not have .values()
- for _, v in nvnc.items():
- for i, j in enumerate(v, start=1):
- assert j.value == i
- # nested std::map<std::map>
- nmnc = m.get_nmnc(5)
- for i in range(1, 6):
- for j in range(10, 60, 10):
- assert nmnc[i][j].value == 10 * j
- vsum = 0
- for _, v_o in nmnc.items():
- for k_i, v_i in v_o.items():
- assert v_i.value == 10 * k_i
- vsum += v_i.value
- assert vsum == 7500
- # nested std::unordered_map<std::unordered_map>
- numnc = m.get_numnc(5)
- for i in range(1, 6):
- for j in range(10, 60, 10):
- assert numnc[i][j].value == 10 * j
- vsum = 0
- for _, v_o in numnc.items():
- for k_i, v_i in v_o.items():
- assert v_i.value == 10 * k_i
- vsum += v_i.value
- assert vsum == 7500
- def test_map_delitem():
- mm = m.MapStringDouble()
- mm["a"] = 1
- mm["b"] = 2.5
- assert list(mm) == ["a", "b"]
- assert list(mm.items()) == [("a", 1), ("b", 2.5)]
- del mm["a"]
- assert list(mm) == ["b"]
- assert list(mm.items()) == [("b", 2.5)]
- um = m.UnorderedMapStringDouble()
- um["ua"] = 1.1
- um["ub"] = 2.6
- assert sorted(list(um)) == ["ua", "ub"]
- assert sorted(list(um.items())) == [("ua", 1.1), ("ub", 2.6)]
- del um["ua"]
- assert sorted(list(um)) == ["ub"]
- assert sorted(list(um.items())) == [("ub", 2.6)]
|