from spectractor.extractor.chromaticpsf import ChromaticPSF
from spectractor.extractor.psf import MoffatGauss
psf = MoffatGauss()
s = ChromaticPSF(psf, Nx=100, Ny=20, deg=4, saturation=20000)
poly_params = s.generate_test_poly_params()
output = s.evaluate(s.set_pixels(mode="1D"), poly_params)
im = plt.imshow(output, origin='lower')
plt.colorbar(im)
plt.show()