σ
ίΘ[c           @` sΗ  d  d l  m Z m Z m Z d  d l Z d  d l Z d  d l Z d  d l Z d  d l	 j
 Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z d d	 l m Z d
 d l m Z d
 d l m Z d
 d l m Z e j j  e j j! e j j" e#  d   Z$ e d    Z% e d    Z& e j' j( d d d Z) e d    Z* e d    Z+ e d    Z, e d    Z- e j' j( d d d Z. e d    Z/ d   Z0 d   Z1 d S(   i    (   t   print_functiont   divisiont   absolute_importNi   (   t   units(   t   WCS(   t   fits(   t   SkyCoord(   t   catch_warnings(   t   ignore_matplotlibrci   (   t   WCSAxes(   t   get_coord_meta(   t   CurvedTransformt   datac          C` sB   t  j d d t t  j d d  }  t |  d d d d g  d  S(   Nt   axest   gridt   figsizei   gΉ?gι?(   i   i   (   t   pltt   rct   Truet   figureR	   (   t   fig(    (    sL   lib/python2.7/site-packages/astropy/visualization/wcsaxes/tests/test_misc.pyt   test_grid_regression   s    c         C` s  t  j d d  } t | d d d d g  } | j |  | j d d  d k sX t  | j d j d  d k sz t  | j d j d  d k s t  | j |  j d	  j	  | j d d  d
 k sΣ t  | j d j d  d k sυ t  | j d j d  d k st  d  S(   NR   i   gΉ?gι?i
   t    i    i   t   nothings   10.0 10.0 (world)s   10.0(   i   i   (
   R   R   R	   t   add_axest   format_coordt   AssertionErrort   coordst   savefigt   joint   strpath(   t   tmpdirR   t   ax(    (    sL   lib/python2.7/site-packages/astropy/visualization/wcsaxes/tests/test_misc.pyt   test_format_coord_regression#   s    """s  
NAXIS   =                    2
NAXIS1  =                  200
NAXIS2  =                  100
CTYPE1  = 'RA---MOL'
CRPIX1  =                  500
CRVAL1  =                180.0
CDELT1  =                 -0.4
CUNIT1  = 'deg     '
CTYPE2  = 'DEC--MOL'
CRPIX2  =                  400
CRVAL2  =                  0.0
CDELT2  =                  0.4
CUNIT2  = 'deg     '
COORDSYS= 'icrs    '
t   seps   
c         C` s―   t  j d d d d t t  } | j t j d	   | j j d d  t	 t
    } t  j |  j d  j  Wd  QXx | D] } t |  q Wt |  d k s« t  d  S(
   Ni   t
   projectionid   iΘ   t   colort   whites   test.pngi    (   id   iΘ   (   R   t   subplotR   t   TARGET_HEADERt   imshowt   npt   zerosR   R   R   t   RuntimeWarningR   R   R   t   printt   lenR   (   R   R    t   wst   w(    (    sL   lib/python2.7/site-packages/astropy/visualization/wcsaxes/tests/test_misc.pyt   test_no_numpy_warningsE   s    !c          C` s€   t  j d d d d t t  }  t j t   } |  j d  Wd  QX| j j	 d d k sb t
  t j t   } t d  Wd  QX| j j	 d d k s  t
  d  S(   Ni   R#   t   bananai    s   Unknown frame: banana(   R   R&   R   R'   t   pytestt   raisest
   ValueErrort   get_coords_overlayt   valuet   argsR   R
   (   R    t   exc(    (    sL   lib/python2.7/site-packages/astropy/visualization/wcsaxes/tests/test_misc.pyt   test_invalid_frame_overlayY   s    !c       
   C` sά   t  j j t d  }  t j j |   }  t j d d  } | j	 d d d d g d t
 |   d d } | j d	 d
  | j d	 d
  t d t j d t j  } t j t  $ | j | d d | j d  Wd  QXd  S(   Nt   2MASS_k_headerR   i   g333333Γ?gι?R#   t   aspectt   equalg      ΰΏg     @gS υΟ*|v@gΨS£>f3Ρ?t   ot	   transformt   galactic(   i   i   (   t   ost   pathR   t   DATAR   t   Headert   fromtextfileR   R   R   R   t   set_xlimt   set_ylimR   t   ut   degR2   R3   t	   TypeErrort
   plot_coordt   get_transform(   t   twoMASS_k_headerR   R    t   c(    (    sL   lib/python2.7/site-packages/astropy/visualization/wcsaxes/tests/test_misc.pyt   test_plot_coord_transformg   s    	c          C` s)  t  j d d d d t t  }  |  j d d d d d |  j d d d	 d d
 |  j d j j   d k su t	  |  j d j j
 d  d k s t	  |  j d j j   d k sΌ t	  |  j d j j   d k sή t	  |  j d j j
 d  d	 k st	  |  j d j j   d
 k s%t	  d  S(   Ni   R#   s   Test x labelt   labelpadi   R$   t   reds   Test y labeli   t   greeni    t   bt   l(   R   R&   R   R'   t
   set_xlabelt
   set_ylabelR   t
   axislabelst   get_textR   t
   get_minpadt	   get_color(   R    (    (    sL   lib/python2.7/site-packages/astropy/visualization/wcsaxes/tests/test_misc.pyt   test_set_label_propertiesx   s    !"%""%s{  
SIMPLE  =                    T / conforms to FITS standard
BITPIX  =                  -32 / array data type
NAXIS   =                    3 / number of array dimensions
NAXIS1  =                   31
NAXIS2  =                 2881
NAXIS3  =                  480
EXTEND  =                    T
CTYPE1  = 'DISTMOD '
CRVAL1  =                  3.5
CDELT1  =                  0.5
CRPIX1  =                  1.0
CTYPE2  = 'GLON-CAR'
CRVAL2  =                180.0
CDELT2  =               -0.125
CRPIX2  =                  1.0
CTYPE3  = 'GLAT-CAR'
CRVAL3  =                  0.0
CDELT3  =                0.125
CRPIX3  =                241.0
c         C` s¨  t  d d  } d d d g | j _ d d d g | j _ d d	 d
 g | j _ d d d g | j _ d d d g | j _ t j d t	  I } t j
   t j d d d d | d d t j |  j d  j  Wd  QXx | D] } t |  qά Wt |  d k st  t  t  } t j d t	  I } t j
   t j d d d d | d d t j |  j d  j  Wd  QXx | D] } t |  qxWt |  d k s€t  d  S(   Nt   naxisi   t   xt   yt   zRH   s   km/sg     4@g     Δ@iM  g      @i   g        g      π?t   recordi   R#   t   slicess   test.pngi    i   (   R\   R]   i   (   R\   R]   i   (   R   t   wcst   ctypet   cunitt   crpixt   cdeltt   crvalt   warningsR   R   t   resetwarningsR   R&   R   R   R   R,   R-   R   t
   GAL_HEADER(   R   t   wcs3dt   warning_linest   warning(    (    sL   lib/python2.7/site-packages/astropy/visualization/wcsaxes/tests/test_misc.pyt   test_slicing_warnings£   s*    

c         C` sJ   t  j d t    t  j d  t  j d  t  j |  j d  j  d  S(   NR#   s   Galactic Longitudes   Galactic Latitudes   test.png(   R   R&   R   t   xlabelt   ylabelR   R   R   (   R   (    (    sL   lib/python2.7/site-packages/astropy/visualization/wcsaxes/tests/test_misc.pyt   test_plt_xlabel_ylabelΝ   s    c         C` sΊ   d t  f d     Y} |   } i d d 6t j t j f d 6d d 6d d	 6} t j   } t | d
 d
 d d g d | d | } | j |  | j	 d d  | j
 |  j d  j  d  S(   Nt   CustomTransformc           B` s   e  Z d    Z RS(   c         S` s   | d S(   NgΝΜΜΜΜΜτ?(    (   t   selft   values(    (    sL   lib/python2.7/site-packages/astropy/visualization/wcsaxes/tests/test_misc.pyR>   β   s    (   t   __name__t
   __module__R>   (    (    (    sL   lib/python2.7/site-packages/astropy/visualization/wcsaxes/tests/test_misc.pyRq   έ   s   t   scalart   typet   unitt   wrapR\   R]   t   namegΉ?gι?R>   t
   coord_metat	   grid_typet   contourss   test.png(   Rv   Rv   (   NN(   R\   R]   (   R   RG   t   mt   st   NoneR   R   R	   R   R   R   R   R   (   R   Rq   R>   R{   R   R    (    (    sL   lib/python2.7/site-packages/astropy/visualization/wcsaxes/tests/test_misc.pyt!   test_grid_type_contours_transformΨ   s    	

(2   t
   __future__R    R   R   R@   Rg   R2   t   numpyR)   t   matplotlib.pyplott   pyplotR   R   R   RG   Ra   R   t   ioR   t   coordinatesR   t   tests.helperR   t   tests.image_testsR   t   coreR	   t   utilsR
   t
   transformsR   RA   t   abspathR   t   dirnamet   __file__RB   R   R!   RC   t
   fromstringR'   R0   R9   RN   RZ   Ri   Rm   Rp   R   (    (    (    sL   lib/python2.7/site-packages/astropy/visualization/wcsaxes/tests/test_misc.pyt   <module>   s8   -			*	