test_sequences_and_iterators.py revision 12037
16691Stjones1@inf.ed.ac.ukimport pytest 26691Stjones1@inf.ed.ac.uk 36691Stjones1@inf.ed.ac.uk 46691Stjones1@inf.ed.ac.ukdef isclose(a, b, rel_tol=1e-05, abs_tol=0.0): 56691Stjones1@inf.ed.ac.uk """Like math.isclose() from Python 3.5""" 66691Stjones1@inf.ed.ac.uk return abs(a - b) <= max(rel_tol * max(abs(a), abs(b)), abs_tol) 76691Stjones1@inf.ed.ac.uk 86691Stjones1@inf.ed.ac.uk 96691Stjones1@inf.ed.ac.ukdef allclose(a_list, b_list, rel_tol=1e-05, abs_tol=0.0): 106691Stjones1@inf.ed.ac.uk return all(isclose(a, b, rel_tol=rel_tol, abs_tol=abs_tol) for a, b in zip(a_list, b_list)) 116691Stjones1@inf.ed.ac.uk 126691Stjones1@inf.ed.ac.uk 136691Stjones1@inf.ed.ac.ukdef test_generalized_iterators(): 146691Stjones1@inf.ed.ac.uk from pybind11_tests.sequences_and_iterators import IntPairs 156691Stjones1@inf.ed.ac.uk 166691Stjones1@inf.ed.ac.uk assert list(IntPairs([(1, 2), (3, 4), (0, 5)]).nonzero()) == [(1, 2), (3, 4)] 176691Stjones1@inf.ed.ac.uk assert list(IntPairs([(1, 2), (2, 0), (0, 3), (4, 5)]).nonzero()) == [(1, 2)] 186691Stjones1@inf.ed.ac.uk assert list(IntPairs([(0, 3), (1, 2), (3, 4)]).nonzero()) == [] 196691Stjones1@inf.ed.ac.uk 206691Stjones1@inf.ed.ac.uk assert list(IntPairs([(1, 2), (3, 4), (0, 5)]).nonzero_keys()) == [1, 3] 216691Stjones1@inf.ed.ac.uk assert list(IntPairs([(1, 2), (2, 0), (0, 3), (4, 5)]).nonzero_keys()) == [1] 226691Stjones1@inf.ed.ac.uk assert list(IntPairs([(0, 3), (1, 2), (3, 4)]).nonzero_keys()) == [] 236691Stjones1@inf.ed.ac.uk 246691Stjones1@inf.ed.ac.uk 256691Stjones1@inf.ed.ac.ukdef test_sequence(): 266691Stjones1@inf.ed.ac.uk from pybind11_tests import ConstructorStats 276691Stjones1@inf.ed.ac.uk from pybind11_tests.sequences_and_iterators import Sequence 286691Stjones1@inf.ed.ac.uk 296691Stjones1@inf.ed.ac.uk cstats = ConstructorStats.get(Sequence) 306691Stjones1@inf.ed.ac.uk 316691Stjones1@inf.ed.ac.uk s = Sequence(5) 326691Stjones1@inf.ed.ac.uk assert cstats.values() == ['of size', '5'] 336691Stjones1@inf.ed.ac.uk 346691Stjones1@inf.ed.ac.uk assert "Sequence" in repr(s) 356691Stjones1@inf.ed.ac.uk assert len(s) == 5 366691Stjones1@inf.ed.ac.uk assert s[0] == 0 and s[3] == 0 376691Stjones1@inf.ed.ac.uk assert 12.34 not in s 386691Stjones1@inf.ed.ac.uk s[0], s[3] = 12.34, 56.78 396691Stjones1@inf.ed.ac.uk assert 12.34 in s 406691Stjones1@inf.ed.ac.uk assert isclose(s[0], 12.34) and isclose(s[3], 56.78) 416691Stjones1@inf.ed.ac.uk 426691Stjones1@inf.ed.ac.uk rev = reversed(s) 436691Stjones1@inf.ed.ac.uk assert cstats.values() == ['of size', '5'] 446691Stjones1@inf.ed.ac.uk 456691Stjones1@inf.ed.ac.uk rev2 = s[::-1] 466691Stjones1@inf.ed.ac.uk assert cstats.values() == ['of size', '5'] 476691Stjones1@inf.ed.ac.uk 486691Stjones1@inf.ed.ac.uk expected = [0, 56.78, 0, 0, 12.34] 496691Stjones1@inf.ed.ac.uk assert allclose(rev, expected) 5012385Sgabeblack@google.com assert allclose(rev2, expected) 516691Stjones1@inf.ed.ac.uk assert rev == rev2 5212385Sgabeblack@google.com 536691Stjones1@inf.ed.ac.uk rev[0::2] = Sequence([2.0, 2.0, 2.0]) 546691Stjones1@inf.ed.ac.uk assert cstats.values() == ['of size', '3', 'from std::vector'] 556691Stjones1@inf.ed.ac.uk 5612616Sgabeblack@google.com assert allclose(rev, [2, 56.78, 2, 0, 2]) 5712616Sgabeblack@google.com 586691Stjones1@inf.ed.ac.uk assert cstats.alive() == 3 596691Stjones1@inf.ed.ac.uk del s 606691Stjones1@inf.ed.ac.uk assert cstats.alive() == 2 616691Stjones1@inf.ed.ac.uk del rev 626691Stjones1@inf.ed.ac.uk assert cstats.alive() == 1 636691Stjones1@inf.ed.ac.uk del rev2 646691Stjones1@inf.ed.ac.uk assert cstats.alive() == 0 656691Stjones1@inf.ed.ac.uk 666691Stjones1@inf.ed.ac.uk assert cstats.values() == [] 676691Stjones1@inf.ed.ac.uk assert cstats.default_constructions == 0 686691Stjones1@inf.ed.ac.uk assert cstats.copy_constructions == 0 696691Stjones1@inf.ed.ac.uk assert cstats.move_constructions >= 1 706691Stjones1@inf.ed.ac.uk assert cstats.copy_assignments == 0 7112385Sgabeblack@google.com assert cstats.move_assignments == 0 7212385Sgabeblack@google.com 736691Stjones1@inf.ed.ac.uk 746691Stjones1@inf.ed.ac.ukdef test_map_iterator(): 756691Stjones1@inf.ed.ac.uk from pybind11_tests.sequences_and_iterators import StringMap 7612616Sgabeblack@google.com 7712616Sgabeblack@google.com m = StringMap({'hi': 'bye', 'black': 'white'}) 786691Stjones1@inf.ed.ac.uk assert m['hi'] == 'bye' 796691Stjones1@inf.ed.ac.uk assert len(m) == 2 807811Ssteve.reinhardt@amd.com assert m['black'] == 'white' 816691Stjones1@inf.ed.ac.uk 826691Stjones1@inf.ed.ac.uk with pytest.raises(KeyError): 83 assert m['orange'] 84 m['orange'] = 'banana' 85 assert m['orange'] == 'banana' 86 87 expected = {'hi': 'bye', 'black': 'white', 'orange': 'banana'} 88 for k in m: 89 assert m[k] == expected[k] 90 for k, v in m.items(): 91 assert v == expected[k] 92 93 94def test_python_iterator_in_cpp(): 95 import pybind11_tests.sequences_and_iterators as m 96 97 t = (1, 2, 3) 98 assert m.object_to_list(t) == [1, 2, 3] 99 assert m.object_to_list(iter(t)) == [1, 2, 3] 100 assert m.iterator_to_list(iter(t)) == [1, 2, 3] 101 102 with pytest.raises(TypeError) as excinfo: 103 m.object_to_list(1) 104 assert "object is not iterable" in str(excinfo.value) 105 106 with pytest.raises(TypeError) as excinfo: 107 m.iterator_to_list(1) 108 assert "incompatible function arguments" in str(excinfo.value) 109 110 def bad_next_call(): 111 raise RuntimeError("py::iterator::advance() should propagate errors") 112 113 with pytest.raises(RuntimeError) as excinfo: 114 m.iterator_to_list(iter(bad_next_call, None)) 115 assert str(excinfo.value) == "py::iterator::advance() should propagate errors" 116 117 l = [1, None, 0, None] 118 assert m.count_none(l) == 2 119 assert m.find_none(l) is True 120 assert m.count_nonzeros({"a": 0, "b": 1, "c": 2}) == 2 121 122 r = range(5) 123 assert all(m.tuple_iterator(tuple(r))) 124 assert all(m.list_iterator(list(r))) 125 assert all(m.sequence_iterator(r)) 126