import pytest
import numpy as np
from astropy import units as u
from astropy import convolution
from astropy.wcs import WCS
from astropy import wcs

from .test_spectral_cube import cube_and_raw
from ..cube_utils import largest_beam, smallest_beam

try:
    from radio_beam import beam, Beam
    RADIO_BEAM_INSTALLED = True
except ImportError:
    RADIO_BEAM_INSTALLED = False

try:
    import reproject
    REPROJECT_INSTALLED = True
except ImportError:
    REPROJECT_INSTALLED = False


@pytest.mark.skipif('not RADIO_BEAM_INSTALLED')
def test_largest_beam():

    cube, data = cube_and_raw('522_delta_beams.fits')

    large_beam = largest_beam(cube.beams)

    assert large_beam == cube.beams[2]


@pytest.mark.skipif('not RADIO_BEAM_INSTALLED')
def test_smallest_beam():

    cube, data = cube_and_raw('522_delta_beams.fits')

    small_beam = smallest_beam(cube.beams)

    assert small_beam == cube.beams[0]

Add tests for largest and smallest beams
Add tests for largest and smallest beams
