ó
›ßÈ[c           @` sâ   d  d l  m Z m Z m Z m Z d  d l Z d  d l Z y d  d l j Z	 Wn e
 k
 rf e Z n Xe Z d d l m Z d d l m Z e j j d ƒ d „  ƒ Z e j j d ƒ d	 „  ƒ Z e j j d ƒ d
 „  ƒ Z d S(   i    (   t   absolute_importt   divisiont   print_functiont   unicode_literalsNi   (   t   unitsi   (   t   quantity_supportu   not HAS_PLTc       	   C` s9  t  j ƒ  t ƒ  t j ƒ  }  t  j d d d g t j d d d g t j d d ƒt  j d d	 d
 g t j	 d d d g t j
 ƒ t  j ƒ  t  j d d g t j d d g t j d d g t j
 ƒ t  j |  d d ƒt  j ƒ  j j ƒ  t j k st ‚ t  j ƒ  j j ƒ  t j k s%t ‚ Wd  QXt  j ƒ  d  S(   Ni   i   i   i   i   t   labelu   labelii   iÒ   i;  iê  iÑ  iÂ  t   formatu   svg(   t   pltt   figureR   t   iot   BytesIOt   plott   ut   mt   kgt   cmt   gt   legendt   fill_betweent   savefigt   gcat   xaxist	   get_unitst   AssertionErrort   yaxist   clf(   t   buff(    (    sE   lib/python2.7/site-packages/astropy/visualization/tests/test_units.pyt
   test_units   s    

60
:$*c       	   C` sá   t  j d d d ƒt j ƒ  t ƒ  « d d d g t j }  d d d g t j } d d d g t j } t j	 ƒ  \ } } | j
 |  | d | ƒ| j j ƒ  t j k s¯ t ‚ | j j ƒ  t j k sÍ t ‚ Wd  QXt j ƒ  d  S(   Nu
   matplotlibt
   minversionu   2.2i   i   i   t   yerr(   t   pytestt   importorskipR   R	   R   R   t   sR   R   t   subplotst   errorbarR   R   R   R   R   (   t   xt   yR   t   figt   ax(    (    sE   lib/python2.7/site-packages/astropy/visualization/tests/test_units.pyt   test_units_errbarr+   s    

$c           C` s{   t  j ƒ  t ƒ  X t  j d d d g t j ƒ t j t j ƒ " t  j d d d g t j	 ƒ Wd  QXWd  QXt  j
 ƒ  d  S(   Ni   i   i   ii   iÒ   i;  (   R   R	   R   R   R   R   R   t   raisest   UnitConversionErrorR   R   (    (    (    sE   lib/python2.7/site-packages/astropy/visualization/tests/test_units.pyt   test_incompatible_units>   s    

)(   t
   __future__R    R   R   R   R
   R   t   matplotlib.pyplott   pyplotR   t   ImportErrort   Falset   HAS_PLTt   Truet    R   R   R   t   markt   skipifR   R(   R+   (    (    (    sE   lib/python2.7/site-packages/astropy/visualization/tests/test_units.pyt   <module>   s   "
