ext/pybind11/tests/test_sequences_and_iterators.py

   1 # -*- coding: utf-8 -*-
   2 import pytest
   3 from pybind11_tests import sequences_and_iterators as m
   4 from pybind11_tests import ConstructorStats
   5
   6
   7 def isclose(a, b, rel_tol=1e-05, abs_tol=0.0):
   8     """Like math.isclose() from Python 3.5"""
   9     return abs(a - b) <= max(rel_tol * max(abs(a), abs(b)), abs_tol)
  10
  11
  12 def allclose(a_list, b_list, rel_tol=1e-05, abs_tol=0.0):
  13     return all(
  14         isclose(a, b, rel_tol=rel_tol, abs_tol=abs_tol) for a, b in zip(a_list, b_list)
  15     )
  16
  17
  18 def test_generalized_iterators():
  19     assert list(m.IntPairs([(1, 2), (3, 4), (0, 5)]).nonzero()) == [(1, 2), (3, 4)]
  20     assert list(m.IntPairs([(1, 2), (2, 0), (0, 3), (4, 5)]).nonzero()) == [(1, 2)]
  21     assert list(m.IntPairs([(0, 3), (1, 2), (3, 4)]).nonzero()) == []
  22
  23     assert list(m.IntPairs([(1, 2), (3, 4), (0, 5)]).nonzero_keys()) == [1, 3]
  24     assert list(m.IntPairs([(1, 2), (2, 0), (0, 3), (4, 5)]).nonzero_keys()) == [1]
  25     assert list(m.IntPairs([(0, 3), (1, 2), (3, 4)]).nonzero_keys()) == []
  26
  27     # __next__ must continue to raise StopIteration
  28     it = m.IntPairs([(0, 0)]).nonzero()
  29     for _ in range(3):
  30         with pytest.raises(StopIteration):
  31             next(it)
  32
  33     it = m.IntPairs([(0, 0)]).nonzero_keys()
  34     for _ in range(3):
  35         with pytest.raises(StopIteration):
  36             next(it)
  37
  38
  39 def test_sliceable():
  40     sliceable = m.Sliceable(100)
  41     assert sliceable[::] == (0, 100, 1)
  42     assert sliceable[10::] == (10, 100, 1)
  43     assert sliceable[:10:] == (0, 10, 1)
  44     assert sliceable[::10] == (0, 100, 10)
  45     assert sliceable[-10::] == (90, 100, 1)
  46     assert sliceable[:-10:] == (0, 90, 1)
  47     assert sliceable[::-10] == (99, -1, -10)
  48     assert sliceable[50:60:1] == (50, 60, 1)
  49     assert sliceable[50:60:-1] == (50, 60, -1)
  50
  51
  52 def test_sequence():
  53     cstats = ConstructorStats.get(m.Sequence)
  54
  55     s = m.Sequence(5)
  56     assert cstats.values() == ["of size", "5"]
  57
  58     assert "Sequence" in repr(s)
  59     assert len(s) == 5
  60     assert s[0] == 0 and s[3] == 0
  61     assert 12.34 not in s
  62     s[0], s[3] = 12.34, 56.78
  63     assert 12.34 in s
  64     assert isclose(s[0], 12.34) and isclose(s[3], 56.78)
  65
  66     rev = reversed(s)
  67     assert cstats.values() == ["of size", "5"]
  68
  69     rev2 = s[::-1]
  70     assert cstats.values() == ["of size", "5"]
  71
  72     it = iter(m.Sequence(0))
  73     for _ in range(3):  # __next__ must continue to raise StopIteration
  74         with pytest.raises(StopIteration):
  75             next(it)
  76     assert cstats.values() == ["of size", "0"]
  77
  78     expected = [0, 56.78, 0, 0, 12.34]
  79     assert allclose(rev, expected)
  80     assert allclose(rev2, expected)
  81     assert rev == rev2
  82
  83     rev[0::2] = m.Sequence([2.0, 2.0, 2.0])
  84     assert cstats.values() == ["of size", "3", "from std::vector"]
  85
  86     assert allclose(rev, [2, 56.78, 2, 0, 2])
  87
  88     assert cstats.alive() == 4
  89     del it
  90     assert cstats.alive() == 3
  91     del s
  92     assert cstats.alive() == 2
  93     del rev
  94     assert cstats.alive() == 1
  95     del rev2
  96     assert cstats.alive() == 0
  97
  98     assert cstats.values() == []
  99     assert cstats.default_constructions == 0
 100     assert cstats.copy_constructions == 0
 101     assert cstats.move_constructions >= 1
 102     assert cstats.copy_assignments == 0
 103     assert cstats.move_assignments == 0
 104
 105
 106 def test_sequence_length():
 107     """#2076: Exception raised by len(arg) should be propagated """
 108
 109     class BadLen(RuntimeError):
 110         pass
 111
 112     class SequenceLike:
 113         def __getitem__(self, i):
 114             return None
 115
 116         def __len__(self):
 117             raise BadLen()
 118
 119     with pytest.raises(BadLen):
 120         m.sequence_length(SequenceLike())
 121
 122     assert m.sequence_length([1, 2, 3]) == 3
 123     assert m.sequence_length("hello") == 5
 124
 125
 126 def test_map_iterator():
 127     sm = m.StringMap({"hi": "bye", "black": "white"})
 128     assert sm["hi"] == "bye"
 129     assert len(sm) == 2
 130     assert sm["black"] == "white"
 131
 132     with pytest.raises(KeyError):
 133         assert sm["orange"]
 134     sm["orange"] = "banana"
 135     assert sm["orange"] == "banana"
 136
 137     expected = {"hi": "bye", "black": "white", "orange": "banana"}
 138     for k in sm:
 139         assert sm[k] == expected[k]
 140     for k, v in sm.items():
 141         assert v == expected[k]
 142
 143     it = iter(m.StringMap({}))
 144     for _ in range(3):  # __next__ must continue to raise StopIteration
 145         with pytest.raises(StopIteration):
 146             next(it)
 147
 148
 149 def test_python_iterator_in_cpp():
 150     t = (1, 2, 3)
 151     assert m.object_to_list(t) == [1, 2, 3]
 152     assert m.object_to_list(iter(t)) == [1, 2, 3]
 153     assert m.iterator_to_list(iter(t)) == [1, 2, 3]
 154
 155     with pytest.raises(TypeError) as excinfo:
 156         m.object_to_list(1)
 157     assert "object is not iterable" in str(excinfo.value)
 158
 159     with pytest.raises(TypeError) as excinfo:
 160         m.iterator_to_list(1)
 161     assert "incompatible function arguments" in str(excinfo.value)
 162
 163     def bad_next_call():
 164         raise RuntimeError("py::iterator::advance() should propagate errors")
 165
 166     with pytest.raises(RuntimeError) as excinfo:
 167         m.iterator_to_list(iter(bad_next_call, None))
 168     assert str(excinfo.value) == "py::iterator::advance() should propagate errors"
 169
 170     lst = [1, None, 0, None]
 171     assert m.count_none(lst) == 2
 172     assert m.find_none(lst) is True
 173     assert m.count_nonzeros({"a": 0, "b": 1, "c": 2}) == 2
 174
 175     r = range(5)
 176     assert all(m.tuple_iterator(tuple(r)))
 177     assert all(m.list_iterator(list(r)))
 178     assert all(m.sequence_iterator(r))
 179
 180
 181 def test_iterator_passthrough():
 182     """#181: iterator passthrough did not compile"""
 183     from pybind11_tests.sequences_and_iterators import iterator_passthrough
 184
 185     values = [3, 5, 7, 9, 11, 13, 15]
 186     assert list(iterator_passthrough(iter(values))) == values
 187
 188
 189 def test_iterator_rvp():
 190     """#388: Can't make iterators via make_iterator() with different r/v policies """
 191     import pybind11_tests.sequences_and_iterators as m
 192
 193     assert list(m.make_iterator_1()) == [1, 2, 3]
 194     assert list(m.make_iterator_2()) == [1, 2, 3]
 195     assert not isinstance(m.make_iterator_1(), type(m.make_iterator_2()))