def test_base64(): import matplotlib matplotlib.use('Agg') import matplotlib.pyplot as plt fig = plt.figure() plt.plot(range(10)) res = jutil.fig2base64(fig, 'png') res = jutil.fig2base64(fig, 'pdf')
def show_fig(self, fig, file_name): """ Save fig object to self.output_folder/filename. Parameters ---------- fig : matplotlib.figure.Figure file_name : str """ self.fig_objs[file_name] = fig if self.output_format in ['pdf', 'png', 'jpg']: fp = os.path.join(self.output_folder, '.'.join([file_name, self.output_format])) jutil.create_dir(fp) fig.savefig(fp) print("Figure saved: {}".format(fp)) elif self.output_format == 'base64': fig_b64 = jutil.fig2base64(fig, 'png') self.fig_data[file_name] = fig_b64 print("Base64 data of figure {} will be stored in dictionary.". format(file_name)) elif self.output_format == 'plot': fig.show() else: raise NotImplementedError("output_format = {}".format( self.output_format))
def show_fig(self, fig, file_name): """ Save fig object to self.output_folder/filename. Parameters ---------- fig : matplotlib.figure.Figure file_name : str """ self.fig_objs[file_name] = fig if self.output_format in ['pdf', 'png', 'jpg']: fp = os.path.join(self.output_folder, '.'.join([file_name, self.output_format])) jutil.create_dir(fp) fig.savefig(fp) print("Figure saved: {}".format(fp)) elif self.output_format == 'base64': fig_b64 = jutil.fig2base64(fig, 'png') self.fig_data[file_name] = fig_b64 print("Base64 data of figure {} will be stored in dictionary.".format(file_name)) elif self.output_format == 'plot': fig.show() else: raise NotImplementedError("output_format = {}".format(self.output_format))