ext/pybind11/tests/test_multiple_inheritance.py - public/gem5 - Git at Google

 # -*- coding: utf-8 -*-
 import pytest

 import env  # noqa: F401

 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.skipif("env.PYPY and env.PY2")
 @pytest.mark.xfail("env.PYPY and not env.PY2")
 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.skipif("env.PYPY and env.PY2")
 @pytest.mark.xfail("env.PYPY and not env.PY2")
 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(object):
         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(object):
         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()
     # + 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

     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

     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

     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()
	# -- coding: utf-8 --
	import pytest

	import env # noqa: F401

	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.skipif("env.PYPY and env.PY2")
	@pytest.mark.xfail("env.PYPY and not env.PY2")
	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.skipif("env.PYPY and env.PY2")
	@pytest.mark.xfail("env.PYPY and not env.PY2")
	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(object):
	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(object):
	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()
	# + 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

	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

	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

	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()