IRIS Tests

"""Test cases for IRISMap.
This particular test file pertains to IRISMap.
@Author: Pritish C. (VaticanCameos)
"""

import os
import glob
import numpy as np

import pytest

from sunpy.map import Map
from sunpy.map.sources.iris import IRISMap
from sunpy.map.mapbase import GenericMap
import sunpy.data.test

path = sunpy.data.test.rootdir
fitspaths = glob.glob(os.path.join(path, "iris_l2_20130801_074720_4040000014_SJI_1400_t000.fits"))
irislist = []
for fits in fitspaths:
    irislist.append(Map(fits))

# IRIS Tests
def test_fitstoIRIS():
    """Tests the creation of IRISMap using FITS."""
    for amap in irislist:
        assert (isinstance(amap, IRISMap) or
                isinstance(amap, GenericMap))

def test_is_datasource_for():
    """Test the is_datasource_for method of IRISMap.
    Note that header data to be provided as an argument
    can be a MapMeta object."""
    for amap in irislist:
        if isinstance(amap, IRISMap):
            assert iris.is_datasource_for(iris.data, iris.meta)

def test_observatory():
    """Tests the observatory property of IRISMap."""
    for amap in irislist:
        if isinstance(amap, IRISMap):
            assert iris.observatory == "IRIS"