def stop(self):
if _memalloc is not None:
try:
_memalloc.stop()
except RuntimeError:
pass
super(MemoryCollector, self).stop()
def test_iter_events():
max_nframe = 32
_memalloc.start(max_nframe, 10000, 512 * 1024)
_allocate_1k()
events, count, alloc_count = _memalloc.iter_events()
_memalloc.stop()
assert count >= 1000
# Watchout: if we dropped samples the test will likely fail
object_count = 0
for (stack, nframe, thread_id), size in events:
assert 0 < len(stack) <= max_nframe
assert nframe >= len(stack)
last_call = stack[0]
assert size >= 1 # size depends on the object size
if last_call[2] == "<listcomp>" and last_call[1] == _ALLOC_LINE_NUMBER:
assert thread_id == nogevent.main_thread_id
assert last_call[0] == __file__
assert stack[1][0] == __file__
assert stack[1][1] == _ALLOC_LINE_NUMBER
assert stack[1][2] == "_allocate_1k"
object_count += 1
assert object_count >= 1000
def test_heap():
max_nframe = 32
_memalloc.start(max_nframe, 10, 1024)
x = _allocate_1k()
# Check that at least one sample comes from the main thread
thread_found = False
for (stack, nframe, thread_id), size in _memalloc.heap():
assert 0 < len(stack) <= max_nframe
assert size > 0
if thread_id == nogevent.main_thread_id:
thread_found = True
assert isinstance(thread_id, int)
if (stack[0][0] == __file__ and stack[0][1] == _ALLOC_LINE_NUMBER
and stack[0][2] == "<listcomp>" and stack[1][0] == __file__
and stack[1][1] == _ALLOC_LINE_NUMBER
and stack[1][2] == "_allocate_1k" and stack[2][0] == __file__
and stack[2][2] == "test_heap"):
break
else:
pytest.fail("No trace of allocation in heap")
assert thread_found, "Main thread not found"
y = _pre_allocate_1k()
for (stack, nframe, thread_id), size in _memalloc.heap():
assert 0 < len(stack) <= max_nframe
assert size > 0
assert isinstance(thread_id, int)
if (stack[0][0] == __file__ and stack[0][1] == _ALLOC_LINE_NUMBER
and stack[0][2] == "<listcomp>" and stack[1][0] == __file__
and stack[1][1] == _ALLOC_LINE_NUMBER
and stack[1][2] == "_allocate_1k" and stack[2][0] == __file__
and stack[2][2] == "_pre_allocate_1k"):
break
else:
pytest.fail("No trace of allocation in heap")
del x
gc.collect()
for (stack, nframe, thread_id), size in _memalloc.heap():
assert 0 < len(stack) <= max_nframe
assert size > 0
assert isinstance(thread_id, int)
if (stack[0][0] == __file__ and stack[0][1] == _ALLOC_LINE_NUMBER
and stack[0][2] == "<listcomp>" and stack[1][0] == __file__
and stack[1][1] == _ALLOC_LINE_NUMBER
and stack[1][2] == "_allocate_1k" and stack[2][0] == __file__
and stack[2][2] == "test_heap"):
pytest.fail("Allocated memory still in heap")
del y
gc.collect()
for (stack, nframe, thread_id), size in _memalloc.heap():
assert 0 < len(stack) <= max_nframe
assert size > 0
assert isinstance(thread_id, int)
if (stack[0][0] == __file__ and stack[0][1] == _ALLOC_LINE_NUMBER
and stack[0][2] == "<listcomp>" and stack[1][0] == __file__
and stack[1][1] == _ALLOC_LINE_NUMBER
and stack[1][2] == "_allocate_1k" and stack[2][0] == __file__
and stack[2][2] == "_pre_allocate_1k"):
pytest.fail("Allocated memory still in heap")
_memalloc.stop()
def _stop_service(self): # type: ignore[override]
# type: (...) -> None
super(MemoryCollector, self)._stop_service()
if _memalloc is not None:
try:
_memalloc.stop()
except RuntimeError:
pass
def test_iter_events_dropped():
max_nframe = 32
_memalloc.start(max_nframe, 100, 512 * 1024)
_allocate_1k()
events, count, alloc_count = _memalloc.iter_events()
_memalloc.stop()
assert count == 100
assert alloc_count >= 1000
def test_heap_stress():
# This should run for a few seconds, and is enough to spot potential segfaults.
_memalloc.start(64, 64, 1024)
x = []
for _ in range(20):
for _ in range(1000):
x.append(object())
_memalloc.heap()
del x[:100]
_memalloc.stop()
def test_iter_events_multi_thread():
max_nframe = 32
t = threading.Thread(target=_allocate_1k)
_memalloc.start(max_nframe, 10000)
_allocate_1k()
t.start()
t.join()
events, count, alloc_count = _memalloc.iter_events()
_memalloc.stop()
assert count >= 1000
# Watchout: if we dropped samples the test will likely fail
count_object = 0
count_thread = 0
for (stack, nframe, thread_id), size in events:
assert 0 < len(stack) <= max_nframe
assert nframe >= len(stack)
last_call = stack[0]
assert size >= 1 # size depends on the object size
if last_call[2] == "_allocate_1k" and last_call[
1] == _ALLOC_LINE_NUMBER:
assert last_call[0] == __file__
if thread_id == _nogevent.main_thread_id:
count_object += 1
assert stack[1][0] == __file__
assert stack[1][1] == 105
assert stack[1][2] == "test_iter_events_multi_thread"
elif thread_id == t.ident:
count_thread += 1
assert stack[1][0] == threading.__file__
assert stack[1][1] > 0
assert stack[1][2] == "run"
assert count_object == 1000
assert count_thread == 1000
def stop(self):
if _memalloc is not None:
_memalloc.stop()
super(MemoryCollector, self).stop()
def test_start_stop():
_memalloc.start(1, 1, 1)
_memalloc.stop()
def test_start_twice():
_memalloc.start(64, 1000, 512)
with pytest.raises(RuntimeError):
_memalloc.start(64, 1000, 512)
_memalloc.stop()
