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')
