from spectractor.simulation.atmosphere import AtmosphereGrid
a = AtmosphereGrid(image_filename='tests/data/reduc_20170530_134_atmsim.fits', pwv_grid=[5, 5, 1],
                   ozone_grid=[400, 400, 1], aerosol_grid=[0.0, 0.1, 2])
atmospheric_grid = a.compute()
a.plot_transmission_image()