Measure BTree memory usage

memt.py

from __future__ import print_function

import sys
import random
import psutil

from BTrees import IIBTree
from BTrees import IBBTree

mem_info = psutil.Process().memory_info()
human = "%s MiB (%s)" % (mem_info.rss / (1024*1024), mem_info)
print(human)

size = 10**8

if sys.argv[1] == "IB":
    print("Using IB, adding %s values" % size)
    data = IBBTree.IBBTree()
else:
    print("Using II, adding %s values" % size)
    data = IIBTree.IIBTree()

for i in range(size):
    value = random.randint(-127, 127)
    data[i] = value

mem_info = psutil.Process().memory_info()
human = "%s MiB (%s)" % (mem_info.rss / (1024*1024), mem_info)
print(human)