66
77from __future__import division
88import os ,time ,sys
9+ import matplotlib .backends as mplbe
10+
911files = (
1012'alignment_test.py' ,
1113'arctest.py' ,
3032'image_demo.py' ,
3133'image_demo2.py' ,
3234'image_demo_na.py' ,
35+ 'image_masked.py' ,
3336'image_origin.py' ,
3437'invert_axes.py' ,
3538'layer_images.py' ,
7477 )
7578
7679
77- #tests known to fail on python22 (require datetime)
78- fail22 = (
79- 'date_demo1.py' ,
80- 'date_demo2.py' ,
81- 'finance_demo.py' ,
82- )
83-
84-
8580# tests known to fail on a given backend
8681
8782failbackend = dict (
@@ -118,7 +113,7 @@ def drive(backend, python='python', switches = []):
118113line_lstrip .startswith ('show' )):
119114continue
120115tmpfile .write (line )
121- if backend in ( 'GTK' , 'WX' , 'TkAgg' ) :
116+ if backend in mplbe . interactive_bk :
122117tmpfile .write ('show()' )
123118else :
124119tmpfile .write ('savefig("%s", dpi=150)' % outfile )
@@ -130,19 +125,15 @@ def drive(backend, python='python', switches = []):
130125
131126if __name__ == '__main__' :
132127times = {}
133- # backends = ['Agg', 'Cairo', 'GDK', 'PS', 'SVG', 'Template']
134- #backends = ['Agg', 'PS', 'SVG', 'Template']
135- # backends = [ 'GTK', 'WX', 'TkAgg']
136128default_backends = ['Agg' ,'PS' ,'SVG' ,'Template' ]
137- #default_backends = ['Agg']
138- #backends = ['Agg']
139129if sys .platform == 'win32' :
140130python = r'c:\Python24\python.exe'
141131else :
142132python = 'python'
133+ backends = []
143134switches = []
144135if sys .argv [1 :]:
145- backends = [b for b in sys .argv [1 :]if b in default_backends ]
136+ backends = [b for b in sys .argv [1 :]if b in mplbe . all_backends ]
146137switches = [s for s in sys .argv [1 :]if s .startswith ('--' )]
147138if not backends :
148139backends = default_backends