ó
 \c           @   sX  d  d l  Z d  d l Z d  d l j Z d  d l m Z d  d l m	 Z	 e
 Z y d  d l m Z Wn; e k
 r y d  d l Z Wq e k
 r e Z q Xn Xe r― d  d l j j j Z n  d   Z d   Z e	 j e d d d	    Z e	 j e d d d
    Z e	 j e d d d    Z e	 j e d d d    Z d   Z d S(   iĸĸĸĸN(   t   data_dir(   t   testing(   t   fitsc           C   sB   y t  j d  Wn t k
 r1 t s> t  n Xt s> t  d  S(   NR   (   t   iot
   use_plugint   ImportErrort   pyfits_availablet   AssertionError(    (    (    s9   lib/python2.7/site-packages/skimage/io/tests/test_fits.pyt   test_fits_plugin_import   s
    c           C   s   t  j   d  S(   N(   R   t   reset_plugins(    (    (    s9   lib/python2.7/site-packages/skimage/io/tests/test_fits.pyt   teardown"   s    t   reasons   pyfits not installedc          C   s\   t  j d  t j j t d  }  t  j |   } t j | t	 j
 |  d  k  sX t  d  S(   NR   s
   multi.fitsi   (   R   R   t   ost   patht   joinR    t   imreadt   npt   allt   pyfitst   getdataR   (   t   testfilet   img(    (    s9   lib/python2.7/site-packages/skimage/io/tests/test_fits.pyt   test_imread_MEF&   s    c          C   s\   t  j d  t j j t d  }  t  j |   } t j | t	 j
 |  d  k  sX t  d  S(   NR   s   simple.fitsi    (   R   R   R   R   R   R    R   R   R   R   R   R   (   R   R   (    (    s9   lib/python2.7/site-packages/skimage/io/tests/test_fits.pyt   test_imread_simple.   s    c          C   s}   t  j d  t j j t d  }  t  j |   } t  j |  d f |  d f |  d f g d t j	 } t
 | |  sy t  d  S(   NR   s
   multi.fitsi   i   i   t	   load_func(   R   R   R   R   R   R    t   imread_collectiont   ImageCollectiont   fplugt   FITSFactoryt   _same_ImageCollectionR   (   R   t   ic1t   ic2(    (    s9   lib/python2.7/site-packages/skimage/io/tests/test_fits.pyt!   test_imread_collection_single_MEF6   s    !c          C   sĄ   t  j d  t j j t d  }  t j j t d  } t  j |  | g  } t  j |  d f |  d f |  d f | d f g d t j	 } t
 | |  s t  d  S(	   NR   s
   multi.fitss   simple.fitsi   i   i   i    R   (   R   R   R   R   R   R    R   R   R   R   R   R   (   t	   testfile1t	   testfile2R   R   (    (    s9   lib/python2.7/site-packages/skimage/io/tests/test_fits.pyt%   test_imread_collection_MEF_and_simpleA   s    c         C   sY   t  |   t  |  k r t Sx6 t |  |  D]% \ } } t j | | k  s, t Sq, Wt S(   sx   Ancillary function to compare two ImageCollection objects, checking
       that their constituent arrays are equal.
    (   t   lent   Falset   zipR   R   t   True(   t   collection1t   collection2t   ext1t   ext2(    (    s9   lib/python2.7/site-packages/skimage/io/tests/test_fits.pyR   M   s    (   t   os.pathR   t   numpyR   t
   skimage.ioR   t   skimageR    t   skimage._sharedR   R'   R   t
   astropy.ioR   R   R   R%   t   skimage.io._plugins.fits_plugint   _pluginst   fits_pluginR   R   R
   t   skipifR   R   R    R#   R   (    (    (    s9   lib/python2.7/site-packages/skimage/io/tests/test_fits.pyt   <module>   s*   		