U
    Evfr  ã                   @   s   d dl Zd dlmZ d dlmZ d dlmZ d dl	m
Z
mZmZ d dlmZmZ edgddd	d
„ ƒZedgdddd„ ƒZdd„ Zedgdddddd„ ƒZedgddddd„ ƒZedgddddd„ ƒZedgddddd „ ƒZed!gddd"d#„ ƒZdS )$é    N)Úimage_comparison)ÚIdentityTransform)ÚAxesZeroÚSubplotZeroÚSubplot)ÚAxesÚSubplotHostzSubplotZero.pngÚdefault)Ústylec                  C   sš   dt jd< t  ¡ } t| dddƒ}|  |¡ |jd  d¡ |jd j d¡ dD ]}|j|  d¡ qPt	 
d	d
t	j d¡}| |t	 |¡¡ | d¡ d S )Né   útext.kerning_factoré   ÚxzeroTz	Axis Zero)ÚtopÚrightFr   é   ç{®Gáz„?ÚTest)ÚpltÚrcParamsÚfigurer   Úadd_subplotÚaxisÚset_visibleÚlabelZset_textÚnpÚarangeÚpiÚplotÚsinÚ
set_ylabel)ÚfigÚaxÚnÚxx© r%   úP/tmp/pip-unpacked-wheel-7vhvci0g/mpl_toolkits/axisartist/tests/test_axislines.pyÚtest_SubplotZero
   s    

r'   zSubplot.pngc                  C   s’   dt jd< t  ¡ } t| dddƒ}|  |¡ t ddtj d¡}| |t 	|¡¡ | 
d¡ |jd j d	¡ |jd
 j d	¡ |jd
  d¡ d S )Nr   r   r   r   r   r   r   r   TÚbottomZTk0)r   r   r   r   r   r   r   r   r   r   r    r   Zmajor_ticksZset_tick_outZ	set_label)r!   r"   r$   r%   r%   r&   Útest_Subplot   s    


r)   c                  C   sT   t  ¡ } t| ddddgƒ}|  |¡ | dddgdddg¡ | d	¡ | j ¡  d S )
Ng333333Ã?gš™™™™™¹?gÍÌÌÌÌÌä?gš™™™™™é?r   r   é   r   Úlog)r   r   r   Zadd_axesr   Z
set_xscaleZcanvasZdraw)r!   r"   r%   r%   r&   Ú	test_Axes3   s    

r,   z!ParasiteAxesAuxTrans_meshplot.pngTg333333³?)Úremove_textr
   Ztolc                  C   s:  dt jd< t d¡} d| d< d| dd d …f< d| dd d …f< d| d d …df< d| d d …df< t d¡}t d¡}t ||¡\}}d	d
dg}t  ¡ }t|ƒD ]ˆ\}}t|dd|d ƒ}	| 	|	¡ |	j
tƒ d d}
| d	¡rüt|
|ƒ||| d d…d d…f ƒ nt|
|ƒ||| ƒ |	 d¡ |	 d¡ qš|
j||| dd d S )NFzpcolormesh.snap)r   r   r   ©r   r   r   éþÿÿÿr   ZpcolorZ
pcolormeshZcontourfr   r*   )Zviewlim_modeéÿÿÿÿ)r   é   Úk)Úcolors)r   r   r   Zonesr   Zmeshgridr   Ú	enumerater   r   Zget_aux_axesr   Ú
startswithÚgetattrZset_xlimZset_ylimZcontour)ÚdataÚxÚyr$   ÚyyZ	funcnamesr!   ÚiÚnameZax1Zax2r%   r%   r&   Útest_ParasiteAxesAuxTrans<   s,    






$
r=   zaxisline_style.pngZmpl20)r-   r
   c                  C   sv   t jdd} | jtd}|jd  d¡ |jd  d¡ |jd  d¡ |jd  d¡ d	D ]}|j|  d
¡ q\d S )Nr.   ©Zfigsize©Z
axes_classr   ú-|>TÚyzeroú->©Úleftr   r(   r   F©r   r   r   r   r   Úset_axisline_styler   ©r!   r"   Ú	directionr%   r%   r&   Útest_axisline_style_   s    rI   zaxisline_style_size_color.pngc                  C   s|   t jdd} | jtd}|jd jdddd |jd  d	¡ |jd
  d¡ |jd
  d	¡ dD ]}|j|  d¡ qbd S )Nr.   r>   r?   r   r@   g       @Úr©ÚsizeZ	facecolorTrA   z->, size=1.5rC   FrE   rG   r%   r%   r&   Útest_axisline_style_size_colorl   s    rM   zaxisline_style_tight.pngc                  C   s„   t jdd} | jtd}|jd jdddd |jd  d	¡ |jd
  d¡ |jd
  d	¡ dD ]}|j|  d¡ qb|  ¡  d S )Nr.   r>   r?   r   r@   r1   ÚgrK   TrA   z
->, size=8rC   F)r   r   r   r   r   rF   r   Ztight_layoutrG   r%   r%   r&   Útest_axisline_style_tightz   s    rO   zsubplotzero_ylabel.pngc                  C   sZ   t  ¡ } | jdtd}|jddddd |jd }| d¡ |jd	  d
¡ | d¡ d S )Néo   r?   )éýÿÿÿé   r8   r9   )ZxlimZylimZxlabelZylabel)r   rA   TrC   FrB   )r   r   r   r   Úsetr   r   rF   )r!   r"   Z	zero_axisr%   r%   r&   Útest_subplotzero_ylabelŠ   s    

rT   )Znumpyr   Zmatplotlib.pyplotZpyplotr   Zmatplotlib.testing.decoratorsr   Zmatplotlib.transformsr   Z!mpl_toolkits.axisartist.axislinesr   r   r   Zmpl_toolkits.axisartistr   r   r'   r)   r,   r=   rI   rM   rO   rT   r%   r%   r%   r&   Ú<module>   s8   

	  ÿ
!
ÿ
ÿ
