def test_frequency_plot(self):
sp = Spectrum(name='Example 1')
sp.from_surface(surface=np.ones([10, 10]))
pl = sp.frequency_plot()
os.makedirs('data/test_data')
pl.savefig('data/test_data/frequency_plot.png')
self.assertEqual(os.path.exists('data/test_data/frequency_plot.png'),
True)
shutil.rmtree('data/test_data/')
def plot_spectra(inputfolder, outputfolder, **kwargs):
"""
Plot bar plots for individual frequency spectra in a given directory.
Parameters
----------
inputfolder : str
Input directory with spectra to plot.
outputfolder : str
Output directory to save the bar plots.
kwargs : key, value pairings
Arbitrary keyword arguments to pass to the Spectrum.frequency_plot function.
"""
files = filelib.list_subfolders(inputfolder, extensions=['csv'])
for fn in files:
s = Spectrum(filename=inputfolder + fn)
pl = s.frequency_plot(title=fn[:-4], **kwargs)
filelib.make_folders([os.path.dirname(outputfolder + fn[:-4])])
pl.savefig(outputfolder + fn[:-4] + '.png')
pl.clf()
