U \ a) @s<ddlZddlmZddlZddlZddlZddlZddlZ ddl m Z ddl mZmZddlmZmZddlmZmZddlmZmZmZedd\ZZejjed  d d Zejjed  d d Zejjed dd Z ejjde j!j"j#kdd Z$idfddZ%ddZ&ej'ddddddgddZ(eej)dedgd d!d"d#Z*eejjed$ d%d ej)ded&gd d!d'd(Z+eeej,d ej)dd)d*Z-eej,d ej)dd+d,Z.eej)ded-gd d!d.d/Z/eej,d ej)dd0d1Z0ej,d ej)dej'd2ej1de gd3ej1d egd3ej1d egd3gd4d5Z2ej,d ej)dej'd2ej1de gd3ej1d egd3ej1d egd3gd6d7Z3ed8d9Z4ed:d;Z5ee$dd?Z7ed@gdAej'dBdCej)ddDdEZ8dS)FN)BytesIO)_has_tex_package_check_for_pgf)compare_imagesImageComparisonFailure)PdfPagescommon_texification)_image_directoriescheck_figures_equalimage_comparisoncCsdS)Nz dummy funcr r r E/tmp/pip-unpacked-wheel-wjyw_3jo/matplotlib/tests/test_backend_pgf.pyrxelatexzxelatex + pgf is required)reasonpdflatexzpdflatex + pgf is requiredlualatexzlualatex + pgf is requiredZepsz*This test needs a ghostscript installationcCsbtjt|}tj|f|tjtd|}ttjt||t |||d}|r^t |dS)Nz expected_%stol) ospathjoin result_dirpltsavefigshutilcopyfile baseline_dirrr)fnamesavefig_kwargsractualexpectederrr r r compare_figuresr$cCsttddd}t||ddt|d|ddtjddgddgd d d d td d ddd gdddddgdtjdgdgdd dtjddddddtjddddt ddt dddS)Nrzb-zg>g?z//Z lightgrayZred)ZhatchZ facecolorZ edgecolorg?bg??ro)Z markersizeu/unicode (ü, °, µ) and math ($\mu_i = x_i^2$)right)ZhaZfontsizez*sans-serif, blue, $\frac{\sqrt{x}}{y^2}$.. sans-serifZblue)familycolor) rfigurenpZlinspaceplotZ fill_betweenfilltextZylabelxlimylim)xr r r create_figure*s&"  r;zplain_text, escaped_text)zquad_sum: $\sum x_i^2$z'quad\_sum: \(\displaystyle \sum x_i^2\))no \$splits \$ herer<)Zwith_underscoreszwith\_underscores)z% not a commentz\% not a comment)z^notz\^notcCst||kstdS)N)rAssertionError)Z plain_textZ escaped_textr r r test_common_texificationDsr>pgfzpgf_xelatex.pdfdefault)stylecCs ddd}tj|tdS)NserifF font.family pgf.rcfonts)mplrcParamsupdater;)Z rc_xelatexr r r test_xelatexPs  rIucsz needs ucs.styzpgf_pdflatex.pdfcCs:tjdrtdddddd}tj|tdS)NZAPPVEYORzBpdflatex test does not work on appveyor due to missing LaTeX fontsrBFrz4\usepackage[utf8x]{inputenc}\usepackage[T1]{fontenc})rDrE pgf.texsystem pgf.preamble) renvirongetpytestZxfailrFrGrHr;) rc_pdflatexr r r test_pdflatex[s   rQc Csdddddddddd d dd d d g}ddg}t|D]r\}}t|ZdD]0\}}|tjdkrNt|sNtd|dqNttd|d||dW5QRXq6dS)Nr0g? Fr)rD font.sizefigure.subplot.leftlines.markersizerErKZ monospaceg?r/rzG\usepackage[utf8x]{inputenc}\usepackage[T1]{fontenc}\usepackage{sfmath})rDrTrUrVrErKrLr))sfmathrX)Zutf8xrJrLzneeds z.styzpgf_rcupdate%d.pdfr%r) enumeraterFZ rc_contextrGrrOskipr;r$)Zrc_setsriZrc_setZ substringpkgr r r test_rcupdatens4  r]cCsRtjdddtddgddgtddtddtjtdd dS) NrBFrCr(g}Ô%ITrr%pdfformat) rFrGrHrr5r8r9rrr r r r test_pathclips   razpgf_mixedmode.pdfcCsPtjdddtjddddddf\}}t|d|dddS) NrBFrCr%yD@r'T)rFrGrHr4ZogridrZpcolorZset_rasterized)YXr r r test_mixedmodesrecCsttjdddtdd\}\}}|td|tdt| |j }t dd|id d dS) NrBFrCr%r'zpgf_bbox_inches.pdf bbox_inchesr)r r) rFrGrHrsubplotsr5range tight_layoutZget_window_extentZ transformedZdpi_scale_transinvertedr$)figax1ax2Zbboxr r r test_bbox_inchess rosystem)Zmarksc Csdd|d}tj|t\}}|td|tjdd\}}|td|tj t d|d}d d d d t j d ddt t dddd}t||d4}|||||||dkstW5QRXdS)NrBF)rDrErKrf)r)r')ZfigsizeZ pdfpages_.pdfmeMultipage PDF with pgf Test pagetest,pdf,multipager%rtzinfoUnknownZAuthorZTitleSubjectZKeywordsZModDateZTrappedmetadatar))rFrGrHrrhr5rirjrrrrdatetimetimezone timedeltarrZ get_pagecountr=) rprPZfig1rmZfig2rnrmdr^r r r test_pdf_pagess:       rc CsNtd}|ddtjd|it\}}|t ddddd t j d d d t t d ddd}t jtd|d}t||d}||W5QRX|j|}dd|jD}W5QRXd|kr|d=d|kr|d=|d} | dtjks |dkrd| ks t|dddtjdd dddd d!ksJtdS)"NpikepdfSOURCE_DATE_EPOCH0rKrfrrrsrtrurvrwr%rrxTruer{Zpdfpages_meta_check_rqr}cSsi|]\}}|t|qSr )str).0kvr r r sz1test_pdf_pages_metadata_check..z/PTEX.FullBannerz/PTEX.Fullbannerz /ProducerzMatplotlib pgf backend vrZLuaTeXzD:19700101000000Zz Matplotlib vz, https://matplotlib.orgzD:19680801000000Zz/True)z/Authorz /CreationDatez/Creatorz /Keywordsz/ModDatez/Subjectz/Titlez/Trapped)rOZ importorskipZsetenvrFrGrHrrhr5rirrrrrrrrrZPdfopenZdocinfoitemspop __version__r=) Z monkeypatchrprrlaxrrr^infoZproducerr r r test_pdf_pages_metadata_checksT      rc Cs^t}|dtt|jtddW5QRXt}|d|jtdddS)Nz\oopsr?r_z this is ok)rr3ZsuptitlerOZraises ValueErrorrr)rlr r r test_tex_restart_after_errors   rcCs:t\}}|ddgddgg|jtdddddS) Nrr%r'r)r^r?Ztight)r`backendrg)rrhZimshowrrrlrr r r test_bbox_inches_tight!s rcCs"t\}}|jtddddS)NZpngr?)r`r)rrhrrrr r r test_png(s rc CsZ|d.dtjd<tdddtjtddW5QRXdd d |jDksVtdS) NWARNINGzthis-font-does-not-existrDr,z hello, worldr?r_z/Ignoring unknown font: this-font-does-not-existcSsg|] }|qSr ) getMessage)rrr r r 6sz%test_unknown_font..) Zat_levelrFrGrZfigtextrrrecordsr=)Zcaplogr r r test_unknown_font0s  rr^) extensions texsystem)rrrcCs@t|st|d|tjd<|ddd|ddddS)Nz + pgf is requiredrKr,z$-1$u$−1$)rrOrZrFrGr7)Zfig_testZfig_refrr r r test_minus_signs_with_tex:s  r)9riorrrZnumpyr4rOZ matplotlibrFZmatplotlib.pyplotZpyplotrZmatplotlib.testingrrZmatplotlib.testing.comparerrZmatplotlib.backends.backend_pgfrrZmatplotlib.testing.decoratorsr r r rrmarkZskipifZ needs_xelatexZneeds_pdflatexZneeds_lualatexZtestingcompare converterZneeds_ghostscriptr$r;Z parametrizer>rrIrQrAr]rareroparamrrrrrrrr r r r s                    #  2