o
    g<                     @   s  d dl 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
Zd dl
mZ d dlmZ d dlmZ d dlmZmZ e	jjed dd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 d d! Z!ed"d# Z"e	j#d$ed%d& Z$d'd( Z%d)d* Z&d+d, Z'd-d. Z(d/d0 Z)d1d2 Z*d3d4 Z+d5d6 Z,d7d8 Z-dS )9    N)BytesIO)dviread)Figure)image_comparisoncheck_figures_equalTz"This test needs a TeX installation)reasonc                  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/cloudmapper/venv/lib/python3.10/site-packages/matplotlib/tests/test_backend_svg.pytest_visibility   s   


r-   zfill_black_with_alpha.svg)remove_textc                  C   s,   t  \} }|jg dg ddddd d S )N)r   皙?   )r   r   r   kr/   i'  )r"   r#   r&   alphas)r   r   scatterr    r!   r+   r+   r,   test_fill_black_with_alpha)   s    r6   noscalec                  C   sR   t t dddt ddd\} }t |d }t \}}|j|ddd d S )N   r0      graynone)cmapinterpolation)r   meshgridaranger   r   r   imshow)XYZr    r!   r+   r+   r,   test_noscale/   s   $rE   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+   r,   rF   8   s   
rF   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   r@   
set_xlabel
set_ylabel	set_titler5   r+   r+   r,   test_bold_font_outputF   s
   
rX   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.fonttyperM   rN   rO   rP   rQ   rS   )	r   rcParamsr   rT   r   r@   rU   rV   rW   r5   r+   r+   r,   (test_bold_font_output_with_none_fonttypeO   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@-rrM   )r&   lwr0   r%   Tr&   rb   
rasterized)r   r@   cosr   r   rT   fig_testfig_reftr"   r#   ax_refax_testr+   r+   r,   test_rasterizedY   s   

rl   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   r^   r_      gg?r`   ra   rM   Trc   r0   r%   Fr:   gm)r&   rb   rd   zorderg?gffffff?g333333?)r   r@   re   r   r   set_xlimset_ylimrT   rf   r+   r+   r,   test_rasterized_orderingh   s"   

 rs   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   rJ   count)r    tagr(   r)   r+   r+   r,   	count_tag   s
   z%test_count_bitmaps.<locals>.count_tagr0   r9   r   r\   zb-F)rd   imagepath   T)r   rH   add_subplotset_axis_offrangerT   set_rasterizedsuppressComposite)rw   fig1ax1nfig2ax2fig3ax3fig4ax4fig5ax5r+   r+   r,   test_count_bitmaps   sP   
r   c              	   C   s   dd }|  tjd| tjddd t \}}|ddd t	 -}t
t |j|d	d
 W d   n1 s<w   Y  W d   dS W d   dS 1 sTw   Y  dS )z:An error is raised if a TeX font lacks a Type-1 equivalentc                  _   s   t jddd d d dS )Ntexfontz	Some Font)texnamepsnameeffectsencodingfilename)r   PsFont)argskwargsr+   r+   r,   psfont   s   z#test_missing_psfont.<locals>.psfont__getitem__textTusetex      ?hellor   r   N)setattrr   
PsfontsMapmplrcr   r   r   tempfileTemporaryFilepytestraises
ValueErrorr   )monkeypatchr   r    r!   tmpfiler+   r+   r,   test_missing_psfont   s   Pr   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 )Nr   z\textwonTr   r   r   zhttp://www.w3.org/2000/svgzSFSS3583-8ez.//{z}path[@d][@id="z"]r0   #z}use)r   r   r   r   r   r   etreeElementTree
fromstringlenfindallfindattribvalues)r    r(   r)   treenswon_idr+   r+   r,   test_unicode_won   s   
$*r   c                  C   s   dt jd< d} t  \}}|tdd|  |tdtd |dd	 t }|j	|d
d |
d |  }W d    n1 sIw   Y  | |v sTJ dD ]}d| d|v sbJ qVd S )Nr<   rY   zUnlikely to appear by chancez
2019-06-30r0   z
2019-01-01z
2019-12-31r   r:   r   r   )familyweightstretchstylesizezfont-:)r   rZ   r   r   r   
datetime64rq   rr   r   r   seekreadrJ   )rL   r    r!   r(   r)   propr+   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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r0   r:   rm   	myscatter)label)r:   rm   r0   myplot   polar)
projection)r   g      ?rm      3dr0   r:   T)include_selftest123_r   r   Nc                    s   t | rdS t |tjr| dkrdS |jd u rdS t |tjrD| \}}t|t|  kr7dkr8dS  t|drB|jd u rDdS t |re|	 }|dkrSdS |
 }|t|k sc|t|kredS dS )NF r0   axesr   T)
isinstancer   Textget_textr   Line2Dget_datar   hasattrget_locget_view_intervalminmax)gidobjxdataydatalocvir   r   r+   r,   include  s,   


ztest_gid.<locals>.include)matplotlib.offsetboxr   matplotlib.axisr   r   rH   r{   rA   r4   rT   legendtwinxbarcanvasdraw	enumeratefindobjget_visible	__class____name__set_gidr   r   r   rJ   items)r    r   ax1ar   r   gdicidxr   r   r(   r)   r   r+   r   r,   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+   r,   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   r9   rz   )zhttp://example.com/foozhttp://example.com/barNr0   rm   rz   r9   zhttp://example.com/bazr   r   )s   foos   bars   bazs   http://example.com/)	r   r   r4   set_urlsrT   set_urlr   r   r   )r    r!   r3   pr%   vr+   r+   r,   rK   2  s   
rK   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http://example.com/r   r   ascii)setenvr   r   r4   r   yaxisget_major_ticksr   label1label2r   r   r   r}   r   encode)r   r   r!   itickr   b1b2r+   r+   r,   test_url_tickF  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   rJ   r   __version__)r   r    r!   r(   r)   r+   r+   r,   test_svg_default_metadataa  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)r   )r  r   r  r   r   r   r   titler   rJ   r   r   r   r   r   tostring)r   metadata_containsSVGNSRDFNSCCNSDCNSr    r!   namer(   r)   rootworkkeydataxmlstrr+   r+   r,   test_svg_clear_default_metadata  s4    r3  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   rJ   r   r   r   r   r   )r    r!   r(   r)   r)  r.  r+   r+   r,   test_svg_clear_all_metadata  s   
r4  c                     sd  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 sLw   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s0J d S )+N)Coverage
IdentifierLanguageRelationSourceTitler  )Contributorr  Keywords	PublisherRightsi     r0   r:   rm   zdescription
text)r  Descriptionc                 S   s   i | ]}|| d qS )z foor+   .0r1   r+   r+   r,   
<dictcomp>  s    z%test_svg_metadata.<locals>.<dictcomp>c                 S   s"   i | ]}|| d | dgqS )z barz bazr+   rA  r+   r+   r,   rC    s   " r   r  r  r  r   r!  r"  r#  RDFc                 S      g | ]}|j qS r+   r   rB  noder+   r+   r,   
<listcomp>      z%test_svg_metadata.<locals>.<listcomp>r&  r:  c                    s   g | ]
}|j   d  qS )resource)r   rG  r*  r+   r,   rI    s    zWork/r  r  r@  c                 S   rE  r+   rF  rG  r+   r+   r,   rI    s    
r<  c                 S   rE  r+   rF  rG  r+   r+   r,   rI    s    /zAgent/c                 S   rE  r+   rF  rG  r+   r+   r,   rI    rJ  r  z1968-08-01/1968-08-02T01:02:03c                 S   rE  r+   rF  rG  r+   r+   r,   rI    rJ  zsubject/zBag/li)datetimer  r   r   r   r   r   rJ   r   r   r   r   r   lower)single_valuemulti_valuer  r    r!   r(   r)   r)  r+  r,  r.  rdftitlestypesr1   r   datesr+   rL  r,   test_svg_metadata  sd   
$""rW  ).rO  ior   r   xml.etree.ElementTreer   xml.parsers.expatnumpyr   r   
matplotlibr   r   matplotlib.figurer   matplotlib.pyplotpyplotr   matplotlib.testing.decoratorsr   r   markskipifcheckdep_usetexneeds_usetexr-   r6   rE   rF   rX   r[   rl   rs   r   r   r   r   r   r   r   rK   r  r  r3  r4  rW  r+   r+   r+   r,   <module>   sZ    




	

9

?"!