mirror of
https://github.com/pybind/pybind11.git
synced 2024-11-28 08:02:00 +00:00
56c1edb46d
This fixes #1251 (patient vector grows without bounds) for the 2.2.2 branch by checking that the vector doesn't already have the given patient. This is a little less elegant than the same fix for `master` (which changes the patients `vector` to an `unordered_set`), but that requires an internals layout change, which this approach avoids.
217 lines
6.5 KiB
Python
217 lines
6.5 KiB
Python
import pytest
|
|
from pybind11_tests import call_policies as m
|
|
from pybind11_tests import ConstructorStats, UserType
|
|
|
|
|
|
def test_keep_alive_argument(capture):
|
|
n_inst = ConstructorStats.detail_reg_inst()
|
|
with capture:
|
|
p = m.Parent()
|
|
assert capture == "Allocating parent."
|
|
with capture:
|
|
p.addChild(m.Child())
|
|
assert ConstructorStats.detail_reg_inst() == n_inst + 1
|
|
assert capture == """
|
|
Allocating child.
|
|
Releasing child.
|
|
"""
|
|
with capture:
|
|
del p
|
|
assert ConstructorStats.detail_reg_inst() == n_inst
|
|
assert capture == "Releasing parent."
|
|
|
|
with capture:
|
|
p = m.Parent()
|
|
assert capture == "Allocating parent."
|
|
with capture:
|
|
p.addChildKeepAlive(m.Child())
|
|
assert ConstructorStats.detail_reg_inst() == n_inst + 2
|
|
assert capture == "Allocating child."
|
|
with capture:
|
|
del p
|
|
assert ConstructorStats.detail_reg_inst() == n_inst
|
|
assert capture == """
|
|
Releasing parent.
|
|
Releasing child.
|
|
"""
|
|
|
|
|
|
def test_keep_alive_return_value(capture):
|
|
n_inst = ConstructorStats.detail_reg_inst()
|
|
with capture:
|
|
p = m.Parent()
|
|
assert capture == "Allocating parent."
|
|
with capture:
|
|
p.returnChild()
|
|
assert ConstructorStats.detail_reg_inst() == n_inst + 1
|
|
assert capture == """
|
|
Allocating child.
|
|
Releasing child.
|
|
"""
|
|
with capture:
|
|
del p
|
|
assert ConstructorStats.detail_reg_inst() == n_inst
|
|
assert capture == "Releasing parent."
|
|
|
|
with capture:
|
|
p = m.Parent()
|
|
assert capture == "Allocating parent."
|
|
with capture:
|
|
p.returnChildKeepAlive()
|
|
assert ConstructorStats.detail_reg_inst() == n_inst + 2
|
|
assert capture == "Allocating child."
|
|
with capture:
|
|
del p
|
|
assert ConstructorStats.detail_reg_inst() == n_inst
|
|
assert capture == """
|
|
Releasing parent.
|
|
Releasing child.
|
|
"""
|
|
|
|
|
|
def test_keep_alive_single():
|
|
"""Issue #1251 - patients are stored multiple times when given to the same nurse"""
|
|
|
|
nurse, p1, p2 = UserType(), UserType(), UserType()
|
|
b = m.refcount(nurse)
|
|
assert [m.refcount(nurse), m.refcount(p1), m.refcount(p2)] == [b, b, b]
|
|
m.add_patient(nurse, p1)
|
|
assert m.get_patients(nurse) == [p1, ]
|
|
assert [m.refcount(nurse), m.refcount(p1), m.refcount(p2)] == [b, b + 1, b]
|
|
m.add_patient(nurse, p1)
|
|
assert m.get_patients(nurse) == [p1, ]
|
|
assert [m.refcount(nurse), m.refcount(p1), m.refcount(p2)] == [b, b + 1, b]
|
|
m.add_patient(nurse, p1)
|
|
assert m.get_patients(nurse) == [p1, ]
|
|
assert [m.refcount(nurse), m.refcount(p1), m.refcount(p2)] == [b, b + 1, b]
|
|
m.add_patient(nurse, p2)
|
|
assert m.get_patients(nurse) == [p1, p2]
|
|
assert [m.refcount(nurse), m.refcount(p1), m.refcount(p2)] == [b, b + 1, b + 1]
|
|
m.add_patient(nurse, p2)
|
|
assert m.get_patients(nurse) == [p1, p2]
|
|
assert [m.refcount(nurse), m.refcount(p1), m.refcount(p2)] == [b, b + 1, b + 1]
|
|
m.add_patient(nurse, p2)
|
|
m.add_patient(nurse, p1)
|
|
assert m.get_patients(nurse) == [p1, p2]
|
|
assert [m.refcount(nurse), m.refcount(p1), m.refcount(p2)] == [b, b + 1, b + 1]
|
|
del nurse
|
|
assert [m.refcount(p1), m.refcount(p2)] == [b, b]
|
|
|
|
|
|
# https://bitbucket.org/pypy/pypy/issues/2447
|
|
@pytest.unsupported_on_pypy
|
|
def test_alive_gc(capture):
|
|
n_inst = ConstructorStats.detail_reg_inst()
|
|
p = m.ParentGC()
|
|
p.addChildKeepAlive(m.Child())
|
|
assert ConstructorStats.detail_reg_inst() == n_inst + 2
|
|
lst = [p]
|
|
lst.append(lst) # creates a circular reference
|
|
with capture:
|
|
del p, lst
|
|
assert ConstructorStats.detail_reg_inst() == n_inst
|
|
assert capture == """
|
|
Releasing parent.
|
|
Releasing child.
|
|
"""
|
|
|
|
|
|
def test_alive_gc_derived(capture):
|
|
class Derived(m.Parent):
|
|
pass
|
|
|
|
n_inst = ConstructorStats.detail_reg_inst()
|
|
p = Derived()
|
|
p.addChildKeepAlive(m.Child())
|
|
assert ConstructorStats.detail_reg_inst() == n_inst + 2
|
|
lst = [p]
|
|
lst.append(lst) # creates a circular reference
|
|
with capture:
|
|
del p, lst
|
|
assert ConstructorStats.detail_reg_inst() == n_inst
|
|
assert capture == """
|
|
Releasing parent.
|
|
Releasing child.
|
|
"""
|
|
|
|
|
|
def test_alive_gc_multi_derived(capture):
|
|
class Derived(m.Parent, m.Child):
|
|
def __init__(self):
|
|
m.Parent.__init__(self)
|
|
m.Child.__init__(self)
|
|
|
|
n_inst = ConstructorStats.detail_reg_inst()
|
|
p = Derived()
|
|
p.addChildKeepAlive(m.Child())
|
|
# +3 rather than +2 because Derived corresponds to two registered instances
|
|
assert ConstructorStats.detail_reg_inst() == n_inst + 3
|
|
lst = [p]
|
|
lst.append(lst) # creates a circular reference
|
|
with capture:
|
|
del p, lst
|
|
assert ConstructorStats.detail_reg_inst() == n_inst
|
|
assert capture == """
|
|
Releasing parent.
|
|
Releasing child.
|
|
Releasing child.
|
|
"""
|
|
|
|
|
|
def test_return_none(capture):
|
|
n_inst = ConstructorStats.detail_reg_inst()
|
|
with capture:
|
|
p = m.Parent()
|
|
assert capture == "Allocating parent."
|
|
with capture:
|
|
p.returnNullChildKeepAliveChild()
|
|
assert ConstructorStats.detail_reg_inst() == n_inst + 1
|
|
assert capture == ""
|
|
with capture:
|
|
del p
|
|
assert ConstructorStats.detail_reg_inst() == n_inst
|
|
assert capture == "Releasing parent."
|
|
|
|
with capture:
|
|
p = m.Parent()
|
|
assert capture == "Allocating parent."
|
|
with capture:
|
|
p.returnNullChildKeepAliveParent()
|
|
assert ConstructorStats.detail_reg_inst() == n_inst + 1
|
|
assert capture == ""
|
|
with capture:
|
|
del p
|
|
assert ConstructorStats.detail_reg_inst() == n_inst
|
|
assert capture == "Releasing parent."
|
|
|
|
|
|
def test_keep_alive_constructor(capture):
|
|
n_inst = ConstructorStats.detail_reg_inst()
|
|
|
|
with capture:
|
|
p = m.Parent(m.Child())
|
|
assert ConstructorStats.detail_reg_inst() == n_inst + 2
|
|
assert capture == """
|
|
Allocating child.
|
|
Allocating parent.
|
|
"""
|
|
with capture:
|
|
del p
|
|
assert ConstructorStats.detail_reg_inst() == n_inst
|
|
assert capture == """
|
|
Releasing parent.
|
|
Releasing child.
|
|
"""
|
|
|
|
|
|
def test_call_guard():
|
|
assert m.unguarded_call() == "unguarded"
|
|
assert m.guarded_call() == "guarded"
|
|
|
|
assert m.multiple_guards_correct_order() == "guarded & guarded"
|
|
assert m.multiple_guards_wrong_order() == "unguarded & guarded"
|
|
|
|
if hasattr(m, "with_gil"):
|
|
assert m.with_gil() == "GIL held"
|
|
assert m.without_gil() == "GIL released"
|