o
    i8                     @   s  d dl 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	Z
d dl	mZmZ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 d d	lmZ d d
lmZmZ d dl m!Z! edgdd Z"ej#$dddgej#$dddgdd Z%dd Z&dd Z'dd Z(dd Z)dd  Z*d!d" Z+d#d$ Z,d%d& Z-d'd( Z.d)d* Z/e!d+d, Z0d-d. Z1ed/gd0d1 Z2ed2gd3d4 Z3e
j45d5ed6d7gd8d9d: Z6e!d;d< Z7d=d> Z8ed?gd@dA Z9dBdC Z:edDgdEdFdGdH Z;edIgdJdFdKdL Z<dS )M    N)Path)pyplotrcParamsfont_manager)_get_data_path)FT2Font)findfontFontProperties)get_glyphs_subset)PdfPages)	Rectangle)check_figures_equalimage_comparison)needs_usetexzpdf_use14corefonts.pdfc                  C   sl   dt d< dt d< dt d< dgt d< d	t d
< d} t \}}|d |jdd| dddd |jddd d S )NTzpdf.use14corefontsz
sans-serifzfont.family   z	font.sizeZ	Helveticazfont.sans-serifr   pdf.compressionu   A three-line text positioned just above a blue line
and containing some French characters and the euro symbol:
"Merci pépé pour les 10 €"z0Test PDF backend with option use14corefonts=Trueg      ?centerbottom   )ZhorizontalalignmentZverticalalignmentZfontsize)Z	linewidth)r   pltsubplots	set_titletextZaxhline)r   figax r   p/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/matplotlib/tests/test_backend_pdf.pytest_use14corefonts   s   

r   zfontname, fontfile)DejaVu SanszDejaVuSans.ttf)WenQuanYi Zen Heiwqy-zenhei.ttcfonttype   *   c                 C   sr   t tt| gdj|krtd| d |td< t \}}|	g d |j
d| d |jt dd	 d S )
NfamilyzFont z may be missingzpdf.fonttype      r"   z
Axes Title)fontpdfformat)r   r   r	   namepytestskipr   r   r   plotr   savefigioBytesIO)ZfontnameZfontfiler!   r   r   r   r   r   test_embed_fonts,   s   r4   c                  C   s   t t 9} |  dksJ t \}}|g d |j| dd |  dks+J |   |  dks7J W d    d S 1 sBw   Y  d S )Nr   r&   r*   r+   r'   r(   )r   r2   r3   Zget_pagecountr   r   r0   r1   )r*   r   r   r   r   r   test_multipage_pagecount<   s   "r5   c                  C   s   t  } t| !}tdD ]}t \}}|d |j|dd qW d    n1 s,w   Y  |  }|	ddks>J t
|dk sFJ d S )N
   zThis is a long titler*   r+   s	   startxrefr'   i@  )r2   r3   r   ranger   r   r   r1   getvaluecountlen)Zpdfior*   ir   r   sr   r   r   test_multipage_properfinalizeG   s   

r=   c              	   C   s  t |  ttj td}W d    n1 sw   Y  W d    n1 s*w   Y  t jds7J ttj tddd}W d    n1 sOw   Y  W d    n1 s^w   Y  t jdskJ tddd}W d    n1 s|w   Y  t jdrJ td}|	t
  W d    n1 sw   Y  t jdsJ ttj% tddd}|	t
  W d    n1 sw   Y  W d    n1 sw   Y  t jdsJ td	dd}|	t
  W d    n1 sw   Y  t jd	sJ d S )
Nza.pdfzb.pdfT)Z
keep_emptyzc.pdfFzd.pdfze.pdfzf.pdf)oschdirr.   warnsmplZMatplotlibDeprecationWarningr   pathexistsr1   r   figure)Ztmp_pathr*   r   r   r   test_multipage_keep_emptyS   sJ   
 
 

 rE   c                  C   s.  t t dddt ddd\} }t |d }t \}}|dd |j|g dd |j|d d d	 g d
d dtjd< t	t
 }|j|dd t|jjdksYJ W d    n1 scw   Y  dtjd< t	t
 }|j|dd t|jjdksJ W d    d S 1 sw   Y  d S )N   r'   r(   r   r"   )r   r'   r   r'   )Zextent)r(   r"   r   r'   Tzimage.composite_imager*   r+   F)npZmeshgridarangesinr   r   Zset_xlimimshowr   r   r2   r3   r1   r:   _fileZ_images)XYZr   r   r*   r   r   r   test_composite_image{   s    $

"rQ   c            
      C   s   t d} tjdtjd}td|d d ddf< dtd< t }|j	|dd t
 }|j|d	d
d | j|(}|j\}|j \}| |}|jsOJ | }t|d}	W d    n1 sew   Y  tj||	 d S )Npikepdf)   r'   r"   )ZdtyperS   r   Tr   )resizer*   rD   )r,   ZdpiRGB)r.   importorskiprI   ZzerosZuint8rJ   r   r   rD   Zfigimager2   r3   r1   PdfopenpagesZimagesvaluesZPdfImageZindexedZas_pil_imageZasarrayconverttestingZassert_array_equal)
rR   datar   bufr*   pageimageZ	pdf_imageZ	pil_imagergbr   r   r   test_indexed_image   s"   


rb   c                 C   s   t d}| dd t \}}|td ddddtjd	d
dtt	dddd}t
 }|j||dd |j|}dd |j D }W d    n1 sUw   Y  |dddtj ddddtj dddd	kssJ d S )NrR   SOURCE_DATE_EPOCH0rG   meMultipage PDF	Test pagetest,pdf,multipage  r   r'   r   tzinfoTrueAuthorTitleSubjectKeywordsModDateTrappedr*   )metadatar,   c                 S      i | ]	\}}|t |qS r   str.0kvr   r   r   
<dictcomp>       z)test_savefig_metadata.<locals>.<dictcomp>D:19700101000000ZMatplotlib v, https://matplotlib.orgD:19680801000000ZMatplotlib pdf backend v/True	z/Authorz/CreationDatez/Creatorz	/Keywordsz/ModDatez	/Producerz/Subjectz/Titlez/Trapped)r.   rV   setenvr   r   r0   r7   datetimetimezone	timedeltar2   r3   r1   rW   rX   docinfoitemsrA   __version__ZmonkeypatchrR   r   r   mdr^   r*   infor   r   r   test_savefig_metadata   s:   
	
r   c                  C   s*  t  \} }tjtdd | jt dddid W d    n1 s$w   Y  tjtdd | jt ddd	id W d    n1 sGw   Y  tjtd
d | jt dddid W d    n1 sjw   Y  tjtdd | jt dddid W d    d S 1 sw   Y  d S )Nz#Unknown infodict keyword: 'foobar'.)matchr*   Zfoobarinvalid)r,   rt   z%not an instance of datetime.datetime.rr   z
1968-08-01z'not one of {"True", "False", "Unknown"}rs   Zfooznot an instance of str.ro   i  )r   r   r.   r@   UserWarningr1   r2   r3   r   r   r   r   r   test_invalid_metadata   s*   "r   c                 C   s  t d}| dd t \}}|td ddddtjd	d
dtt	dddd}t
 }t||d}|| || W d    n1 sMw   Y  |j|}dd |j D }W d    n1 smw   Y  |dddtj ddddtj dddd	ksJ d S )NrR   rc   rd   rG   re   rf   rg   rh   ri   r   r'   r   rj   rl   rm   )rt   c                 S   ru   r   rv   rx   r   r   r   r|      r}   z+test_multipage_metadata.<locals>.<dictcomp>r~   r   r   r   r   r   r   )r.   rV   r   r   r   r0   r7   r   r   r   r2   r3   r   r1   rW   rX   r   r   rA   r   r   r   r   r   test_multipage_metadata   s@   
	

r   c               	      s2  t d} dtjdd}|jddd dd |jdd	d
 dd t f}|j|dd | j	|@}|j
d j}dD ]/\} t fdd|D d }|d usWJ t|dd d u saJ |jd t|d ksoJ q@W d    n1 szw   Y  W d    d S W d    d S 1 sw   Y  d S )NrR   &https://test_text_urls.matplotlib.org/r(   r'   Zfigsize皙?ztest plain 123plain)urlg?ztest mathtext $123$mathtextr*   r+   r   ))z0.1r   )z0.4r   c                 3   s(    | ]}|j j   kr|V  qd S NAURIry   afragmenttest_urlr   r   	<genexpr>  s   & z!test_text_urls.<locals>.<genexpr>
QuadPointsr'   H   )r.   rV   r   rD   r   r2   r3   r1   rW   rX   rY   AnnotsnextgetattrRectdecimalDecimal)rR   r   fdr*   annotsyannotr   r   r   test_text_urls  s,   

"r   c               	      s  t d} d tjdd}|jdddd  d t a}|j|d	d
 | j	|;}|j
d j}t fdd|D d }|d usCJ t|dd d usMJ |jd |jd td ks^J W d    n1 shw   Y  W d    d S W d    d S 1 sw   Y  d S )NrR   r   )r'   r'   r   r   N-   )Zrotationr   r*   r+   r   c                 3   s"    | ]}|j j  kr|V  qd S r   r   r   r   r   r   r   .  s     z)test_text_rotated_urls.<locals>.<genexpr>r      z0.00001)r.   rV   r   rD   r   r2   r3   r1   rW   rX   rY   r   r   r   r   r   r   r   rR   r   r   r*   r   r   r   r   r   test_text_rotated_urls  s*   


"r   c               	      s   t d} d tjdd}|jdddd  d	d
 t T}|j|dd | j	|.}|j
d j}t fdd|D d }|d usEJ |jd tdd ksSJ W d    n1 s]w   Y  W d    d S W d    d S 1 suw   Y  d S )NrR   r   r   r   r   gffffff?ztest tex $123$Ttex)usetexr   r*   r+   r   c                 3   s&    | ]}|j j  d kr|V  qdS )r   Nr   r   r   r   r   r   I  s   $ z%test_text_urls_tex.<locals>.<genexpr>r'   z0.7r   )r.   rV   r   rD   r   r2   r3   r1   rW   rX   rY   r   r   r   r   r   r   r   r   r   test_text_urls_tex7  s$   

"r   c                  C   sB   t ttj} | t  W d    d S 1 sw   Y  d S r   )r   r   r>   devnullr1   r   rD   )r*   r   r   r   test_pdfpages_fspathP  s   "r   zhatching_legend.pdfc                  C   sT   t jdd} tddgddddd}tddgddddd}| ||||gg d d	S )
z.Test for correct hatching on patches in legend)r'   r(   r   r   greenZXXXX)Z	facecolorZhatchblue) r   r   r   N)r   rD   r   Zlegend)r   r   br   r   r   test_hatching_legendU  s   r   zgrayscale_alpha.pdfc                  C   sx   t jddddddf \} }t | d |d   }t j||dk < t \}}|j|ddd |g  |g  dS )z9Masking images with NaN did not work for grayscale imagesr(   r   noneZgray_r)interpolationZcmapN)	rI   Zogridexpnanr   r   rL   Z
set_xticksZ
set_yticks)xr   ddr   r   r   r   r   test_grayscale_alpha`  s   
r   defaultr*   Zeps)
extensionsc                 C   sD   |   }|  |jttdddddd |  }|  d S )NrF   rG   d   r{   r   )c)Zadd_subplotZset_axis_offr0   rI   rK   Zlinspace)Zfig_testZfig_refZax_testZax_refr   r   r   'test_pdf_eps_savefig_when_color_is_nonel  s
    r   c                   C   sP   t jddd tt t jt dd W d   dS 1 s!w   Y  dS )z"Test failing latex subprocess callz$22_2_2$T)r   r*   r+   N)r   Zxlabelr.   ZraisesRuntimeErrorr1   r2   r3   r   r   r   r   test_failing_latexv  s   "r   c                  C   s2   t  \} }|jg g dd | jt dd d S )NT)Z
rasterizedr*   r+   )r   r   r0   r1   r2   r3   r   r   r   r   test_empty_rasterized~  s   r   zkerning.pdfc                  C   s4   t  } d}| jdd|dd | jdd|dd d S )Nu   AVAVAVAVAVAVAVAV€AAVVr   g      ?rG   )sizeg      ?   )r   rD   r   )r   r<   r   r   r   test_kerning  s   r   c                  C   s   t td} d}t| }|| tt| |}|| | }| }h |dd |D ks2J t|t|k s<J | | ksFJ d S )Nzfonts/ttf/DejaVuSerif.ttfz%these should be subsetted! 1234567890c                 S   s   h | ]}t |qS r   )chr)ry   keyr   r   r   	<setcomp>  s    z%test_glyphs_subset.<locals>.<setcomp>)rw   r   r   Zset_textr
   Zget_charmapr:   Zget_num_glyphs)ZfpathcharsZ	nosubfontZsubfontZ	nosubcmapZsubcmapr   r   r   test_glyphs_subset  s   

r   zmulti_font_type3.pdfgffffff@)Ztolc                  C   h   t jdgd} tt | jdkrtd tjdddgdd tjd	d
d t	 }|
ddd d S )Nr   r$   r    Font may be missingr)   r      r%   r   r*   r"   r!   333333?ffffff?"   There are 几个汉字 in between!fmr	   r   r   r-   r.   r/   r   rcrD   r   fpr   r   r   r   test_multi_font_type3     
r   zmulti_font_type42.pdfg@c                  C   r   )Nr   r$   r    r   r)   r   r   r   r*   r#   r   r   r   r   r   r   r   r   r   test_multi_font_type42  r   r   )=r   r   r2   r>   pathlibr   numpyrI   r.   Z
matplotlibrA   r   r   r   r   r   Zmatplotlib.cbookr   Zmatplotlib.ft2fontr   Zmatplotlib.font_managerr   r	   Z#matplotlib.backends._backend_pdf_psr
   Zmatplotlib.backends.backend_pdfr   Zmatplotlib.patchesr   Zmatplotlib.testing.decoratorsr   r   Zmatplotlib.testing._markersr   r   markZparametrizer4   r5   r=   rE   rQ   rb   r   r   r   r   r   r   r   r   r   stylecontextr   r   r   r   r   r   r   r   r   r   r   <module>   sn    
(#%







