ó
›ßÈ[c           @   s·   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 e f d „  ƒ  YZ d e f d „  ƒ  YZ d S(   iÿÿÿÿNi   (   t   WCS(   t   remote_datai   (   t   WCSAxes(   t	   BaseFrame(   t   IMAGE_REFERENCE_DIRi   (   t   BaseImageTestst   HexagonalFramec           B   s   e  Z d  Z d „  Z RS(   t   abcdefc         C   sL  |  j  j ƒ  \ } } |  j  j ƒ  \ } } d | | } | | d } | | d d } t j | | g | | g f ƒ |  d _ t j | | g | | g f ƒ |  d _ t j | | g | | g f ƒ |  d _ t j | | g | | g f ƒ |  d _ t j | | g | | g f ƒ |  d _ t j | | g | | g f ƒ |  d	 _ d  S(
   Ng      à?g      @g      @t   at   bt   ct   dt   et   f(   t   parent_axest   get_xlimt   get_ylimt   npt   arrayt   data(   t   selft   xmint   xmaxt   ymint   ymaxt   ymidt   xmid1t   xmid2(    (    sM   lib/python2.7/site-packages/astropy/visualization/wcsaxes/tests/test_frame.pyt   update_spines   s    ((((((   t   __name__t
   __module__t   spine_namesR   (    (    (    sM   lib/python2.7/site-packages/astropy/visualization/wcsaxes/tests/test_frame.pyR      s   t	   TestFramec        
   B   s  e  Z e d  d ƒ e j j d e d d d d d i  ƒ d „  ƒ ƒ Z e d  d ƒ e j j d e d d	 d d d i  ƒ d
 „  ƒ ƒ Z e d  d ƒ e j j d e d d d d d i  ƒ d „  ƒ ƒ Z	 e d  d ƒ e j j d e d d d d d i  ƒ d „  ƒ ƒ Z
 d „  Z RS(   t   sourcet   astropyt   baseline_dirt   filenames   custom_frame.pngt	   tolerancei    t   stylec      
   C   s–  t  |  j ƒ } t j d d# ƒ } t | d d d d g d | d t ƒ} | j | ƒ | j j d d ƒ | j	 t
 j d$ ƒ d
 d d d d d d t j j ƒ} i  } d | d <| d <d | d <| d <| d <| d <| j d j d d | ƒ| j d j d ƒ | j d j d d | ƒ| j d j d ƒ | j d j d ƒ | j d j d ƒ | j d  d! ƒ | j d  d" ƒ | j | j j j ƒ | S(%   Nt   figsizei   g333333Ã?gffffffæ?t   wcst   frame_classt   colort   whitei•   t   vming        t   vmaxg       @t   origint   lowert   cmapi   R   R   g      @R	   R
   R   R   t   glont	   Longitudet   minpadt   adt   glatt   Latitudet   bcefg      @g      Y@g      [@(   i   i   (   i•   i•   (   R    t
   msx_headert   pltt   figureR   R   t   add_axest   coordst   gridt   imshowR   t   onest   cmt	   gist_heatt   set_axislabelt   set_axislabel_positiont   set_ticklabel_positiont   set_xlimt   set_ylimt   set_clip_patht   framet   patch(   R   R(   t   figt   axt   imR3   (    (    sM   lib/python2.7/site-packages/astropy/visualization/wcsaxes/tests/test_frame.pyt   test_custom_frame(   s,    	!"s    update_clip_path_rectangular.pngc         C   s­   t  j ƒ  } t | d d d d g d d ƒ} | j | ƒ | j d d ƒ | j d d ƒ | j | j d ƒ j ƒ | j	 t
 j d ƒ ƒ | j d
 d ƒ | j d
 d ƒ | S(   Ngš™™™™™¹?gš™™™™™é?t   aspectt   equalg        g       @t   nothingi   i   g      à¿g      @g      '@(   i   i   (   R9   R:   R   R;   RE   RF   t   savefigt   joint   strpathR>   R   t   zeros(   R   t   tmpdirRJ   RK   (    (    sM   lib/python2.7/site-packages/astropy/visualization/wcsaxes/tests/test_frame.pyt!   test_update_clip_path_rectangularR   s    !s#   update_clip_path_nonrectangular.pngc         C   s³   t  j ƒ  } t | d d d d g d d d t ƒ} | j | ƒ | j d d ƒ | j d d ƒ | j | j d ƒ j	 ƒ | j
 t j d ƒ ƒ | j d d ƒ | j d d ƒ | S(   Ngš™™™™™¹?gš™™™™™é?RN   RO   R)   g        g       @RP   i   i   g      à¿g      @g      '@(   i   i   (   R9   R:   R   R   R;   RE   RF   RQ   RR   RS   R>   R   RT   (   R   RU   RJ   RK   (    (    sM   lib/python2.7/site-packages/astropy/visualization/wcsaxes/tests/test_frame.pyt$   test_update_clip_path_nonrectangularj   s    	s   update_clip_path_change_wcs.pngc         C   s·   t  j ƒ  } t | d d d d g d d ƒ} | j | ƒ | j d d ƒ | j d d ƒ | j | j d ƒ j ƒ | j	 ƒ  | j
 t j d ƒ ƒ | j d
 d ƒ | j d
 d ƒ | S(   Ngš™™™™™¹?gš™™™™™é?RN   RO   g        g       @RP   i   i   g      à¿g      @g      '@(   i   i   (   R9   R:   R   R;   RE   RF   RQ   RR   RS   t	   reset_wcsR>   R   RT   (   R   RU   RJ   RK   (    (    sM   lib/python2.7/site-packages/astropy/visualization/wcsaxes/tests/test_frame.pyt    test_update_clip_path_change_wcsƒ   s    	!
c         C   s¤   t  j ƒ  } t | d d d d g ƒ } | j | ƒ | j j j d ƒ | j j j d ƒ | j ƒ  | j j j	 ƒ  d k s‚ t
 ‚ | j j j ƒ  d k s  t
 ‚ d  S(   Ngš™™™™™¹?gš™™™™™é?i   t   purple(   R9   R:   R   R;   R<   RH   t   set_linewidtht	   set_colorRX   t   get_linewidtht   AssertionErrort	   get_color(   R   RJ   RK   (    (    sM   lib/python2.7/site-packages/astropy/visualization/wcsaxes/tests/test_frame.pyt%   test_copy_frame_properties_change_wcs    s    
(   R   R   R   t   pytestt   markt   mpl_image_compareR   RM   RV   RW   RY   R`   (    (    (    sM   lib/python2.7/site-packages/astropy/visualization/wcsaxes/tests/test_frame.pyR    &   s"   '(   Ra   t   numpyR   t   matplotlib.pyplott   pyplotR9   R(   R    t   tests.helperR   t    R   RH   R   t   tests.image_testsR   t   test_imagesR   R   R    (    (    (    sM   lib/python2.7/site-packages/astropy/visualization/wcsaxes/tests/test_frame.pyt   <module>   s   