matplotlib/unit/memleak_hawaii3.py at master · ryanfreckleton/matplotlib · GitHub

Repository navigation

#!/usr/bin/env python

from __future__ import print_function

import os, sys, time, gc

import matplotlib

matplotlib.use('PDF')

from matplotlib.cbook import report_memory

import numpy as np

from pylab import figure, show, close

# take a memory snapshot on indStart and compare it with indEnd

rand = np.random.rand

indStart, indEnd = 200, 401

for i in range(indEnd):

fig = figure(1)

fig.clf()

t1 = np.arange(0.0, 2.0, 0.01)

y1 = np.sin(2*np.pi*t1)

y2 = rand(len(t1))

X = rand(50,50)

ax = fig.add_subplot(221)

ax.plot(t1, y1, '-')

ax.plot(t1, y2, 's')

ax = fig.add_subplot(222)

ax.imshow(X)

ax = fig.add_subplot(223)

ax.scatter(rand(50), rand(50), s=100*rand(50), c=rand(50))

ax = fig.add_subplot(224)

ax.pcolor(10*rand(50,50))

fig.savefig('tmp%d' % i, dpi = 75)

close(1)

gc.collect()

val = report_memory(i)

print(i, val)

if i==indStart: start = val # wait a few cycles for memory usage to stabilize

end = val

print('Average memory consumed per loop: %1.4fk bytes\n' % ((end-start)/float(indEnd-indStart)))