o
    -ñh	  ã                   @   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¡ q(t	 
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ÚlabelÚset_textÚnpÚarangeÚpiÚplotÚsinÚ
set_ylabel)ÚfigÚaxÚnÚxx© r&   ú`/var/www/vscode/kcb/lib/python3.10/site-packages/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ÚbottomÚTk0)r   r   r   r   r   r   r   r   r   r    r!   r   Úmajor_ticksÚset_tick_outÚ	set_label)r"   r#   r%   r&   r&   r'   Útest_Subplot   s   


r.   c                  C   sL   t  ¡ } t| g d¢ƒ}|  |¡ | g d¢g d¢¡ | d¡ | j ¡  d S )N)g333333Ã?gš™™™™™¹?gÍÌÌÌÌÌä?gš™™™™™é?)r   r   é   )r   r   r   Úlog)r   r   r   Úadd_axesr   Ú
set_xscaleÚcanvasÚdraw)r"   r#   r&   r&   r'   Ú	test_Axes3   s   

r5   z!ParasiteAxesAuxTrans_meshplot.pngTg333333³?)Úremove_textr
   Útolc                  C   s.  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  ||¡\}}g d¢}t ¡ }t|ƒD ]D\}}t|dd	|d ƒ}	| |	¡ |	j	t
ƒ d d
}
| d¡rxt|
|ƒ||| d d…d d…f ƒ n	t|
|ƒ||| ƒ |	 d¡ |	 d¡ qG|
j||| dd d S )N)r   r   r   ©r   r   r   éþÿÿÿr   )ÚpcolorÚ
pcolormeshÚcontourfr   r/   )Úviewlim_moder:   éÿÿÿÿ)r   é   Úk)Úcolors)r   Úonesr   Úmeshgridr   r   Ú	enumerater   r   Úget_aux_axesr   Ú
startswithÚgetattrÚset_xlimÚset_ylimÚcontour)ÚdataÚxÚyr%   ÚyyÚ	funcnamesr"   ÚiÚnameÚax1Úax2r&   r&   r'   Útest_ParasiteAxesAuxTrans<   s*   




$
rT   zaxisline_style.pngÚmpl20)r6   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 )Nr8   ©Úfigsize©Ú
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   ÿrc   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¡ q1d S )Nr8   rV   rX   r   rZ   g       @Úr©ÚsizeÚ	facecolorTr[   z->, size=1.5r]   Fr_   ra   r&   r&   r'   Útest_axisline_style_size_colori   s   ÿrh   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¡ q1|  ¡  d S )Nr8   rV   rX   r   rZ   r?   Úgre   Tr[   z
->, size=8r]   F)r   r   r   r   r   r`   r   Útight_layoutra   r&   r&   r'   Útest_axisline_style_tightw   s   rk   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   rX   )éýÿÿÿé   rL   rM   )ÚxlimÚylimÚxlabelÚylabel)r   r[   Tr]   Fr\   )r   r   r   r   Úsetr   r   r`   )r"   r#   Ú	zero_axisr&   r&   r'   Útest_subplotzero_ylabel‡   s   

ru   )Únumpyr   Úmatplotlib.pyplotÚpyplotr   Úmatplotlib.testing.decoratorsr   Úmatplotlib.transformsr   Ú!mpl_toolkits.axisartist.axislinesr   r   r   Úmpl_toolkits.axisartistr   r   r(   r.   r5   rT   rc   rh   rk   ru   r&   r&   r&   r'   Ú<module>   s6    

	ÿ

ÿ
ÿ
