def save_example_fit(fit): """ Save fit result to a json file and a plot to an svg file. """ json_directory = os.path.join('examples', 'json') plot_directory = os.path.join('examples', 'plots') if not os.path.isdir(json_directory): os.makedirs(json_directory) if not os.path.isdir(plot_directory): os.makedirs(plot_directory) fit.to_json(os.path.join(json_directory, fit.name + '.json'), meta=fit.metadata) plot = Plot(fit) plot.save(os.path.join(plot_directory, fit.name + '.svg')) plot.close()
def test_save(self): plot = Plot(fit=self.get_fit_for_fitting()) plot.save('test/test.svg') assert os.path.exists('test/test.svg')