U \ a<@sddlZddlmZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl mZddlmZddlmZmZe jje d ddZd d Zed gdd d dZedgdd ddZddZedgddZedgddZeddddZeddZ d d!Z!ed"d#Z"e j#d$ed%d&Z$d'd(Z%d)d*Z&d+d,Z'd-d.Z(d/d0Z)d1d2Z*d3d4Z+d5d6Z,d7d8Z-dS)9N)BytesIO)dviread)Figure)image_comparisoncheck_figures_equalTz"This test needs a TeX installation)reasonc Cst\}}tddtjd}t|}t|}|j|||dd\}}}|D]}|dqPt } |j | dd| } W5QRXt j j} | | dS) Nr2ko)yerrfmtFsvgformat)pltsubplotsnpZlinspacepisinZ ones_likeZerrorbarZ set_visiblersavefiggetvaluexmlparsersexpat ParserCreateParse) figaxxyr abcZartistfdbufparserr&E/tmp/pip-unpacked-wheel-wjyw_3jo/matplotlib/tests/test_backend_svg.pytest_visibilitys     r(zfill_black_with_alpha.svg)Z remove_textcCs0t\}}|jdddgdddgdddddS)Nrg?ki')rrr"alphas)rrscatterrrr&r&r'test_fill_black_with_alpha)s r/ZnoscalecCsRttdddtddd\}}t|d}t\}}|j|ddddS)Nr)Zgraynone)Zcmap interpolation)rZmeshgridarangerrrimshow)XYZrrr&r&r' test_noscale/s$ r:c Cs`t}d}|jd|dt }|j|dd|}W5QRXd|}||ks\tdS)Nz$http://test_text_urls.matplotlib.orgtest_text_urls)urlr rz) rfigureZsuptitlerrrdecoderAssertionError)rtest_urlr#r$expectedr&r&r'r;8s r;zbold_font_output.svgcCsNt\}}|tdtd|d|jddd|jddddS)N nonbold-xlabel bold-ylabelboldZ fontweight bold-title)rrplotrr5 set_xlabel set_ylabel set_titler.r&r&r'test_bold_font_outputFs   rLz'bold_font_output_with_none_fonttype.svgcCsXdtjd<t\}}|tdtd|d|jddd|jddddS) Nr3 svg.fonttyperBrCrDrErFrG) rrcParamsrrHrr5rIrJrKr.r&r&r'(test_bold_font_output_with_none_fonttypeOs    rO)ZtolcCstddd}t|}t|}|}|j||dddd|j|d|dd dd|}|j||dddd d |j|d|dd dd d dS) Nrdffffff@-rrB)r"lwr)r!Tr"rU rasterized)rr5cosrrrHZfig_testZfig_reftrrZax_refZax_testr&r&r'test_rasterizedYs  r[c Cs8tddd}t|}t|}|}|dd|dd|j||ddd d d |j|d |dd d dd |j|d|ddd d d |j|d|ddd d d |}|dd|dd|j||ddd d dd|j|d|ddd d dd|j|d|ddd d dd|j|d |dd d ddddS)NrrQrRgg?rSrTrBTrVr)r!Fr2gm)r"rUrWZzorderg?gffffff?g333333?)rr5rXrrset_xlimset_ylimrHrYr&r&r'test_rasterized_orderinghs"      rac Cstdd}t}|ddd}|tdD]}|jddgd|gddd q.||d dks`t||d d ksrtt}|ddd}|tdD]}|jddgd|gdd d q||d dkst||d dkstt}|ddd}|tdD]:}|jddg|dgddd |jddgd|gdd d q||d dksRt||d d ksftt}|ddd} | | d tdD]:}| jddg|dgddd | jddgd|gdd d q||d dkst||d dkstt} d | _| ddd} | tdD] }| jddgd|gdd d q&|| d dks\t|| d dksptdS)Nc Ss<t }|j|dd|}W5QRX|d|S)Nr r<)rrrr>count)rtagr#r$r&r&r' count_tagsz%test_count_bitmaps..count_tagr)r1rrPzb-F)rWimagepathT) rr= add_subplotZ set_axis_offrangerHr?Zset_rasterizedZsuppressComposite) refig1ax1nfig2ax2Zfig3ax3Zfig4Zax4Zfig5Zax5r&r&r'test_count_bitmapssP      rqc Cs|dd}|tjd|tjdddt\}}|dddt *}t t |j |d d W5QRXW5QRXd S) z:An error is raised if a TeX font lacks a Type-1 equivalentc_stjddddddS)NZtexfontz Some Font)ZtexnameZpsnameZeffectsencodingfilename)rZPsFont)argskwargsr&r&r'psfonts z#test_missing_psfont..psfont __getitem__textTZusetex?Zhellor rN)setattrrZ PsfontsMapmplrcrrrxtempfile TemporaryFilepytestZraises ValueErrorr) monkeypatchrvrrZtmpfiler&r&r'test_missing_psfonts rdefaultc Cst}|jdddddt}|j|dd|}W5QRXtjj|}d}d}t | d |d |d d kszt d || d |dj kst dS)Nrzz\textwonTryr rzhttp://www.w3.org/2000/svgz SFSS3583-8ez.//{z}path[@d][@id="z"]r)#z}use)rrxrrrretree ElementTree fromstringlenfindallr?findattribvalues)rr#r$treensZwon_idr&r&r'test_unicode_wons$rc Csdtjd<d}t\}}|tdd||tdtd|dd t*}|j |d d | d| }W5QRX||kst d D]}d |d|kst qdS)Nr3rMzUnlikely to appear by chancez 2019-06-30r)z 2019-01-01z 2019-12-31rr2r r)familyZweightZstretchstylesizezfont-:)rrNrrxrZ datetime64r_r`rrseekreadr>r?)rArrr#r$propr&r&r'"test_svgnone_with_data_coordinatess     rc sddlmddlmt}|d}|jddgddggdd |jd d d gd d d gd d|j d d d gdd| | }| d d d gd d d g|jddd}| ddd gd d d g|jddd}| d d gd d gd d g|j i}t|jddD]:\}}|rd|jjd|}|||<||qt } |j| dd| } W5QRXfdd} |D]$\}}| ||r|| kstqdS)z+Test that object gid appears in output svg.r) OffsetBox)Tickg?g@g@auto)Zaspectr)r2r\Z myscatter)labelZmyplotZpolar)Z projectiong?Z3dT)Z include_selfZtest123_r rcst|rdSt|tjr8|dkr*dS|jdkr8dSt|tjr|\}}t|t|krldkrtnndSt|dr|jdkrdSt|r| }|dkrdS| }|t |ks|t |krdSdS)NFr)axesrT) isinstancerTextZget_textrZLine2Dget_datarhasattrZget_locZget_view_intervalminmax)gidobjZxdataZydatalocvirrr&r'includes*        ztest_gid..includeN)Zmatplotlib.offsetboxrZmatplotlib.axisrrr=rir6r-rHZlegendZtwinxbarZcanvasZdraw enumerateZfindobjZ get_visible __class____name__Zset_gidrrrr>itemsr?) rrlZax1arorpZgdicidxrrr#r$rr&rr'test_gids8      rcCstjtddddS)NZsvgzZtight)rZ bbox_inches)rrrr&r&r&r'test_savefig_tight,srcCst\}}|dddgdddg}|dddgtddgddg\}|d t}|j|d d |}d D]}d ||kstt qtdS)Nr)r2r\rr1rhzhttp://example.com/foozhttp://example.com/barzhttp://example.com/bazr r)sfoosbarsbazshttp://example.com/) rrr-Zset_urlsrHset_urlrrrr?)rrr,pr!vr&r&r'r@2s  r@cCs6|ddt\}}|dddgdddgt|jD]\}}|d |q>t\}}|dddgdddgt|jD],\}}|jd ||j d |qt }|j |d d | }t }|j |d d | }t t|jD] }d |d |kstq||ks2tdS) NSOURCE_DATE_EPOCH19680801r)r2r\rr1rhzhttp://example.com/r rascii)setenvrrr-rZyaxisZget_major_ticksrlabel1label2rrrrjrencoder?)rrkriZtickrnZb1Zb2r&r&r' test_url_tickFs&   rc Cs|ddt\}}t }|j|dd|}W5QRXtj|ksRt d|ks^t d|ksjt d|ksvt t,}|j|ddddddd |}W5QRXtj|kst d|kst d|kst d|kst dS) Nrrr r 1970-08-16 image/svg+xml StillImageDateCreatorZFormatTypermetadata) rrrrrrr>r| __version__r?)rrrr#r$r&r&r'test_svg_default_metadataas&         rc Cs|ddtjdddd}d}d}d }d }t\}}|D]}t*} |j| d |did |  } W5QRXt j j | } | d |d|d|d\} |D]R} | d || }| |kr|rtq|\}t j j j|dd}|| |kstqqrrrrrr?tostring)rZmetadata_containsSVGNSRDFNSCCNSDCNSrrnamer#r$rootZworkkeydataZxmlstrr&r&r'test_svg_clear_default_metadatas0   rc Cspt\}}t,}|j|ddddddd|}W5QRXd}tjj |}| d|drlt dS)Nr rrrrr) rrrrrr>rrrrrr?)rrr#r$rrr&r&r'test_svg_clear_all_metadatas   rcspdddddddg}dd d d d g}td ddtd dddddgdddd|Ddd|D}t\}}t"}|j|d|d|}W5QRXd}dd}d} tj j |} | d|dd\} d d!| d|d"D} | |dgkst fd#d!| d|d$| d%D} | |dgks>t d2|D]P}|dkrXqFd'd!| d|d$| |D}|||gksFt qF|D]\}|d krqd(d!| d|d$| |d)|d*| d" D}|||kst qd+d!| d|d$| d,D}|d-gks,t d.d!| d|d$| d/d0d1 D}||d kslt dS)3NZCoverageZ IdentifierZLanguageZRelationZSourceZTitlerZ ContributorrZKeywordsZ PublisherZRightsir)r2r\zdescription text)r DescriptioncSsi|]}||dqS)z foor&.0r*r&r&r' sz%test_svg_metadata..cSs"i|]}||d|dgqS)z barz bazr&rr&r&r'rsr rrrrrrrZRDFcSsg|] }|jqSr&rxrnoder&r&r' sz%test_svg_metadata..rcsg|]}|jdqS)resource)rrrr&r'rszWork/rrcSsg|] }|jqSr&rrr&r&r'rscSsg|] }|jqSr&rrr&r&r'rs/zAgent/cSsg|] }|jqSr&rrr&r&r'rsrz1968-08-01/1968-08-02T01:02:03cSsg|] }|jqSr&rrr&r&r'rszsubject/zBag/li)r)datetimerrrrrrr>rrrrrr?lower)Z single_valueZ multi_valuerrrr#r$rrrrrdftitlestypesr*rdatesr&rr'test_svg_metadatasf         $""r).riorr~xml.etree.ElementTreerxml.parsers.expatZnumpyrrZ matplotlibr|rZmatplotlib.figurerZmatplotlib.pyplotZpyplotrZmatplotlib.testing.decoratorsrrmarkZskipifZcheckdep_usetexZ needs_usetexr(r/r:r;rLrOr[rarqrrrrrrr@rrrrrr&r&r&r'sX             9   ?"!