Merge pull request #46 from gshiba/close-fig

Close figure after saving

Author: astrofrog

pytest_mpl/plugin.py

@@ -172,6 +172,7 @@ def item_function_wrapper(*args, **kwargs):
                     test_image = os.path.abspath(os.path.join(result_dir, filename))
 
                     fig.savefig(test_image, **savefig_kwargs)
+                    plt.close(fig)
 
                     # Find path to baseline image
                     if baseline_remote:
@@ -202,6 +203,7 @@ def item_function_wrapper(*args, **kwargs):
                         os.makedirs(self.generate_dir)
 
                     fig.savefig(os.path.abspath(os.path.join(self.generate_dir, filename)), **savefig_kwargs)
+                    plt.close(fig)
                     pytest.skip("Skipping test, since generating data")
 
         if item.cls is not None: