mirror of
https://github.com/pybind/pybind11.git
synced 2024-11-14 17:43:53 +00:00
748ae2270b
* Add missing error handling to module_::def_submodule * Add test_def_submodule_failures * PyPy only: Skip test with trigger for PyModule_GetName() failure. * Reapply minor fix that accidentally got lost in transfer from PR #3964
122 lines
4.1 KiB
Python
122 lines
4.1 KiB
Python
import pytest
|
|
|
|
import env
|
|
from pybind11_tests import ConstructorStats
|
|
from pybind11_tests import modules as m
|
|
from pybind11_tests.modules import subsubmodule as ms
|
|
|
|
|
|
def test_nested_modules():
|
|
import pybind11_tests
|
|
|
|
assert pybind11_tests.__name__ == "pybind11_tests"
|
|
assert pybind11_tests.modules.__name__ == "pybind11_tests.modules"
|
|
assert (
|
|
pybind11_tests.modules.subsubmodule.__name__
|
|
== "pybind11_tests.modules.subsubmodule"
|
|
)
|
|
assert m.__name__ == "pybind11_tests.modules"
|
|
assert ms.__name__ == "pybind11_tests.modules.subsubmodule"
|
|
|
|
assert ms.submodule_func() == "submodule_func()"
|
|
|
|
|
|
def test_reference_internal():
|
|
b = ms.B()
|
|
assert str(b.get_a1()) == "A[1]"
|
|
assert str(b.a1) == "A[1]"
|
|
assert str(b.get_a2()) == "A[2]"
|
|
assert str(b.a2) == "A[2]"
|
|
|
|
b.a1 = ms.A(42)
|
|
b.a2 = ms.A(43)
|
|
assert str(b.get_a1()) == "A[42]"
|
|
assert str(b.a1) == "A[42]"
|
|
assert str(b.get_a2()) == "A[43]"
|
|
assert str(b.a2) == "A[43]"
|
|
|
|
astats, bstats = ConstructorStats.get(ms.A), ConstructorStats.get(ms.B)
|
|
assert astats.alive() == 2
|
|
assert bstats.alive() == 1
|
|
del b
|
|
assert astats.alive() == 0
|
|
assert bstats.alive() == 0
|
|
assert astats.values() == ["1", "2", "42", "43"]
|
|
assert bstats.values() == []
|
|
assert astats.default_constructions == 0
|
|
assert bstats.default_constructions == 1
|
|
assert astats.copy_constructions == 0
|
|
assert bstats.copy_constructions == 0
|
|
# assert astats.move_constructions >= 0 # Don't invoke any
|
|
# assert bstats.move_constructions >= 0 # Don't invoke any
|
|
assert astats.copy_assignments == 2
|
|
assert bstats.copy_assignments == 0
|
|
assert astats.move_assignments == 0
|
|
assert bstats.move_assignments == 0
|
|
|
|
|
|
def test_importing():
|
|
from collections import OrderedDict
|
|
|
|
from pybind11_tests.modules import OD
|
|
|
|
assert OD is OrderedDict
|
|
assert str(OD([(1, "a"), (2, "b")])) == "OrderedDict([(1, 'a'), (2, 'b')])"
|
|
|
|
|
|
def test_pydoc():
|
|
"""Pydoc needs to be able to provide help() for everything inside a pybind11 module"""
|
|
import pydoc
|
|
|
|
import pybind11_tests
|
|
|
|
assert pybind11_tests.__name__ == "pybind11_tests"
|
|
assert pybind11_tests.__doc__ == "pybind11 test module"
|
|
assert pydoc.text.docmodule(pybind11_tests)
|
|
|
|
|
|
def test_duplicate_registration():
|
|
"""Registering two things with the same name"""
|
|
|
|
assert m.duplicate_registration() == []
|
|
|
|
|
|
def test_builtin_key_type():
|
|
"""Test that all the keys in the builtin modules have type str.
|
|
|
|
Previous versions of pybind11 would add a unicode key in python 2.
|
|
"""
|
|
if hasattr(__builtins__, "keys"):
|
|
keys = __builtins__.keys()
|
|
else: # this is to make pypy happy since builtins is different there.
|
|
keys = __builtins__.__dict__.keys()
|
|
|
|
assert {type(k) for k in keys} == {str}
|
|
|
|
|
|
@pytest.mark.xfail("env.PYPY", reason="PyModule_GetName()")
|
|
def test_def_submodule_failures():
|
|
sm = m.def_submodule(m, b"ScratchSubModuleName") # Using bytes to show it works.
|
|
assert sm.__name__ == m.__name__ + "." + "ScratchSubModuleName"
|
|
malformed_utf8 = b"\x80"
|
|
if env.PYPY:
|
|
# It is not worth the effort finding a trigger for a failure when running with PyPy.
|
|
pytest.skip("Sufficiently exercised on platforms other than PyPy.")
|
|
else:
|
|
# Meant to trigger PyModule_GetName() failure:
|
|
sm_name_orig = sm.__name__
|
|
sm.__name__ = malformed_utf8
|
|
try:
|
|
with pytest.raises(Exception):
|
|
# Seen with Python 3.9: SystemError: nameless module
|
|
# But we do not want to exercise the internals of PyModule_GetName(), which could
|
|
# change in future versions of Python, but a bad __name__ is very likely to cause
|
|
# some kind of failure indefinitely.
|
|
m.def_submodule(sm, b"SubSubModuleName")
|
|
finally:
|
|
# Clean up to ensure nothing gets upset by a module with an invalid __name__.
|
|
sm.__name__ = sm_name_orig # Purely precautionary.
|
|
# Meant to trigger PyImport_AddModule() failure:
|
|
with pytest.raises(UnicodeDecodeError):
|
|
m.def_submodule(sm, malformed_utf8)
|