ó
›ßÈ[c           @   s©   d  d l  Z  d  d l Z d d l m Z y; d  d l Z d  d l j Z e	 Z
 d d l m Z m Z Wn e k
 r| e Z
 n Xe  j j d ƒ d e f d „  ƒ  Yƒ Z d S(	   iÿÿÿÿNi   (   t   fitsi   (   t   fits2bitmapt   mains   not HAS_MATPLOTLIBt   TestFits2Bitmapc           B   sb   e  Z d  „  Z d „  Z d „  Z d „  Z d „  Z e j j	 d d d g ƒ d „  ƒ Z
 d	 „  Z RS(
   c         C   s   d |  _  d  S(   Ns	   test.fits(   t   filename(   t   self(    (    sS   lib/python2.7/site-packages/astropy/visualization/scripts/tests/test_fits2bitmap.pyt   setup_class   s    c         C   s<   | j  |  j ƒ j } t j | t j d ƒ ƒ t | ƒ d  S(   Ni€   (   i€   i€   (   t   joinR   t   strpathR    t   writetot   npt   onesR   (   R   t   tmpdirR   (    (    sS   lib/python2.7/site-packages/astropy/visualization/scripts/tests/test_fits2bitmap.pyt   test_function   s    c         C   sE   | j  |  j ƒ j } t j | t j d ƒ ƒ t | d d g ƒ d  S(   Ni€   s   -et   0(   i€   i€   (   R   R   R   R    R	   R
   R   R   (   R   R   R   (    (    sS   lib/python2.7/site-packages/astropy/visualization/scripts/tests/test_fits2bitmap.pyt   test_script   s    c         C   sx   | j  |  j ƒ j } t j d ƒ } t j ƒ  } t j | ƒ } t j | | g ƒ } | j	 | ƒ t
 | d d g ƒ d  S(   Nid   s   -et   1(   id   id   (   R   R   R   R
   R   R    t
   PrimaryHDUt   ImageHDUt   HDUListR	   R   (   R   R   R   t   datat   hdu1t   hdu2t   hdulist(    (    sS   lib/python2.7/site-packages/astropy/visualization/scripts/tests/test_fits2bitmap.pyt   test_exten_num    s    c         C   s‹   | j  |  j ƒ j } t j d ƒ } t j ƒ  } d } t j | ƒ } | | j d <t j	 | | g ƒ } | j
 | ƒ t | d | g ƒ d  S(   Nid   t   SCIt   EXTNAMEs   -e(   id   id   (   R   R   R   R
   R   R    R   R   t   headerR   R	   R   (   R   R   R   R   R   t   extnameR   R   (    (    sS   lib/python2.7/site-packages/astropy/visualization/scripts/tests/test_fits2bitmap.pyt   test_exten_name)   s    t
   file_extens   .gzs   .bz2c         C   sF   | j  d | ƒ j } t j | t j d ƒ ƒ t | d d g ƒ d  S(   Ns	   test.fitsi€   s   -eR   (   i€   i€   (   R   R   R    R	   R
   R   R   (   R   R   R   R   (    (    sS   lib/python2.7/site-packages/astropy/visualization/scripts/tests/test_fits2bitmap.pyt   test_compressed_fits4   s    c         C   sÀ   | j  |  j ƒ j } d } | j  | ƒ j } t j d ƒ } d | d d … d d … f <t j | | ƒ t | d d d	 | g ƒ t j	 | ƒ } | d d k s¦ t
 ‚ | d d k s¼ t
 ‚ d S(   sQ   
        Regression test to check the image vertical orientation/origin.
        s   fits2bitmap_test.pngi    g      ð?i    i   Ns   -eR   s   -oi   i   (   i    i    (   i    i    i    (   i   i   i    (   R   R   R   R
   t   zerosR    R	   R   t   mpimgt   imreadt   AssertionError(   R   R   R   t   out_filenameR   t   img(    (    sS   lib/python2.7/site-packages/astropy/visualization/scripts/tests/test_fits2bitmap.pyt   test_orientation:   s    (   t   __name__t
   __module__R   R   R   R   R   t   pytestt   markt   parametrizeR   R&   (    (    (    sS   lib/python2.7/site-packages/astropy/visualization/scripts/tests/test_fits2bitmap.pyR      s   						$(   R)   t   numpyR
   t   ioR    t
   matplotlibt   matplotlib.imaget   imageR!   t   Truet   HAS_MATPLOTLIBR   R   t   ImportErrort   FalseR*   t   skipift   objectR   (    (    (    sS   lib/python2.7/site-packages/astropy/visualization/scripts/tests/test_fits2bitmap.pyt   <module>   s   
