o
    ˆ‚¡e»  ã                   @   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 d dlmZ d dlmZmZ d dlmZ d dlmZ dd„ Zed	gd
dddd„ ƒZedgd
dddd„ ƒZdd„ ZdS )é    N)Úimage_comparison)ÚSubplot)ÚFloatingAxesÚGridHelperCurveLinear)ÚFixedLocator)Úangle_helperc                  C   s$   t jdd} t| dƒ}|  |¡ d S )N©é   r	   ©Úfigsizeéo   )ÚpltÚfigurer   Úadd_subplot©ÚfigÚax© r   ú_/var/www/myenv/lib/python3.10/site-packages/mpl_toolkits/axisartist/tests/test_floating_axes.pyÚtest_subplot   s   
r   zcurvelinear3.pngÚdefaultr	   )ÚstyleÚtolc                  C   s„  t jdd} t ¡  tjd d¡tj 	¡  }t
|dt d¡tg d¢ƒt ¡ d d}| jt|d	}d
}t ¡  dd| ¡| }t
|ddd
| d| ftg d¢ƒd}|jd|d |jd< }|jd j d¡ |jd j d¡ |jd  d¡ |jdd|dd}||jd< |jddd |j d¡ |j d¡ |j d¡ | |¡}g d¢g d ¢}	}
| |	|
¡ | |	|
d!¡\}| |j¡ d S )"Nr   r
   é´   é   )r   éh  é
   é   é   ©é   é   é   é   r   ©ÚextremesÚgrid_locator1Úgrid_locator2Útick_formatter1Útick_formatter2©Ú
axes_classÚgrid_helperr   r   r   r   )é   é<   éZ   )r%   r'   Úright)ÚaxesÚleftúTest 1úTest 2)r2   r0   Fé   Úbottom©r1   Úaxis_directionÚzT©ÚallÚlabelúz = ?ú0.5©éC   r/   éK   r-   ©r    r	   r#   r!   úk-)r   r   ÚmtransformsÚAffine2DÚscaleÚnpÚpiÚmprojectionsÚ	PolarAxesÚPolarTransformr   r   Ú
LocatorDMSr   ÚFormatterDMSr   r   Únew_fixed_axisÚaxisr<   Úset_textÚset_visibleÚnew_floating_axisÚtoggleÚlineÚ	set_colorÚget_aux_axesÚscatterÚplotÚset_clip_pathÚpatch)r   Útrr,   Úax1Úr_scaleÚtr2Úgrid_helper2rO   Úax2ÚxxÚyyÚlr   r   r   Útest_curvelinear3   sJ   ÿ
ú
ý
ÿ

rd   zcurvelinear4.pnggÍÌÌÌÌÌì?c            	      C   sH  dt jd< t jdd} t ¡  tjd d¡tj	 
¡  }t|dt d¡tg d	¢ƒt ¡ d d
}| jt|d}| ¡  |jd j d¡ |jd j d¡ |jd  d¡ |jdd|dd}||jd< |jddd |j d¡ |j d¡ |j d¡ |j d¡ | |¡}g d¢g d¢}}| ||¡ | ||d¡\}| |j ¡ d S )Nr"   ztext.kerning_factorr   r
   r   r   )éx   r-   r   r   r	   r   r$   r*   r2   r3   r0   r4   ÚtopFéF   r6   r7   r9   Tr:   r=   r>   r?   rB   rC   )!r   ÚrcParamsr   rD   rE   rF   rG   rH   rI   rJ   rK   r   r   rL   r   rM   r   r   ÚclearrO   r<   rP   rQ   rR   rS   Úset_axis_directionrT   rU   rV   rW   rX   rY   rZ   )	r   r[   r,   r\   rO   r`   ra   rb   rc   r   r   r   Útest_curvelinear4D   s@   
ÿ
ú
ÿ

rk   c                  C   sJ   t  ¡ } t| dƒ}|  |¡ |jdddd|jd< |jd jdks#J ‚d S )Nr   r   r   r2   )Ú	nth_coordÚvaluer8   Úy)r   r   r   r   rR   rO   Ú_axis_directionr   r   r   r   Útest_axis_directionl   s   

ÿrp   )ÚnumpyrG   Úmatplotlib.pyplotÚpyplotr   Úmatplotlib.projectionsÚprojectionsrI   Úmatplotlib.transformsÚ
transformsrD   Úmatplotlib.testing.decoratorsr   Ú!mpl_toolkits.axisartist.axislinesr   Ú%mpl_toolkits.axisartist.floating_axesr   r   Ú#mpl_toolkits.axisartist.grid_finderr   Úmpl_toolkits.axisartistr   r   rd   rk   rp   r   r   r   r   Ú<module>   s    
-
'