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 make_2d_arc_image

model_deg2 = models.Legendre1D(degree=2, c0=50, c1=0, c2=100)
im = make_2d_arc_image(
    linelists=['HeI', 'ArI', 'ArII'],
    line_fwhm=3,
    tilt_func=model_deg2
)
fig = plt.figure(figsize=(10, 6))
plt.imshow(im)