import numpy as np

from gpaw.utilities.memory import maxrss, _VmB

M = 1024 * 1024

print maxrss() / M

itemsize = np.zeros(1).itemsize

# allocate 128 MiB
x = np.zeros(128 * M / itemsize)

print maxrss() / M


# 'names' taken from gpaw.utilities.memory

names = ('VmHWM:',  # Peak resident set size ("high water mark")
         'VmRss:',  # Resident set size
         'VmPeak:', # Peak virtual memory size
         'VmSize:', # Virtual memory size
         )

for name in names:
    print name, _VmB(name) / M