from spectractor.fit.fit_spectrogram import SpectrogramFitWorkspace
file_name = 'tests/data/reduc_20170530_134_spectrum.fits'
atmgrid_file_name = file_name.replace('spectrum', 'atmsim')
fit_workspace = SpectrogramFitWorkspace(file_name, atmgrid_file_name=atmgrid_file_name, verbose=True)
A1, A2, aerosols, ozone, pwv, D, shift_x, shift_y, angle, *psf = fit_workspace.p
lambdas, model, model_err = fit_workspace.simulation.simulate(A1, A2, aerosols, ozone, pwv, D, shift_x,
                                                              shift_y, angle, psf)
fit_workspace.lambdas = lambdas
fit_workspace.model = model
fit_workspace.model_err = model_err
fit_workspace.plot_fit()