1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
# mode: run
# tag: cpp, werror, cpp11, no-cpp-locals
from libcpp.memory cimport unique_ptr, shared_ptr, default_delete, dynamic_pointer_cast
from libcpp cimport nullptr
cdef extern from "cpp_smart_ptr_helper.h":
cdef cppclass CountAllocDealloc:
CountAllocDealloc(int*, int*)
cdef cppclass FreePtr[T]:
pass
ctypedef const CountAllocDealloc const_CountAllocDealloc
def test_unique_ptr():
"""
>>> test_unique_ptr()
"""
cdef int alloc_count = 0, dealloc_count = 0
cdef unique_ptr[CountAllocDealloc] x_ptr
x_ptr.reset(new CountAllocDealloc(&alloc_count, &dealloc_count))
assert alloc_count == 1
x_ptr.reset()
assert alloc_count == 1
assert dealloc_count == 1
##Repeat the above test with an explicit default_delete type
alloc_count = 0
dealloc_count = 0
cdef unique_ptr[CountAllocDealloc,default_delete[CountAllocDealloc]] x_ptr2
x_ptr2.reset(new CountAllocDealloc(&alloc_count, &dealloc_count))
assert alloc_count == 1
x_ptr2.reset()
assert alloc_count == 1
assert dealloc_count == 1
alloc_count = 0
dealloc_count = 0
cdef unique_ptr[CountAllocDealloc,FreePtr[CountAllocDealloc]] x_ptr3
x_ptr3.reset(new CountAllocDealloc(&alloc_count, &dealloc_count))
assert x_ptr3.get() != nullptr;
x_ptr3.reset()
assert x_ptr3.get() == nullptr;
def test_const_shared_ptr():
"""
>>> test_const_shared_ptr()
"""
cdef int alloc_count = 0, dealloc_count = 0
cdef shared_ptr[const CountAllocDealloc] ptr = shared_ptr[const_CountAllocDealloc](
new CountAllocDealloc(&alloc_count, &dealloc_count))
assert alloc_count == 1
assert dealloc_count == 0
cdef shared_ptr[const CountAllocDealloc] ptr2 = ptr
assert alloc_count == 1
assert dealloc_count == 0
ptr.reset()
assert alloc_count == 1
assert dealloc_count == 0
ptr2.reset()
assert alloc_count == 1
assert dealloc_count == 1
cdef cppclass A:
void some_method(): # Force this to be a polymorphic class for dynamic cast.
pass
cdef cppclass B(A):
pass
cdef cppclass C(B):
pass
cdef shared_ptr[A] holding_subclass = shared_ptr[A](new C())
def test_assignment_to_base_class():
"""
>>> test_assignment_to_base_class()
"""
cdef shared_ptr[C] derived = shared_ptr[C](new C())
cdef shared_ptr[A] base = derived
def test_dynamic_pointer_cast():
"""
>>> test_dynamic_pointer_cast()
"""
cdef shared_ptr[B] b = shared_ptr[B](new B())
cdef shared_ptr[A] a = dynamic_pointer_cast[A, B](b)
assert a.get() == b.get()
a = shared_ptr[A](new A())
b = dynamic_pointer_cast[B, A](a)
assert b.get() == NULL
|
