Test py::iterable/py::iterator representation in docstrings

This commit is contained in:
Sergei Izmailov 2020-06-05 21:29:16 +03:00 committed by Wenzel Jakob
parent 4f1531c454
commit ab323e04f3
2 changed files with 14 additions and 0 deletions

View File

@ -13,6 +13,10 @@
TEST_SUBMODULE(pytypes, m) {
// test_int
m.def("get_int", []{return py::int_(0);});
// test_iterator
m.def("get_iterator", []{return py::iterator();});
// test_iterable
m.def("get_iterable", []{return py::iterable();});
// test_list
m.def("get_list", []() {
py::list list;

View File

@ -5,9 +5,19 @@ import sys
from pybind11_tests import pytypes as m
from pybind11_tests import debug_enabled
def test_int(doc):
assert doc(m.get_int) == "get_int() -> int"
def test_iterator(doc):
assert doc(m.get_iterator) == "get_iterator() -> Iterator"
def test_iterable(doc):
assert doc(m.get_iterable) == "get_iterable() -> Iterable"
def test_list(capture, doc):
with capture:
lst = m.get_list()