o
    ˆ‚¡e‡B  ã                   @   s¾  d dl Z d dlmZmZ d dlmZmZmZmZm	Z	m
Z
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Zd dlZdZdd	„ Zd
d„ Zdd„ Zdd„ Zej d¡dd„ ƒZdd„ Z ej !dg d¢¡dd„ ƒZ"dd„ Z#dd„ Z$dd„ Z%ej !dd d!g¡ej !d"d#d$g¡ej !d%d&ej&d'fd&ej'd(fd)ej&d*fd)ej'd+fg¡d,d-„ ƒƒƒZ(d.d/„ Z)d0d1„ Z*ej !d2d3d4dej+d5ed6g¡d7d8„ ƒZ,ej !d9g d:¢¡d;d<„ ƒZ-d=d>„ Z.d?d@„ Z/dAdB„ Z0dS )Cé    N)ÚpathÚ
transforms)ÚFigureCanvasBaseÚKeyEventÚLocationEventÚMouseButtonÚ
MouseEventÚNavigationToolbar2ÚRendererBase)ÚRubberbandBase©ÚFigure)Úneeds_pgf_xelatexzƒTreat the new Tool classes introduced in v[0-9]*.[0-9]* as experimental for now; the API and rcParam may change in future versions.c               	      sR  t  ¡ ‰ dd„ tddƒD ƒ} ‡ fdd„tddƒD ƒ}t d¡ d	¡}d
dg}d
dg}dd„ }|ˆ | ||||ƒ |ˆ | dd… ||||ƒ |ˆ g ||||ƒ |ˆ | |dd… |||ƒ |ˆ | g |||ƒ td|jd ƒD ]}|ˆ | ||d|…d d …f ||ƒ qk|ˆ | ||g |ƒ |ˆ | |||g ƒ |ˆ | ||g g ƒ |ˆ | |||dd… |ƒ d S )Nc                 S   s   g | ]}t j |¡‘qS © )r   ÚPathÚunit_regular_polygon©Ú.0Úir   r   úR/var/www/myenv/lib/python3.10/site-packages/matplotlib/tests/test_backend_bases.pyÚ
<listcomp>   s    z&test_uses_per_path.<locals>.<listcomp>é   é   c                    s   g | ]}ˆ   |¡ ¡  ¡ ‘qS r   )ÚrotateÚ
get_matrixÚcopyr   ©Úidr   r   r      s    é   é   é   )é
   é   ÚredÚgreenc                 S   s¢   t ƒ }t| | ||¡ƒ}| ¡ }dd„ | |tt|ƒƒ|t | ¡||g g dgg d¡D ƒ}	| 	|||||¡}
|rMt
j|	t|ƒd}t|ƒ |
d |
g¡sOJ ‚d S d S )Nc                 S   s   g | ]	\}}}}}|‘qS r   r   )r   ÚxoÚyoÚpath_idÚgc0ÚrgbFacer   r   r   r   $   s    z5test_uses_per_path.<locals>.check.<locals>.<listcomp>FÚscreen)Ú	minlengthr   )r
   ÚlistÚ_iter_collection_raw_pathsÚnew_gcÚ_iter_collectionÚrangeÚlenr   ÚAffineDeltaTransformÚ_iter_collection_uses_per_pathÚnpÚbincountÚsetÚissubset)Úmaster_transformÚpathsÚall_transformsÚoffsetsÚ
facecolorsÚ
edgecolorsÚrbÚ	raw_pathsÚgcÚidsÚusesÚseenr   r   r   Úcheck   s*   ÿüÿ
ÿþz!test_uses_per_path.<locals>.checkr   )r   ÚAffine2Dr0   r4   ÚarangeÚreshapeÚshape)r9   Útforms_matricesr;   r<   r=   rD   Únr   r   r   Útest_uses_per_path   s(   ÿrK   c                   C   s   t tƒ jtƒs	J ‚d S ©N)Ú
isinstancer   Úfigurer   r   r   r   r   Útest_canvas_ctor>   s   rO   c                   C   s   t  ¡ j ¡ dksJ ‚d S )Nz	image.png)ÚpltrN   ÚcanvasÚget_default_filenamer   r   r   r   Útest_get_default_filenameB   s   rS   c                  C   s.   t  ¡ } t| ƒ}t  | ¡ t  | j¡rJ ‚d S rL   )rP   rN   r   ÚcloseÚfignum_existsÚnumber)ÚfigrQ   r   r   r   Útest_canvas_changeF   s   
rX   Úpdfc                 C   sÎ   t  ¡  |  dd¡ t t¡}t  ¡  t|ƒdksJ ‚dt|d j	ƒv s'J ‚W d   ƒ n1 s1w   Y  t t¡"}t  
¡  ¡  t|ƒdksJJ ‚dt|d j	ƒv sUJ ‚W d   ƒ d S 1 s`w   Y  d S )NÚDISPLAYz:999r   z<FigureCanvasPdf is non-interactive, and thus cannot be shownr   )rP   ÚsubplotsÚsetenvÚpytestÚwarnsÚUserWarningÚshowr1   ÚstrÚmessageÚgcf)ÚmonkeypatchÚrecr   r   r   Útest_non_gui_warningO   s    
ÿý
ÿ"ýrf   c                  C   sD   t  ¡ \} }| j |¡ | jj|ksJ ‚|  ¡  | jjd u s J ‚d S rL   )rP   r[   rQ   Ú
grab_mouseÚmouse_grabberÚclear)rW   Úaxr   r   r   Útest_grab_clearb   s
   rk   zx, y))é*   é   )Nrl   )NN)éÈ   gq=
×£ Y@)g     ¸i@g       @c                 C   s
  t  ¡ \}}t|ƒ}td|| |ƒ}| d u r|jd u sJ ‚n|jt| ƒks&J ‚t|jtƒs.J ‚|d u r:|jd u s9J ‚n|jt|ƒksCJ ‚t|jtƒsKJ ‚| d ur|d urt 	d| 
| ¡› d| |¡› | | |¡¡skJ ‚dd„  |_|_t 	d| | |¡¡sƒJ ‚d S d S d S )NÚ
test_eventzx=z +y=c                 S   s   dS )NÚfoor   )Úxr   r   r   Ú<lambda>   s    z.test_location_event_position.<locals>.<lambda>zx=foo +y=foo)rP   r[   r   r   rq   ÚintrM   ÚyÚreÚmatchÚformat_xdataÚformat_ydataÚformat_coordÚ	fmt_xdataÚ	fmt_ydata)rq   rt   rW   rj   rQ   Úeventr   r   r   Útest_location_event_positionl   s(   
þûr}   c                     s¤   t  ¡ } | jddddddd | j ¡  g ‰ ‡ fdd„}| j d|¡ td	| jd
ƒ ¡  td| jg| j	 
d¡¢tj‘R Ž  ¡  td| jd
ƒ ¡  tˆ ƒdksPJ ‚d S )Nç      à?ÚhelloÚcenterT)ÚhaÚvaÚpickerc                    s   | j jdksJ ‚ˆ  | ¡ d S )NÚa)Ú
mouseeventÚkeyÚappend)r|   ©Úpicksr   r   Úhandle_pick‹   s   ztest_pick.<locals>.handle_pickÚ
pick_eventÚkey_press_eventr„   Úbutton_press_event©r~   r~   Úkey_release_eventr   )rP   rN   ÚtextrQ   ÚdrawÚmpl_connectr   Ú_processr   ÚtransFigureÚ	transformr   ÚLEFTr1   )rW   rŠ   r   rˆ   r   Ú	test_pick…   s   


ÿþr—   c                  C   sÎ  t  ¡ \} }|jdd | ¡ d u sJ ‚t| jƒ}| ¡  | ¡ dks%J ‚| ¡ }| ¡ }d}d}|j	 
|¡ t¡}|j	 
|¡ t¡}td| jg|¢tj‘R Ž }	| jj |	j|	¡ td| jg|¢tj‘R Ž }
| jj |
j|
¡ | ¡ |	j|
jfks{J ‚| ¡ |	j|
jfks‡J ‚td| jg|¢tj‘R Ž }	| jj |	j|	¡ td| jg|¢tj‘R Ž }
| jj |
j|
¡ | ¡ tj|dd	d
ksÁJ ‚| ¡ tj|dd	d
ksÏJ ‚| ¡  | ¡ d u sÛJ ‚| ¡ sã| ¡ råJ ‚d S )NÚlogit)ÚxscaleÚZOOM)gíµ ÷Æ°>çš™™™™™¹?)gwJëÿï?çš™™™™™é?r   Úbutton_release_eventr   g»½×Ùß|Û=)ÚrelÚabs)rP   r[   r6   Úget_navigate_moder	   rQ   ÚzoomÚget_xlimÚget_ylimÚ	transDatar•   Úastypers   r   r   r–   Ú	callbacksÚprocessÚnameÚxdataÚydataÚRIGHTr]   ÚapproxÚget_autoscalex_onÚget_autoscaley_on)rW   rj   ÚtbÚxlim0Úylim0Úd0Úd1Ús0Ús1Ústart_eventÚ
stop_eventr   r   r   Útest_interactive_zoom˜   sf   
ÿÿÿÿÿÿÿÿÿÿÿÿr¸   c                  C   sj   t  ¡ \} }| ddgddg¡ | j |¡ t| jƒ}| ¡  | ¡ d u s'J ‚| ¡  | ¡ d u s3J ‚d S )Nr   r   )	rP   r[   ÚplotrQ   Ú
widgetlockr	   r¡   r    Úpan)rW   rj   r¯   r   r   r   Útest_widgetlock_zoompanÈ   s   
r¼   Ú	plot_funcÚimshowÚcontourfÚorientationÚverticalÚ
horizontalztool,button,expectedr¡   ©é   é   )iìÿÿÿé   r»   )éþÿÿÿé   )g…ëQ¸…÷?g…ëQ¸@c                 C   s†  t  ¡ \}}t d¡ d¡}d\}}	t|| ƒ|||	d}
|j|
||d}| dkr1|j ¡ r/J ‚d S |j ¡ s8J ‚d\}}|df}|df}|d	krV|d d d
… }|d d d
… }|jj	 
|¡ t¡}|jj	 
|¡ t¡}td|jg|¢|‘R Ž }td|jg|¢|‘R Ž }t|jƒ}|dkrŸ| ¡  | |¡ | |¡ | |¡ n| ¡  | |¡ | |¡ | |¡ |j|jftj|ddksÁJ ‚d S )Né   )rÄ   r   ©r   r!   )ÚvminÚvmax)rj   rÀ   r¿   rÃ   r~   rÁ   éÿÿÿÿr   r   r¡   g333333Ã?©rŸ   )rP   r[   r4   rF   rG   ÚgetattrÚcolorbarrj   Úget_navigater¤   r•   r¥   rs   r   rQ   r	   r¡   Ú
press_zoomÚ	drag_zoomÚrelease_zoomr»   Ú	press_panÚdrag_panÚrelease_panrË   rÌ   r]   r¬   )r½   rÀ   ÚtoolÚbuttonÚexpectedrW   rj   ÚdataÚvmin0Úvmax0ÚcollÚcbrË   rÌ   r²   r³   r´   rµ   r¶   r·   r¯   r   r   r   Útest_interactive_colorbarÓ   sT   ÿÿÿÿÿÿ





"rà   c                  C   s˜   t jttd dtjd< W d   ƒ n1 sw   Y  t ¡ } |  ¡ d u s(J ‚| jj	j
j d¡ |  ¡ dks9J ‚| jj	j
j d¡ |  ¡ dksJJ ‚d S )N©rv   ÚtoolmanagerÚtoolbarr¡   rš   r»   ÚPAN)r]   r^   r_   Ú_EXPECTED_WARNING_TOOLMANAGERrP   ÚrcParamsÚgcar    rN   rQ   Úmanagerrâ   Útrigger_tool)rj   r   r   r   Útest_toolbar_zoompan  s   ÿrê   c                  C   s¬  t  ¡ \} }| tdƒtdƒ¡ t| jƒ}| ¡  td| jdddƒ ¡  td| jdddƒ ¡  | 	¡ | 
¡   kr>dksAJ ‚ J ‚td| jdddƒ ¡  td| jdddƒ ¡  | 	¡ | 
¡   krfdksiJ ‚ J ‚|j dd	g¡\}}td
| jg|¢tj‘R Ž  ¡  td| jg|¢tj‘R Ž  ¡  td| jdƒ ¡  | 	¡ | 
¡   kr©dks¬J ‚ J ‚td| jdddƒ ¡  td| jdddƒ ¡  | 	¡ | 
¡   krÑdksÔJ ‚ J ‚d S )Né   rŒ   Úkéd   Úl)r   r!   rÊ   )r"   r"   )r   r   r   r   Úh)rP   r[   r¹   r0   r	   rQ   r¡   r   r“   r¢   r£   r¤   r•   r   r   r–   )rW   rj   r¯   ÚstartÚstopr   r   r   Ú$test_toolbar_home_restores_autoscale  s$   
$$$(rò   ÚbackendÚsvgÚpsÚpgf)Úmarksc                 C   s  ddl m} ddlm} t d| › ¡}|j}|dd}||ƒ | dd¡}|dd}||ƒ | dd¡}dd	„ | ¡ D ƒ}	|j 	¡  |j 	¡  d
d	„ | ¡ D ƒ}
dd	„ | ¡ D ƒ}t
|	|
ƒD ]\}}tj||ddrmJ ‚q_t
||
ƒD ]\}}tjj||dd qsd S )Nr   r   )ÚFigureCanvaszmatplotlib.backends.backend_T)Úconstrained_layoutr"   c                 S   ó   g | ]}|  ¡ ‘qS r   ©Úget_position©r   rj   r   r   r   r   N  ó    ztest_draw.<locals>.<listcomp>c                 S   rú   r   rû   rý   r   r   r   r   S  rþ   c                 S   rú   r   rû   rý   r   r   r   r   T  rþ   g{®Gázt?)Úatol)Úmatplotlib.figurer   Úmatplotlib.backends.backend_aggrø   r]   Úimportorskipr[   ÚravelrQ   r‘   Úzipr4   ÚallcloseÚtestingÚassert_allclose)ró   r   rø   Útest_backendÚ
TestCanvasÚfig_testÚ	axes_testÚfig_aggÚaxes_aggÚinit_posÚlayed_out_pos_testÚlayed_out_pos_aggÚinitÚplacedÚrefÚtestr   r   r   Ú	test_draw9  s,   ÿ



ÿr  z&key,mouseend,expectedxlim,expectedylim))N©çš™™™™™É?r  ©gìQ¸…ë@g{®Gáú(@©gš™™™™™@gffffff'@)N)r  r~   r  ©r   é	   )N)r~   r  r  r  )NrŽ   r  r  )N©rœ   ç      Ð?©gÃõ(\ÂÀg…ëQ¸@©g      @g     €&@)N©r  r  r  r  )N)rœ   ç333333ë?r  ©g…ëQ¸	Àgq=
×£p@)N©r  r!  r  r"  )Úshift)r  çš™™™™™Ù?r  r  )r$  ©r%  r  r  r  )r$  r   r  r  )r$  r  r  )gÃõ(\Â@gq=
×£ð(@)r$  )rœ   gÍÌÌÌÌÌì?©g¤p=
×£Àg¤p=
×£@r'  )r$  r#  r  )gìQ¸…ëÀg
×£p=
@)rq   )r  r›   r  r  )rt   )r›   r  r  r  )Úcontrolr  r  r  )r(  r&  ©gÃõ(\Â@gq=
×£p'@r)  c                 C   s  t  ¡ \}}| t d¡¡ | ¡ sJ ‚| d¡ d}|j |¡ 	t
¡}|j |¡ 	t
¡}td|jg|¢R tj| dœŽ}	td|jg|¢R tj| dœŽ}
t|jƒ}| ¡  | |	¡ | |
¡ | |
¡ t| ¡ ƒtj|ddksrJ ‚t| ¡ ƒtj|ddksJ ‚d S )	Nr!   ÚequalrŽ   r   )rÙ   r†   r   g{®Gáz”?rÎ   )rP   r[   r¹   r4   rF   rÑ   Ú
set_aspectr¤   r•   r¥   rs   r   rQ   r   r–   r	   r»   rÕ   rÖ   r×   Útupler¢   r]   r¬   r£   )r†   ÚmouseendÚexpectedxlimÚexpectedylimrW   rj   Ú
mousestartÚsstartÚsendr¶   r·   r¯   r   r   r   Útest_interactive_pan]  s<   
ÿÿþÿÿþ



"r3  c                  C   s¨   t jttd dtjd< W d   ƒ n1 sw   Y  t ¡ } t| jj	j
jƒ}d| jj	j
jv s2J ‚| jj	j
 d¡ t| jj	j
jƒ|d ksHJ ‚d| jj	j
jvsRJ ‚d S )Nrá   râ   rã   Úforwardr   )r]   r^   r_   rå   rP   ræ   rc   r1   rQ   rè   râ   ÚtoolsÚremove_tool)rW   Úinitial_lenr   r   r   Útest_toolmanager_remove“  s   ÿr8  c                  C   s$  t jttd dtjd< W d   ƒ n1 sw   Y  t ¡ } | jjj	 
d¡}t|tƒs/J ‚| jjj	 
|¡|u s;J ‚t jtdd | jjj	 
d¡d u sOJ ‚W d   ƒ n1 sYw   Y  | jjj	j
dddd u slJ ‚t jtdd | jjj	 d¡d u s€J ‚W d   ƒ d S 1 s‹w   Y  d S )	Nrá   râ   rã   Ú
rubberbandz'ToolManager does not control tool 'foo'rp   F)Úwarn)r]   r^   r_   rå   rP   ræ   rc   rQ   rè   râ   Úget_toolrM   r   ré   )rW   r9  r   r   r   Útest_toolmanager_get_toolž  s$   ÿÿþÿ"þr<  c                  C   sø   t jttd dtjd< W d   ƒ n1 sw   Y  t ¡ } d| jjj	 
d¡v s,J ‚t jtdd | jjj	 dd¡ W d   ƒ n1 sGw   Y  | jjj	 
d¡dgksYJ ‚t jtdd | jjj	 d	d¡ W d   ƒ d S 1 suw   Y  d S )
Nrá   râ   rã   Úvr4  z"Key c changed from back to forwardÚcz'foo' not in Toolsrp   )r]   r^   r_   rå   rP   ræ   rc   rQ   rè   râ   Úget_tool_keymapÚupdate_keymapÚraisesÚKeyError)rW   r   r   r   Útest_toolmanager_update_keymap¯  s   ÿÿþ"ÿrC  )1ru   Ú
matplotlibr   r   Úmatplotlib.backend_basesr   r   r   r   r   r	   r
   Úmatplotlib.backend_toolsr   r   r   Úmatplotlib.testing._markersr   Úmatplotlib.pyplotÚpyplotrP   Únumpyr4   r]   rå   rK   rO   rS   rX   Úmarkró   rf   rk   Úparametrizer}   r—   r¸   r¼   r–   r«   rà   rê   rò   Úparamr  r3  r8  r<  rC  r   r   r   r   Ú<module>   sf    $ÿ(
	

ÿ
0



ýÿ6ÿÿ
 þ
!