U
    \	¸av  ã                   @   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   úT/tmp/pip-unpacked-wheel-wjyw_3jo/mpl_toolkits/tests/test_axisartist_floating_axes.pyÚtest_subplot   s    
r   zcurvelinear3.pngÚdefaultg{®Gáz„?)ÚstyleZtolc                  C   sº  t jdd} t ¡  tjd d¡tj 	¡  }t
 d¡}t
 ¡ }tdddd	d
gƒ}t|d|||d d}t| d|d}|  |¡ d
}t ¡  dd| ¡| }tdddgƒ}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$¡ | |¡}d%dd&dgdd'd	dg }}| ||¡ | ||d(¡\}| |j¡ d S ))Nr   r
   é´   é   é   é   é   é   é   é
   )r   éh  r   é   ©ÚextremesÚgrid_locator1Úgrid_locator2Útick_formatter1Ztick_formatter2r   ©Úgrid_helperé   é<   é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   éK   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_helper2rF   ÚanÚax2ÚxxÚyyÚlr   r   r   Útest_curvelinear3   sX    ÿ
û
 ÿý
ÿ

rY   zcurvelinear4.pngg¸…ëQ¸Ž?c                  C   sn  dt jd< t jdd} t ¡  tjd d¡tj	 
¡  }t d¡}t ¡ }tdd	dd
dgƒ}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¡ | |¡}	dd d!d"gddd
d	g }
}|	 |
|¡ |	 |
|d#¡\}| |j¡ d S )$Nr   ztext.kerning_factorr   r
   r   r   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'   r;   ) r   ZrcParamsr   r<   r=   r>   r?   r@   rA   rB   rC   r   rD   rE   r   r   r   r   rF   r6   rG   rH   rI   rJ   Zset_axis_directionrK   rL   rM   rN   rO   rP   rQ   )r   rR   r"   r$   r#   r&   rS   rT   rF   rU   rV   rW   rX   r   r   r   Útest_curvelinear4K   sF    
ÿ
û

ÿ

r]   c                  C   sJ   t  ¡ } t| dƒ}|  |¡ |jdddd|jd< |jd jdksFt‚d S )Nr   r   r   r,   )Z	nth_coordÚvaluer2   Úy)r   r   r   r   rI   rF   Z_axis_directionÚAssertionErrorr   r   r   r   Útest_axis_direction{   s    

ÿra   )Znumpyr?   Zmatplotlib.pyplotZpyplotr   Zmatplotlib.projectionsZprojectionsrA   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   rY   r]   ra   r   r   r   r   Ú<module>   s   
5
/