mirror of https://github.com/pybind/pybind11
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
500 lines
12 KiB
500 lines
12 KiB
from __future__ import annotations |
|
|
|
import pytest |
|
|
|
import env |
|
from pybind11_tests import ConstructorStats |
|
from pybind11_tests import multiple_inheritance as m |
|
|
|
|
|
def test_multiple_inheritance_cpp(): |
|
mt = m.MIType(3, 4) |
|
|
|
assert mt.foo() == 3 |
|
assert mt.bar() == 4 |
|
|
|
|
|
@pytest.mark.xfail("env.PYPY") |
|
def test_multiple_inheritance_mix1(): |
|
class Base1: |
|
def __init__(self, i): |
|
self.i = i |
|
|
|
def foo(self): |
|
return self.i |
|
|
|
class MITypePy(Base1, m.Base2): |
|
def __init__(self, i, j): |
|
Base1.__init__(self, i) |
|
m.Base2.__init__(self, j) |
|
|
|
mt = MITypePy(3, 4) |
|
|
|
assert mt.foo() == 3 |
|
assert mt.bar() == 4 |
|
|
|
|
|
def test_multiple_inheritance_mix2(): |
|
class Base2: |
|
def __init__(self, i): |
|
self.i = i |
|
|
|
def bar(self): |
|
return self.i |
|
|
|
class MITypePy(m.Base1, Base2): |
|
def __init__(self, i, j): |
|
m.Base1.__init__(self, i) |
|
Base2.__init__(self, j) |
|
|
|
mt = MITypePy(3, 4) |
|
|
|
assert mt.foo() == 3 |
|
assert mt.bar() == 4 |
|
|
|
|
|
@pytest.mark.xfail("env.PYPY") |
|
def test_multiple_inheritance_python(): |
|
class MI1(m.Base1, m.Base2): |
|
def __init__(self, i, j): |
|
m.Base1.__init__(self, i) |
|
m.Base2.__init__(self, j) |
|
|
|
class B1: |
|
def v(self): |
|
return 1 |
|
|
|
class MI2(B1, m.Base1, m.Base2): |
|
def __init__(self, i, j): |
|
B1.__init__(self) |
|
m.Base1.__init__(self, i) |
|
m.Base2.__init__(self, j) |
|
|
|
class MI3(MI2): |
|
def __init__(self, i, j): |
|
MI2.__init__(self, i, j) |
|
|
|
class MI4(MI3, m.Base2): |
|
def __init__(self, i, j): |
|
MI3.__init__(self, i, j) |
|
# This should be ignored (Base2 is already initialized via MI2): |
|
m.Base2.__init__(self, i + 100) |
|
|
|
class MI5(m.Base2, B1, m.Base1): |
|
def __init__(self, i, j): |
|
B1.__init__(self) |
|
m.Base1.__init__(self, i) |
|
m.Base2.__init__(self, j) |
|
|
|
class MI6(m.Base2, B1): |
|
def __init__(self, i): |
|
m.Base2.__init__(self, i) |
|
B1.__init__(self) |
|
|
|
class B2(B1): |
|
def v(self): |
|
return 2 |
|
|
|
class B3: |
|
def v(self): |
|
return 3 |
|
|
|
class B4(B3, B2): |
|
def v(self): |
|
return 4 |
|
|
|
class MI7(B4, MI6): |
|
def __init__(self, i): |
|
B4.__init__(self) |
|
MI6.__init__(self, i) |
|
|
|
class MI8(MI6, B3): |
|
def __init__(self, i): |
|
MI6.__init__(self, i) |
|
B3.__init__(self) |
|
|
|
class MI8b(B3, MI6): |
|
def __init__(self, i): |
|
B3.__init__(self) |
|
MI6.__init__(self, i) |
|
|
|
mi1 = MI1(1, 2) |
|
assert mi1.foo() == 1 |
|
assert mi1.bar() == 2 |
|
|
|
mi2 = MI2(3, 4) |
|
assert mi2.v() == 1 |
|
assert mi2.foo() == 3 |
|
assert mi2.bar() == 4 |
|
|
|
mi3 = MI3(5, 6) |
|
assert mi3.v() == 1 |
|
assert mi3.foo() == 5 |
|
assert mi3.bar() == 6 |
|
|
|
mi4 = MI4(7, 8) |
|
assert mi4.v() == 1 |
|
assert mi4.foo() == 7 |
|
assert mi4.bar() == 8 |
|
|
|
mi5 = MI5(10, 11) |
|
assert mi5.v() == 1 |
|
assert mi5.foo() == 10 |
|
assert mi5.bar() == 11 |
|
|
|
mi6 = MI6(12) |
|
assert mi6.v() == 1 |
|
assert mi6.bar() == 12 |
|
|
|
mi7 = MI7(13) |
|
assert mi7.v() == 4 |
|
assert mi7.bar() == 13 |
|
|
|
mi8 = MI8(14) |
|
assert mi8.v() == 1 |
|
assert mi8.bar() == 14 |
|
|
|
mi8b = MI8b(15) |
|
assert mi8b.v() == 3 |
|
assert mi8b.bar() == 15 |
|
|
|
|
|
def test_multiple_inheritance_python_many_bases(): |
|
class MIMany14(m.BaseN1, m.BaseN2, m.BaseN3, m.BaseN4): |
|
def __init__(self): |
|
m.BaseN1.__init__(self, 1) |
|
m.BaseN2.__init__(self, 2) |
|
m.BaseN3.__init__(self, 3) |
|
m.BaseN4.__init__(self, 4) |
|
|
|
class MIMany58(m.BaseN5, m.BaseN6, m.BaseN7, m.BaseN8): |
|
def __init__(self): |
|
m.BaseN5.__init__(self, 5) |
|
m.BaseN6.__init__(self, 6) |
|
m.BaseN7.__init__(self, 7) |
|
m.BaseN8.__init__(self, 8) |
|
|
|
class MIMany916( |
|
m.BaseN9, |
|
m.BaseN10, |
|
m.BaseN11, |
|
m.BaseN12, |
|
m.BaseN13, |
|
m.BaseN14, |
|
m.BaseN15, |
|
m.BaseN16, |
|
): |
|
def __init__(self): |
|
m.BaseN9.__init__(self, 9) |
|
m.BaseN10.__init__(self, 10) |
|
m.BaseN11.__init__(self, 11) |
|
m.BaseN12.__init__(self, 12) |
|
m.BaseN13.__init__(self, 13) |
|
m.BaseN14.__init__(self, 14) |
|
m.BaseN15.__init__(self, 15) |
|
m.BaseN16.__init__(self, 16) |
|
|
|
class MIMany19(MIMany14, MIMany58, m.BaseN9): |
|
def __init__(self): |
|
MIMany14.__init__(self) |
|
MIMany58.__init__(self) |
|
m.BaseN9.__init__(self, 9) |
|
|
|
class MIMany117(MIMany14, MIMany58, MIMany916, m.BaseN17): |
|
def __init__(self): |
|
MIMany14.__init__(self) |
|
MIMany58.__init__(self) |
|
MIMany916.__init__(self) |
|
m.BaseN17.__init__(self, 17) |
|
|
|
# Inherits from 4 registered C++ classes: can fit in one pointer on any modern arch: |
|
a = MIMany14() |
|
for i in range(1, 4): |
|
assert getattr(a, "f" + str(i))() == 2 * i |
|
|
|
# Inherits from 8: requires 1/2 pointers worth of holder flags on 32/64-bit arch: |
|
b = MIMany916() |
|
for i in range(9, 16): |
|
assert getattr(b, "f" + str(i))() == 2 * i |
|
|
|
# Inherits from 9: requires >= 2 pointers worth of holder flags |
|
c = MIMany19() |
|
for i in range(1, 9): |
|
assert getattr(c, "f" + str(i))() == 2 * i |
|
|
|
# Inherits from 17: requires >= 3 pointers worth of holder flags |
|
d = MIMany117() |
|
for i in range(1, 17): |
|
assert getattr(d, "f" + str(i))() == 2 * i |
|
|
|
|
|
def test_multiple_inheritance_virtbase(): |
|
class MITypePy(m.Base12a): |
|
def __init__(self, i, j): |
|
m.Base12a.__init__(self, i, j) |
|
|
|
mt = MITypePy(3, 4) |
|
assert mt.bar() == 4 |
|
assert m.bar_base2a(mt) == 4 |
|
assert m.bar_base2a_sharedptr(mt) == 4 |
|
|
|
|
|
def test_mi_static_properties(): |
|
"""Mixing bases with and without static properties should be possible |
|
and the result should be independent of base definition order""" |
|
|
|
for d in (m.VanillaStaticMix1(), m.VanillaStaticMix2()): |
|
assert d.vanilla() == "Vanilla" |
|
assert d.static_func1() == "WithStatic1" |
|
assert d.static_func2() == "WithStatic2" |
|
assert d.static_func() == d.__class__.__name__ |
|
|
|
m.WithStatic1.static_value1 = 1 |
|
m.WithStatic2.static_value2 = 2 |
|
assert d.static_value1 == 1 |
|
assert d.static_value2 == 2 |
|
assert d.static_value == 12 |
|
|
|
d.static_value1 = 0 |
|
assert d.static_value1 == 0 |
|
d.static_value2 = 0 |
|
assert d.static_value2 == 0 |
|
d.static_value = 0 |
|
assert d.static_value == 0 |
|
|
|
|
|
def test_mi_dynamic_attributes(): |
|
"""Mixing bases with and without dynamic attribute support""" |
|
|
|
for d in (m.VanillaDictMix1(), m.VanillaDictMix2()): |
|
d.dynamic = 1 |
|
assert d.dynamic == 1 |
|
|
|
|
|
def test_mi_unaligned_base(): |
|
"""Returning an offset (non-first MI) base class pointer should recognize the instance""" |
|
|
|
n_inst = ConstructorStats.detail_reg_inst() |
|
|
|
c = m.I801C() |
|
d = m.I801D() |
|
if not env.GRAALPY: |
|
# + 4 below because we have the two instances, and each instance has offset base I801B2 |
|
assert ConstructorStats.detail_reg_inst() == n_inst + 4 |
|
b1c = m.i801b1_c(c) |
|
assert b1c is c |
|
b2c = m.i801b2_c(c) |
|
assert b2c is c |
|
b1d = m.i801b1_d(d) |
|
assert b1d is d |
|
b2d = m.i801b2_d(d) |
|
assert b2d is d |
|
|
|
if env.GRAALPY: |
|
pytest.skip("ConstructorStats is incompatible with GraalPy.") |
|
|
|
assert ConstructorStats.detail_reg_inst() == n_inst + 4 # no extra instances |
|
del c, b1c, b2c |
|
assert ConstructorStats.detail_reg_inst() == n_inst + 2 |
|
del d, b1d, b2d |
|
assert ConstructorStats.detail_reg_inst() == n_inst |
|
|
|
|
|
def test_mi_base_return(): |
|
"""Tests returning an offset (non-first MI) base class pointer to a derived instance""" |
|
|
|
n_inst = ConstructorStats.detail_reg_inst() |
|
|
|
c1 = m.i801c_b1() |
|
assert type(c1) is m.I801C |
|
assert c1.a == 1 |
|
assert c1.b == 2 |
|
|
|
d1 = m.i801d_b1() |
|
assert type(d1) is m.I801D |
|
assert d1.a == 1 |
|
assert d1.b == 2 |
|
|
|
if not env.GRAALPY: |
|
assert ConstructorStats.detail_reg_inst() == n_inst + 4 |
|
|
|
c2 = m.i801c_b2() |
|
assert type(c2) is m.I801C |
|
assert c2.a == 1 |
|
assert c2.b == 2 |
|
|
|
d2 = m.i801d_b2() |
|
assert type(d2) is m.I801D |
|
assert d2.a == 1 |
|
assert d2.b == 2 |
|
|
|
if not env.GRAALPY: |
|
assert ConstructorStats.detail_reg_inst() == n_inst + 8 |
|
|
|
del c2 |
|
assert ConstructorStats.detail_reg_inst() == n_inst + 6 |
|
del c1, d1, d2 |
|
assert ConstructorStats.detail_reg_inst() == n_inst |
|
|
|
# Returning an unregistered derived type with a registered base; we won't |
|
# pick up the derived type, obviously, but should still work (as an object |
|
# of whatever type was returned). |
|
e1 = m.i801e_c() |
|
assert type(e1) is m.I801C |
|
assert e1.a == 1 |
|
assert e1.b == 2 |
|
|
|
e2 = m.i801e_b2() |
|
assert type(e2) is m.I801B2 |
|
assert e2.b == 2 |
|
|
|
|
|
def test_diamond_inheritance(): |
|
"""Tests that diamond inheritance works as expected (issue #959)""" |
|
|
|
# Issue #959: this shouldn't segfault: |
|
d = m.D() |
|
|
|
# Make sure all the various distinct pointers are all recognized as registered instances: |
|
assert d is d.c0() |
|
assert d is d.c1() |
|
assert d is d.b() |
|
assert d is d.c0().b() |
|
assert d is d.c1().b() |
|
assert d is d.c0().c1().b().c0().b() |
|
|
|
|
|
def test_pr3635_diamond_b(): |
|
o = m.MVB() |
|
assert o.b == 1 |
|
|
|
assert o.get_b_b() == 1 |
|
|
|
|
|
def test_pr3635_diamond_c(): |
|
o = m.MVC() |
|
assert o.b == 1 |
|
assert o.c == 2 |
|
|
|
assert o.get_b_b() == 1 |
|
assert o.get_c_b() == 1 |
|
|
|
assert o.get_c_c() == 2 |
|
|
|
|
|
def test_pr3635_diamond_d0(): |
|
o = m.MVD0() |
|
assert o.b == 1 |
|
assert o.c == 2 |
|
assert o.d0 == 3 |
|
|
|
assert o.get_b_b() == 1 |
|
assert o.get_c_b() == 1 |
|
assert o.get_d0_b() == 1 |
|
|
|
assert o.get_c_c() == 2 |
|
assert o.get_d0_c() == 2 |
|
|
|
assert o.get_d0_d0() == 3 |
|
|
|
|
|
def test_pr3635_diamond_d1(): |
|
o = m.MVD1() |
|
assert o.b == 1 |
|
assert o.c == 2 |
|
assert o.d1 == 4 |
|
|
|
assert o.get_b_b() == 1 |
|
assert o.get_c_b() == 1 |
|
assert o.get_d1_b() == 1 |
|
|
|
assert o.get_c_c() == 2 |
|
assert o.get_d1_c() == 2 |
|
|
|
assert o.get_d1_d1() == 4 |
|
|
|
|
|
def test_pr3635_diamond_e(): |
|
o = m.MVE() |
|
assert o.b == 1 |
|
assert o.c == 2 |
|
assert o.d0 == 3 |
|
assert o.d1 == 4 |
|
assert o.e == 5 |
|
|
|
assert o.get_b_b() == 1 |
|
assert o.get_c_b() == 1 |
|
assert o.get_d0_b() == 1 |
|
assert o.get_d1_b() == 1 |
|
assert o.get_e_b() == 1 |
|
|
|
assert o.get_c_c() == 2 |
|
assert o.get_d0_c() == 2 |
|
assert o.get_d1_c() == 2 |
|
assert o.get_e_c() == 2 |
|
|
|
assert o.get_d0_d0() == 3 |
|
assert o.get_e_d0() == 3 |
|
|
|
assert o.get_d1_d1() == 4 |
|
assert o.get_e_d1() == 4 |
|
|
|
assert o.get_e_e() == 5 |
|
|
|
|
|
def test_pr3635_diamond_f(): |
|
o = m.MVF() |
|
assert o.b == 1 |
|
assert o.c == 2 |
|
assert o.d0 == 3 |
|
assert o.d1 == 4 |
|
assert o.e == 5 |
|
assert o.f == 6 |
|
|
|
assert o.get_b_b() == 1 |
|
assert o.get_c_b() == 1 |
|
assert o.get_d0_b() == 1 |
|
assert o.get_d1_b() == 1 |
|
assert o.get_e_b() == 1 |
|
assert o.get_f_b() == 1 |
|
|
|
assert o.get_c_c() == 2 |
|
assert o.get_d0_c() == 2 |
|
assert o.get_d1_c() == 2 |
|
assert o.get_e_c() == 2 |
|
assert o.get_f_c() == 2 |
|
|
|
assert o.get_d0_d0() == 3 |
|
assert o.get_e_d0() == 3 |
|
assert o.get_f_d0() == 3 |
|
|
|
assert o.get_d1_d1() == 4 |
|
assert o.get_e_d1() == 4 |
|
assert o.get_f_d1() == 4 |
|
|
|
assert o.get_e_e() == 5 |
|
assert o.get_f_e() == 5 |
|
|
|
assert o.get_f_f() == 6 |
|
|
|
|
|
def test_python_inherit_from_mi(): |
|
"""Tests extending a Python class from a single inheritor of a MI class""" |
|
|
|
class PyMVF(m.MVF): |
|
g = 7 |
|
|
|
def get_g_g(self): |
|
return self.g |
|
|
|
o = PyMVF() |
|
|
|
assert o.b == 1 |
|
assert o.c == 2 |
|
assert o.d0 == 3 |
|
assert o.d1 == 4 |
|
assert o.e == 5 |
|
assert o.f == 6 |
|
assert o.g == 7 |
|
|
|
assert o.get_g_g() == 7
|
|
|