mirror of https://github.com/pybind/pybind11.git
99 lines
2.6 KiB
Python
99 lines
2.6 KiB
Python
import pytest
|
|
|
|
from pybind11_tests import class_sh_void_ptr_capsule as m
|
|
|
|
|
|
class Valid:
|
|
def __init__(self):
|
|
self.capsule_generated = False
|
|
|
|
def as_pybind11_tests_class_sh_void_ptr_capsule_Valid(self): # noqa: N802
|
|
self.capsule_generated = True
|
|
return m.create_test_void_ptr_capsule()
|
|
|
|
|
|
class NoConversion:
|
|
def __init__(self):
|
|
self.capsule_generated = False
|
|
|
|
|
|
class NoCapsuleReturned:
|
|
def __init__(self):
|
|
self.capsule_generated = False
|
|
|
|
def as_pybind11_tests_class_sh_void_ptr_capsule_NoCapsuleReturned( # noqa: N802
|
|
self,
|
|
):
|
|
pass
|
|
|
|
|
|
class AsAnotherObject:
|
|
def __init__(self):
|
|
self.capsule_generated = False
|
|
|
|
def as_pybind11_tests_class_sh_void_ptr_capsule_Valid(self): # noqa: N802
|
|
self.capsule_generated = True
|
|
return m.create_test_void_ptr_capsule()
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"ctor, caller, expected, capsule_generated",
|
|
[
|
|
(Valid, m.get_from_valid_capsule, 1, True),
|
|
(AsAnotherObject, m.get_from_valid_capsule, 1, True),
|
|
],
|
|
)
|
|
def test_valid_as_void_ptr_capsule_function(ctor, caller, expected, capsule_generated):
|
|
obj = ctor()
|
|
assert caller(obj) == expected
|
|
assert obj.capsule_generated == capsule_generated
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"ctor, caller, expected, capsule_generated",
|
|
[
|
|
(NoConversion, m.get_from_no_conversion_capsule, 2, False),
|
|
(NoCapsuleReturned, m.get_from_no_capsule_returned, 3, False),
|
|
],
|
|
)
|
|
def test_invalid_as_void_ptr_capsule_function(
|
|
ctor, caller, expected, capsule_generated
|
|
):
|
|
obj = ctor()
|
|
with pytest.raises(TypeError):
|
|
caller(obj)
|
|
assert obj.capsule_generated == capsule_generated
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"ctor, caller, pointer_type, capsule_generated",
|
|
[
|
|
(AsAnotherObject, m.get_from_shared_ptr_valid_capsule, "shared_ptr", True),
|
|
(AsAnotherObject, m.get_from_unique_ptr_valid_capsule, "unique_ptr", True),
|
|
],
|
|
)
|
|
def test_as_void_ptr_capsule_unsupported(ctor, caller, pointer_type, capsule_generated):
|
|
obj = ctor()
|
|
with pytest.raises(RuntimeError) as excinfo:
|
|
caller(obj)
|
|
assert pointer_type in str(excinfo.value)
|
|
assert obj.capsule_generated == capsule_generated
|
|
|
|
|
|
def test_type_with_getattr():
|
|
obj = m.TypeWithGetattr()
|
|
assert obj.get_42() == 42
|
|
assert obj.something == "GetAttr: something"
|
|
|
|
|
|
def test_multiple_inheritance_getattr():
|
|
d1 = m.Derived1()
|
|
assert d1.foo() == 0
|
|
assert d1.bar() == 1
|
|
assert d1.prop1 == "Base GetAttr: prop1"
|
|
|
|
d2 = m.Derived2()
|
|
assert d2.foo() == 0
|
|
assert d2.bar() == 2
|
|
assert d2.prop2 == "Base GetAttr: prop2"
|