o
    -ñh¤  ã                   @   sœ   d dl mZ d dlmZ d dlmZ d dlmZm	Z	m
Z
mZmZ edgdddd	„ ƒZed
gdddd„ ƒZedgdddd„ ƒZedgdddd„ ƒZdS )é    N)Úimage_comparison)ÚAxisArtistHelperRectlinear)Ú
AxisArtistÚ	AxisLabelÚ	LabelBaseÚTicksÚ
TickLabelszaxis_artist_ticks.pngÚdefault)Ústylec                  C   s„   t  ¡ \} }|j d¡ |j d¡ dd„ tddƒD ƒ}td|jd}| |¡ | |¡ tddd	|jd
}| |¡ | |¡ d S )NFc                 S   s    g | ]}|d  df|d f‘qS )é
   g        é   © )Ú.0Úir   r   úb/var/www/vscode/kcb/lib/python3.10/site-packages/mpl_toolkits/axisartist/tests/test_axis_artist.pyÚ
<listcomp>   s     ztest_ticks.<locals>.<listcomp>éÿÿÿÿé   r   ©ÚticksizeÚaxisTÚC3)r   Útick_outÚcolorr   )	ÚpltÚsubplotsÚxaxisÚset_visibleÚyaxisÚranger   Úset_locs_anglesÚ
add_artist)ÚfigÚaxÚlocs_anglesÚticks_inÚ	ticks_outr   r   r   Ú
test_ticks	   s   


r'   zaxis_artist_labelbase.pngc                  C   sf   dt jd< t  ¡ \} }| dgdgd¡ tdddƒ}d|_d|_| d¡ |jdd	d
 | 	|¡ d S )Né   útext.kerning_factorç      à?ÚoÚTesté¦ÿÿÿé2   ÚcenterÚtop)ÚhaÚva)
r   ÚrcParamsr   Úplotr   Ú
_ref_angleÚ_offset_radiusÚset_rotationÚsetr!   )r"   r#   Úlabelr   r   r   Útest_labelbase   s   

r:   zaxis_artist_ticklabels.pngc                  C   sú   dt jd< t  ¡ \} }|j d¡ |j d¡ | ddgddgd¡ td|jd	}| |¡ d
dg}dd„ |D ƒ}| 	|¡ t
dd}||_| d¡ | |¡ | dgdgd¡ tdddƒ}d|_d|_| d¡ | |¡ | dd¡ | dd¡ d S )Nr(   r)   Fçš™™™™™É?çš™™™™™Ù?r*   r+   r   r   ))r;   r*   r-   z0.2))r<   r*   iˆÿÿÿz0.4c                 S   s   g | ]\}}}||d  f‘qS )é´   r   )r   ÚxyÚaÚlr   r   r   r   <   s    z#test_ticklabels.<locals>.<listcomp>Úleft)Úaxis_directionÚsr,   é   r   Úbottomé   )r   r3   r   r   r   r   r4   r   r!   r    r   Ú_locs_angles_labelsÚset_padr   r6   r5   Úset_axis_directionÚset_xlimÚset_ylim)r"   r#   ÚticksÚlocs_angles_labelsÚtick_locs_anglesÚ
ticklabelsÚ	axislabelr   r   r   Útest_ticklabels,   s0   

ÿ





rQ   zaxis_artist.pngc                  C   s   dt jd< t  ¡ \} }|j d¡ |j d¡ dD ]}tj||d}t||d |d}| 	|¡ q| 
d¡ |j d¡ |j d¡ | d	¡ d S )
Nr(   r)   F)rA   ÚrightrE   )Úloc)ÚoffsetrB   ÚTTTé   r,   )r   r3   r   r   r   r   r   ÚFixedr   r!   Ú	set_labelÚmajor_ticksÚset_tick_outr9   rH   Ú
set_ylabel)r"   r#   rS   ÚhelperÚaxisliner   r   r   Útest_axis_artistO   s   

r^   )Úmatplotlib.pyplotÚpyplotr   Úmatplotlib.testing.decoratorsr   Úmpl_toolkits.axisartistr   Ú#mpl_toolkits.axisartist.axis_artistr   r   r   r   r   r'   r:   rQ   r^   r   r   r   r   Ú<module>   s    


"