o
    ä¸Gf 1  ã                   @   sÊ  d dl Z d dlZd dlmZ d dlZd dlZd dlm	Z	 d dl
mZ d dlmZmZ d dlmZ dRdd„Ze	d	gd
ddd„ ƒZe	dgƒdd„ ƒZe	dgƒdd„ ƒZe	dgddddd„ ƒZe	dgƒdd„ ƒZe	dgƒdd„ ƒZe	d gd
dd!d"„ ƒZe	d#gƒd$d%„ ƒZe	d&gƒd'd(„ ƒZd)d*„ ZdSd.d/„Ze	d0d1gƒd2d3„ ƒZd4d5„ Zej  !d6d7d8g¡d9d:„ ƒZ"d;d<„ Z#d=d>„ Z$d?d@„ Z%dAdB„ Z&dCdD„ Z'ej  (dE¡dFdG„ ƒZ)dHdI„ Z*dJdK„ Z+dLdM„ Z,dNdO„ Z-dPdQ„ Z.dS )Té    N)Úassert_array_equal)Úimage_comparison)ÚAnchoredOffsetboxÚDrawingArea)Ú	Rectangleé   c                 C   sH   |   ddg¡ | jdd | jd|d | jd|d | jd|d d S )	Né   é   é   ©Znbinsúx-label©Úfontsizeúy-labelZTitle)ÚplotÚlocator_paramsÚ
set_xlabelÚ
set_ylabelÚ	set_title)Úaxr   © r   úAlib/python3.10/site-packages/matplotlib/tests/test_tightlayout.pyÚexample_plot   s
   r   Ztight_layout1gffffffþ?)Útolc                  C   s$   t  ¡ \} }t|dd t  ¡  dS )z'Test tight_layout for a single subplot.é   r   N©ÚpltÚsubplotsr   Útight_layout©Úfigr   r   r   r   Útest_tight_layout1   s   r!   Ztight_layout2c                  C   sJ   t jddd\} \\}}\}}t|ƒ t|ƒ t|ƒ t|ƒ t  ¡  dS )ú(Test tight_layout for multiple subplots.r	   )ZnrowsZncolsNr   )r    Úax1Úax2Úax3Úax4r   r   r   Útest_tight_layout2   s   r'   Ztight_layout3c                  C   sB   t  d¡} t  d¡}t  d¡}t| ƒ t|ƒ t|ƒ t  ¡  dS )r"   éÝ   éß   éz   N)r   Úsubplotr   r   )r#   r$   r%   r   r   r   Útest_tight_layout3)   s   


r,   Ztight_layout4)z2.5.5z2.6.1ç¸…ëQ¸Ž?)Zfreetype_versionr   c                  C   sj   t  dd¡} t jdddd}t jddddd}t jdddd	}t| ƒ t|ƒ t|ƒ t|ƒ t  ¡  d
S )z#Test tight_layout for subplot2grid.)r
   r
   ©r   r   )r   r   r	   )Úcolspan)r   r   ©r/   Úrowspan)r   r	   )r1   N)r   Úsubplot2gridr   r   )r#   r$   r%   r&   r   r   r   Útest_tight_layout45   s   r3   Ztight_layout5c                  C   s2   t  ¡ } t d¡ d¡}| j|dd t  ¡  dS )zTest tight_layout for image.éd   ©é
   r6   Znone)ZinterpolationN)r   r+   ÚnpÚarangeÚreshapeZimshowr   )r   Úarrr   r   r   Útest_tight_layout5D   s   r;   Ztight_layout6c            	   	   C   s`  t  ¡ ¢ t  dt¡ t ¡ } tj dd¡}|  	|d ¡}|  	|d ¡}t
|ƒ t
|ƒ |j| g d¢d tj dd¡}|D ]}|  	|¡}t
|ƒ | d¡ | d¡ q>|jd	d
d |j| g d¢dd t|j|jƒ}t|j|jƒ}|j| dd||j  dd|j|  gd |j| dd||j  dd|j|  gdd W d  ƒ dS 1 s©w   Y  dS )zTest tight_layout for gridspec.Úignorer	   r   r   )r   r   ç      à?r   ©Úrectr
   Ú r   r   r   )r=   r   r   r   gÍÌÌÌÌÌÜ?)r?   Zh_padNr=   )ÚwarningsÚcatch_warningsÚsimplefilterÚUserWarningr   ÚfigureÚmplZgridspecZGridSpecÚadd_subplotr   r   r   r   ÚminÚtopÚmaxÚbottom)	r    Zgs1r#   r$   Zgs2Ússr   rI   rK   r   r   r   Útest_tight_layout6M   s8   


ÿÿþ"ârM   Ztight_layout7c                  C   sr   d} t  ¡ \}}| ddg¡ |jdd |jd| d |jd| d |jd	d
| d |jdd| d t  ¡  d S )Nr   r   r	   r
   r   r   r   r   z
Left TitleÚleft)Úlocr   zRight TitleÚright)r   r   r   r   r   r   r   r   )r   r    r   r   r   r   Útest_tight_layout7w   s   rQ   Ztight_layout8c                  C   s6   t  ¡ } | jddd |  ¡ }t|dd |  ¡  dS )ú#Test automatic use of tight_layout.Ztightçš™™™™™¹?)ZlayoutÚpadr   r   N)r   rE   Zset_layout_enginerG   r   Údraw_without_renderingr   r   r   r   Útest_tight_layout8…   s
   rV   Ztight_layout9c                  C   s.   t  dd¡\} }|d d  d¡ t  ¡  d S )Nr	   r   F)r   r   Úset_visibler   )ÚfZaxarrr   r   r   Útest_tight_layout9   s   rY   c                  C   sŠ  t  ¡ } |  d¡}|jjdddd |jjdddd |jjdddddd	 |jjdddddd	 |jjd
gdd |jjd
gdd |  d¡}|jjdddd |jjdddd |  d¡}|jjdddd |jjdddd |  d¡}|jjdddd |jjdddd t  ¡  ddgddggddgddggddgddggddgddggg}t| j	ƒD ]\}}t
t | ¡  ¡ d¡|| ƒ q¯dS )rR   r(   Úouté   r
   )ÚtickdirÚlengthÚwidthé    TÚminor)r\   r]   r^   Ztick1OnÚwhichr   )r`   éÞ   Úinr)   Zinoutéà   g²ï§ÆK·?g•C‹lã?gƒÀÊ¡E¶Û?gB`åÐ"Ûí?gºI+‡â?g/Ý$í?gìQ¸…ëÁ?gmçû©ñÒÝ?N)r   rE   rG   ZxaxisZset_tick_paramsZyaxisZ	set_ticksr   Ú	enumerateÚaxesr   r7   ÚroundÚget_positionZ
get_points)r    r   ZansÚnnr   r   r   Útest_outward_ticks˜   s@   

ÿ
ÿ


ýÿÿrj   r6   rS   Úblackc              
   C   s°   |d| }}| | f| df| |f|dfd|f||fd| f|| fd| fg	}|D ](}t ||ƒ}td|||dddd}	| |	¡ td|d	d|| jd	d
}
|  |
¡ q-|
S )z&
    Surround ax with OffsetBoxes
    r   r=   r.   ÚNoner   F)r^   ÚheightZ	facecolorZ	edgecolorZ	linewidthZantialiasedÚcenterg        )rO   ÚchildrT   ZframeonZbbox_to_anchorZbbox_transformZ	borderpad)r   r   Z
add_artistr   Ú	transAxes)r   ÚsizeZmarginÚcolorÚmZmpZanchor_pointsZpointÚdaZ
backgroundZanchored_boxr   r   r   Úadd_offsetboxes¹   s4   þ
û
ùru   Ztight_layout_offsetboxes1Ztight_layout_offsetboxes2c                     sŠ   d ‰‰g d¢‰ ddg ‰‰‡ ‡‡‡‡fdd„} | ƒ }t  ¡  | ƒ }|ˆd d ˆ… D ]}| ¡ D ]}t|tƒr=| d¡ q1q+t  ¡  d S )Nr	   )ZredZblueZgreenZyellowr   r   c                     sL   t  ˆˆ¡\} }|j}t|ˆ ƒD ]\}}|jˆˆ|d t|d|d q|S )N)rr   é   )r   r   ZflatÚzipr   ru   )Ú_Úaxsr   rr   ©ZcolorsZcolsZrowsÚxÚyr   r   Ú	_subplotsé   s   z0test_tight_layout_offsetboxes.<locals>._subplotsF)r   r   Zget_childrenÚ
isinstancer   rW   )r}   ry   r   ro   r   rz   r   Útest_tight_layout_offsetboxes×   s   	

€þr   c                  C   s   t  ¡ } |  ¡  dS )zETest that tight layout doesn't cause an error when there are no axes.N)r   Zgcfr   )r    r   r   r   Útest_empty_layoutÿ   s   r€   ÚlabelZxlabelZylabelc                 C   s(   t jdd\}}|jdi | di¤Ž dS )z8Test that no warning emitted when xlabel/ylabel too big.©r
   r	   ©ZfigsizeZdaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaNr   )r   r   Úset)r   r    r   r   r   r   Útest_verybig_decorators  s   r…   c                  C   s4   t jdddd\} }|d  d¡ |d  d¡ dS )	z+Test that doesn't warn when xlabel too big.r   r	   r‚   rƒ   r   ZaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaZbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbN)r   r   r   ©r    ry   r   r   r   Útest_big_decorators_horizontal  ó   r‡   c                  C   s4   t jdddd\} }|d  d¡ |d  d¡ dS )	z+Test that doesn't warn when ylabel too big.r	   r   r‚   rƒ   r   ZaaaaaaaaaaaaaaaaaaaaZbbbbbbbbbbbbbbbbbbbbN)r   r   r   r†   r   r   r   Útest_big_decorators_vertical  rˆ   r‰   c                   C   sV   t  dd¡ t jddddd t t¡ t  ¡  W d   ƒ d S 1 s$w   Y  d S )N)é   é   r.   )r‹   r‹   )r   r
   r
   r‹   r0   )r   r2   ÚpytestÚwarnsrD   r   r   r   r   r   Útest_badsubplotgrid  s
   
"ÿrŽ   c                  C   sÐ   t jdd\} }| ddg¡ | ddg¡ |jddddd	 | ¡ }t t¡ t  	¡  | ¡ }|j
|j
ks9J ‚W d   ƒ n1 sCw   Y  t t¡ t j	g d
¢d W d   ƒ d S 1 saw   Y  d S )NT©r   r   r   zBIG LONG STRING)g      ô?r	   )g      %@g      ü?F)ZxyZxytextZannotation_clip)r   r   çš™™™™™é?r   r>   )r   r   Zset_xlimZset_ylimZannotaterh   rŒ   r   rD   r   r^   )r    r   Zp1Zp2r   r   r   Útest_collapsed#  s   
ÿý"ÿr‘   c                  C   sJ   t jdd\} }|  d¡}| d¡}| j ¡  | ¡ j| ¡ jks#J ‚d S )NTr   ZfooZbar)	r   r   Zsuptitler   ZcanvasZdrawZget_window_extentÚy0Zy1)r    r   ÚstÚtr   r   r   Útest_suptitle8  s
   


r•   Zpdfc                    s@   t jjj‰ ‡ fdd„}|  t jjd|¡ t ¡ \}}| ¡  d S )Nc                    s.   t | tjjjƒs
J ‚ˆ | g|¢R i |¤Ž d S )N)r~   rF   ZbackendsZbackend_pdfZRendererPdf)ÚselfÚargsÚkwargs©Zunpatched_initr   r   Ú__init__D  s   z'test_non_agg_renderer.<locals>.__init__rš   )rF   Zbackend_basesZRendererBaserš   Úsetattrr   r   r   )ZmonkeypatchZrecwarnrš   r    r   r   r™   r   Útest_non_agg_renderer@  s
   
rœ   c                  C   s¢   t  dd¡\} }|d jddgddgddgd}|d  ¡ }|  |jd |jd|jg¡}| j||d t	j
td	d
 |  ¡  W d   ƒ d S 1 sJw   Y  d S )Nr   r	   r   r‹   )Úcg{®Gázt?r-   )ÚcaxzThis figure includes Axes©Úmatch)r   r   Zscatterrh   Zadd_axesZx1r’   rm   ZcolorbarrŒ   r   rD   r   )r    rf   ZptsZax_rectrž   r   r   r   Útest_manual_colorbarO  s    ÿ
"ÿr¡   c            	      C   sæ   t  d¡ d¡} tjdd}|jddd}|jdd	d}|jd
dd}|||fD ]H}| d¡ | | d d …df ¡\}| | ¡}| 	¡ sFJ ‚| 	¡ sLJ ‚t
ddd|jd}| |¡ | | ¡ | ¡ ¡ | 	¡ rjJ ‚| 	¡ rpJ ‚q(d S )Nr4   r5   )é   r	   rƒ   éƒ   Zrectilinear)Z
projectioné„   Z	mollweideé…   ZpolarFr   r.   r=   )Z	transform)r7   r8   r9   r   rE   rG   Zgridr   ZpcolorZ_fully_clipped_to_axesr   rp   Zset_clip_pathZget_pathZget_transform)	r:   r    r#   r$   r%   r   Úhrs   r?   r   r   r   Útest_clipped_to_axes\  s"   


ôr§   c                  C   sT   t  ¡ \} }tjtdd |  ddi¡ W d   ƒ n1 sw   Y  |  ¡  d S )Nzwill be deprecatedrŸ   rT   ç333333Ã?)r   r   rŒ   r   ÚPendingDeprecationWarningÚset_tight_layoutrU   r   r   r   r   Útest_tight_padst  s   ÿþr«   c                  C   s    t jddid\} }|  ¡  d S )NrT   r¨   r   )r   r   rU   r   r   r   r   Útest_tight_kwargs|  s   r¬   c                  C   s~   t  ¡ \} }t t¡* |  d¡ |  ¡ sJ ‚|  d¡ |  ¡ r"J ‚|  d¡ |  ¡ s-J ‚W d   ƒ d S 1 s8w   Y  d S )NTF)r   r   rŒ   r   r©   rª   Zget_tight_layoutr   r   r   r   Útest_tight_toggle  s   


"úr­   )r   )r6   rS   rk   )/rA   Znumpyr7   Znumpy.testingr   rŒ   Z
matplotlibrF   Zmatplotlib.testing.decoratorsr   Zmatplotlib.pyplotZpyplotr   Zmatplotlib.offsetboxr   r   Zmatplotlib.patchesr   r   r!   r'   r,   r3   r;   rM   rQ   rV   rY   rj   ru   r   r€   ZmarkZparametrizer…   r‡   r‰   rŽ   r‘   r•   Zbackendrœ   r¡   r§   r«   r¬   r­   r   r   r   r   Ú<module>   sb    




ÿ


)

	

!

'
	

