ó m[c@`sdZddlmZmZmZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlmZddlZddlZddlmZddlmZddlmZddlmZdd lmZd d d gZd „Zddd„Zd„Zd#d„Zd„Z e j!dƒj"Z#d„Z$de%fd„ƒYZ&d„Z'iZ(e'ƒd„Z)d„Z*iZ+e,rĢej-ƒrĢd„e+dN(s R[(titert functoolstpartialRUR@R5tlent itertoolstteet enumeratetnexttisliceRR4ttupletmaptFalsetpoll(RYtstreamtprompttntitstitittwindow((s9lib/python2.7/site-packages/matplotlib/testing/compare.pyt_read_to_promptšs$    c C`s²|j s|jjƒdk r½tjjƒ}|jddƒtj|tdƒ×ss --export-png=iRQ(RURhRR tenvirontcopytpoptdevnullRzRt_stderrRRCRDRptOSErrorRytAttributeErrorRfRSRNRttwritetflushtseekRR5tdecodeRvRw(RYtorigtdestRuRytorig_btdest_b((s9lib/python2.7/site-packages/matplotlib/testing/compare.pyt__call__°s6      $   cC`sr|jrn|jjƒdkr>|jjdƒ|jjƒn|jjjƒ|jjjƒ|jjƒndS(Nsquit ( RURhRRERFRttcloseR@R€(RY((s9lib/python2.7/site-packages/matplotlib/testing/compare.pyRXås (t__name__t __module__RZRpR‹RX(((s9lib/python2.7/site-packages/matplotlib/testing/compare.pyRTs  5c`svtjƒ\‰}|dk rP‡fd†}t|ƒtdRRtshutiltcopyfile( tfilenametcacheRt extensiontreasonR™RRštnewnameR,t hash_valuetnew_extt cached_file((s9lib/python2.7/site-packages/matplotlib/testing/compare.pyRM s8     $    cC`sdddd|gS(Ntxmllints--valids --nowarnings--noout((R¤((s9lib/python2.7/site-packages/matplotlib/testing/compare.pyR{GsR’s2.1c C`stjj|ƒs%td|ƒ‚n|jddƒ\}}tj|d ƒ}|d k r ||ƒ}tj |dt dtj dtj ƒ}|j ƒ\}}|j ƒ}|dkr dd j|ƒ} |rā| d |7} n|rł| d |7} nt| ƒ‚q nd S( s7Verify the file through some sort of verification tool.s'%s' does not existR˜iR?R@RAis%File verification command failed: %s RBsStandard output: %s sStandard error: %s N(R R R&R(Rœt verifierstgetRRRCR4RDRERFR( R¤RR¦tverifierRLRJR@RARKR((s9lib/python2.7/site-packages/matplotlib/testing/compare.pytverifyKs"    c C`s»|dd!dkr±|dd!dkr±|j\}}}|j\}}} |t|d|dƒt|d|dƒ…t|d|dƒt|d|dƒ…f}n||fS(Nił’’’iü’’’R‘Ri(tshapetint( t actual_patht actual_imaget expected_pathtexpected_imagetawtahtadtewtehted((s9lib/python2.7/site-packages/matplotlib/testing/compare.pyt crop_to_sameas &LcC`sW|j|jkr3tdj|j|jƒƒ‚ntj||jtƒdjƒƒS(sHCalculate the per-pixel errors, then compute the root mean square error.s;Image sizes do not match expected size: {0} actual size {1}i(R±RRtnptsqrttastypetfloattmean(t expectedImaget actualImage((s9lib/python2.7/site-packages/matplotlib/testing/compare.pyt calculate_rmsls c C`sHtjj|ƒs%td|ƒ‚ntj|ƒjdkrPtd|ƒ‚n|jdƒd}tjj|ƒsˆtd|ƒ‚n|dkrµt|t ƒ}t|t ƒ}nt j |ƒ}t j |ƒ}|dd…dd…dd …f}|dd…dd…dd …f}t ||||ƒ\}}t|d ƒ}|d krltj||ƒrldSn|jtjƒ}|jtjƒ}t||ƒ}||krÆdSt|||ƒtd |d t|ƒdt|ƒdt|ƒd|ƒ} |sDddddddg} djg| D]} | j| ^q#ƒ} n| S(s_ Compare two "image" files checking differences within a tolerance. The two given filenames may point to files which are convertible to PNG via the `.converter` dictionary. The underlying RMS is calculated with the `.calculate_rms` function. Parameters ---------- expected : str The filename of the expected image. actual :str The filename of the actual image. tol : float The tolerance (a color value difference, where 255 is the maximal difference). The test fails if the average pixel difference is greater than this value. in_decorator : bool If called from image_comparison decorator, this should be True. (default=False) Examples -------- img1 = "./baseline/plot.png" img2 = "./output/plot.png" compare_images( img1, img2, 0.001 ): sOutput image %s does not exist.isOutput image file %s is empty.R˜i’’’’s!Baseline image %r does not exist.R–Nis failed-diffgtrmsRRtdiffttols!Error: Image files did not match.sRMS Value: {rms}sExpected: {expected}sActual: {actual}sDifference: {diff}sTolerance: {tol}s (R R R&t ExceptionR tst_sizetsplitR(RMRgR4Rt read_png_intR½RR¾t array_equalRRĄtint16RÅtsave_diff_imagetdictRzRR( RRRČt in_decoratorR¦RĆRÄt diff_imageRĘtresultsR!R"((s9lib/python2.7/site-packages/matplotlib/testing/compare.pyR vsJ %%  ! .c C`sztj|ƒ}tj|ƒ}t||||ƒ\}}tj|ƒjtƒ}tj|ƒjtƒ}|j|jkrœtdj |j|jƒƒ‚ntj ||ƒ}|d9}tj |ddƒjtj ƒ}|j\}}} | dkrGtj ||dfdtj ƒ} || dd…dd…dd…f<| }nd|dd…dd…dfRNtcompiletsearchRPRStobjectRTR•R“R RMR­Rgtcheckdep_xmllintt deprecatedR°R½RÅR RĻ(((s9lib/python2.7/site-packages/matplotlib/testing/compare.pytsP            +    _  6 W