ó
<Zc           @   s)   d  Z  d d l m Z m Z d   Z d S(   sj   
Limited tests of the visualization module. Right now it just makes
sure that passing custom Axes works.

i˙˙˙˙(   t   mpt   fpc          C   s˘  yy d d  l  }  |  j j d  d } | j d  d  } g  | D] } t |  ^ q? d d g k  rl t  n  d d  l } Wn t k
 r d GHd  SX| j   } | j d  } xc t t	 g D]U } | j
 d	   d d
 g d | | j   d k sř t  | j   d k sť t  qť W| j   } | j d  } x< t t	 g D]. } | j d   d d g d d g d | q<W| j   d k st  | j   d k st  d  S(   Ni˙˙˙˙t   -i    t   .i   ic   s8   
Skipping test (pylab not available or too old version)
io   c         S   s   |  d S(   Ni   (    (   t   x(    (    s>   lib/python2.7/site-packages/mpmath/tests/test_visualization.pyt   <lambda>   s    i   t   axesR   s   f(x)c         S   s   |  S(   N(    (   t   z(    (    s>   lib/python2.7/site-packages/mpmath/tests/test_visualization.pyR      s    iţ˙˙˙iö˙˙˙i
   s   Re(z)s   Im(z)(   t
   matplotlibt   __version__t   splitt   intt   ImportErrort   pylabt   figuret   add_subplotR    R   t   plott
   get_xlabelt   AssertionErrort
   get_ylabelt   cplot(   R   t   versiont   _R   t   figR   t   ctx(    (    s>   lib/python2.7/site-packages/mpmath/tests/test_visualization.pyt	   test_axes	   s,    +	,N(   t   __doc__t   mpmathR    R   R   (    (    (    s>   lib/python2.7/site-packages/mpmath/tests/test_visualization.pyt   <module>   s   