test_stl_binders.py revision 11986
111986Sandreas.sandberg@arm.comdef test_vector_int(): 211986Sandreas.sandberg@arm.com from pybind11_tests import VectorInt 311986Sandreas.sandberg@arm.com 411986Sandreas.sandberg@arm.com v_int = VectorInt([0, 0]) 511986Sandreas.sandberg@arm.com assert len(v_int) == 2 611986Sandreas.sandberg@arm.com assert bool(v_int) is True 711986Sandreas.sandberg@arm.com 811986Sandreas.sandberg@arm.com v_int2 = VectorInt([0, 0]) 911986Sandreas.sandberg@arm.com assert v_int == v_int2 1011986Sandreas.sandberg@arm.com v_int2[1] = 1 1111986Sandreas.sandberg@arm.com assert v_int != v_int2 1211986Sandreas.sandberg@arm.com 1311986Sandreas.sandberg@arm.com v_int2.append(2) 1411986Sandreas.sandberg@arm.com v_int2.append(3) 1511986Sandreas.sandberg@arm.com v_int2.insert(0, 1) 1611986Sandreas.sandberg@arm.com v_int2.insert(0, 2) 1711986Sandreas.sandberg@arm.com v_int2.insert(0, 3) 1811986Sandreas.sandberg@arm.com assert str(v_int2) == "VectorInt[3, 2, 1, 0, 1, 2, 3]" 1911986Sandreas.sandberg@arm.com 2011986Sandreas.sandberg@arm.com v_int.append(99) 2111986Sandreas.sandberg@arm.com v_int2[2:-2] = v_int 2211986Sandreas.sandberg@arm.com assert v_int2 == VectorInt([3, 2, 0, 0, 99, 2, 3]) 2311986Sandreas.sandberg@arm.com del v_int2[1:3] 2411986Sandreas.sandberg@arm.com assert v_int2 == VectorInt([3, 0, 99, 2, 3]) 2511986Sandreas.sandberg@arm.com del v_int2[0] 2611986Sandreas.sandberg@arm.com assert v_int2 == VectorInt([0, 99, 2, 3]) 2711986Sandreas.sandberg@arm.com 2811986Sandreas.sandberg@arm.com 2911986Sandreas.sandberg@arm.comdef test_vector_custom(): 3011986Sandreas.sandberg@arm.com from pybind11_tests import El, VectorEl, VectorVectorEl 3111986Sandreas.sandberg@arm.com 3211986Sandreas.sandberg@arm.com v_a = VectorEl() 3311986Sandreas.sandberg@arm.com v_a.append(El(1)) 3411986Sandreas.sandberg@arm.com v_a.append(El(2)) 3511986Sandreas.sandberg@arm.com assert str(v_a) == "VectorEl[El{1}, El{2}]" 3611986Sandreas.sandberg@arm.com 3711986Sandreas.sandberg@arm.com vv_a = VectorVectorEl() 3811986Sandreas.sandberg@arm.com vv_a.append(v_a) 3911986Sandreas.sandberg@arm.com vv_b = vv_a[0] 4011986Sandreas.sandberg@arm.com assert str(vv_b) == "VectorEl[El{1}, El{2}]" 4111986Sandreas.sandberg@arm.com 4211986Sandreas.sandberg@arm.com 4311986Sandreas.sandberg@arm.comdef test_vector_bool(): 4411986Sandreas.sandberg@arm.com from pybind11_tests import VectorBool 4511986Sandreas.sandberg@arm.com 4611986Sandreas.sandberg@arm.com vv_c = VectorBool() 4711986Sandreas.sandberg@arm.com for i in range(10): 4811986Sandreas.sandberg@arm.com vv_c.append(i % 2 == 0) 4911986Sandreas.sandberg@arm.com for i in range(10): 5011986Sandreas.sandberg@arm.com assert vv_c[i] == (i % 2 == 0) 5111986Sandreas.sandberg@arm.com assert str(vv_c) == "VectorBool[1, 0, 1, 0, 1, 0, 1, 0, 1, 0]" 5211986Sandreas.sandberg@arm.com 5311986Sandreas.sandberg@arm.com 5411986Sandreas.sandberg@arm.comdef test_map_string_double(): 5511986Sandreas.sandberg@arm.com from pybind11_tests import MapStringDouble, UnorderedMapStringDouble 5611986Sandreas.sandberg@arm.com 5711986Sandreas.sandberg@arm.com m = MapStringDouble() 5811986Sandreas.sandberg@arm.com m['a'] = 1 5911986Sandreas.sandberg@arm.com m['b'] = 2.5 6011986Sandreas.sandberg@arm.com 6111986Sandreas.sandberg@arm.com assert list(m) == ['a', 'b'] 6211986Sandreas.sandberg@arm.com assert list(m.items()) == [('a', 1), ('b', 2.5)] 6311986Sandreas.sandberg@arm.com assert str(m) == "MapStringDouble{a: 1, b: 2.5}" 6411986Sandreas.sandberg@arm.com 6511986Sandreas.sandberg@arm.com um = UnorderedMapStringDouble() 6611986Sandreas.sandberg@arm.com um['ua'] = 1.1 6711986Sandreas.sandberg@arm.com um['ub'] = 2.6 6811986Sandreas.sandberg@arm.com 6911986Sandreas.sandberg@arm.com assert sorted(list(um)) == ['ua', 'ub'] 7011986Sandreas.sandberg@arm.com assert sorted(list(um.items())) == [('ua', 1.1), ('ub', 2.6)] 7111986Sandreas.sandberg@arm.com assert "UnorderedMapStringDouble" in str(um) 7211986Sandreas.sandberg@arm.com 7311986Sandreas.sandberg@arm.com 7411986Sandreas.sandberg@arm.comdef test_map_string_double_const(): 7511986Sandreas.sandberg@arm.com from pybind11_tests import MapStringDoubleConst, UnorderedMapStringDoubleConst 7611986Sandreas.sandberg@arm.com 7711986Sandreas.sandberg@arm.com mc = MapStringDoubleConst() 7811986Sandreas.sandberg@arm.com mc['a'] = 10 7911986Sandreas.sandberg@arm.com mc['b'] = 20.5 8011986Sandreas.sandberg@arm.com assert str(mc) == "MapStringDoubleConst{a: 10, b: 20.5}" 8111986Sandreas.sandberg@arm.com 8211986Sandreas.sandberg@arm.com umc = UnorderedMapStringDoubleConst() 8311986Sandreas.sandberg@arm.com umc['a'] = 11 8411986Sandreas.sandberg@arm.com umc['b'] = 21.5 8511986Sandreas.sandberg@arm.com 8611986Sandreas.sandberg@arm.com str(umc) 8711986Sandreas.sandberg@arm.com 8811986Sandreas.sandberg@arm.com 8911986Sandreas.sandberg@arm.comdef test_noncopyable_vector(): 9011986Sandreas.sandberg@arm.com from pybind11_tests import get_vnc 9111986Sandreas.sandberg@arm.com 9211986Sandreas.sandberg@arm.com vnc = get_vnc(5) 9311986Sandreas.sandberg@arm.com for i in range(0, 5): 9411986Sandreas.sandberg@arm.com assert vnc[i].value == i + 1 9511986Sandreas.sandberg@arm.com 9611986Sandreas.sandberg@arm.com for i, j in enumerate(vnc, start=1): 9711986Sandreas.sandberg@arm.com assert j.value == i 9811986Sandreas.sandberg@arm.com 9911986Sandreas.sandberg@arm.com 10011986Sandreas.sandberg@arm.comdef test_noncopyable_deque(): 10111986Sandreas.sandberg@arm.com from pybind11_tests import get_dnc 10211986Sandreas.sandberg@arm.com 10311986Sandreas.sandberg@arm.com dnc = get_dnc(5) 10411986Sandreas.sandberg@arm.com for i in range(0, 5): 10511986Sandreas.sandberg@arm.com assert dnc[i].value == i + 1 10611986Sandreas.sandberg@arm.com 10711986Sandreas.sandberg@arm.com i = 1 10811986Sandreas.sandberg@arm.com for j in dnc: 10911986Sandreas.sandberg@arm.com assert(j.value == i) 11011986Sandreas.sandberg@arm.com i += 1 11111986Sandreas.sandberg@arm.com 11211986Sandreas.sandberg@arm.com 11311986Sandreas.sandberg@arm.comdef test_noncopyable_map(): 11411986Sandreas.sandberg@arm.com from pybind11_tests import get_mnc 11511986Sandreas.sandberg@arm.com 11611986Sandreas.sandberg@arm.com mnc = get_mnc(5) 11711986Sandreas.sandberg@arm.com for i in range(1, 6): 11811986Sandreas.sandberg@arm.com assert mnc[i].value == 10 * i 11911986Sandreas.sandberg@arm.com 12011986Sandreas.sandberg@arm.com vsum = 0 12111986Sandreas.sandberg@arm.com for k, v in mnc.items(): 12211986Sandreas.sandberg@arm.com assert v.value == 10 * k 12311986Sandreas.sandberg@arm.com vsum += v.value 12411986Sandreas.sandberg@arm.com 12511986Sandreas.sandberg@arm.com assert vsum == 150 12611986Sandreas.sandberg@arm.com 12711986Sandreas.sandberg@arm.com 12811986Sandreas.sandberg@arm.comdef test_noncopyable_unordered_map(): 12911986Sandreas.sandberg@arm.com from pybind11_tests import get_umnc 13011986Sandreas.sandberg@arm.com 13111986Sandreas.sandberg@arm.com mnc = get_umnc(5) 13211986Sandreas.sandberg@arm.com for i in range(1, 6): 13311986Sandreas.sandberg@arm.com assert mnc[i].value == 10 * i 13411986Sandreas.sandberg@arm.com 13511986Sandreas.sandberg@arm.com vsum = 0 13611986Sandreas.sandberg@arm.com for k, v in mnc.items(): 13711986Sandreas.sandberg@arm.com assert v.value == 10 * k 13811986Sandreas.sandberg@arm.com vsum += v.value 13911986Sandreas.sandberg@arm.com 14011986Sandreas.sandberg@arm.com assert vsum == 150 141