a
    s=ic}f                     @   s  d dl m Z  d dlZ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 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 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!dd Z"edgdd Z#dd Z$ej%&de'e(dfe'd d gd fe)d!d"fe)d#d$fe)d%d&fe)d'd(ffd)d* Z*ed+gd,d- Z+ed.gddd/d0 Z,ed1gd2d3 Z-d4d5 Z.ej%j&d6d7d8gd9d:gd;d<d= Z/d>d? Z0d@dA Z1dBdC Z2dDdE Z3dFdG Z4dHdI Z5ej%&dJg dKej%&dLg dMdNdO Z6edPgdQdR Z7edSgdTdU Z8edVgdWdX Z9dYdZ Z:ed[gd\d] Z;d^d_ Z<ed`gdadb Z=ej%&dcg dddedf Z>dgdh Z?didj Z@dkdl ZAedmdn ZBedodp ZCej%&dqg drdsdt ZDedugdvdw ZEedxgdydz ZFd{d| ZGd}d~ ZHedgddd ZIedgdddd ZJdd ZKdd ZLdd ZMedgddd ZNdd ZOdd ZPdd ZQej%&dddggdd ZRdd ZSedgdddd ZTdd ZUdd ZVdd ZWedgdddd ZXedgdddd ZYedd ZZdS )    )datetimeN)assert_almost_equal)
MouseEvent)FontProperties)check_figures_equalimage_comparison)needs_usetex)TextZfont_stylesc            	         s^   fdd} ddl m m tjddtdd t \}}| d	d
d
dd}|jddd|d}|	 dksnJ |
 d
ks~J | d
ksJ | d
ksJ | d
ksJ | dd
d
dddd}|jddd|d | ddd
dddd}|jddd|d | d	d
d
dddd}|jddd|d | d	d
d
dd dd}|jd!d"d|d |g  |g  d S )#Nc                     s(    f i | }|t  d} |dS )N)	directory)fname)mplZget_data_path)kwproppathr   findfont k/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/matplotlib/tests/test_text.pyfind_matplotlib_font   s    z.test_font_styles.<locals>.find_matplotlib_fontr   r   ignorez>findfont: Font family \[u?'Foo'\] not found. Falling back to .zmatplotlib.font_manager)modulez
sans-serifnormal   )familystylevariantsizezNormal Font)皙?r   axes fraction)xycoordsfontpropertieszDejaVu SansZFooboldi  )r   r   r   weightZstretchr   z	Bold Font)r   皙?z
sans serifitalici  zBold Italic Font)r   333333?   z
Light Font)r   皙?d   zCondensed Font)r         ?)matplotlib.font_managerr   r   warningsfilterwarningsUserWarningpltsubplotsannotateZget_fontnameZget_fontstyleZget_fontvariantZ
get_weightZget_stretch
set_xticks
set_yticks)	r   figaxZnormal_fontaZ	bold_fontZbold_italic_fontZ
light_fontZcondensed_fontr   r   r   test_font_styles   s    
r6   	multilinec                  C   s   t   t ddd} | d t jddddddd	 t jddd
dddd	 t jddddddd	 t dd t dd | g  | g  d S )N   zmultiline
text alignmentr#   r)   zTpTpTp
$M$
TpTpTp   centertop)r   havazTpTpTp
$M^{M^{M^{M}}}$
TpTpTpg?zTpTpTp
$M_{q_{q_{q}}}$
TpTpTpr   )	r.   figuresubplot	set_titletextZxlimZylimr1   r2   r4   r   r   r   test_multilinem   s$    

rC   Z
multiline2Zmpl20)r   c               	      s  dt jd< t  \} }|ddg |ddg |jdddd	 g d
}| j   fdd}d}t|D ].\}}|j	d| d d||dd}||| ql|j	ddddd |jdddd	 t|D ].\}}|j	d| d d||dd}||| q|j	ddddd |jdddd	 t|D ]0\}}|j	d| d d||dd}||| q |j	ddddd |jdddd	 t|D ]2\}}|j	d| d d||ddd}||| q||j	ddddd d S )N   text.kerning_factorr   gffffff?   r)   ZC2r%   )colorZ	linewidth)Linez2 Lineg
 2 Lgz$\sum_i x $zhi $\sum_i x $
testztest
 $\sum_i x $z$\sum_i x $
 $\sum_i x $c                    sB   t jdddd| jd}|| | j j | | d S )Nr   r   r8   F)clip_on	transform)	mpatchesZ	Rectangle	transAxesZ
set_boundsget_window_extentZtransformedinvertedZboundsZ	add_patch)r4   ttrrendererr   r   draw_box   s    
z!test_multiline2.<locals>.draw_boxleftr#   r   bottomhorizontalalignmentverticalalignmentg333333?zBottom alignrG   g?r;   z	Top aligng?baselinezBaseline alignr9   )rX   rY   rotationzBot align, rot20)
r.   rcParamsr/   set_xlimset_ylimZaxhlinecanvasget_renderer	enumeraterA   )r3   r4   stsrT   ZhoralnnstrP   r   rR   r   test_multiline2   sH    

	rf   zantialiased.pngc                  C   sB   dt jd< tjdd} | jdddddd	 | jdd
dddd	 d S )NTztext.antialiased)g      @      ?figsizer)   rg   Zantialiasedr:   rW         ?z
$\sqrt{x}$)r   r]   r.   r>   rA   r3   r   r   r   test_antialiasing   s    
rl   c                  C   s\   t jjddd} t| d}t j|}W d    n1 s<0    Y  |ddksXJ d S )NZ	Helveticaafm)ZfontextrbZVAVAVAVAVAVA)g     @i  )r   font_managerr   openZ_afmZAFMZstring_width_height)fnfhrm   r   r   r   test_afm_kerning   s    *rs   ztext_contains.pngc                  C   s   t  } t  }td| jdddd }tddd}tddd}t||\}}t jdddd	ddd
}| j	  t
|j|jD ]f\}}t  j||g\|_|_||\}}	|rdnd}
|j }|j||d|
d |j| q~d S )Nbutton_press_eventr)   r8   rj   rg      r'   zhello worldr:   )r<   fontsizer\   yellowredorZ   )r.   r>   Zaxesr   r`   npZlinspaceZmeshgridrA   drawzipZflatgcarM   rK   xycontainsZviewLimfrozenplotset)r3   r4   ZmeventZxsZystxtr~   r   r   _rG   Zvlr   r   r   test_contains   s"    

r   c                  C   sb   t  \} }|jdddddid}| j  td| jg|jdR  }||d	i fks^J d S )
NZhello)r'   r'   )333333?r   Z
arrowstylez->)xyxytextZ
arrowpropsrt   )r)   r   F)	r.   r/   r0   r`   r{   r   Z	transDatarK   r   )r3   r4   anneventr   r   r   test_annotation_contains   s    

r   zerr, xycoords, matchzUnknown return typez!Unknown coordinate type: \[0, 0\]fooz$'foo' is not a recognized coordinatezfoo barz('foo bar' is not a recognized coordinatez
offset fooz'xycoords cannot be an offset coordinatezaxes fooz'foo' is not a recognized unitc                 C   sZ   t  \}}tj| |d, |jddd|d |j  W d    n1 sL0    Y  d S )Nmatchr   rI   r)   r)   )r   r   )r.   r/   pytestraisesr0   r`   r{   )errr   r   r3   r4   r   r   r   test_annotate_errors   s    	r   titlesc                  C   sJ   t   t ddd} | jddd | jddd | g  | g  d S )Nr8   z
left titlerU   loczright titleright)r.   r>   r?   r@   r1   r2   rB   r   r   r   test_titles	  s    
r   Ztext_alignmentc                  C   s   t   t ddd} d}dD ]N}dD ]D}| j|d|d ||tdddd	d
 | j|dd||d |d7 }q&q| ddgddg | ddgddg | ddg | ddg | g  | 	g  d S )Nr8   r   )r   ru   )r;   rV   r[   r:   r)   z TjroundZwheat)boxstyleZ	facecoloralpha)r=   r\   bboxg      ?z$\sum_{i=0}^{j}$)r=   r\   r   g      ?)
r.   r>   r?   rA   dictr   r^   r_   r1   r2   )r4   r~   r\   	alignmentr   r   r   test_alignment  s&    

r   zaxes_titles.pngc                  C   sP   t   t ddd} | jddddd | jddddd | jd	d	ddd d S )
Nr8   r:   r9   i  )r   rv   Z
fontweightrU      i  r   )r.   r>   r?   r@   rB   r   r   r   test_axes_titles,  s
    r   c                  C   s  t  \} }|jddddd}| j  || jj}d}|||f | j  || jj}t|j	|j	D ]\}}|| |ksnJ qn|jddddd}| j  || jj}d}||f|_
| j  || jj}t|j	|j	D ]\}}|| |ksJ qd S )NtestrI   figure pixels)r   
textcoords   )r.   r/   r0   r`   r{   rN   rS   Zset_positionr|   minZxyann)r3   r4   r   Zinit_posZ	shift_valZpost_posr5   br   r   r   test_set_position6  s.    




r   rA    Oemptyz	non-empty)Zidsc                 C   sx   t  \}}|jdd| ddd}|j  |j}| }|j|d d}tjj	|
 |
 d dd |j|kstJ d S )	Nr)   rU   rV   )r<   r=   
   )dpi皙?Zrtol)r.   r/   rA   r`   r{   r   rN   rz   testingassert_allclose
get_points)rA   r3   r4   t1r   Zbbox1Zbbox2r   r   r   test_non_default_dpiV  s    
r   c                   C   s0   t dd dksJ t dd dks,J d S )N
horizontalr\           verticalg     V@r	   get_rotationr   r   r   r   test_get_rotation_stringf  s    r   c                  C   s$   dD ]} t | d | ksJ qd S )N)g      .@g333330@gYS@r   r   ir   r   r   test_get_rotation_floatk  s    r   c                  C   s(   dD ]} t | d t| ksJ qd S )N)C      )   r   )r	   r   floatr   r   r   r   test_get_rotation_intp  s    r   c                   C   s8   t t tdd W d    n1 s*0    Y  d S )NZ
hozirontalr   r   r   
ValueErrorr	   r   r   r   r   test_get_rotation_raisesu  s    r   c                   C   s   t d d dksJ d S )Nr   r   r   r   r   r   r   test_get_rotation_nonez  s    r   c                  C   s4   t g dg dD ]\} }tt| d | qd S )N)g     v@g     w@g	@)r   g      1@gfffff&f@r   )r|   r   r	   r   )r   jr   r   r   test_get_rotation_mod360~  s    r   r<   )r:   r   rU   r=   )r:   r;   rV   r[   Zcenter_baselinec                 C   sv   t  \}}td|| d}|jdddi|}|jdddi|}|j  t||jj	 ||jj	  d S )Nr   )r\   r=   r<   )r)   r)   r   Zrotation_modeanchordefault)
r.   r/   r   rA   r`   r{   r   rN   rS   r   )r<   r=   r3   r4   r   t0r   r   r   r   %test_null_rotation_with_rotation_mode  s    
r   Ztext_bboxclipc                  C   s8   t jdddddd t jddddd	} | d
di d S )Ng?r#   zIs bbox clipped?rQ   T)ZbackgroundcolorrJ   r)   zIs fancy bbox clipped?rJ   r   zround, pad=0.1)r.   rA   Zset_bbox)tr   r   r   test_bbox_clipping  s    r   z!annotation_negative_ax_coords.pngc               	   C   s   t  \} }|jdddgdddgddd |jdddgdddgddd	d
 |jdddgdddgddd |jdddgdddgddd	d
 |jdddgdddgddd |jdddgdddgddd	d
 d S )N+ ptsru   r9   zaxes points    r   r   r   r   rv   - ptsir;   r   r   r   r   rv   r=   + fracrg   r   r   - frac皙+ pixels      zaxes pixels- pixelsir.   r/   r0   r3   r4   r   r   r   "test_annotation_negative_ax_coords  s8    





r   z"annotation_negative_fig_coords.pngc               	   C   s   t  \} }|jdddgdddgddd |jddd	gddd	gddd
d |jdddgdddgddd |jdddgdddgddd
d |jdddgdddgddd |jdddgdddgddd
d d S )Nr   r   x   zfigure pointsr   r   r      r;   r   r   r   g?zfigure fractionr   r   r)   r   2   r   r   ir(   r   r   r   r   r   #test_annotation_negative_fig_coords  s8    





r   c                  C   s   t dd\} \}}t   |jr&J |jr0J | jr:J |ddd}|jsRJ |js\J | jsfJ |jdddgd}|jsJ |jsJ | jsJ t   |jrJ |jrJ | jrJ d S )Nr8   rF   r)   Zaardvarkr   )r.   r/   Zdraw_allstalerA   r0   )r3   ax1ax2Ztxt1Zann1r   r   r   test_text_stale  s"    










r   zagg_text_clip.pngc                  C   sZ   t jd td\} \}}t jddD ](\}}|j||ddd |||d q,d S )Nr8   rF   r   r   Tr   )rz   randomseedr.   r/   ZrandrA   )r3   r   r   r~   r   r   r   r   test_agg_text_clip  s
    r   c                  C   s>   dt jd< t jjdd} |  }dt jd< ||  ks:J d S )Nr   z	font.sizeZlarger   r(   )r   r]   ro   r   Zget_size_in_points)fpZsz1r   r   r   test_text_size_binding  s
    

r   zfont_scaling.pdfc                  C   s   dt jd< tjdd\} }|jt  |jt  |dd t	t
ddd	D ]&\}}|jd
|d dj|d|d qVd S )N*   pdf.fonttype)g@g(@rh   r   iX     +   rF   r   ru   z{fs} pt font size)fs)rv   )r   r]   r.   r/   xaxisZset_major_locatorZNullLocatoryaxisr_   rb   rangerA   format)r3   r4   r   r   r   r   r   test_font_scaling  s    
r   zspacing1, spacing2))r'   rF   )rF   r'   )rF   rF   c           	      C   s   d}t  }|j }|jdd|| d}|jdd||d}|j  |j|d}|j|d}|j|jkslJ | |kr|j|jksJ n|j|jksJ d S )Nzline1
line2rj   r)   ZlinespacingrR   )	r.   r>   r`   ra   rA   r{   rN   widthheight)	Zspacing1Zspacing2Ztext_stringr3   rS   text1text2Zbox1Zbox2r   r   r   test_two_2line_texts  s    

r   c                   C   s@   t t" tjddddd W d    n1 s20    Y  d S )Nrj   r)   r   abcr   )r   r   	TypeErrorr.   rA   r   r   r   r   test_validate_linespacing  s    r  c                  C   s:   t  \} }|dtjd |tjdd | j  d S )Nr   naninf)r.   r/   rA   rz   r  r  r`   r{   r   r   r   r   test_nonfinite_pos  s    r  c                  C   sj   dt jd< t  } | ddd}| jt dd | j}| jt dd t	j
j| j|dd	 d S )
Nr8   ztext.hinting_factorr)   z	some textsvgr   pngr   r   )r.   r]   r>   rA   savefigioBytesIOrN   Z	intervalxrz   r   r   )r3   r   expectedr   r   r   test_hinting_factor_backends%  s    

r  c                  C   sx   t  } dt jd< | d}dt jd< | d}| j  |df|dffD ](\}}|jjD ]}|j	 |ksZJ qZqJd S )NFtext.usetexy   Tz   )
r.   r>   r]   add_subplotr`   r{   r   
majorTickslabel1Z
get_usetex)r3   r   r   r4   usetexr   r   r   r   test_usetex_is_copied3  s    




r  c                  C   s(   t  } | jddddd | j  d S )Nr)   z	$\frac12$Tr  )r.   r>   rA   r`   r{   rk   r   r   r   test_single_artist_usetexB  s    r  fmt)r	  Zpdfr  c                 C   s:   dt jd< t  }|jddddd |jt | d d S )NTr  r)   Z2_2_2Fr  r  )r.   r]   r>   rA   r
  r  r  )r  r3   r   r   r   test_single_artist_usenotexL  s    
r  ztext_as_path_opacity.svgc                   C   sP   t   t    t jddddd t jddddd t jddd	dd
d d S )Nrj   cr   r   r   r)   rZ   r)   r5   r   rg   r~   r   r   r   r8   r   rG   )r.   r>   r}   set_axis_offrA   r   r   r   r   test_text_as_path_opacityX  s
    r!  ztext_as_text_opacity.svgc                   C   sZ   dt jd< t  t   tjddddd tjddddd	 tjdd
dddd d S )Nnonezsvg.fonttyperj   z50% using `color`r  rZ   r)   z50% using `alpha`r  rg   z"50% using `alpha` and 100% `color`r  r  )r   r]   r.   r>   r}   r   rA   r   r   r   r   test_text_as_text_opacitya  s    
r#  c                   C   s,   t ddgddg tt dgdd d S )NABr8   rF   r)   Boo)r.   r   reprrA   r   r   r   r   test_text_reprl  s    r(  c                  C   sf   t dd\} }|jddd}|| j }|   || j }tj|	 |	 ddrbJ d S )Nr8   
annotationr   r   gư>r   )
r.   r/   r0   rN   r`   ra   Ztight_layoutrz   Zallcloser   )r3   r4   anZextent1Zextent2r   r   r   test_annotation_updater  s    r+  r	  )
extensionsc                 C   sl   |   }|t dd |jdt dfdddd |  }|t dd |jdt dfdd	 d S )
Nr8   ry   r~   r)   )datar   rI   zoffset points)r   r   r   )r   )r  r   r   nowr0   Zfig_testZfig_refr4   r   r   r   test_annotation_units}  s    r0  zlarge_subscript_title.pngc                  C   s   dt jd< d t jd< t jddddd\} }|d	 }|d
 |jddd |g  |d }|jd
dd |jddd |g  d S )NrD   rE   zaxes.titleyr8   rF   )	   g      @T)ri   Zconstrained_layoutr   z$\sum_{i} x_i$zNew wayrU   r   g)\(?)r   zOld Way)r.   r]   r/   r@   Zset_xticklabels)r3   Zaxsr4   r   r   r   test_large_subscript_title  s    



r2  c                  C   s@   t jdd} d}| jdd|dd}| j  | dks<J d S )	NrD   r   rh   z?This is a very long text that should be wrapped multiple times.gffffff?r)   Twrapz?This is a very long
text that should be
wrapped multiple
times.r.   r>   rA   r`   r{   Z_get_wrapped_text)r3   srA   r   r   r   	test_wrap  s
    
r8  c                  C   s<   t jdd} | jddddd}| j  | dks8J d S )Nr3  rh   g      #@   ZAlonglineoftexttowrapTr4  r6  r3   rA   r   r   r   test_long_word_wrap  s    
r;  c                  C   s<   t jdd} | jddddd}| j  | dks8J d S )Nr3  rh   r   znon wrapped textTr4  r6  r:  r   r   r   test_wrap_no_wrap  s    
r<  c                 C   s`   |   }|ddg |ddg |jjd jd |  }|ddg |ddg d S )Nr   r8   u   €r5   wr   )r  r2   Zset_yticklabelsr   r  r  Z	set_colorr/  r   r   r   test_buffer_size  s    r>  c                  C   sL   t   t jdddd} t jdddd}|  dks8J | dksHJ dS )z>Test that kwargs take precedence over fontproperties defaults.valuezTimes New Romang      D@)r    r   counts)r   r    N)r.   r>   ZxlabelZylabelget_size)r   r   r   r   r   $test_fontproperties_kwarg_precedence  s
    rB  c                  C   s@   t  } t d}| jddd|dd}| }t|d d S )Nru   r   r   T)rK   Ztransform_rotates_text)r.   r}   mtransformsZAffine2DZ
rotate_degrA   r   r   )r4   rK   rA   resultr   r   r   test_transform_rotates_text  s    rE  c                  C   sV   t tddd d} t | }t }||  | d |d ks>J | d |d ksRJ d S )Nr!   )r"   )r    r   r    r   )r   r   r	   update)inpcacher   r   r   r   test_update_mutate_input  s    

rI  r\   zinvalid stringZ   c                 C   sB   t jtdd  tddd| d W d    n1 s40    Y  d S )Nz5rotation must be 'vertical', 'horizontal' or a numberr   r   r   r   r   r   r   r   r   test_invalid_rotation_values  s
    rK  c                   C   s@   t t" tjddddd W d    n1 s20    Y  d S )Nr)   r   Zfoobar)r  )r   r   r   r.   figtextr   r   r   r   test_invalid_color  s    rM  ztext_pdf_kerning.pdfc                   C   s   t   t jddddd d S )Nr   r)   ZATATATATATATATATATAru   r   )r.   r>   rL  r   r   r   r   test_pdf_kerning  s    rN  c                 C   sT   t  }|ddd |j  tdd | D s6J dd | D ddgksPJ d S )	Nr)   u   ০c                 s   s   | ]}t |jtV  qd S )N)
isinstancemessager-   .0warnr   r   r   	<genexpr>      z*test_unsupported_script.<locals>.<genexpr>c                 S   s   g | ]}|j jqS r   )rP  argsrQ  r   r   r   
<listcomp>  rU  z+test_unsupported_script.<locals>.<listcomp>)z>Glyph 2534 (\N{BENGALI DIGIT ZERO}) missing from current font.)z7Matplotlib currently does not support Bengali natively.)r.   r>   rA   r`   r{   all)Zrecwarnr3   r   r   r   test_unsupported_script  s    
rY  c                  C   sv   t  \} }|jddddd | j  |jddddd tjtdd | j  W d    n1 sh0    Y  d S )Nr   $ \wrong{math} $F)Z
parse_mathTUnknown symbolr   )r.   r/   rA   r`   r{   r   r   r   r   r   r   r   test_parse_math  s    
r\  c                  C   s   t  \} }|ddd tjtdd | j  W d    n1 sH0    Y  t	ddi4 t  \} }|ddd | j  W d    n1 s0    Y  d S )Nr   rZ  r[  r   ztext.parse_mathF)
r.   r/   rA   r   r   r   r`   r{   r   Z
rc_contextr   r   r   r   test_parse_math_rcparams  s    (r]  ztext_pdf_font42_kerning.pdfc                   C   s(   dt jd< t   t jddddd d S )Nr   r   r   r)   ZATAVATAVATAVATAVATAru   r   r.   r]   r>   rL  r   r   r   r   test_pdf_font42_kerning  s    
r_  ztext_pdf_chars_beyond_bmp.pdfc                   C   s2   dt jd< dt jd< t   t jddddd	 d S )
Nr   r   Zstixsanszmathtext.fontsetr   r)   u   Mass $m$ 𐌈ru   r   r^  r   r   r   r   test_pdf_chars_beyond_bmp  s    

r`  c                  C   s~   t jj  t } | ddd | ddd | jddddd | jddddd | j  t jj }|j	|j
kszJ d S )Nr%   r)   zfoo
barTr  )r   rA   Z!_get_text_metrics_with_cache_implcache_clearr.   r>   r`   r{   
cache_infohitsmisses)r3   infor   r   r   test_metrics_cache#  s    
rf  )[r   r  r+   numpyrz   Znumpy.testingr   r   Z
matplotlibr   Zmatplotlib.backend_basesr   r*   r   Zmatplotlib.patchesZpatchesrL   Zmatplotlib.pyplotZpyplotr.   Zmatplotlib.transformsZ
transformsrC  Zmatplotlib.testing.decoratorsr   r   Zmatplotlib.testing._markersr   Zmatplotlib.textr	   r6   rC   rf   rl   rs   r   r   markZparametrizeRuntimeErrorprintr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r  r  r  r  r  r!  r#  r(  r+  r0  r2  r8  r;  r<  r>  rB  rE  rI  rK  rM  rN  rY  r\  r]  r_  r`  rf  r   r   r   r   <module>   s   
X

3






	 





	



	








		





