Added option (-c) to display cycles including uncollectable garbage.

mdboom · mdboom · commit 79d28921dfe1 · 2007-07-03T14:25:12.000Z
Displays the total number of Python objects in existence at each
iteration.

svn path=/trunk/matplotlib/; revision=3451
diff --git a/unit/memleak_gui.py b/unit/memleak_gui.py
@@ -34,7 +34,8 @@
 parser.add_option("-d", "--backend", dest="backend",
                   default='',
                   help="backend")
-
+parser.add_option("-c", "--cycles", dest="cycles",
+                  default=False, action="store_true")
 
 options, args = parser.parse_args()
 
@@ -47,6 +48,8 @@
 import pylab
 import matplotlib.cbook as cbook
 
+print indEnd
+
 for i in range(indEnd+1):
 
     fig = pylab.figure()
@@ -57,7 +60,8 @@
     val = cbook.report_memory(i)
     if options.verbose:
         if i % 10 == 0:
-            print i, val
+            print ("iter: %4d OS memory: %8d Python objects: %8d" % 
+                   (i, val, len(gc.get_objects())))
     if i==indStart: start = val # wait a few cycles for memory usage to stabilize
 
 gc.collect()
@@ -73,3 +77,5 @@
     print '# Memory went from %dk to %dk' % (start, end)
     print '# Average memory consumed per loop: %1.4fk bytes\n' % ((end-start)/float(indEnd-indStart))
 
+if options.cycles:
+    cbook.print_cycles(gc.garbage)
