SF.net SVN: matplotlib: [5160] trunk/matplotlib/examples/tests/ backend_driver.py

[<jd...@us...>]

Revision: 5160
          http://matplotlib.svn.sourceforge.net/matplotlib/?rev=5160&view=rev
Author:   jdh2358
Date:     2008-05-16 13:34:35 -0700 (Fri, 16 May 2008)

Log Message:
-----------
added the new backend driver

Added Paths:
-----------
    trunk/matplotlib/examples/tests/backend_driver.py

Added: trunk/matplotlib/examples/tests/backend_driver.py
===================================================================
--- trunk/matplotlib/examples/tests/backend_driver.py	                        (rev 0)
+++ trunk/matplotlib/examples/tests/backend_driver.py	2008-05-16 20:34:35 UTC (rev 5160)
@@ -0,0 +1,238 @@
+#!/usr/bin/env python
+"""
+This is used to drive many of the examples across the backends, for
+regression testing, and comparing backend efficiency.
+
+The script takes one or more arguments specifying backends
+to be tested, e.g.
+
+    python backend_driver.py agg ps cairo.png cairo.ps
+
+would test the agg and ps backends, and the cairo backend with
+output to png and ps files.
+
+If no arguments are given, a default list of backends will be
+tested.
+"""
+
+from __future__ import division
+import os, time, sys
+import matplotlib.backends as mplbe
+
+pylab_dir = os.path.join('..', 'pylab')
+pylab_files = [
+    'alignment_test.py',
+    'arctest.py',
+    'arrow_demo.py',
+    'auto_layout.py',
+    'axes_demo.py',
+    'axhspan_demo.py',
+    'bar_stacked.py',
+    'barchart_demo.py',
+    'boxplot_demo.py',
+    'broken_barh.py',
+    'barh_demo.py',
+    'color_demo.py',
+    'cohere_demo.py',
+    'contour_demo.py',
+    'contourf_demo.py',
+    'csd_demo.py',
+    'custom_ticker1.py',
+    'customize_rc.py',
+    'date_demo1.py',
+    'date_demo2.py',
+    'equal_aspect_ratio.py',
+    'errorbar_limits.py',
+    'figimage_demo.py',
+    'figlegend_demo.py',
+    'figtext.py',
+    'fill_demo.py',
+    'finance_demo.py',
+    'fonts_demo_kw.py',
+    'hexbin_demo.py',
+    'histogram_demo.py',
+    'hline_demo.py',
+    'image_demo.py',
+    'image_demo2.py',
+    'image_masked.py',
+    'image_origin.py',
+    'invert_axes.py',
+    'layer_images.py',
+    'legend_auto.py',
+    'legend_demo.py',
+    'legend_demo2.py',
+    'line_collection.py',
+    'line_collection2.py',
+    'line_styles.py',
+    'log_demo.py',
+    'log_test.py',
+    'major_minor_demo1.py',
+    'major_minor_demo2.py',
+    'masked_demo.py',
+    'mathtext_demo.py',
+    'mri_with_eeg.py',
+    'multiple_figs_demo.py',
+    'nan_test.py',
+    'pcolor_demo.py',
+    'pcolor_demo2.py',
+    'pcolor_small.py',
+    'pie_demo.py',
+    'polar_demo.py',
+    'polar_scatter.py',
+    'psd_demo.py',
+    'quadmesh_demo.py',
+    'quiver_demo.py',
+    'scatter_demo.py',
+    'scatter_demo2.py',
+    'scatter_star_poly.py',
+    'shared_axis_demo.py',
+    'shared_axis_across_figures.py',
+    'simple_plot.py',
+    'specgram_demo.py',
+    'spy_demos.py',
+    'stem_plot.py',
+    'step_demo.py',
+    'stock_demo.py',
+    'subplot_demo.py',
+#    'set_and_get.py',
+    'table_demo.py',
+    'text_handles.py',
+    'text_rotation.py',
+    'text_themes.py',
+#    'tex_demo.py',
+    'two_scales.py',
+    'unicode_demo.py',
+    'vline_demo.py',
+    'xcorr_demo.py',
+    'zorder_demo.py',
+    ]
+
+
+api_dir = os.path.join('..', 'api')
+api_files = [
+    'colorbar_only.py',
+]
+
+
+files = [os.path.join(pylab_dir, fname) for fname in pylab_files] +\
+    [os.path.join(api_dir, fname) for fname in api_files]
+# tests known to fail on a given backend
+
+
+failbackend = dict(
+    SVG = ('tex_demo.py,'),
+    )
+
+try:
+    import subprocess
+    def run(arglist):
+        try:
+            subprocess.call(arglist)
+        except KeyboardInterrupt:
+            sys.exit()
+except ImportError:
+    def run(arglist):
+        os.system(' '.join(arglist))
+
+def drive(backend, python=['python'], switches = []):
+    exclude = failbackend.get(backend, [])
+
+    # Clear the destination directory for the examples
+    path = backend
+    if os.path.exists(path):
+        import glob
+        for fname in os.listdir(path):
+            os.unlink(os.path.join(path,fname))
+    else:
+        os.mkdir(backend)
+
+    for fullpath in files:
+
+        print ('\tdriving %-40s' % (fullpath)),
+        sys.stdout.flush()
+
+        fpath, fname = os.path.split(fullpath)
+
+        if fname in exclude:
+            print '\tSkipping %s, known to fail on backend: %s'%backend
+            continue
+
+        basename, ext = os.path.splitext(fname)
+        outfile = os.path.join(path,basename)
+        tmpfile_name = '_tmp_%s.py' % basename
+        tmpfile = file(tmpfile_name, 'w')
+
+        for line in file(fullpath):
+            line_lstrip = line.lstrip()
+            if line_lstrip.startswith("#"):
+                tmpfile.write(line)
+            else:
+                break
+
+        tmpfile.writelines((
+            'from __future__ import division\n',
+            'import sys\n',
+            'sys.path.append("%s")\n'%fpath,
+            'import matplotlib\n',
+            'matplotlib.use("%s")\n' % backend,
+            'from pylab import savefig\n',
+            ))
+        for line in file(fullpath):
+            line_lstrip = line.lstrip()
+            if (line_lstrip.startswith('from __future__ import division') or
+                line_lstrip.startswith('matplotlib.use') or
+                line_lstrip.startswith('savefig') or
+                line_lstrip.startswith('show')):
+                continue
+            tmpfile.write(line)
+        if backend in mplbe.interactive_bk:
+            tmpfile.write('show()')
+        else:
+            tmpfile.write('savefig("%s", dpi=150)' % outfile)
+
+        tmpfile.close()
+        start_time = time.time()
+        program = [x % {'name': basename} for x in python]
+        run(program + [tmpfile_name, switchstring])
+        end_time = time.time()
+        print (end_time - start_time)
+        #os.system('%s %s %s' % (python, tmpfile_name, switchstring))
+        os.remove(tmpfile_name)
+
+if __name__ == '__main__':
+    times = {}
+    default_backends = ['Agg', 'PS', 'SVG', 'PDF', 'Template']
+    if '--coverage' in sys.argv:
+        python = ['coverage.py', '-x']
+        sys.argv.remove('--coverage')
+    elif '--valgrind' in sys.argv:
+        python = ['valgrind', '--tool=memcheck', '--leak-check=yes',
+                  '--log-file=%(name)s', 'python']
+        sys.argv.remove('--valgrind')
+    elif sys.platform == 'win32':
+        python = [r'c:\Python24\python.exe']
+    else:
+        python = ['python']
+    all_backends = [b.lower() for b in mplbe.all_backends]
+    all_backends.extend(['cairo.png', 'cairo.ps', 'cairo.pdf', 'cairo.svg'])
+    backends = []
+    switches = []
+    if sys.argv[1:]:
+        backends = [b.lower() for b in sys.argv[1:] if b.lower() in all_backends]
+        switches = [s for s in sys.argv[1:] if s.startswith('--')]
+    if not backends:
+        backends = default_backends
+    for backend in backends:
+        switchstring = ' '.join(switches)
+        print 'testing %s %s' % (backend, switchstring)
+        t0 = time.time()
+        drive(backend, python, switches)
+        t1 = time.time()
+        times[backend] = (t1-t0)/60.0
+
+    # print times
+    for backend, elapsed in times.items():
+        print 'Backend %s took %1.2f minutes to complete' % ( backend, elapsed)
+        if 'Template' in times:
+            print '\ttemplate ratio %1.3f, template residual %1.3f' % (
+                elapsed/times['Template'], elapsed-times['Template'])


This was sent by the SourceForge.net collaborative development platform, the world's largest Open Source development site.