2 from pytest
import approx
4 from pybind11_tests
import ConstructorStats
5 from pybind11_tests
import sequences_and_iterators
as m
9 assert m.make_forward_slice_size_t() ==
slice(0, -1, 1)
10 assert m.make_reversed_slice_object() ==
slice(
None,
None, -1)
13 @pytest.mark.skipif(
not m.has_optional, reason=
"no <optional>")
15 assert m.make_reversed_slice_size_t_optional() ==
slice(
None,
None, -1)
16 assert m.make_reversed_slice_size_t_optional_verbose() ==
slice(
None,
None, -1)
20 assert list(m.IntPairs([(1, 2), (3, 4), (0, 5)]).nonzero()) == [(1, 2), (3, 4)]
21 assert list(m.IntPairs([(1, 2), (2, 0), (0, 3), (4, 5)]).nonzero()) == [(1, 2)]
22 assert list(m.IntPairs([(0, 3), (1, 2), (3, 4)]).nonzero()) == []
24 assert list(m.IntPairs([(1, 2), (3, 4), (0, 5)]).nonzero_keys()) == [1, 3]
25 assert list(m.IntPairs([(1, 2), (2, 0), (0, 3), (4, 5)]).nonzero_keys()) == [1]
26 assert list(m.IntPairs([(0, 3), (1, 2), (3, 4)]).nonzero_keys()) == []
28 assert list(m.IntPairs([(1, 2), (3, 4), (0, 5)]).nonzero_values()) == [2, 4]
29 assert list(m.IntPairs([(1, 2), (2, 0), (0, 3), (4, 5)]).nonzero_values()) == [2]
30 assert list(m.IntPairs([(0, 3), (1, 2), (3, 4)]).nonzero_values()) == []
33 it = m.IntPairs([(0, 0)]).nonzero()
35 with pytest.raises(StopIteration):
38 it = m.IntPairs([(0, 0)]).nonzero_keys()
40 with pytest.raises(StopIteration):
45 pairs = m.IntPairs([(1, 2), (3, 4), (0, 5)])
46 assert list(pairs.nonref()) == [(1, 2), (3, 4), (0, 5)]
47 assert list(pairs.nonref_keys()) == [1, 3, 0]
48 assert list(pairs.nonref_values()) == [2, 4, 5]
52 assert list(m.IntPairs([(1, 2), (3, 4), (0, 5)]).simple_iterator()) == [
57 assert list(m.IntPairs([(1, 2), (3, 4), (0, 5)]).simple_keys()) == [1, 3, 0]
58 assert list(m.IntPairs([(1, 2), (3, 4), (0, 5)]).simple_values()) == [2, 4, 5]
62 """Test that iterators reference rather than copy their referents."""
63 vec = m.VectorNonCopyableInt()
66 assert [
int(x)
for x
in vec] == [3, 5]
70 assert [
int(x)
for x
in vec] == [4, 6]
72 vec = m.VectorNonCopyableIntPair()
75 assert [
int(x)
for x
in vec.keys()] == [3, 5]
76 assert [
int(x)
for x
in vec.values()] == [4, 7]
79 for x
in vec.values():
81 assert [
int(x)
for x
in vec.keys()] == [4, 6]
82 assert [
int(x)
for x
in vec.values()] == [14, 17]
86 sliceable = m.Sliceable(100)
87 assert sliceable[::] == (0, 100, 1)
88 assert sliceable[10::] == (10, 100, 1)
89 assert sliceable[:10:] == (0, 10, 1)
90 assert sliceable[::10] == (0, 100, 10)
91 assert sliceable[-10::] == (90, 100, 1)
92 assert sliceable[:-10:] == (0, 90, 1)
93 assert sliceable[::-10] == (99, -1, -10)
94 assert sliceable[50:60:1] == (50, 60, 1)
95 assert sliceable[50:60:-1] == (50, 60, -1)
102 assert cstats.values() == [
"of size",
"5"]
104 assert "Sequence" in repr(s)
108 assert 12.34
not in s
109 s[0], s[3] = 12.34, 56.78
111 assert s[0] == approx(12.34, rel=1e-05)
112 assert s[3] == approx(56.78, rel=1e-05)
115 assert cstats.values() == [
"of size",
"5"]
118 assert cstats.values() == [
"of size",
"5"]
120 it =
iter(m.Sequence(0))
122 with pytest.raises(StopIteration):
124 assert cstats.values() == [
"of size",
"0"]
126 expected = [0, 56.78, 0, 0, 12.34]
127 assert rev == approx(expected, rel=1e-05)
128 assert rev2 == approx(expected, rel=1e-05)
131 rev[0::2] = m.Sequence([2.0, 2.0, 2.0])
132 assert cstats.values() == [
"of size",
"3",
"from std::vector"]
134 assert rev == approx([2, 56.78, 2, 0, 2], rel=1e-05)
136 assert cstats.alive() == 4
138 assert cstats.alive() == 3
140 assert cstats.alive() == 2
142 assert cstats.alive() == 1
144 assert cstats.alive() == 0
146 assert cstats.values() == []
147 assert cstats.default_constructions == 0
148 assert cstats.copy_constructions == 0
149 assert cstats.move_constructions >= 1
150 assert cstats.copy_assignments == 0
151 assert cstats.move_assignments == 0
155 """#2076: Exception raised by len(arg) should be propagated"""
157 class BadLen(RuntimeError):
161 def __getitem__(self, i):
167 with pytest.raises(BadLen):
168 m.sequence_length(SequenceLike())
170 assert m.sequence_length([1, 2, 3]) == 3
171 assert m.sequence_length(
"hello") == 5
175 sm = m.StringMap({
"hi":
"bye",
"black":
"white"})
176 assert sm[
"hi"] ==
"bye"
178 assert sm[
"black"] ==
"white"
180 with pytest.raises(KeyError):
182 sm[
"orange"] =
"banana"
183 assert sm[
"orange"] ==
"banana"
185 expected = {
"hi":
"bye",
"black":
"white",
"orange":
"banana"}
187 assert sm[k] == expected[k]
188 for k, v
in sm.items():
189 assert v == expected[k]
190 assert list(sm.values()) == [expected[k]
for k
in sm]
192 it =
iter(m.StringMap({}))
194 with pytest.raises(StopIteration):
200 assert m.object_to_list(t) == [1, 2, 3]
201 assert m.object_to_list(
iter(t)) == [1, 2, 3]
202 assert m.iterator_to_list(
iter(t)) == [1, 2, 3]
204 with pytest.raises(TypeError)
as excinfo:
206 assert "object is not iterable" in str(excinfo.value)
208 with pytest.raises(TypeError)
as excinfo:
209 m.iterator_to_list(1)
210 assert "incompatible function arguments" in str(excinfo.value)
213 raise RuntimeError(
"py::iterator::advance() should propagate errors")
215 with pytest.raises(RuntimeError)
as excinfo:
216 m.iterator_to_list(
iter(bad_next_call,
None))
217 assert str(excinfo.value) ==
"py::iterator::advance() should propagate errors"
219 lst = [1,
None, 0,
None]
220 assert m.count_none(lst) == 2
221 assert m.find_none(lst)
is True
222 assert m.count_nonzeros({
"a": 0,
"b": 1,
"c": 2}) == 2
225 assert all(m.tuple_iterator(
tuple(r)))
226 assert all(m.list_iterator(
list(r)))
227 assert all(m.sequence_iterator(r))
231 """#181: iterator passthrough did not compile"""
232 from pybind11_tests.sequences_and_iterators
import iterator_passthrough
234 values = [3, 5, 7, 9, 11, 13, 15]
235 assert list(iterator_passthrough(
iter(values))) == values
239 """#388: Can't make iterators via make_iterator() with different r/v policies"""
240 import pybind11_tests.sequences_and_iterators
as m
242 assert list(m.make_iterator_1()) == [1, 2, 3]
243 assert list(m.make_iterator_2()) == [1, 2, 3]
244 assert not isinstance(m.make_iterator_1(),
type(m.make_iterator_2()))
248 """#4100: Check for proper iterator overload with C-Arrays"""
250 arr_h = m.CArrayHolder(*args_gt)
252 assert args_gt == args