123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466 |
- # -*- coding: utf-8 -*-
- import pytest
- import env # noqa: F401
- from pybind11_tests import class_ as m
- from pybind11_tests import UserType, ConstructorStats
- def test_repr():
- # In Python 3.3+, repr() accesses __qualname__
- assert "pybind11_type" in repr(type(UserType))
- assert "UserType" in repr(UserType)
- def test_instance(msg):
- with pytest.raises(TypeError) as excinfo:
- m.NoConstructor()
- assert msg(excinfo.value) == "m.class_.NoConstructor: No constructor defined!"
- instance = m.NoConstructor.new_instance()
- cstats = ConstructorStats.get(m.NoConstructor)
- assert cstats.alive() == 1
- del instance
- assert cstats.alive() == 0
- def test_type():
- assert m.check_type(1) == m.DerivedClass1
- with pytest.raises(RuntimeError) as execinfo:
- m.check_type(0)
- assert "pybind11::detail::get_type_info: unable to find type info" in str(
- execinfo.value
- )
- assert "Invalid" in str(execinfo.value)
- # Currently not supported
- # See https://github.com/pybind/pybind11/issues/2486
- # assert m.check_type(2) == int
- def test_type_of_py():
- assert m.get_type_of(1) == int
- assert m.get_type_of(m.DerivedClass1()) == m.DerivedClass1
- assert m.get_type_of(int) == type
- def test_type_of_classic():
- assert m.get_type_classic(1) == int
- assert m.get_type_classic(m.DerivedClass1()) == m.DerivedClass1
- assert m.get_type_classic(int) == type
- def test_type_of_py_nodelete():
- # If the above test deleted the class, this will segfault
- assert m.get_type_of(m.DerivedClass1()) == m.DerivedClass1
- def test_as_type_py():
- assert m.as_type(int) == int
- with pytest.raises(TypeError):
- assert m.as_type(1) == int
- with pytest.raises(TypeError):
- assert m.as_type(m.DerivedClass1()) == m.DerivedClass1
- def test_docstrings(doc):
- assert doc(UserType) == "A `py::class_` type for testing"
- assert UserType.__name__ == "UserType"
- assert UserType.__module__ == "pybind11_tests"
- assert UserType.get_value.__name__ == "get_value"
- assert UserType.get_value.__module__ == "pybind11_tests"
- assert (
- doc(UserType.get_value)
- == """
- get_value(self: m.UserType) -> int
- Get value using a method
- """
- )
- assert doc(UserType.value) == "Get/set value using a property"
- assert (
- doc(m.NoConstructor.new_instance)
- == """
- new_instance() -> m.class_.NoConstructor
- Return an instance
- """
- )
- def test_qualname(doc):
- """Tests that a properly qualified name is set in __qualname__ (even in pre-3.3, where we
- backport the attribute) and that generated docstrings properly use it and the module name"""
- assert m.NestBase.__qualname__ == "NestBase"
- assert m.NestBase.Nested.__qualname__ == "NestBase.Nested"
- assert (
- doc(m.NestBase.__init__)
- == """
- __init__(self: m.class_.NestBase) -> None
- """
- )
- assert (
- doc(m.NestBase.g)
- == """
- g(self: m.class_.NestBase, arg0: m.class_.NestBase.Nested) -> None
- """
- )
- assert (
- doc(m.NestBase.Nested.__init__)
- == """
- __init__(self: m.class_.NestBase.Nested) -> None
- """
- )
- assert (
- doc(m.NestBase.Nested.fn)
- == """
- fn(self: m.class_.NestBase.Nested, arg0: int, arg1: m.class_.NestBase, arg2: m.class_.NestBase.Nested) -> None
- """ # noqa: E501 line too long
- )
- assert (
- doc(m.NestBase.Nested.fa)
- == """
- fa(self: m.class_.NestBase.Nested, a: int, b: m.class_.NestBase, c: m.class_.NestBase.Nested) -> None
- """ # noqa: E501 line too long
- )
- assert m.NestBase.__module__ == "pybind11_tests.class_"
- assert m.NestBase.Nested.__module__ == "pybind11_tests.class_"
- def test_inheritance(msg):
- roger = m.Rabbit("Rabbit")
- assert roger.name() + " is a " + roger.species() == "Rabbit is a parrot"
- assert m.pet_name_species(roger) == "Rabbit is a parrot"
- polly = m.Pet("Polly", "parrot")
- assert polly.name() + " is a " + polly.species() == "Polly is a parrot"
- assert m.pet_name_species(polly) == "Polly is a parrot"
- molly = m.Dog("Molly")
- assert molly.name() + " is a " + molly.species() == "Molly is a dog"
- assert m.pet_name_species(molly) == "Molly is a dog"
- fred = m.Hamster("Fred")
- assert fred.name() + " is a " + fred.species() == "Fred is a rodent"
- assert m.dog_bark(molly) == "Woof!"
- with pytest.raises(TypeError) as excinfo:
- m.dog_bark(polly)
- assert (
- msg(excinfo.value)
- == """
- dog_bark(): incompatible function arguments. The following argument types are supported:
- 1. (arg0: m.class_.Dog) -> str
- Invoked with: <m.class_.Pet object at 0>
- """
- )
- with pytest.raises(TypeError) as excinfo:
- m.Chimera("lion", "goat")
- assert "No constructor defined!" in str(excinfo.value)
- def test_inheritance_init(msg):
- # Single base
- class Python(m.Pet):
- def __init__(self):
- pass
- with pytest.raises(TypeError) as exc_info:
- Python()
- expected = "m.class_.Pet.__init__() must be called when overriding __init__"
- assert msg(exc_info.value) == expected
- # Multiple bases
- class RabbitHamster(m.Rabbit, m.Hamster):
- def __init__(self):
- m.Rabbit.__init__(self, "RabbitHamster")
- with pytest.raises(TypeError) as exc_info:
- RabbitHamster()
- expected = "m.class_.Hamster.__init__() must be called when overriding __init__"
- assert msg(exc_info.value) == expected
- def test_automatic_upcasting():
- assert type(m.return_class_1()).__name__ == "DerivedClass1"
- assert type(m.return_class_2()).__name__ == "DerivedClass2"
- assert type(m.return_none()).__name__ == "NoneType"
- # Repeat these a few times in a random order to ensure no invalid caching is applied
- assert type(m.return_class_n(1)).__name__ == "DerivedClass1"
- assert type(m.return_class_n(2)).__name__ == "DerivedClass2"
- assert type(m.return_class_n(0)).__name__ == "BaseClass"
- assert type(m.return_class_n(2)).__name__ == "DerivedClass2"
- assert type(m.return_class_n(2)).__name__ == "DerivedClass2"
- assert type(m.return_class_n(0)).__name__ == "BaseClass"
- assert type(m.return_class_n(1)).__name__ == "DerivedClass1"
- def test_isinstance():
- objects = [tuple(), dict(), m.Pet("Polly", "parrot")] + [m.Dog("Molly")] * 4
- expected = (True, True, True, True, True, False, False)
- assert m.check_instances(objects) == expected
- def test_mismatched_holder():
- import re
- with pytest.raises(RuntimeError) as excinfo:
- m.mismatched_holder_1()
- assert re.match(
- 'generic_type: type ".*MismatchDerived1" does not have a non-default '
- 'holder type while its base ".*MismatchBase1" does',
- str(excinfo.value),
- )
- with pytest.raises(RuntimeError) as excinfo:
- m.mismatched_holder_2()
- assert re.match(
- 'generic_type: type ".*MismatchDerived2" has a non-default holder type '
- 'while its base ".*MismatchBase2" does not',
- str(excinfo.value),
- )
- def test_override_static():
- """#511: problem with inheritance + overwritten def_static"""
- b = m.MyBase.make()
- d1 = m.MyDerived.make2()
- d2 = m.MyDerived.make()
- assert isinstance(b, m.MyBase)
- assert isinstance(d1, m.MyDerived)
- assert isinstance(d2, m.MyDerived)
- def test_implicit_conversion_life_support():
- """Ensure the lifetime of temporary objects created for implicit conversions"""
- assert m.implicitly_convert_argument(UserType(5)) == 5
- assert m.implicitly_convert_variable(UserType(5)) == 5
- assert "outside a bound function" in m.implicitly_convert_variable_fail(UserType(5))
- def test_operator_new_delete(capture):
- """Tests that class-specific operator new/delete functions are invoked"""
- class SubAliased(m.AliasedHasOpNewDelSize):
- pass
- with capture:
- a = m.HasOpNewDel()
- b = m.HasOpNewDelSize()
- d = m.HasOpNewDelBoth()
- assert (
- capture
- == """
- A new 8
- B new 4
- D new 32
- """
- )
- sz_alias = str(m.AliasedHasOpNewDelSize.size_alias)
- sz_noalias = str(m.AliasedHasOpNewDelSize.size_noalias)
- with capture:
- c = m.AliasedHasOpNewDelSize()
- c2 = SubAliased()
- assert capture == ("C new " + sz_noalias + "\n" + "C new " + sz_alias + "\n")
- with capture:
- del a
- pytest.gc_collect()
- del b
- pytest.gc_collect()
- del d
- pytest.gc_collect()
- assert (
- capture
- == """
- A delete
- B delete 4
- D delete
- """
- )
- with capture:
- del c
- pytest.gc_collect()
- del c2
- pytest.gc_collect()
- assert capture == ("C delete " + sz_noalias + "\n" + "C delete " + sz_alias + "\n")
- def test_bind_protected_functions():
- """Expose protected member functions to Python using a helper class"""
- a = m.ProtectedA()
- assert a.foo() == 42
- b = m.ProtectedB()
- assert b.foo() == 42
- class C(m.ProtectedB):
- def __init__(self):
- m.ProtectedB.__init__(self)
- def foo(self):
- return 0
- c = C()
- assert c.foo() == 0
- def test_brace_initialization():
- """ Tests that simple POD classes can be constructed using C++11 brace initialization """
- a = m.BraceInitialization(123, "test")
- assert a.field1 == 123
- assert a.field2 == "test"
- # Tests that a non-simple class doesn't get brace initialization (if the
- # class defines an initializer_list constructor, in particular, it would
- # win over the expected constructor).
- b = m.NoBraceInitialization([123, 456])
- assert b.vec == [123, 456]
- @pytest.mark.xfail("env.PYPY")
- def test_class_refcount():
- """Instances must correctly increase/decrease the reference count of their types (#1029)"""
- from sys import getrefcount
- class PyDog(m.Dog):
- pass
- for cls in m.Dog, PyDog:
- refcount_1 = getrefcount(cls)
- molly = [cls("Molly") for _ in range(10)]
- refcount_2 = getrefcount(cls)
- del molly
- pytest.gc_collect()
- refcount_3 = getrefcount(cls)
- assert refcount_1 == refcount_3
- assert refcount_2 > refcount_1
- def test_reentrant_implicit_conversion_failure(msg):
- # ensure that there is no runaway reentrant implicit conversion (#1035)
- with pytest.raises(TypeError) as excinfo:
- m.BogusImplicitConversion(0)
- assert (
- msg(excinfo.value)
- == """
- __init__(): incompatible constructor arguments. The following argument types are supported:
- 1. m.class_.BogusImplicitConversion(arg0: m.class_.BogusImplicitConversion)
- Invoked with: 0
- """
- )
- def test_error_after_conversions():
- with pytest.raises(TypeError) as exc_info:
- m.test_error_after_conversions("hello")
- assert str(exc_info.value).startswith(
- "Unable to convert function return value to a Python type!"
- )
- def test_aligned():
- if hasattr(m, "Aligned"):
- p = m.Aligned().ptr()
- assert p % 1024 == 0
- # https://foss.heptapod.net/pypy/pypy/-/issues/2742
- @pytest.mark.xfail("env.PYPY")
- def test_final():
- with pytest.raises(TypeError) as exc_info:
- class PyFinalChild(m.IsFinal):
- pass
- assert str(exc_info.value).endswith("is not an acceptable base type")
- # https://foss.heptapod.net/pypy/pypy/-/issues/2742
- @pytest.mark.xfail("env.PYPY")
- def test_non_final_final():
- with pytest.raises(TypeError) as exc_info:
- class PyNonFinalFinalChild(m.IsNonFinalFinal):
- pass
- assert str(exc_info.value).endswith("is not an acceptable base type")
- # https://github.com/pybind/pybind11/issues/1878
- def test_exception_rvalue_abort():
- with pytest.raises(RuntimeError):
- m.PyPrintDestructor().throw_something()
- # https://github.com/pybind/pybind11/issues/1568
- def test_multiple_instances_with_same_pointer(capture):
- n = 100
- instances = [m.SamePointer() for _ in range(n)]
- for i in range(n):
- # We need to reuse the same allocated memory for with a different type,
- # to ensure the bug in `deregister_instance_impl` is detected. Otherwise
- # `Py_TYPE(self) == Py_TYPE(it->second)` will still succeed, even though
- # the `instance` is already deleted.
- instances[i] = m.Empty()
- # No assert: if this does not trigger the error
- # pybind11_fail("pybind11_object_dealloc(): Tried to deallocate unregistered instance!");
- # and just completes without crashing, we're good.
- # https://github.com/pybind/pybind11/issues/1624
- def test_base_and_derived_nested_scope():
- assert issubclass(m.DerivedWithNested, m.BaseWithNested)
- assert m.BaseWithNested.Nested != m.DerivedWithNested.Nested
- assert m.BaseWithNested.Nested.get_name() == "BaseWithNested::Nested"
- assert m.DerivedWithNested.Nested.get_name() == "DerivedWithNested::Nested"
- def test_register_duplicate_class():
- import types
- module_scope = types.ModuleType("module_scope")
- with pytest.raises(RuntimeError) as exc_info:
- m.register_duplicate_class_name(module_scope)
- expected = (
- 'generic_type: cannot initialize type "Duplicate": '
- "an object with that name is already defined"
- )
- assert str(exc_info.value) == expected
- with pytest.raises(RuntimeError) as exc_info:
- m.register_duplicate_class_type(module_scope)
- expected = 'generic_type: type "YetAnotherDuplicate" is already registered!'
- assert str(exc_info.value) == expected
- class ClassScope:
- pass
- with pytest.raises(RuntimeError) as exc_info:
- m.register_duplicate_nested_class_name(ClassScope)
- expected = (
- 'generic_type: cannot initialize type "DuplicateNested": '
- "an object with that name is already defined"
- )
- assert str(exc_info.value) == expected
- with pytest.raises(RuntimeError) as exc_info:
- m.register_duplicate_nested_class_type(ClassScope)
- expected = 'generic_type: type "YetAnotherDuplicateNested" is already registered!'
- assert str(exc_info.value) == expected
|