o
    gY                     @   sh  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 ZOdS )    )datetimeN)assert_almost_equal)
MouseEvent)FontProperties)check_figures_equalimage_comparison)TextTz"This test needs a TeX installation)reason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(    di | }|t  d} |dS )N)	directory)fname )mplget_data_path)kwproppathr   findfontr   X/home/ubuntu/cloudmapper/venv/lib/python3.10/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fontpropertiesFooboldi  )r   r   r   weight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axnormal_font	bold_fontbold_italic_font
light_fontcondensed_fontr   r   r   test_font_styles   s   
r>   	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   )	r2   figuresubplot	set_titletextxlimylimr5   r6   r8   r   r   r   test_multilinel   s$   

rM   
multiline2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}||| q6|j	ddddd |jdddd	 t|D ]\}}|j	d| d d||dd}||| qc|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 ]\}}|j	d| d d||ddd}||| q|j	ddddd d S )N   text.kerning_factorr   gffffff?   r-   C2r)   )color	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   r@   F)clip_on	transform)	mpatches	Rectangle	transAxes
set_boundsget_window_extenttransformedinvertedbounds	add_patch)r8   ttrrendererr   r   draw_box   s   z!test_multiline2.<locals>.draw_boxleftr'   r   bottomhorizontalalignmentverticalalignmentg333333?zBottom alignrT   g?rC   z	Top aligng?baselinezBaseline alignrA   )rk   rl   rotationzBot align, rot20)
r2   rcParamsr3   set_xlimset_ylimaxhlinecanvasget_renderer	enumeraterI   )r7   r8   stsrg   horalnnstrc   r   re   r   test_multiline2   sH   

	r{   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-   r|   antialiasedrB   rj         ?z
$\sqrt{x}$)r   rp   r2   rF   rI   r7   r   r   r   test_antialiasing   s   

r   c                  C   s\   t jjddd} t| d}t j|}W d    n1 sw   Y  |ddks,J d S )N	Helveticaafm)fontextrbVAVAVAVAVAVA)g     @i  )r   font_managerr   openr   AFMstring_width_height)fnfhr   r   r   r   test_afm_kerning   s
   r   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 ]3\}}t  j||g\|_|_||\}}	|r\dnd}
|j }|j||d|
d |j| q?d S )Nbutton_press_eventr-   r@   r   r|      r+   zhello worldrB   )rD   fontsizero   yellowredorm   )r2   rF   axesr   rt   nplinspacemeshgridrI   drawzipflatgcar\   rY   xycontainsviewLimfrozenplotset)r7   r8   meventxsystxtr   r   r   _rT   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 )
Nhello)r+   r+   )333333?r   
arrowstylez->)xyxytext
arrowpropsr   )r-   r   F)	r2   r3   r4   rt   r   r   	transDatarY   r   )r7   r8   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 )Nr@   z
left titlerh   loczright titleright)r2   rF   rG   rH   r5   r6   rL   r   r   r   test_titles   s   
r   text_alignmentc                  C   s   t   t ddd} d}d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 )Nr@   r   )r   r   )rC   ri   rn   rB   r-   z Tjroundwheat)boxstyle	facecoloralpha)rE   ro   bboxg      ?z$\sum_{i=0}^{j}$)rE   ro   r   g      ?)
r2   rF   rG   rI   dictr   rq   rr   r5   r6   )r8   r   ro   	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 )
Nr@   rB   rA   i  )r   r   
fontweightrh      i  r   )r2   rF   rG   rH   rL   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sCJ q7|jddddd}| j  || jj}d}||f|_
| j  || jj}t|j	|j	D ]\}}|| |ksJ qsd S )NtestrW   figure pixels)r   
textcoords   )r2   r3   r4   rt   r   r^   rf   set_positionr   minxyann)r7   r8   r   init_pos	shift_valpost_posabr   r   r   test_set_position&  s0   




r   rI    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s:J d S )	Nr-   rh   ri   )rD   rE   
   )dpi皙?rtol)r2   r3   rI   rt   r   r   r^   r   testingassert_allclose
get_points)rI   r7   r8   t1r   bbox1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 )N
horizontal        verticalg     V@z15.      .@r   rI   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   rI   r   floatr   r   r   r   test_get_rotation_inta  s   r   c                   C   s<   t t tjd W d    d S 1 sw   Y  d S )N
hozirontal)pytestraises
ValueErrorr   rI   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@)r   r   r   rI   r   )r   jr   r   r   test_get_rotation_mod360o  s   r   rD   )rB   r   rh   rE   )rB   rC   ri   rn   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   )ro   rE   rD   )r-   r-   r   rotation_modeanchordefault)
r2   r3   r   rI   rt   r   r   r^   rf   r   )rD   rE   r7   r8   r   t0r   r   r   r   %test_null_rotation_with_rotation_modet  s   
r   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?rd   T)backgroundcolorrX   r-   zIs fancy bbox clipped?rX   r   zround, pad=0.1)r2   rI   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+ ptsr   rA   zaxes points    r   r   r   r!   r   - ptsirC   r   r   r   r!   r   rE   + fracr|   r   r    - frac皙+ pixels      zaxes pixels- pixelsir2   r3   r4   r7   r8   r   r   r   "test_annotation_negative_ax_coords  8   






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
     rC   r  r  r   g?zfigure fractionr  r  r-   r  2   r   r  ir,   r  r  r   r   r   #test_annotation_negative_fig_coords  r  r  c                  C   s   t dd\} \}}t   |jrJ |jrJ | jrJ |ddd}|js)J |js.J | js3J |jdddgd}|jsAJ |jsFJ | jsKJ t   |jrTJ |jrYJ | jr^J d S )Nr@   rR   r-   aardvarkr   )r2   r3   draw_allstalerI   r4   )r7   ax1ax2txt1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 )Nr@   rR   r   fooTr  )r   randomseedr2   r3   randrI   )r7   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.sizelarger   r,   )r   rp   r   r   get_size_in_points)fp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 q+d S )N*   zpdf.fonttype)g@g(@r}   r  iX     +   rR   r   r   z{fs} pt font size)fs)r   )r   rp   r2   r3   xaxisset_major_locatorNullLocatoryaxisrr   rv   rangerI   format)r7   r8   r   r3  r   r   r   test_font_scaling  s   
 r:  zspacing1, spacing2))r+   rR   )rR   r+   )rR   rR   c           	      C   s   d}t  }|j }t jdd|| d}t jdd||d}|j  |j|d}|j|d}|j|jks6J | |krD|j|jksBJ d S |j|jksLJ d S )Nzline1
line2r   r-   )linespacingre   )	r2   rF   rt   ru   rI   r   r^   widthheight)	spacing1spacing2text_stringr7   rf   text1text2box1box2r   r   r   test_two_2line_texts  s   

rE  c                  C   s:   t  \} }|dtjd |tjdd | j  d S )Nr   naninf)r2   r3   rI   r   rF  rG  rt   r   r  r   r   r   test_nonfinite_pos
  s   rH  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 )
Nr@   ztext.hinting_factorr-   z	some textsvgr9  pngr   r   )r2   rp   rF   rI   savefigioBytesIOr^   	intervalxr   r   r   )r7   r  expectedr   r   r   test_hinting_factor_backends  s   


rQ  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s8J q-q%d S )NFtext.usetexy   Tz   )
r2   rF   rp   add_subplotrt   r   r4  
majorTickslabel1
get_usetex)r7   r   r!  r8   usetexr  r   r   r   test_usetex_is_copied  s   




rZ  c                  C   s(   t  } | jddddd | j  d S )Nr-   z	$\frac12$TrY  )r2   rF   rI   rt   r   r   r   r   r   test_single_artist_usetex.  s   r\  fmt)rK  pdfrI  c                 C   s:   dt jd< t  }|jddddd |jt | d d S )NTrR  r-   2_2_2Fr[  rJ  )r2   rp   rF   rI   rL  rM  rN  )r]  r7   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 )Nr   cr   r   r   r-   rm   r-   r   r   r|   r   r   r   r   r@   r   rT   )r2   rF   r   set_axis_offrI   r   r   r   r   test_text_as_path_opacityD  s
   rg  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.fonttyper   z50% using `color`rb  rm   r-   z50% using `alpha`rc  r|   z"50% using `alpha` and 100% `color`rd  re  )r   rp   r2   rF   r   rf  rI   r   r   r   r   test_text_as_text_opacityM  s   

ri  c                   C   s,   t ddgddg tt dgdd d S )NABr@   rR   r-   Boo)r2   r   reprrI   r   r   r   r   test_text_reprX  s   rn  c                  C   sf   t dd\} }|jddd}|| j }|   || j }tj|	 |	 ddr1J d S )Nr@   
annotation)r-   r-   r  gư>r   )
r2   r3   r4   r^   rt   ru   tight_layoutr   allcloser   )r7   r8   anextent1extent2r   r   r   test_annotation_update^  s   ru  rK  )
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 )
Nr@   r   r   r-   )datar    rW   zoffset points)r!   r   r   )r!   )rU  r   r   nowr4   fig_testfig_refr8   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 |d |d }|jd
dd |jddd |d d S )NrP   rQ   zaxes.titleyr@   rR   )	   g      @T)r~   constrained_layoutr   z$\sum_{i} x_i$zNew wayrh   r   r   g)\(?)r   zOld Way)r2   rp   r3   rH   set_xticklabels)r7   axsr8   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rP   r1  r}   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.r2   rF   rI   rt   r   _get_wrapped_text)r7   srI   r   r   r   	test_wrap  s
   
r  c                  C   s<   t jdd} | jddddd}| j  | dksJ d S )Nr  r}   g      #@   AlonglineoftexttowrapTr  r  r7   rI   r   r   r   test_long_word_wrap     
r  c                  C   s<   t jdd} | jddddd}| j  | dksJ d S )Nr  r}   r   znon wrapped textTr  r  r  r   r   r   test_wrap_no_wrap  r  r  c                 C   s`   |   }|ddg |ddg |jjd jd |  }|ddg |ddg d S )Nr   r@   u   €r   wr   )rU  r6   set_yticklabelsr7  rV  rW  	set_colorry  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sJ | dks$J dS )z>Test that kwargs take precedence over fontproperties defaults.valuezTimes New Romang      D@)r"   r   counts)r   r"   N)r2   rF   xlabelylabelget_size)rA  rB  r   r   r   $test_fontproperties_kwarg_precedence  s
   r  c                  C   s@   t  } t d}| jddd|dd}| }t|d d S )Nr   r   r   T)rY   transform_rotates_text)r2   r   mtransformsAffine2D
rotate_degrI   r   r   )r8   rY   rI   resultr   r   r   test_transform_rotates_text  s   r  c                  C   sV   t tddd d} t | }t }||  | d |d ksJ | d |d ks)J 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   sB   t t tjddddd W d    d S 1 sw   Y  d S )Nr-   r%  foobar)ra  )r   r   r   r2   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-   ATATATATATATATATATAr   r+  )r2   rF   r  r   r   r   r   test_pdf_kerning  s   r  )Pr   rM  r/   numpyr   numpy.testingr   r   
matplotlibr   matplotlib.backend_basesr   r.   r   matplotlib.patchespatchesrZ   matplotlib.pyplotpyplotr2   matplotlib.transforms
transformsr  matplotlib.testing.decoratorsr   r   matplotlib.textr   markskipifcheckdep_usetexneeds_usetexr>   rM   r{   r   r   r   r   r   r   r   r   parametrizer   r   r   r   r   r   r   r   r  r  r  r$  r)  r/  r:  rE  rH  rQ  rZ  r\  r`  rg  ri  rn  ru  r|  r  r  r  r  r  r  r  r  r  r  r   r   r   r   <module>   s    

S

3





	 





	



	








		
