def test_collection_reset():
counter = memory._gc_nbytes_counter
op = HomothetyOperator(2)
op.delete()
assert memory._gc_nbytes_counter - counter == 8
memory.garbage_collect()
assert memory._gc_nbytes_counter == 0
def test_collection():
with setting(config, 'GC_NBYTES_THRESHOLD', 8000):
memory.garbage_collect()
counter = 0
for i in range(10):
data = np.arange(100)
counter += data.nbytes
op = DiagonalOperator(data)
op.delete()
if i < 9:
assert memory._gc_nbytes_counter == counter
assert memory._gc_nbytes_counter == 0
