import matplotlib.pyplot as plt
import numpy as np
from astropy.modeling import models
import astropy.units as u
from specreduce.utils.synth_data import SynthImage

model_deg2 = models.Legendre1D(degree=2, c0=50, c1=0, c2=100)
im = (
    SynthImage(nx=3000, ny=1000)
    .add_background(5)
    .add_arcs(['HeI', 'ArI', 'ArII'], line_fwhm=3, tilt_func=model_deg2)
    .add_poisson_noise()
    .to_ccddata()
)
fig = plt.figure(figsize=(10, 6))
plt.imshow(im)