| Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 1 | def test_vector_int(): |
| 2 | from pybind11_tests import VectorInt |
| 3 | |
| 4 | v_int = VectorInt([0, 0]) |
| 5 | assert len(v_int) == 2 |
| 6 | assert bool(v_int) is True |
| 7 | |
| 8 | v_int2 = VectorInt([0, 0]) |
| 9 | assert v_int == v_int2 |
| 10 | v_int2[1] = 1 |
| 11 | assert v_int != v_int2 |
| 12 | |
| 13 | v_int2.append(2) |
| 14 | v_int2.append(3) |
| 15 | v_int2.insert(0, 1) |
| 16 | v_int2.insert(0, 2) |
| 17 | v_int2.insert(0, 3) |
| 18 | assert str(v_int2) == "VectorInt[3, 2, 1, 0, 1, 2, 3]" |
| 19 | |
| 20 | v_int.append(99) |
| 21 | v_int2[2:-2] = v_int |
| 22 | assert v_int2 == VectorInt([3, 2, 0, 0, 99, 2, 3]) |
| 23 | del v_int2[1:3] |
| 24 | assert v_int2 == VectorInt([3, 0, 99, 2, 3]) |
| 25 | del v_int2[0] |
| 26 | assert v_int2 == VectorInt([0, 99, 2, 3]) |
| 27 | |
| 28 | |
| 29 | def test_vector_custom(): |
| 30 | from pybind11_tests import El, VectorEl, VectorVectorEl |
| 31 | |
| 32 | v_a = VectorEl() |
| 33 | v_a.append(El(1)) |
| 34 | v_a.append(El(2)) |
| 35 | assert str(v_a) == "VectorEl[El{1}, El{2}]" |
| 36 | |
| 37 | vv_a = VectorVectorEl() |
| 38 | vv_a.append(v_a) |
| 39 | vv_b = vv_a[0] |
| 40 | assert str(vv_b) == "VectorEl[El{1}, El{2}]" |
| 41 | |
| 42 | |
| 43 | def test_vector_bool(): |
| 44 | from pybind11_tests import VectorBool |
| 45 | |
| 46 | vv_c = VectorBool() |
| 47 | for i in range(10): |
| 48 | vv_c.append(i % 2 == 0) |
| 49 | for i in range(10): |
| 50 | assert vv_c[i] == (i % 2 == 0) |
| 51 | assert str(vv_c) == "VectorBool[1, 0, 1, 0, 1, 0, 1, 0, 1, 0]" |
| Sergey Lyskov | 7520418 | 2016-08-29 22:50:38 -0400 | [diff] [blame] | 52 | |
| 53 | |
| 54 | def test_map_string_double(): |
| 55 | from pybind11_tests import MapStringDouble, UnorderedMapStringDouble |
| 56 | |
| 57 | m = MapStringDouble() |
| 58 | m['a'] = 1 |
| 59 | m['b'] = 2.5 |
| 60 | |
| 61 | keys = [] |
| 62 | for k in m: keys.append(k) |
| 63 | assert keys == ['a', 'b'] |
| 64 | |
| 65 | key_values = [] |
| 66 | for k, v in m.items(): key_values.append( (k, v) ) |
| 67 | assert key_values == [('a', 1), ('b', 2.5) ] |
| 68 | |
| 69 | assert str(m) == "MapStringDouble{a: 1, b: 2.5}" |
| 70 | |
| 71 | |
| 72 | um = UnorderedMapStringDouble() |
| 73 | um['ua'] = 1.1 |
| 74 | um['ub'] = 2.6 |
| 75 | |
| 76 | keys = [] |
| 77 | for k in um: keys.append(k) |
| 78 | assert sorted(keys) == ['ua', 'ub'] |
| 79 | |
| 80 | key_values = [] |
| 81 | for k, v in um.items(): key_values.append( (k, v) ) |
| 82 | assert sorted(key_values) == [('ua', 1.1), ('ub', 2.6) ] |
| 83 | |
| 84 | str(um) |
| 85 | |
| 86 | |
| 87 | def test_map_string_double_const(): |
| 88 | from pybind11_tests import MapStringDoubleConst, UnorderedMapStringDoubleConst |
| 89 | |
| 90 | mc = MapStringDoubleConst() |
| 91 | mc['a'] = 10 |
| 92 | mc['b'] = 20.5 |
| 93 | assert str(mc) == "MapStringDoubleConst{a: 10, b: 20.5}" |
| 94 | |
| 95 | umc = UnorderedMapStringDoubleConst() |
| 96 | umc['a'] = 11 |
| 97 | umc['b'] = 21.5 |
| 98 | |
| 99 | str(umc) |