from matplotlib import pyplot as plt
import numpy as np
from astropy import units as u
from astropy.visualization import quantity_support

from m4opt.synphot.background import ZodiacalBackground

quantity_support()

wave = np.linspace(1000, 11000) * u.angstrom
ax = plt.axes()
for key in ['high', 'mid', 'low']:
    surf = getattr(ZodiacalBackground, key)()(wave, flux_unit=u.ABmag)
    ax.plot(wave, surf, label=f'ZodiacalBackground.{key}()')
ax.invert_yaxis()
ax.legend()