U
    FvfFR                     @   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\d]d^d_d`dagej67dbddcgddde Z<dfdg Z=dS )h    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 qPt	 }	| j
|	dd |	 }
W 5 Q R X tjj }||
 d S )	Nr      2   ko)yerrfmtFsvgformat)pltsubplotsnpZlinspacepisinZ	ones_likeZerrorbarZset_visibler   savefiggetvaluexmlparsersexpatParserCreateParse)figaxxyr   abcZartistfdbufparser r*   E/tmp/pip-unpacked-wheel-7vhvci0g/matplotlib/tests/test_backend_svg.pytest_visibility   s    

r,   zfill_black_with_alpha.svgT)Zremove_textc                  C   s0   t  \} }|jdddgdddgdddd d S )Nr   皙?   ki'  )r"   r#   r&   alphas)r   r   scatterr    r!   r*   r*   r+   test_fill_black_with_alpha(   s    r4   Znoscalec                  C   sR   t t dddt ddd\} }t |d }t \}}|j|ddd d S )N   r.      Zgraynone)Zcmapinterpolation)r   Zmeshgridaranger   r   r   imshow)XYZr    r!   r*   r*   r+   test_noscale.   s    $r?   c               	   C   s`   t  } d}| jd|d t  }| j|dd |  }W 5 Q R X d|}||ks\td S )Nz$http://test_text_urls.matplotlib.orgtest_text_urls)urlr   r   z<a xlink:href="{0}">)	r   figureZsuptitler   r   r   decoder   AssertionError)r    test_urlr'   r(   expectedr*   r*   r+   r@   7   s    
r@   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Z
fontweight
bold-title)r   r   plotr   r:   
set_xlabel
set_ylabel	set_titler3   r*   r*   r+   test_bold_font_outputE   s
    
rQ   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 )	Nr8   svg.fonttyperG   rH   rI   rJ   rK   rL   )	r   rcParamsr   rM   r   r:   rN   rO   rP   r3   r*   r*   r+   (test_bold_font_output_with_none_fonttypeN   s    

rT      )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@-rrG   )r&   lwr.   r%   Tr&   r[   
rasterized)r   r:   cosr   r   rM   fig_testfig_reftr"   r#   ax_refax_testr*   r*   r+   test_rasterizedX   s    

re   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   rW   rX      gg?rY   rZ   rG   Tr\   r.   r%   Fr7   gm)r&   r[   r]   zorderg?gffffff?g333333?)r   r:   r^   r   r   set_xlimset_ylimrM   r_   r*   r*   r+   test_rasterized_orderingg   s"    

rl   r6   r   Zpdf)rV   
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   r.   r"   Zblackr7   )markerr&   ri   r<   g @)ri   T)rn   r&   ri   r]   )r   rM   mplZ	offsetboxZTextArear   
add_artist)r`   ra   locrc   r%   Zaboxrd   r*   r*   r+   test_prevent_rasterization~   s     
rr   c                  C   st  dd } t  }|ddd}|  tdD ]}|jddgd|gddd	 q.| |d
dks`t| |ddksrtt  }|ddd}|  tdD ]}|jddgd|gddd	 q| |d
dkst| |ddkst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sRt| |ddksft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st| |ddkstt  }
d|
_|
ddd}|  tdD ] }|jddgd|gddd	 q&| |
d
dks\t| |
ddksptd S )Nc              	   S   s<   t   }| j|dd |  }W 5 Q R X |d| S )Nr   r   <)r   r   r   rC   count)r    tagr'   r(   r*   r*   r+   	count_tag   s    z%test_count_bitmaps.<locals>.count_tagr.   r6   r   rU   zb-F)r]   imagepath   T)	r   rB   add_subplotZset_axis_offrangerM   rD   Zset_rasterizedZsuppressComposite)rv   fig1ax1nfig2ax2Zfig3ax3Zfig4Zax4Zfig5Zax5r*   r*   r+   test_count_bitmaps   sP    
r   defaultc               	   C   s   t  } | jddddd t }| j|dd | }W 5 Q R X tjj|}d}d}t	|
d	| d
| ddksztd| |d	| dj kstd S )N      ?z\textwonT)Zusetexr   r   http://www.w3.org/2000/svgzSFSS3583-8e.//{z}path[@d][@id="z"]r.   #z}use)r   textr   r   r   r   etreeElementTree
fromstringlenfindallrD   findattribvalues)r    r'   r(   treensZwon_idr*   r*   r+   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 5 Q R X | |krd|kstd S )Nr8   Z	condensed)rR   zfont.stretchzUnlikely to appear by chancez
2019-06-30r.   z
2019-01-01z
2019-12-31r   r7   r   r   )r   rS   updater   r   r   Z
datetime64rj   rk   r   r   seekreadrC   rD   )rF   r    r!   r'   r(   r*   r*   r+   "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d
ddgd
ddgdd |j	ddd
gdd |
  | }|d
ddgd
ddg | jddd}|	dddgd
ddg | jddd}|	d
dgd
dgd
dg | j  i }t| jddD ]:\}}| rd|jj d| }|||< || qt  }	| j|	dd |	  }
W 5 Q R X  fdd}| D ]$\}}|||r||
kstqdS )z+Test that object gid appears in output svg.r   )	OffsetBox)Tick   g      ?g       @g      @auto)Zaspectr.   r7   rf   Z	myscatter)labelZmyplot   Zpolar)Z
projectiong      ?   Z3dT)Zinclude_selfZtest123_r   r   c                    s   t | rdS t |tr6| dkr(dS |jd kr6dS t |tjr| \}}t|t|  krjdkrrn ndS t|dr|jd krdS t |r|	 }|dkrdS |
 }|t|k s|t|krdS dS )NF r.   axesr   T)
isinstancer   Zget_textr   r   ZLine2Dget_datar   hasattrZget_locZget_view_intervalminmax)gidobjZxdataZydatarq   vir   r   r*   r+   include  s*    


 
ztest_gid.<locals>.includeN)matplotlib.offsetboxr   Zmatplotlib.axisr   r   rB   rz   r;   r2   rM   ZlegendZtwinxbarZcanvasZdraw	enumerateZfindobjZget_visible	__class____name__set_gidr   r   r   rC   itemsrD   )r    r}   Zax1ar   r   Zgdicidxr   r   r'   r(   r   r*   r   r+   test_gid   s8    


r   c                   C   s   t jt ddd d S )NZsvgzZtight)r   Zbbox_inches)r   r   r   r*   r*   r*   r+   test_savefig_tight.  s    r   c                  C   s   t  \} }|dddgdddg}|ddd g t ddgddg\}|d	 t }| j|d
d | }dD ]}d| |kstt	qtd S )Nr.   r7   rf   r   r6   ry   zhttps://example.com/foozhttps://example.com/barzhttps://example.com/bazr   r   )s   foos   bars   bazs   https://example.com/)
r   r   r2   Zset_urlsrM   set_urlr   r   r   rD   )r    r!   r1   pr%   vr*   r*   r+   rE   4  s    
rE   c                 C   s6  |  dd t \}}|dddgdddg t|j D ]\}}|d	|  q>t \}}|dddgdddg t|j D ],\}}|jd	|  |j	d	|  qt
 }|j|d
d | }t
 }|j|d
d | }tt|j D ] }d	| d|kstq||ks2td S )NSOURCE_DATE_EPOCH19680801r.   r7   rf   r   r6   ry   zhttps://example.com/r   r   ascii)setenvr   r   r2   r   ZyaxisZget_major_ticksr   label1label2r   r   r   r{   r   encoderD   )monkeypatchr|   r!   iZtickr   Zb1Zb2r*   r*   r+   test_url_tickH  s&    r   c              	   C   s   |  dd t \}}t  }|j|dd |  }W 5 Q R X tj|ksRt	d|ks^t	d|ksjt	d|ksvt	t ,}|j|dd d d d dd	 |  }W 5 Q R X tj|kst	d|kst	d|kst	d|kst	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   rC   ro   __version__rD   )r   r    r!   r'   r(   r*   r*   r+   test_svg_default_metadatac  s&     
r   c              
   C   s  |  dd tjdddd}d}d}d	}d
}t \}}|D ]}t *}	|j|	d| d id |	 	 }
W 5 Q R X t
jj|
}|d| d| d| d\}|D ]R}|d| | }||kr|rtq|\}t
jjj|dd}|| |kstqq<d S )Nr   r   r   r   r   )Z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/ZWorkunicode)encoding)r   ro   r   r   r   r   r   titler   rC   r   r   r   r   r   rD   tostring)r   Zmetadata_containsSVGNSRDFNSCCNSDCNSr    r!   namer'   r(   rootZworkkeydataZxmlstrr*   r*   r+   test_svg_clear_default_metadata  s0      r   c               	   C   sp   t  \} }t ,}| j|dd d d d dd |  }W 5 Q R X d}tjj	|}|
d| drltd S )Nr   r   r   r   r   r   )r   r   r   r   r   rC   r   r   r   r   r   rD   )r    r!   r'   r(   r   r   r*   r*   r+   test_svg_clear_all_metadata  s     
r   c                     sj  dddddddg} dd	d
ddg}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 5 Q R X d}d d}d}tj	j
|}	|	d| d  d\}
d d! |	d| d"D }||d gkst fd#d!|
d| d$| d%D }||d gks8td2| D ]P}|dkrRq@d'd! |
d| d$| |  D }||| gks@tq@|D ]\}|d
krqd(d! |
d| d$| |  d)| d*| d"
D }||| kstqd+d! |
d| d$| d,D }|d-gks&td.d! |
d| d$| d/  d0  d1	D }||d
 ksftd S )3NZCoverageZ
IdentifierZLanguageZRelationZSourceTitler   ZContributorr   KeywordsZ	PublisherZRightsi     r.   r7   rf   zdescription
text)r   Descriptionc                 S   s   i | ]}|| d qS )z foor*   .0r/   r*   r*   r+   
<dictcomp>  s      z%test_svg_metadata.<locals>.<dictcomp>c                 S   s"   i | ]}|| d | dgqS )z barz bazr*   r   r*   r*   r+   r     s      r   r   r   r   r   r   r   r   ZRDFc                 S   s   g | ]
}|j qS r*   r   r   noder*   r*   r+   
<listcomp>  s     z%test_svg_metadata.<locals>.<listcomp>r   c                    s   g | ]}|j   d  qS )resource)r   r   r   r*   r+   r     s   zWork/r   r   c                 S   s   g | ]
}|j qS r*   r   r   r*   r*   r+   r     s   c                 S   s   g | ]
}|j qS r*   r   r   r*   r*   r+   r     s   /zAgent/c                 S   s   g | ]
}|j qS r*   r   r   r*   r*   r+   r     s     r   z1968-08-01/1968-08-02T01:02:03c                 S   s   g | ]
}|j qS r*   r   r   r*   r*   r+   r     s     zsubject/zBag/li)r   )datetimer   r   rB   r   r   r   rC   r   r   r   r   r   rD   lower)Zsingle_valueZmulti_valuer   r    r'   r(   r   r   r   r   rdftitlestypesr/   r   datesr*   r   r+   test_svg_metadata  sf    
 


$""r   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   r   sizer   rx   Zfonttype333333?ffffff?"   There are 几个汉字 in between!)fmFontPropertiesr   findfontr   pytestskipr   rcrB   r   fpr    r*   r*   r+   test_multi_font_type3  s    
r  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 )Nr   r   r   r   r   r8   r  r   r  r  r  r  r  r  )r	  r
  r   r  r   r  r  r   rB   r  r   r  r*   r*   r+   test_multi_font_type42  s    
r  zmetadata,error,messager   r.   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 strZFooBarzUnknown metadata keyc              
   C   sH   t j||d0 t }t }|j|d| d W 5 Q R X W 5 Q R X d S )N)matchr   r   )r  Zraisesr   r   rB   r   )r   errormessager'   r    r*   r*   r+   test_svg_incorrect_metadata	  s    r  c               	   C   sV   t  } | jddddd t ,}| j|dd |  }d|ksHtW 5 Q R X d S )Nr   z<'"&>)r   r   r   z&lt;&apos;&quot;&amp;&gt;")r   rB   r   r   r   r   rC   rD   )r    r'   r(   r*   r*   r+   test_svg_escape!  s    r  font_strz7'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rdddg}	n|g}	|	D ]:}
|jddd||
gdd |jddd|f||
fdd q|d t }|j|dd | }W 5 Q R X tjj|}d}d}|d| dD ]N}|d7 }tdd t| d dD d  }|| d!|  ksBtqB|t	|jkstd S )"Nr   r   r   r   c                 S   s
   |  dS )N'stripr"   r*   r*   r+   <lambda>=      z&test_svg_font_string.<locals>.<lambda>z, zfont.z	font.sizer8   rR   z
sans-serifZsansz
sans serifr   r  center)r   Zhar-   offr   r   r   r   r   z}textr.   c                 s   s&   | ]}t d d | dV  qdS )c                 S   s   |   S )Nr  r  r*   r*   r+   r  _  r   z0test_svg_font_string.<locals>.<genexpr>.<lambda>:N)mapr  split)r   r   r*   r*   r+   	<genexpr>^  s   z'test_svg_font_string.<locals>.<genexpr>style;r   zpx )r	  r
  r   r  r   r  r  r$  r%  r   r   rS   r   r   Zaxisr   r   r   r   r   r   r   r   dictr   rD   Ztexts)r  r  r  explicitrestZgenericr  r    r!   Zgeneric_optionsZgeneric_namer'   r(   r   r   Z
text_countZtext_elementZ	font_infor*   r*   r+   test_svg_font_string*  sT    
 



 
 
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 5 Q R X d}||kstd S )N)    r-  )g333333?g?r-   )Zzoom)g      ^@g      Tr   zoffset pointsr   z->zangle,angleA=0,angleB=90,rad=3)Z
arrowstyleZconnectionstyle)ZxyboxZxycoordsZ	boxcoordspadZ
arrowpropsza test for issue 20044r   r   zutf-8z<g id="a test for issue 20044">)r   rB   rz   r   Zonesr
   rw   r   r   r)  r   rp   r   r   r   rC   rD   )	r    r!   Zarr_imgZxyZimageboxabr'   r(   rF   r*   r*   r+   test_annotationbbox_gidg  s.    
	

r0  )>r   ior   pathlibr   xml.etree.ElementTreer   xml.parsers.expatr  Znumpyr   Z
matplotlibro   Zmatplotlib.figurer   Zmatplotlib.textr   Zmatplotlib.pyplotZpyplotr   Zmatplotlib.testing.decoratorsr   r   Zmatplotlib.testing._markersr   r	   r	  r   r
   r   r,   r4   r?   r@   rQ   rT   re   rl   rr   r   r'  contextr   r   r   r   rE   r   r   r   r   r   r  r  markZparametrize	TypeError
ValueErrorr  r  r,  r0  r*   r*   r*   r+   <module>   s   



	


:
?"!9





	0