a
    ضa^^                     @   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 ejje	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#dd Z$edgdd Z%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jj+d+d,d-gd.d/gd0d1d2 Z,d3d4 Z-d5d6 Z.d7d8 Z/d9d: Z0d;d< Z1d=d> Z2ej+d?g d@ej+dAg dBdCdD Z3edEgdFdG Z4edHgdIdJ Z5edKgdLdM Z6dNdO Z7edPgdQdR Z8dSdT Z9edUgdVdW Z:ej+dXg dYdZd[ Z;d\d] Z<d^d_ Z=ed`da Z>edbdc Z?ej+ddg dedfdg Z@edhgdidj ZAedkgdldm ZBdndo ZCdpdq ZDedrgdsdtdu ZEedvgdddwdx ZFdydz ZGd{d| ZHd}d~ ZIedrgdsdd ZJdd ZKdd ZLdd ZMdd ZNedgdddd ZOdd ZPdd ZQedgdddd ZRedgdddd ZSdS )    )datetimeN)assert_almost_equal)
MouseEvent)FontProperties)check_figures_equalimage_comparison)TextTz"This test needs a TeX installation)reasonZ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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 j/Users/vegardjervell/Documents/master/model/venv/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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annotate
set_xticks
set_yticks)r   figaxZnormal_fontZ	bold_fontZbold_italic_fontZ
light_fontZcondensed_fontr   r   r   test_font_styles   s    
r5   	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_multilinel   s$    

rB   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)Z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   r7   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rF   g?r:   z	Top aligng?baselinezBaseline alignr8   )rV   rW   rotationzBot align, rot20)
r.   rcParamsr/   set_xlimset_ylimZaxhlinecanvasget_renderer	enumerater@   )r3   r4   stsrR   ZhoralnnstrN   r   rP   r   test_multiline2   sH    

	rd   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)   re   Zantialiasedr9   rU         ?z
$\sqrt{x}$)r   r[   r.   r=   r@   r3   r   r   r   test_antialiasing   s    
rj   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   openrk   ZAFMZstring_width_height)fnfhrk   r   r   r   test_afm_kerning   s    *rq   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)   r7   rh   re      r'   zhello worldr9   )r;   fontsizerZ   yellowredorX   )r.   r=   Zaxesr   r^   npZlinspaceZmeshgridr@   drawzipZflatgcarK   rI   xycontainsZviewLimfrozenplotset)r3   r4   ZmeventxsZystxtr|   r}   r~   _rF   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
arrowpropsrr   )r)   r   F)	r.   r/   r0   r^   ry   r   Z	transDatarI   r~   )r3   r4   anneventr   r   r   test_annotation_contains   s    

r   titlesc                  C   sJ   t   t ddd} | jddd | jddd | g  | g  d S )Nr7   z
left titlerS   loczright titleright)r.   r=   r>   r?   r1   r2   rA   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 )Nr7   r   )r   rs   )r:   rT   rY   r9   r)   z TjroundZwheat)boxstyleZ	facecoloralpha)r<   rZ   bboxg      ?z$\sum_{i=0}^{j}$)r<   rZ   r   g      ?)
r.   r=   r>   r@   dictr   r\   r]   r1   r2   )r4   r|   rZ   	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 )
Nr7   r9   r8   i  )r   rt   Z
fontweightrS      i  r   )r.   r=   r>   r?   rA   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testrG   figure pixels)r   
textcoords   )r.   r/   r0   r^   ry   rL   rQ   Zset_positionrz   minZxyann)r3   r4   r   Zinit_posZ	shift_valZpost_posabr   r   r   test_set_position&  s.    




r   r@    Oemptyz	non-empty)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)   rS   rT   )r;   r<   
   )dpi皙?Zrtol)r.   r/   r@   r^   ry   r   rL   rx   testingassert_allclose
get_points)r@   r3   r4   t1r   Zbbox1Zbbox2r   r   r   test_non_default_dpiF  s    
r   c                   C   s@   t jddksJ t jddks(J t jddks<J d S )NZ
horizontal        Zverticalg     V@z15.      .@r   r@   get_rotationr   r   r   r   test_get_rotation_stringV  s    r   c                  C   s"   dD ]} t j| | ksJ qd S )N)r   g333330@gYS@r   ir   r   r   test_get_rotation_float\  s    r   c                  C   s&   dD ]} t j| t| ksJ qd S )N)C      )   )r   r@   r   floatr   r   r   r   test_get_rotation_inta  s    r   c                   C   s:   t t tjd W d    n1 s,0    Y  d S )NZ
hozirontal)pytestraises
ValueErrorr   r@   r   r   r   r   r   test_get_rotation_raisesf  s    r   c                   C   s   t jd dksJ d S )Nr   r   r   r   r   r   test_get_rotation_nonek  s    r   c                  C   s2   t g dg dD ]\} }ttj| | qd S )N)g     v@g     w@g	@)r   g      1@gfffff&f@)rz   r   r   r@   r   )r   jr   r   r   test_get_rotation_mod360o  s    r   r;   )r9   r   rS   r<   )r9   r:   rT   rY   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   )rZ   r<   r;   )r)   r)   r   Zrotation_modeanchordefault)
r.   r/   r   r@   r^   ry   r   rL   rQ   r   )r;   r<   r3   r4   r   t0r   r   r   r   %test_null_rotation_with_rotation_modet  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?rO   T)ZbackgroundcolorrH   r)   zIs fancy bbox clipped?rH   r   zround, pad=0.1)r.   r@   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+ ptsrs   r8   zaxes points    r   r   r   r   rt   - ptsir:   r   r   r   r   rt   r<   + fracre   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 )Nr7   rE   r)   Zaardvarkr   )r.   r/   Zdraw_allstaler@   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 )Nr7   rE   r   fooTr   )rx   randomseedr.   r/   Zrandr@   )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[   rm   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(@rf   r   iX     +   rE   r   rs   z{fs} pt font size)fs)rt   )r   r[   r.   r/   xaxisZset_major_locatorZNullLocatoryaxisr]   r`   ranger@   format)r3   r4   r   r   r   r   r   test_font_scaling  s    
r   zspacing1, spacing2))r'   rE   )rE   r'   )rE   rE   c           	      C   s   d}t  }|j }t jdd|| d}t 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
line2rh   r)   )ZlinespacingrP   )	r.   r=   r^   r_   r@   ry   rL   widthheight)	Zspacing1Zspacing2Ztext_stringr3   rQ   text1text2Zbox1Zbox2r   r   r   test_two_2line_texts  s    

r   c                  C   s:   t  \} }|dtjd |tjdd | j  d S )Nr   naninf)r.   r/   r@   rx   r   r   r^   ry   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 )
Nr7   ztext.hinting_factorr)   z	some textsvgr   pngr   r   )r.   r[   r=   r@   savefigioBytesIOrL   Z	intervalxrx   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^   ry   r   
majorTickslabel1Z
get_usetex)r3   r   r   r4   usetexr   r   r   r   test_usetex_is_copied  s    




r  c                  C   s(   t  } | jddddd | j  d S )Nr)   z	$\frac12$Tr
  )r.   r=   r@   r^   ry   ri   r   r   r   test_single_artist_usetex.  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=   r@   r   r   r  )r  r3   r   r   r   test_single_artist_usenotex8  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 )Nrh   cr   r   r   r)   rX   r)   r   r   re   r|   r   r   r   r7   r   rF   )r.   r=   r{   set_axis_offr@   r   r   r   r   test_text_as_path_opacityD  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.fonttyperh   z50% using `color`r  rX   r)   z50% using `alpha`r  re   z"50% using `alpha` and 100% `color`r  r  )r   r[   r.   r=   r{   r  r@   r   r   r   r   test_text_as_text_opacityM  s    
r  c                   C   s,   t ddgddg tt dgdd d S )NABr7   rE   r)   ZBoo)r.   r   reprr@   r   r   r   r   test_text_reprX  s    r  c                  C   sf   t dd\} }|jddd}|| j }|   || j }tj|	 |	 ddrbJ d S )Nr7   
annotation)r)   r)   r   gư>r   )
r.   r/   r0   rL   r^   r_   Ztight_layoutrx   Zallcloser   )r3   r4   ZanZextent1Zextent2r   r   r   test_annotation_update^  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 )
Nr7   rw   r|   r)   )datar   rG   zoffset points)r   r   r   )r   )r  r   r   nowr0   Zfig_testZfig_refr4   r   r   r   test_annotation_unitsi  s    r#  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 )NrC   rD   zaxes.titleyr7   rE   )	   g      @T)rg   Zconstrained_layoutr   z$\sum_{i} x_i$zNew wayrS   r   g)\(?)r}   zOld Way)r.   r[   r/   r?   Zset_xticklabels)r3   Zaxsr4   r   r   r   test_large_subscript_titleu  s    



r%  c                  C   s@   t jdd} d}| jdd|dd}| j  | dks<J d S )	NrC   r   rf   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=   r@   r^   ry   Z_get_wrapped_text)r3   sr@   r   r   r   	test_wrap  s
    
r+  c                  C   s<   t jdd} | jddddd}| j  | dks8J d S )Nr&  rf   g      #@   ZAlonglineoftexttowrapTr'  r)  r3   r@   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 )Nr&  rf   r   znon wrapped textTr'  r)  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   r7   u   €r   wr   )r  r2   Zset_yticklabelsr   r  r	  Z	set_colorr"  r   r   r   test_buffer_size  s    r1  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
    r5  c                  C   s@   t  } t d}| jddd|dd}| }t|d d S )Nrs   r   r   T)rI   Ztransform_rotates_text)r.   r{   mtransformsZAffine2DZ
rotate_degr@   r   r   )r4   rI   r@   resultr   r   r   test_transform_rotates_text  s    r8  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    

r<  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    r>  ztext_pdf_kerning.pdfc                   C   s   t   t jddddd d S )Nr   r)   ZATATATATATATATATATArs   r   )r.   r=   r=  r   r   r   r   test_pdf_kerning  s    r?  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   )rA  argsrB  r   r   r   
<listcomp>  rF  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=   r@   r^   ry   all)Zrecwarnr3   r   r   r   test_unsupported_script  s    
rJ  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   z$ \wrong{math} $F)Z
parse_mathTzUnknown symbol)match)r.   r/   r@   r^   ry   r   r   r   r   r   r   r   test_parse_math  s    
rL  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ATAVATAVATAVATAVATArs   r   r.   r[   r=   r=  r   r   r   r   test_pdf_font42_kerning  s    
rN  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$ 𐌈rs   r   rM  r   r   r   r   test_pdf_chars_beyond_bmp  s    

rO  )Tr   r   r+   Znumpyrx   Znumpy.testingr   r   Z
matplotlibr   Zmatplotlib.backend_basesr   r*   r   Zmatplotlib.patchesZpatchesrJ   Zmatplotlib.pyplotZpyplotr.   Zmatplotlib.transformsZ
transformsr6  Zmatplotlib.testing.decoratorsr   r   Zmatplotlib.textr   markZskipifZcheckdep_usetexZneeds_usetexr5   rB   rd   rj   rq   r   r   r   r   r   r   Zparametrizer   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/  r1  r5  r8  r<  r>  r?  rJ  rL  rN  rO  r   r   r   r   <module>   s   

S

3



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






		



