matplotlib/unit/memleak.py at master · rryan/matplotlib · GitHub

Repository navigation

History

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

#!/usr/bin/env python

from __future__ import print_function

import gc

try:

import tracemalloc

except ImportError:

raise ImportError("This script requires Python 3.4 or later")

try:

import psutil

except ImportError:

raise ImportError("This script requires psutil")

import numpy as np

def run_memleak_test(bench, iterations, report):

tracemalloc.start()

starti = min(50, iterations // 2)

endi = iterations

malloc_arr = np.empty((endi,), dtype=np.int64)

rss_arr = np.empty((endi,), dtype=np.int64)

rss_peaks = np.empty((endi,), dtype=np.int64)

nobjs_arr = np.empty((endi,), dtype=np.int64)

garbage_arr = np.empty((endi,), dtype=np.int64)

open_files_arr = np.empty((endi,), dtype=np.int64)

rss_peak = 0

p = psutil.Process()

for i in range(endi):

bench()

gc.collect()

rss = p.memory_info().rss

malloc, peak = tracemalloc.get_traced_memory()

nobjs = len(gc.get_objects())

garbage = len(gc.garbage)

open_files = len(p.open_files())

print("{0: 4d}: pymalloc {1: 10d}, rss {2: 10d}, nobjs {3: 10d}, garbage {4: 4d}, files: {5: 4d}".format(

i, malloc, rss, nobjs, garbage, open_files))

malloc_arr[i] = malloc

rss_arr[i] = rss

if rss > rss_peak:

rss_peak = rss

rss_peaks[i] = rss_peak

nobjs_arr[i] = nobjs

garbage_arr[i] = garbage

open_files_arr[i] = open_files

print('Average memory consumed per loop: %1.4f bytes\n' %

(np.sum(rss_peaks[starti+1:] - rss_peaks[starti:-1]) / float(endi - starti)))

from matplotlib import pyplot as plt

fig, (ax1, ax2, ax3) = plt.subplots(3)

ax1b = ax1.twinx()

ax1.plot(malloc_arr, 'r')

ax1b.plot(rss_arr, 'b')

ax1.set_ylabel('pymalloc', color='r')

ax1b.set_ylabel('rss', color='b')

ax2b = ax2.twinx()

ax2.plot(nobjs_arr, 'r')

ax2b.plot(garbage_arr, 'b')

ax2.set_ylabel('total objects', color='r')

ax2b.set_ylabel('garbage objects', color='b')

ax3.plot(open_files_arr)

ax3.set_ylabel('open file handles')

if not report.endswith('.pdf'):

report = report + '.pdf'

fig.tight_layout()

fig.savefig(report, format='pdf')

class MemleakTest(object):

def __init__(self, empty):

self.empty = empty

def __call__(self):

import matplotlib.pyplot as plt

fig = plt.figure(1)

if not self.empty:

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

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

y2 = np.random.rand(len(t1))

X = np.random.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(np.random.rand(50), np.random.rand(50),

s=100 * np.random.rand(50), c=np.random.rand(50))

ax = fig.add_subplot(224)

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

fig.savefig('tmp', dpi=75)

plt.close(1)

if __name__ == '__main__':

import argparse

parser = argparse.ArgumentParser('Run memory leak tests')

parser.add_argument('backend', type=str, nargs=1,

help='backend to test')

parser.add_argument('iterations', type=int, nargs=1,

help='number of iterations')

parser.add_argument('report', type=str, nargs=1,

help='filename to save report')

parser.add_argument('--empty', action='store_true',

help="Don't plot any content, just test creating "

"and destroying figures")

parser.add_argument('--interactive', action='store_true',

help="Turn on interactive mode to actually open "

"windows. Only works with some GUI backends.")

args = parser.parse_args()

import matplotlib

matplotlib.use(args.backend[0])

if args.interactive:

from matplotlib import pyplot as plt

plt.ion()

run_memleak_test(MemleakTest(args.empty), args.iterations[0], args.report[0])

matplotlib/unit/memleak.py at master · rryan/matplotlib · GitHub

Repository navigation

Expand file tree

Breadcrumbs

memleak.py

Latest commit

History

Breadcrumbs

memleak.py

File metadata and controls