| 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 | |
| # Requires PyPy 6+ | |
| 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 |