o
    liFR                     @   s  d dl Z d dlmZ d dlmZ d dlZd dlZd dl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mZ d dlmZ d dlmZ d d	lmZmZ d
d Zedgdddd Zedgdddd Z dd Z!edgdd Z"edgdd Z#edddd Z$e d d! Z%ed"d#d$gd%d&d' Z&d(d) Z'ej()d*ed+d, Z*d-d. Z+d/d0 Z,d1d2 Z-d3d4 Z.d5d6 Z/d7d8 Z0d9d: Z1d;d< Z2d=d> Z3ed?gd@ddAdB Z4edCgdDdE Z5ej67dFdGdHie8dIfdGdHgie8dJfdKdHie8dLfdKdHgie8dMfdNdHie8dOfdNdHgie8dPfdQdHie8dRfdSdHie8dTfdUdVie9dWfg	dXdY Z:dZd[ Z;ej67d\g d]ej67d^dd_gd`da Z<dbdc Z=dS )d    N)BytesIO)Path)Figure)Text)check_figures_equalimage_comparison)needs_usetex)font_manager)OffsetImageAnnotationBboxc                  C   s   t  \} }tddtj d}t|}t|}|j|||dd\}}}|D ]}|d q(t	 }	| j
|	dd |	 }
W d    n1 sIw   Y  tjj }||
 d S )	Nr      2   ko)yerrfmtFsvgformat)pltsubplotsnplinspacepisin	ones_likeerrorbarset_visibler   savefiggetvaluexmlparsersexpatParserCreateParse)figaxxyr   abcartistfdbufparser r/   _/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/matplotlib/tests/test_backend_svg.pytest_visibility   s   


r1   zfill_black_with_alpha.svgT)remove_textc                  C   s,   t  \} }|jg dg ddddd d S )N)r   皙?   )r   r   r   kr3   i'  )r&   r'   r*   alphas)r   r   scatterr$   r%   r/   r/   r0   test_fill_black_with_alpha(   s    r:   noscalec                  C   sR   t t dddt ddd\} }t |d }t \}}|j|ddd d S )N   r4      graynone)cmapinterpolation)r   meshgridaranger   r   r   imshow)XYZr$   r%   r/   r/   r0   test_noscale.   s   $rI   c                  C   st   t  } d}| jd|d t }| j|dd |  }W d    n1 s(w   Y  d|}||v s8J d S )Nz$http://test_text_urls.matplotlib.orgtest_text_urls)urlr   r   z<a xlink:href="{0}">)r   figuresuptitler   r   r   decoder   )r$   test_urlr,   r-   expectedr/   r/   r0   rJ   7   s   
rJ   zbold_font_output.svgc                  C   sN   t  \} }|tdtd |d |jddd |jddd d S )N
   nonbold-xlabelbold-ylabelbold
fontweight
bold-title)r   r   plotr   rD   
set_xlabel
set_ylabel	set_titler9   r/   r/   r0   test_bold_font_outputE   s
   
r\   z'bold_font_output_with_none_fonttype.svgc                  C   sX   dt jd< t  \} }|tdtd |d |jddd |jddd d S )	Nr@   svg.fonttyperQ   rR   rS   rT   rU   rW   )	r   rcParamsr   rX   r   rD   rY   rZ   r[   r9   r/   r/   r0   (test_bold_font_output_with_none_fonttypeN   s   

r_      )tolc                 C   s   t ddd }t |}t |}| }|j||dddd |j|d |dd	dd |  }|j||dddd
d |j|d |dd	dd
d d S )Nr   d   ffffff@-rrQ   )r*   lwr4   r)   Tr*   rf   
rasterized)r   rD   cosr   r   rX   fig_testfig_reftr&   r'   ax_refax_testr/   r/   r0   test_rasterizedX   s   

rp   c              	   C   s8  t ddd }t |}t |}| }|dd |dd |j||ddd	d
d |j|d |ddd	dd |j|d |ddd	d
d |j|d |ddd	d
d |  }|dd |dd |j||ddd	d
dd |j|d |ddd	d
dd |j|d |ddd	d
dd |j|d |ddd	ddd d S )Nr   rb   rc      gg?rd   re   rQ   Trg   r4   r)   Fr>   gm)r*   rf   rh   zorderg?gffffff?g333333?)r   rD   ri   r   r   set_xlimset_ylimrX   rj   r/   r/   r0   test_rasterized_orderingg   s"   

 rw   r=   r   pdf)ra   
extensionsc                 C   s   ddg}|  }|j|d g|d gdddd tjd}tjj||d	d
}|| |   }|j|d g|d gddddd tjd}tjj||d	d
}|| d S )Ng?r   r4   r&   blackr>   )markerr*   rt   rF   g @)rt   T)r{   r*   rt   rh   )r   rX   mpl	offsetboxTextArear   
add_artist)rk   rl   locrn   r)   aboxro   r/   r/   r0   test_prevent_rasterization~   s    
r   c                  C   sh  dd } t  }|ddd}|  tdD ]}|jddgd|gddd	 q| |d
dks0J | |ddks9J t  }|ddd}|  tdD ]}|jddgd|gddd	 qL| |d
dkseJ | |ddksnJ t  }|ddd}|  tdD ]}|jddg|dgddd	 |jddgd|gddd	 q| |d
dksJ | |ddksJ t  }|ddd}	|	  |	d tdD ]}|	jddg|dgddd	 |	jddgd|gddd	 q| |d
dksJ | |ddksJ t  }
d|
_|
ddd}|  tdD ]}|jddgd|gddd	 q| |
d
dks(J | |
ddks2J d S )Nc                 S   sP   t  }| j|dd |  }W d    n1 sw   Y  |d| S )Nr   r   <)r   r   r   rN   count)r$   tagr,   r-   r/   r/   r0   	count_tag   s
   z%test_count_bitmaps.<locals>.count_tagr4   r=   r   r`   zb-F)rh   imagepath   T)r   rL   add_subplotset_axis_offrangerX   set_rasterizedsuppressComposite)r   fig1ax1nfig2ax2fig3ax3fig4ax4fig5ax5r/   r/   r0   test_count_bitmaps   sP   
r   defaultc                  C   s   t  } | jddddd t }| j|dd | }W d    n1 s%w   Y  tjj|}d}d}t	|
d	| d
| ddksGJ d| |d	| dj v sZJ d S )N      ?z\textwonT)usetexr   r   http://www.w3.org/2000/svgzSFSS3583-8e.//{z}path[@d][@id="z"]r4   #z}use)r   textr   r   r   r   etreeElementTree
fromstringlenfindallfindattribvalues)r$   r,   r-   treenswon_idr/   r/   r0   test_unicode_won   s   
$*r   c                  C   s   t jddd d} t  \}}|tdd|  |tdtd |d	d
 t	 }|j
|dd |d	 |  }W d    n1 sMw   Y  | |v rZd|v s\J d S )Nr@   	condensed)r]   zfont.stretchzUnlikely to appear by chancez
2019-06-30r4   z
2019-01-01z
2019-12-31r   r>   r   r   )r   r^   updater   r   r   
datetime64ru   rv   r   r   seekreadrN   )rP   r$   r%   r,   r-   r/   r/   r0   "test_svgnone_with_data_coordinates   s   
r   c                     s  ddl m  ddlm t } | d}|jddgddggdd	 |jg d
g d
dd |j	g ddd |
  | }|g d
g d
 | jddd}|	g dg d
 | jddd}|	ddgddgddg | j  i }t| jddD ]\}}| rd|jj d| }|||< || q}t }	| j|	dd |	  }
W d   n1 sw   Y   fdd}| D ]\}}|||r||
v sJ qdS ) z+Test that object gid appears in output svg.r   )	OffsetBox)Tick   g      ?g       @g      @auto)aspectr4   r>   rq   	myscatter)label)r>   rq   r4   myplot   polar)
projection)r   g      ?rq      3dr4   r>   T)include_selftest123_r   r   Nc                    s   t | rdS t |tr| dkrdS |jd u rdS t |tjrC| \}}t|t|  kr6dkr7dS  t|drA|jd u rCdS t |rd|	 }|dkrRdS |
 }|t|k sb|t|krddS dS )NF r4   axesr   T)
isinstancer   get_textr   r   Line2Dget_datar   hasattrget_locget_view_intervalminmax)gidobjxdataydatar   vir   r   r/   r0   include  s,   



ztest_gid.<locals>.include)matplotlib.offsetboxr   matplotlib.axisr   r   rL   r   rE   r8   rX   legendtwinxbarcanvasdraw	enumeratefindobjget_visible	__class____name__set_gidr   r   r   rN   items)r$   r   ax1ar   r   gdicidxr   r   r,   r-   r   r/   r   r0   test_gid   s@   



r   c                   C   s   t jt ddd d S )Nsvgztight)r   bbox_inches)r   r   r   r/   r/   r/   r0   test_savefig_tight.  s   r   c                  C   s   t  \} }|g dg d}|g d t ddgddg\}|d t }| j|d	d
 | }dD ]
}d| |v sAJ q7d S )Nr   r   r=   r   )zhttps://example.com/foozhttps://example.com/barNr4   rq   r   r=   zhttps://example.com/bazr   r   )s   foos   bars   bazs   https://example.com/)	r   r   r8   set_urlsrX   set_urlr   r   r   )r$   r%   r7   pr)   vr/   r/   r0   rO   4  s   
rO   c                 C   s(  |  dd t \}}|g dg d t|j D ]\}}|d|  qt \}}|g dg d t|j D ]\}}|jd|  |j	d|  qAt
 }|j|dd | }t
 }|j|dd | }tt|j D ]}d| d|v sJ q}||ksJ d S )	NSOURCE_DATE_EPOCH19680801r   r   zhttps://example.com/r   r   ascii)setenvr   r   r8   r   yaxisget_major_ticksr   label1label2r   r   r   r   r   encode)monkeypatchr   r%   itickr   b1b2r/   r/   r0   test_url_tickH  s&   r   c              	   C   s  |  dd t \}}t }|j|dd |  }W d    n1 s'w   Y  tj|v s3J d|v s9J d|v s?J d|v sEJ t }|j|dd d d d dd	 |  }W d    n1 sfw   Y  tj|vsrJ d|vsxJ d|vs~J d|vsJ d S )
Nr   r   r   r   
1970-08-16image/svg+xml
StillImageDateCreatorFormatTyper   metadata)	r   r   r   r   r   r   rN   r|   __version__)r   r$   r%   r,   r-   r/   r/   r0   test_svg_default_metadatac  s(   
r  c              
   C   s  |  dd tjdddd}d}d}d	}d
}t \}}|D ]j}t }	|j|	d| d id |	 	 }
W d    n1 s@w   Y  t
jj|
}|d| d| d| d\}|D ])}|d| | }||krs|rrJ q^|\}t
jjj|dd}|| |v sJ q^qd S )Nr   r   r  r  r  )creatordater   type{http://www.w3.org/2000/svg}-{http://www.w3.org/1999/02/22-rdf-syntax-ns#} {http://creativecommons.org/ns#}"{http://purl.org/dc/elements/1.1/}r   r	  ./	metadata/zRDF/Workunicode)encoding)r   r|   r  r   r   r   r   titler   rN   r   r   r   r   r   tostring)r   metadata_containsSVGNSRDFNSCCNSDCNSr$   r%   namer,   r-   rootworkkeydataxmlstrr/   r/   r0   test_svg_clear_default_metadata  s4    r&  c               	   C   s   t  \} }t }| j|dd d d d dd |  }W d    n1 s'w   Y  d}tjj	|}|
d| dr@J d S )Nr   r  r	  r  r  r
  )r   r   r   r   r   rN   r   r   r   r   r   )r$   r%   r,   r-   r  r!  r/   r/   r0   test_svg_clear_all_metadata  s   
r'  c                     s`  g d} g d}t dddt  ddddddgdd	d
d | D dd |D }t }t }|j|d|d |  }W d    n1 sJw   Y  d}d d}d}tj	j
|}	|	d| d  d\}
dd |	d| dD }||d gksJ  fdd|
d| d| dD }||d gksJ dg| D ]$}|dkrqdd |
d| d| |  D }||| gksJ q|D ]*}|d krqd!d |
d| d| |  d"| d#| d
D }||| ksJ qd$d |
d| d| d%D }|d&gksJ d'd |
d| d| d(  d)  d*	D }||d  ks.J d S )+N)Coverage
IdentifierLanguageRelationSourceTitler  )Contributorr  Keywords	PublisherRightsi     r4   r>   rq   zdescription
text)r  Descriptionc                 S   s   i | ]}|| d qS )z foor/   .0r5   r/   r/   r0   
<dictcomp>  s    z%test_svg_metadata.<locals>.<dictcomp>c                 S   s"   i | ]}|| d | dgqS )z barz bazr/   r4  r/   r/   r0   r6    s   " r   r	  r  r  r  r  r  r  RDFc                 S      g | ]}|j qS r/   r   r5  noder/   r/   r0   
<listcomp>      z%test_svg_metadata.<locals>.<listcomp>r  r-  c                    s   g | ]
}|j   d  qS )resource)r   r:  r  r/   r0   r<    s    zWork/r  r  r3  c                 S   r8  r/   r9  r:  r/   r/   r0   r<    s    
r/  c                 S   r8  r/   r9  r:  r/   r/   r0   r<    s    /zAgent/c                 S   r8  r/   r9  r:  r/   r/   r0   r<    r=  r  z1968-08-01/1968-08-02T01:02:03c                 S   r8  r/   r9  r:  r/   r/   r0   r<    r=  zsubject/zBag/li)datetimer  r   rL   r   r   r   rN   r   r   r   r   r   lower)single_valuemulti_valuer
  r$   r,   r-   r  r  r  r!  rdftitlestypesr5   r   datesr/   r?  r0   test_svg_metadata  sd   
$""rJ  zmulti_font_aspath.svgg?c                  C   sh   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 )NWenQuanYi Zen Heifamilywqy-zenhei.ttcFont may be missingfontDejaVu Sans   rM  sizer   r   fonttype333333?ffffff?"   There are 几个汉字 in between!)fmFontPropertiesr   findfontr   pytestskipr   rcrL   r   fpr$   r/   r/   r0   test_multi_font_type3  s   
rb  zmulti_font_astext.svgc                  C   sh   t jdgd} tt | jdkrtd t }tj	ddd tj	dd	dgd
d |
ddd d S )NrK  rL  rN  rO  r   r@   rU  rP  rQ  rR  rS  rW  rX  rY  )rZ  r[  r   r\  r   r]  r^  r   rL   r_  r   r`  r/   r/   r0   test_multi_font_type42  s   
rc  zmetadata,error,messager  r4   z,Invalid type for Date metadata. Expected strz1Invalid type for Date metadata. Expected iterabler/  z0Invalid type for Keywords metadata. Expected strz5Invalid type for Keywords metadata. Expected iterabler  z/Invalid type for Creator metadata. Expected strz4Invalid type for Creator metadata. Expected iterabler-  z-Invalid type for Title metadata. Expected strr  z.Invalid type for Format metadata. Expected strFooBarzUnknown metadata keyc              	   C   s   t j||d0 t }t }|j|d| d W d    n1 s"w   Y  W d    d S W d    d S 1 s:w   Y  d S )N)matchr   r	  )r]  raisesr   r   rL   r   )r
  errormessager,   r$   r/   r/   r0   test_svg_incorrect_metadata	  s   Prj  c                  C   sl   t  } | jddddd t }| j|dd |  }d|v s$J W d    d S 1 s/w   Y  d S )Nr   z<'"&>)r   r   r   z&lt;&apos;&quot;&amp;&gt;")r   rL   r   r   r   r   rN   )r$   r,   r-   r/   r/   r0   test_svg_escape!  s   "rk  font_str)z7'DejaVu Sans', 'WenQuanYi Zen Hei', 'Arial', sans-serifz='DejaVu Serif', 'WenQuanYi Zen Hei', 'Times New Roman', serifz%'Arial', 'WenQuanYi Zen Hei', cursivez&'Impact', 'WenQuanYi Zen Hei', fantasyzA'DejaVu Sans Mono', 'WenQuanYi Zen Hei', 'Courier New', monospaceinclude_genericFc                 C   s  t jdgd}tt |jdkrtd tdd | d^}}}t	|}|r/||g }|t
jd| < |t
jd	< d
t
jd< t
 \}}|dkrPg d}	n|g}	|	D ]}
|jddd||
gdd |jddd|g||
dd qU|d t }|j|dd | }W d    n1 sw   Y  tjj|}d}d}|d| dD ]%}|d7 }tdd t| d dD d }|| d |  ksJ q|t	|jksJ d S )!NrK  rL  rN  rO  c                 S   s
   |  dS )N'stripr&   r/   r/   r0   <lambda>=  s   
 z&test_svg_font_string.<locals>.<lambda>z, zfont.z	font.sizer@   r]   
sans-serif)sansrs  z
sans serifr   rY  center)rM  har3   offr   r   r   r   r   z}textr4   c                 s   s(    | ]}t d d | dV  qdS )c                 S   s   |   S )Nro  rq  r/   r/   r0   rr  _  s    z0test_svg_font_string.<locals>.<genexpr>.<lambda>:N)maprp  split)r5  r   r/   r/   r0   	<genexpr>^  s
    
z'test_svg_font_string.<locals>.<genexpr>style;rP  zpx )rZ  r[  r   r\  r   r]  r^  ry  rz  r   r   r^   r   r   axisr   r   r   r   r   r   r   r   dictr   texts)rl  rm  ra  explicitrestgenericrT  r$   r%   generic_optionsgeneric_namer,   r-   r   r   
text_counttext_element	font_infor/   r/   r0   test_svg_font_string*  sP   








r  c            	      C   s   t  } |  }td}d}t|dd}||j_t||ddddt	d	d
dd}|
d || t }| j|dd | d}W d    n1 sNw   Y  d}||v s[J d S )N)    r  )g333333?g?r3   )zoom)g      ^@g      Tr$  zoffset pointsr   z->zangle,angleA=0,angleB=90,rad=3)
arrowstyleconnectionstyle)xyboxxycoords	boxcoordspad
arrowpropsza test for issue 20044r   r   zutf-8z<g id="a test for issue 20044">)r   rL   r   r   onesr
   r   r   r   r  r   r   r   r   r   rN   )	r$   r%   arr_imgxyimageboxabr,   r-   rP   r/   r/   r0   test_annotationbbox_gidg  s0   

	
r  )>rB  ior   pathlibr   xml.etree.ElementTreer   xml.parsers.expatr]  numpyr   
matplotlibr|   matplotlib.figurer   matplotlib.textr   matplotlib.pyplotpyplotr   matplotlib.testing.decoratorsr   r   matplotlib.testing._markersr   r	   rZ  r   r
   r   r1   r:   rI   rJ   r\   r_   rp   rw   r   r   r|  contextr   r   r   r   rO   r   r  r&  r'  rJ  rb  rc  markparametrize	TypeError
ValueErrorrj  rk  r  r  r/   r/   r/   r0   <module>   s    



	



:?"!9





	0