diff options
Diffstat (limited to 'ext/pybind11/tests/test_docstring_options.py')
-rw-r--r-- | ext/pybind11/tests/test_docstring_options.py | 34 |
1 files changed, 15 insertions, 19 deletions
diff --git a/ext/pybind11/tests/test_docstring_options.py b/ext/pybind11/tests/test_docstring_options.py index 5e40f6868..0dbca609e 100644 --- a/ext/pybind11/tests/test_docstring_options.py +++ b/ext/pybind11/tests/test_docstring_options.py @@ -1,42 +1,38 @@ +from pybind11_tests import docstring_options as m def test_docstring_options(): - from pybind11_tests import (test_function1, test_function2, test_function3, - test_function4, test_function5, test_function6, - test_function7, DocstringTestFoo, - test_overloaded1, test_overloaded2, test_overloaded3) - # options.disable_function_signatures() - assert not test_function1.__doc__ + assert not m.test_function1.__doc__ - assert test_function2.__doc__ == "A custom docstring" + assert m.test_function2.__doc__ == "A custom docstring" # docstring specified on just the first overload definition: - assert test_overloaded1.__doc__ == "Overload docstring" + assert m.test_overloaded1.__doc__ == "Overload docstring" # docstring on both overloads: - assert test_overloaded2.__doc__ == "overload docstring 1\noverload docstring 2" + assert m.test_overloaded2.__doc__ == "overload docstring 1\noverload docstring 2" # docstring on only second overload: - assert test_overloaded3.__doc__ == "Overload docstr" + assert m.test_overloaded3.__doc__ == "Overload docstr" # options.enable_function_signatures() - assert test_function3.__doc__ .startswith("test_function3(a: int, b: int) -> None") + assert m.test_function3.__doc__ .startswith("test_function3(a: int, b: int) -> None") - assert test_function4.__doc__ .startswith("test_function4(a: int, b: int) -> None") - assert test_function4.__doc__ .endswith("A custom docstring\n") + assert m.test_function4.__doc__ .startswith("test_function4(a: int, b: int) -> None") + assert m.test_function4.__doc__ .endswith("A custom docstring\n") # options.disable_function_signatures() # options.disable_user_defined_docstrings() - assert not test_function5.__doc__ + assert not m.test_function5.__doc__ # nested options.enable_user_defined_docstrings() - assert test_function6.__doc__ == "A custom docstring" + assert m.test_function6.__doc__ == "A custom docstring" # RAII destructor - assert test_function7.__doc__ .startswith("test_function7(a: int, b: int) -> None") - assert test_function7.__doc__ .endswith("A custom docstring\n") + assert m.test_function7.__doc__ .startswith("test_function7(a: int, b: int) -> None") + assert m.test_function7.__doc__ .endswith("A custom docstring\n") # Suppression of user-defined docstrings for non-function objects - assert not DocstringTestFoo.__doc__ - assert not DocstringTestFoo.value_prop.__doc__ + assert not m.DocstringTestFoo.__doc__ + assert not m.DocstringTestFoo.value_prop.__doc__ |