@@ -142,7 +142,7 @@ def do_test():
142142 if self ._remove_text :
143143 self .remove_text (figure )
144144
145- figure .savefig (actual_fname )
145+ figure .savefig (actual_fname , ** self . _savefig_kwarg )
146146
147147 err = compare_images (expected_fname , actual_fname ,
148148 self ._tol , in_decorator = True )
@@ -166,7 +166,8 @@ def do_test():
166166 yield (do_test ,)
167167
168168def image_comparison (baseline_images = None , extensions = None , tol = 1e-3 ,
169- freetype_version = None , remove_text = False ):
169+ freetype_version = None , remove_text = False ,
170+ savefig_kwarg = None ):
170171 """
171172 call signature::
172173
@@ -199,6 +200,10 @@ def image_comparison(baseline_images=None, extensions=None, tol=1e-3,
199200 Remove the title and tick text from the figure before
200201 comparison. This does not remove other, more deliberate,
201202 text, such as legends and annotations.
203+
204+ *savefig_kwarg*: dict
205+ Optional arguments that are passed to the savefig method.
206+
202207 """
203208
204209 if baseline_images is None :
@@ -208,6 +213,10 @@ def image_comparison(baseline_images=None, extensions=None, tol=1e-3,
208213 # default extensions to test
209214 extensions = ['png' , 'pdf' , 'svg' ]
210215
216+ if savefig_kwarg is None :
217+ #default no kwargs to savefig
218+ savefig_kwarg = dict ()
219+
211220 def compare_images_decorator (func ):
212221 # We want to run the setup function (the actual test function
213222 # that generates the figure objects) only once for each type
@@ -231,7 +240,8 @@ def compare_images_decorator(func):
231240 '_extensions' : extensions ,
232241 '_tol' : tol ,
233242 '_freetype_version' : freetype_version ,
234- '_remove_text' : remove_text })
243+ '_remove_text' : remove_text ,
244+ '_savefig_kwarg' : savefig_kwarg })
235245
236246 return new_class
237247 return compare_images_decorator
0 commit comments