ó ±éRc@siddlmZddlmZmZmZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl m Z ddl mZddl mZddl mZddlZddlmZmZmZddlZddd „Zd efd „ƒYZd „Zd „Zdefd„ƒYZdddde dd„Z!d„Z"dS(iÿÿÿÿ(tprint_function(tKnownFailureTesttKnownFailureDidNotFailTesttImageComparisonFailureN(tcbook(tticker(tpyplot(tft2font(tcomparable_formatstcompare_imagestmake_test_filenamecs.ˆdkrd‰n‡‡‡fd†}|S(sD Assume a will fail if *fail_condition* is True. *fail_condition* may also be False or the string 'indeterminate'. *msg* is the error message displayed for the test. If *known_exception_class* is not None, the failure is only known if the exception is an instance of this class. (Default = None) sTest known to failcs:ddl}‡‡‡‡fd†}|jjˆƒ|ƒS(NiÿÿÿÿcsŠyˆ||Ž}WnOtk rd}ˆr^ˆdk rOt|ˆƒsO‚qOntˆƒ‚qe‚nXˆr†ˆdkr†tˆƒ‚n|S(Nt indeterminate(t ExceptiontNonet isinstanceRR(targstkwargstresultterr(tftfail_conditiontknown_exception_classtmsg(sW/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/testing/decorators.pytfailer%s  (tnosettoolstmake_decorator(RRR(RRR(RsW/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/testing/decorators.pytknown_fail_decorator!s N(R (RRRR((RRRsW/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/testing/decorators.pytknownfailureifs  t CleanupTestcBs/eZed„ƒZed„ƒZd„ZRS(cCstjjjƒ|_dS(N(t matplotlibtunitstregistrytcopytoriginal_units_registry(tcls((sW/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/testing/decorators.pyt setup_class:scCsNtjdƒtjjƒtjjjƒtjjj|j ƒt j ƒdS(Ntall( tplttcloseRtteststsetupRR tcleartupdateR"twarningst resetwarnings(R#((sW/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/testing/decorators.pytteardown_class>s   cCs|jƒdS(N(t_func(tself((sW/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/testing/decorators.pyttestHs(t__name__t __module__t classmethodR$R.R1(((sW/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/testing/decorators.pyR9s cCsG|j}t|ƒ}d|jdƒ_t|tfi|d6ƒ}|S(Nt_privateiR/(R2t staticmethodt__get__ttypeR(tfunctnamet new_class((sW/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/testing/decorators.pytcleanupKs  cCs|dkrtSddlm}t|tƒr>||f}ng|D]}|j|ƒ^qE}|jtjƒ}||dko||dkS(Niÿÿÿÿ(tversionii( R tTruet distutilsR=Rtstrt StrictVersionRt__freetype_version__(tverR=txtfound((sW/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/testing/decorators.pytcheck_freetype_versionUs "tImageComparisonTestcBs/eZed„ƒZed„ƒZd„ZRS(cCstjƒ|jƒdS(N(RR$R/(R#((sW/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/testing/decorators.pyR$bs cCs|jdƒxy|jƒD]k}|jdƒ|jjtjƒƒ|jjtjƒƒ|jjtjƒƒ|jjtjƒƒqWdS(Nt( tsuptitletget_axest set_titletxaxistset_major_formatterRt NullFormattertset_minor_formattertyaxis(tfiguretax((sW/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/testing/decorators.pyt remove_texths  c #s–tˆjƒ\}}xzttjƒˆjƒD]`\}}tj|ƒ‰xBˆjD]7}|tƒk}|r{d|}nd}t j j ||ƒd|}|dkrÙt j j |ƒ rÙt j j ||ƒd}nt t j j |t j j|ƒƒdƒ‰t j j ||ƒd|‰t j j |ƒrEtj|ˆƒnt}dˆ}t||dtƒ‡‡‡‡fd †ƒ} | fVqSWq.WdS( Ns&Cannot compare %s files on this systemsNo failure expectedt.tepss.pdftexpectedsDo not have baseline image %sRcsÒˆjrˆjˆƒnˆjˆˆjtˆˆˆjdtƒ}yBtjj ˆƒsot dˆƒ‚n|rˆt d|ƒ‚nWnBt k rÍt ˆj ƒsÇt dˆj tjfƒ‚n‚nXdS(Nt in_decoratorsimage does not exist: %ss=images not close: %(actual)s vs. %(expected)s (RMS %(rms).3f)sBMismatched version of freetype. Test requires '%s', you have '%s'(t _remove_textRStsavefigt_savefig_kwargR t_tolR>tostpathtexistsRRFt_freetype_versionRRRB(R(t actual_fnametexpected_fnameRQR0(sW/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/testing/decorators.pytdo_test‹s&   (t_image_directoriesR/tzipR&t get_fignumst_baseline_imagesRQt _extensionsRR\R]tjoinR^R tbasenametshutiltcopyfileR>RR( R0t baseline_dirt result_dirtfignumtbaselinet extensiont will_failtfail_msgtorig_expected_fnameRb((R`RaRQR0sW/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/testing/decorators.pyR1rs,%    !(R2R3R4R$R6RSR1(((sW/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/testing/decorators.pyRGas i cssˆdkrtdƒ‚nˆdkr9dddg‰nˆdkrQtƒ‰n‡‡‡‡‡‡fd†}|S(sz call signature:: image_comparison(baseline_images=['my_figure'], extensions=None) Compare images generated by the test with those specified in *baseline_images*, which must correspond else an ImageComparisonFailure exception will be raised. Keyword arguments: *baseline_images*: list A list of strings specifying the names of the images generated by calls to :meth:`matplotlib.figure.savefig`. *extensions*: [ None | list ] If *None*, default to all supported extensions. Otherwise, a list of extensions to test. For example ['png','pdf']. *tol*: (default 13) The RMS threshold above which the test is considered failed. *freetype_version*: str or tuple The expected freetype version or range of versions for this test to pass. *remove_text*: bool Remove the title and tick text from the figure before comparison. This does not remove other, more deliberate, text, such as legends and annotations. *savefig_kwarg*: dict Optional arguments that are passed to the savefig method. s!baseline_images must be specifiedtpngtpdftsvgcsq|j}t|ƒ}d|jdƒ_t|tfi|d6ˆd6ˆd6ˆd6ˆd6ˆd6ˆd 6ƒ}|S( NR5iR/RfRgR[R_RXRZ(R2R6R7R8RG(R9R:R;(tbaseline_imagest extensionstfreetype_versionRSt savefig_kwargttol(sW/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/testing/decorators.pytcompare_images_decoratorÝs    N(R t ValueErrortdict(RwRxR{RyRSRzR|((RwRxRyRSRzR{sW/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/testing/decorators.pytimage_comparison©s)    c sg|j}|dkrvtjdƒtjd}tjjtjj|ƒƒ}tjj tjj |ƒdƒd}nŒ|j dƒ}|j dƒ|j dƒdks­t ‚tjj |Œ}ddl‰d‡fd †}||jƒd}tjj|ƒ}tjj |d |ƒ}tjjtjj d |ƒƒ} tjj| ƒs]tj| ƒn|| fS( s„ Compute the baseline and result image directories for testing *func*. Create the result directory if it doesn't exist. t__main__s<test module run as script. guessing baseline image locationsiiRTR(iÿÿÿÿNcsed}xX|jdƒD]G}ˆj||ƒ}\}}}|g}|dk r|jƒqqW|S(s9A version of imp which can handle dots in the module nameRTN(R tsplitt find_moduleR'(t module_nameR]trestsub_modtfilet_(timp(sW/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/testing/decorators.pytfind_dotted_modules  Rwt result_images(R3R,twarntsystargvR\R]tabspathtdirnametsplitextRtpoptAssertionErrorRhRˆR R^Rtmkdirs( R9Rƒt script_nametbasedirtsubdirtmodsR‰tmod_fileRlRm((RˆsW/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/testing/decorators.pyRcús&    )   !(#t __future__Rtmatplotlib.testing.noseclassesRRRR\RŒRjRRtmatplotlib.teststmatplotlib.unitsRRRR&Rtnumpytnptmatplotlib.testing.compareRR R R,R RtobjectRR<RFRGtFalseRRc(((sW/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/testing/decorators.pyts,$      ' H  O