a
    fø1bQ  ã                   @   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)ÚFloatingSubplotÚGridHelperCurveLinear)ÚFixedLocator)Úangle_helperc                  C   s$   t jdd} t| dƒ}|  |¡ d S )N©é   r	   ©Zfigsizeéo   )ÚpltÚfigurer   Úadd_subplot©ÚfigZax© r   úOlib/python3.9/site-packages/mpl_toolkits/tests/test_axisartist_floating_axes.pyÚtest_subplot   s    
r   zcurvelinear3.pngÚdefaultr	   )ZstyleZtolc                  C   s¨  t jdd} t ¡  tjd d¡tj 	¡  }t
 d¡}t
 ¡ }tg d¢ƒ}t|d|||d d}t| d	|d
}|  |¡ d}t ¡  dd| ¡| }tg d¢ƒ}t|ddd| d| f|d}	|	jd|d |jd< }
|jd j d¡ |jd j d¡ dD ]}|j|  d¡ qü|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_formatter1Ztick_formatter2r   ©Úgrid_helperr   )é   é<   éZ   r   r   r   )r!   r#   Úright)ÚaxesÚleftúTest 1úTest 2)r,   r*   Fé   Úbottom©r+   Ú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   Ú
LocatorDMSÚFormatterDMSr   r   r   r   Znew_fixed_axisÚaxisr6   Úset_textÚset_visibleÚnew_floating_axisÚtoggleÚlineÚ	set_colorÚget_aux_axesÚscatterÚplotÚset_clip_pathÚpatch)r   Útrr"   r$   r#   r&   Úax1Zr_scaleZtr2Zgrid_helper2rH   ÚanÚax2ÚxxÚyyÚlr   r   r   Útest_curvelinear3   sV    ÿ
û
ÿý
ÿ

r[   zcurvelinear4.pnggÍÌÌÌÌÌì?c                  C   s`  dt jd< t jdd} t ¡  tjd d¡tj	 
¡  }t d¡}t ¡ }tg d¢ƒ}t|d	|||d d
}t| d|d}|  |¡ |jd j d¡ |jd j d¡ dD ]}|j|  d¡ q¨|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   r	   r   )éx   r'   r   r   r    r   r%   r,   r-   r*   r.   )ÚtopFéF   r0   r1   r3   Tr4   r]   r7   r8   r9   r<   r=   ) r   ZrcParamsr   r>   r?   r@   rA   rB   rC   rD   rE   r   rF   rG   r   r   r   r   rH   r6   rI   rJ   rK   rL   Zset_axis_directionrM   rN   rO   rP   rQ   rR   rS   )r   rT   r"   r$   r#   r&   rU   rV   rH   rW   rX   rY   rZ   r   r   r   Útest_curvelinear4O   sF    
ÿ
û

ÿ

r_   c                  C   sJ   t  ¡ } t| dƒ}|  |¡ |jdddd|jd< |jd jdksFJ ‚d S )Nr   r   r   r,   )Z	nth_coordÚvaluer2   Úy)r   r   r   r   rK   rH   Z_axis_directionr   r   r   r   Útest_axis_direction   s    

ÿrb   )ZnumpyrA   Zmatplotlib.pyplotZpyplotr   Zmatplotlib.projectionsZprojectionsrC   Zmatplotlib.transformsZ
transformsr>   Zmatplotlib.testing.decoratorsr   Z!mpl_toolkits.axisartist.axislinesr   Z%mpl_toolkits.axisartist.floating_axesr   r   Z#mpl_toolkits.axisartist.grid_finderr   Zmpl_toolkits.axisartistr   r   r[   r_   rb   r   r   r   r   Ú<module>   s   
7
/