def test_plot_spectra(): mock_fig = plt.figure(); mock_fig.plot = MagicMock() ds = create_fake_dataset() sp = spp.Spectrum(ds) spp.plot_spectra(mock_fig, [sp, sp.copy()]) assert mock_fig.plot.call_count == 2
def test_plot_spectra(): mock_axes = MagicMock() mock_axes.plot = MagicMock() ds = create_fake_dataset() sp = spp.Spectrum(ds) spp.plot_spectra(sp, sp.copy(), axes=mock_axes) assert mock_axes.plot.call_count == 2
def make_plots(): """ Function that loads two spectra and plots them in the different subplots. """ fig = plt.figure(1) ax1 = fig.add_subplot(2, 1, 1) filename1 = 'sampledata/01-CD-Aviv62DS/CSA/CSA.CD' filename2 = 'sampledata/01-CD-Aviv62DS/CSA/blank.CD' imp1 = spp.AvivImporter(filename1) sp1 = spp.Spectrum(imp1.dataset) imp2 = spp.AvivImporter(filename2) sp2 = spp.Spectrum(imp2.dataset) spp.plot_spectra(ax1, [sp1, sp2]) ax2 = fig.add_subplot(2, 1, 2) filename1 = 'sampledata/02-CD-Mos500/csa.bka' filename2 = 'sampledata/02-CD-Mos500/p07-10tfe.bka' imp1 = spp.MosImporter(filename1) sp1 = spp.Spectrum(imp1.dataset) imp2 = spp.MosImporter(filename2) sp2 = spp.Spectrum(imp2.dataset) spp.plot_spectra(ax2, [sp1, sp2], '--o', color='g') plt.show()
# -*- coding: utf-8 -*- """ Created on Thu Mar 20 22:21:45 2014 @author: amaniada """ import spectraplotpy as spp import matplotlib.pyplot as plt filename1 = 'sampledata/01-CD-Aviv62DS/CSA/CSA.CD' imp1 = spp.AvivImporter(filename1) sp1 = spp.Spectrum(imp1.dataset) sp2 = sp1 * 2 sp_aver = spp.average_spectra([sp1, sp2]) spp.plot_spectra(plt, [sp1, sp2, sp_aver],) plt .title("Plotting test spectra") plt.xlabel("x") plt.ylabel("y") plt.legend( ("spectum 1", "spectrum 2", "average spectrum"),loc = 'upper left') plt.show() csve = spp.CSVExporter(sp_aver.dataset) csve.save('csa.csv')