U
    K?h                     @   s    d Z ddlmZmZ dd ZdS )zj
Limited tests of the visualization module. Right now it just makes
sure that passing custom Axes works.

    )mpfpc                  C   s.  zPdd l } | jdd }|dd d }dd |D ddgk rFtdd l}W n tk
rn   td Y d S X | }|d	}tt	fD ]<}|j
d
d ddg|d | dkst| dkstq| }|d	}tt	fD ]"}|jdd ddgddg|d q| dkst| dks*td S )Nr   -.   c                 S   s   g | ]}t |qS  )int).0_r   r   Q/var/www/html/venv/lib/python3.8/site-packages/mpmath/tests/test_visualization.py
<listcomp>   s     ztest_axes.<locals>.<listcomp>c   z8
Skipping test (pylab not available or too old version)
o   c                 S   s   | d S )Nr   r   )xr   r   r   <lambda>       ztest_axes.<locals>.<lambda>   )axesr   zf(x)c                 S   s   | S )Nr   )zr   r   r   r      r   i
   zRe(z)zIm(z))
matplotlib__version__splitImportErrorpylabprintZfigureZadd_subplotr   r   ZplotZ
get_xlabelAssertionErrorZ
get_ylabelZcplot)r   versionr   Zfigr   ctxr   r   r   	test_axes	   s,    

 r    N)__doc__Zmpmathr   r   r    r   r   r   r   <module>   s   