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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
|
from .memory_allocator cimport MemoryAllocator
cdef class TestMemoryAllocator():
cdef MemoryAllocator mem
def __init__(self):
self.mem = MemoryAllocator()
def malloc(self, size):
"""
TESTS::
>>> from memory_allocator.test import TestMemoryAllocator
>>> mem = TestMemoryAllocator()
>>> _ = mem.malloc(100)
>>> mem.malloc(mem.size_t_max())
Traceback (most recent call last):
...
MemoryError: failed to allocate ... bytes
"""
return <size_t> self.mem.malloc(size)
def calloc(self, nmemb, size):
"""
TESTS::
>>> from memory_allocator.test import TestMemoryAllocator
>>> mem = TestMemoryAllocator()
>>> _ = mem.calloc(100, 10)
>>> mem.calloc(mem.size_t_max(), 1)
Traceback (most recent call last):
...
MemoryError: failed to allocate ... * 1 bytes
"""
return <size_t> self.mem.calloc(nmemb, size)
def allocarray(self, nmemb, size):
"""
TESTS::
>>> from memory_allocator.test import TestMemoryAllocator
>>> mem = TestMemoryAllocator()
>>> _ = mem.allocarray(100, 10)
>>> mem.allocarray(mem.size_t_max(), 1)
Traceback (most recent call last):
...
MemoryError: failed to allocate ... * 1 bytes
"""
return <size_t> self.mem.allocarray(nmemb, size)
def realloc(self, size_t ptr, size):
"""
TESTS::
>>> from memory_allocator.test import TestMemoryAllocator
>>> def test_realloc_good():
... mem = TestMemoryAllocator()
... ptr = mem.malloc(20)
... mem.realloc(ptr, 21)
>>> def test_realloc_NULL():
... mem = TestMemoryAllocator()
... mem.realloc(0, 21)
>>> def test_realloc_bad():
... mem = TestMemoryAllocator()
... mem2 = TestMemoryAllocator()
... ptr = mem.malloc(20)
... mem2.realloc(ptr, 21)
>>> test_realloc_good()
>>> test_realloc_NULL()
>>> test_realloc_bad()
Traceback (most recent call last):
...
ValueError: given pointer not found in MemoryAllocator
"""
return <size_t> self.mem.realloc(<void*> ptr, size)
def reallocarray(self, size_t ptr, nmemb, size):
"""
TESTS::
>>> from memory_allocator.test import TestMemoryAllocator
>>> def test_reallocarray_good():
... mem = TestMemoryAllocator()
... ptr = mem.allocarray(20, 8)
... mem.reallocarray(ptr, 21, 8)
>>> def test_reallocarray_NULL():
... mem = TestMemoryAllocator()
... mem.reallocarray(0, 21, 8)
>>> def test_reallocarray_bad():
... mem = TestMemoryAllocator()
... mem2 = TestMemoryAllocator()
... ptr = mem.allocarray(20, 8)
... mem2.reallocarray(ptr, 21, 8)
>>> test_reallocarray_good()
>>> test_reallocarray_NULL()
>>> test_reallocarray_bad()
Traceback (most recent call last):
...
ValueError: given pointer not found in MemoryAllocator
"""
return <size_t> self.mem.reallocarray(<void*> ptr, nmemb, size)
def aligned_malloc(self, alignment, size):
"""
TESTS::
>>> from memory_allocator.test import TestMemoryAllocator
>>> mem = TestMemoryAllocator()
>>> for i in range(12):
... ptr = mem.aligned_malloc(2**i, 4048)
... assert ptr == ptr & ~(2**i-1)
"""
return <size_t> self.mem.aligned_malloc(alignment, size)
def aligned_calloc(self, alignment, nmemb, size):
"""
TESTS::
>>> from memory_allocator.test import TestMemoryAllocator
>>> mem = TestMemoryAllocator()
>>> for i in range(12):
... ptr = mem.aligned_calloc(2**i, i, 4048)
... assert ptr == ptr & ~(2**i-1)
"""
return <size_t> self.mem.aligned_calloc(alignment, nmemb, size)
def aligned_allocarray(self, alignment, nmemb, size):
"""
TESTS::
>>> from memory_allocator.test import TestMemoryAllocator
>>> mem = TestMemoryAllocator()
>>> for i in range(12):
... ptr = mem.aligned_allocarray(2**i, i, 4048)
... assert ptr == ptr & ~(2**i-1)
"""
return <size_t> self.mem.aligned_allocarray(alignment, nmemb, size)
def n(self):
return self.mem.n
def size(self):
return self.mem.size
def size_t_max(self):
return <size_t> -1
|
