a
    ü
þdr  ã                   @   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%   úu/var/www/html/stable-diffusion-webui/venv/lib/python3.9/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Ú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   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   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   s8  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 ||¡\}}g d	¢}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   )ÚpcolorZ
pcolormeshZcontourfr   r*   )Zviewlim_moder0   éÿÿÿÿ)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    rJ   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   rA   g       @Úr©ÚsizeZ	facecolorTrB   z->, size=1.5rD   FrF   rH   r%   r%   r&   Útest_axisline_style_size_colorl   s    rN   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   rA   r2   ÚgrL   TrB   z
->, size=8rD   F)r   r   r   r   r   rG   r   Ztight_layoutrH   r%   r%   r&   Útest_axisline_style_tightz   s    rP   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@   )éýÿÿÿé   r9   r:   )ZxlimZylimZxlabelZylabel)r   rB   TrD   FrC   )r   r   r   r   Úsetr   r   rG   )r!   r"   Z	zero_axisr%   r%   r&   Útest_subplotzero_ylabelŠ   s    

rU   )Ú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>   rJ   rN   rP   rU   r%   r%   r%   r&   Ú<module>   s4   

	ÿ
!
ÿ
ÿ
