ó ,º]c@@s8ddlmZmZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z mZmZmZmZddlmZmZmZerÓddlZnd„Zd„Zdefd „ƒYZd efd „ƒYZd efd „ƒYZdejfd„ƒYZejedƒZd„Z d„Z!d„Z"d„Z#e$ejdƒ ržejj%ej_&ne'd„Z(e$ejdƒ rÓe ejde(ƒnde)fd„ƒYZ*de*fd„ƒYZ+e'd„Z,e$ejdƒ r4e ejde,ƒndS(i(tprint_functiontabsolute_importN(tdedent(t bind_methodtPY26tPY3tPY2tPY27(t check_outputtSTDOUTtCalledProcessErrorcC@s&|jdƒr|d}nt|ƒS(s, Removes any leading and dedents. s i(t startswithR(tcode((s5/usr/lib/python2.7/site-packages/future/tests/base.pyt reformat_codes cC@s|jdƒ}gt|ƒD]!\}}|jdƒr|^q}gt|ƒD]0\}}|jdƒsz|jdƒrP|^qP}gt|ƒD]!\}}|jdƒr“|^q“}d„}d„}tg|D]}||^qÖƒ} tt|| ƒƒ} tg|D]}||^qƒ} tt|| ƒƒ} tg|D]}||^qFƒ} tt|| ƒƒ}g}x‹tt|ƒƒD]w}||kr³|j| |ƒq||krÓ|j| |ƒq||kró|j||ƒq|j||ƒqWdj |ƒS(s Returns the code block with any ``__future__`` import lines sorted, and then any ``future`` import lines sorted, then any ``builtins`` import lines sorted. This only sorts the lines within the expected blocks. See test_order_future_lines() for an example. s sfrom __future__ import s from futures from pasts from builtinscS@s t|ƒdkrt|ƒSdS(Ni(tlentmax(tnumbers((s5/usr/lib/python2.7/site-packages/future/tests/base.pytmymax7scS@s&t|ƒdkrt|ƒStdƒS(Nitinf(Rtmintfloat(R((s5/usr/lib/python2.7/site-packages/future/tests/base.pytmymin:s( tsplitt enumerateR tsortedtdicttziptrangeRtappendtjoin(R tlinestitlinetuufuture_line_numberstfuture_line_numberstbuiltins_line_numbersRRtuultsorted_uufuture_linestfltsorted_future_linestbltsorted_builtins_linest new_lines((s5/usr/lib/python2.7/site-packages/future/tests/base.pytorder_future_liness4   ###   tVerboseCalledProcessErrorcB@s#eZdZdd„Zd„ZRS(s… Like CalledProcessError, but it displays more information (message and script output) for diagnosing test failures etc. cC@s(||_||_||_||_dS(N(tmsgt returncodetcmdtoutput(tselfR-R.R/R0((s5/usr/lib/python2.7/site-packages/future/tests/base.pyt__init__`s   cC@s d|j|j|j|jfS(Ns>Command '%s' failed with exit status %d Message: %s Output: %s(R/R.R-R0(R1((s5/usr/lib/python2.7/site-packages/future/tests/base.pyt__str__fsN(t__name__t __module__t__doc__tNoneR2R3(((s5/usr/lib/python2.7/site-packages/future/tests/base.pyR,[s t FuturizeErrorcB@seZRS((R4R5(((s5/usr/lib/python2.7/site-packages/future/tests/base.pyR8jstPasteurizeErrorcB@seZRS((R4R5(((s5/usr/lib/python2.7/site-packages/future/tests/base.pyR9mst CodeHandlercB@s­eZdZd„Zdeeeeed„Zed„Zd„Zdeeeeed„Z d„Z d d „Z d d „Z d deeed „Z d ejd „ZRS(st Handy mixin for test classes for writing / reading / futurizing / running .py files in the test suite. cC@s™tdƒ|_tdƒ|_tjg|_tjƒtj j |_ tj dƒ}|ritj ƒtj|d6|_nitj ƒd6|_dS(si The outputs from the various futurize stages should have the following headers: sŽ from __future__ import absolute_import from __future__ import division from __future__ import print_function s4 from __future__ import absolute_import from __future__ import division from __future__ import print_function from __future__ import unicode_literals from future import standard_library standard_library.install_aliases() from builtins import * t PYTHONPATHN(R theaders1theaders2tsyst executablet interpretersttempfiletmkdtemptostpathtsepttempdirtgetenvtgetcwdtpathseptenv(R1tpypath((s5/usr/lib/python2.7/site-packages/future/tests/base.pytsetUpvs   $iic C@sƒ|rt|ƒ}n|j|ƒ|jd|d|d|d|ƒ|jƒ}|rx&|jD]} |jd| ƒ} q`Wn|S(s• Converts the code block using ``futurize`` and returns the resulting code. Passing stages=[1] or stages=[2] passes the flag ``--stage1`` or ``stage2`` to ``futurize``. Passing both stages runs ``futurize`` with both stages by default. If from3 is False, runs ``futurize``, converting from Python 2 to both 2 and 3. If from3 is True, runs ``pasteurize`` to convert from Python 3 to both 2 and 3. Optionally reformats the code block first using the reformat() function. If run is True, runs the resulting code under all Python interpreters in self.interpreters. tstagest all_importstfrom3t conservativet interpreter(R t_write_test_scriptt_futurize_test_scriptt_read_test_scriptR@t_run_test_script( R1R RMRNROtreformattrunRPR0RQt_((s5/usr/lib/python2.7/site-packages/future/tests/base.pytconvert¡s   cC@s¯|r'|j|ƒ}|j|ƒ}nt|tƒrXt|tƒ rX|jdƒ}nt|tƒr‰t|tƒ r‰|jdƒ}n|jt|jƒƒ|jƒƒdS(s› Compares whether the code blocks are equal. If not, raises an exception so the test fails. Ignores any trailing whitespace like blank lines. If ignore_imports is True, passes the code blocks into the strip_future_imports method. If one code block is a unicode string and the other a byte-string, it assumes the byte-string is encoded as utf-8. sutf-8N(tstrip_future_importst isinstancetbytestdecodet assertEqualR+trstrip(R1R0texpectedtignore_imports((s5/usr/lib/python2.7/site-packages/future/tests/base.pytcompare¿s cC@s‘g}x{|jdƒD]j}|jdƒpm|jdƒpm|jdƒpmd|kpmd|kpm|jdƒs|j|ƒqqWdj|ƒS(s Strips any of these import lines: from __future__ import from future from future. from builtins or any line containing: install_hooks() or: install_aliases() Limitation: doesn't handle imports split across multiple lines like this: from __future__ import (absolute_import, division, print_function, unicode_literals) s sfrom __future__ import s from future sfrom builtins sinstall_hooks()sinstall_aliases()s from future.(RR RR(R1R R0R ((s5/usr/lib/python2.7/site-packages/future/tests/base.pyRZÕs  c C@s™|j|d|d|d|d|d|ƒ} |rTd|krH|jn|j} nd} t|ƒ} | | kr{d} n|j| | | d|ƒd S( s< Convenience method that calls convert() and compare(). Reformats the code blocks automatically using the reformat_code() function. If all_imports is passed, we add the appropriate import headers for the stage(s) selected to the ``expected`` code-block, so they needn't appear repeatedly in the test code. If ignore_imports is True, ignores the presence of any lines beginning: from __future__ import ... from future import ... for the purpose of the comparison. RMRNRORWRPitRaN(RYR=R<R Rb( R1tbeforeR`RMRNRaRORWRPR0theaderst reformatted((s5/usr/lib/python2.7/site-packages/future/tests/base.pyt convert_check÷s!   cK@s|j|||dS(se Convenience method to ensure the code is unchanged by the futurize process. N(Rg(R1R tkwargs((s5/usr/lib/python2.7/site-packages/future/tests/base.pyt unchangedssmytestscript.pycC@s`t|tƒr!|jdƒ}ntj|j|dddƒ}|jt|ƒƒWdQXdS(sœ Dedents the given code (a multiline string) and writes it out to a file in a temporary folder like /tmp/tmpUDCn7x/mytestscript.py. sutf-8twttencodingN(R[R\R]tiotopenRFtwriteR(R1R tfilenametf((s5/usr/lib/python2.7/site-packages/future/tests/base.pyRR!s"cC@s8tj|j|dddƒ}|jƒ}WdQX|S(NtrtRksutf-8(RlRmRFtread(R1RoRpt newsource((s5/usr/lib/python2.7/site-packages/future/tests/base.pyRT,s"c C@sg}t|ƒ}|r(|jdƒn|r7d}nZd}|dgkr\|jdƒn|dgkr{|jdƒn|r‘|jdƒn|j|}tj|g|d |g} yt| d td |jƒ} Wn¬tk rˆ} t |ƒ7} d d j | ƒd|j|d| j ƒf} WdQXd|krCt nt }t| dƒsdd| _n|| | j| jd| jƒ‚nX| S(Ns --all-importss pasteurize.pys futurize.pyis--stage1is--stage2s--conservatives-wtstderrRJs8Error running the command %s %s Contents of file %s: %st senv=%ss ---- %s ----tfuturizeR0(tlistRRFR>R?RR RJR RmRRrR8R9thasattrR7R0R.R/(R1RoRMRNRORPtparamstscripttfnt call_argsR0teRpR-t ErrorClass((s5/usr/lib/python2.7/site-packages/future/tests/base.pyRS1s:      %c C@sÓ|j|}y%t||gd|jdtƒ}Wnštk rÎ}t|ƒ=}ddj||gƒd|j|d|jƒf}WdQXt|dƒsªd|_ nt ||j |j d|j ƒ‚nX|S(NRJRts8Error running the command %s %s Contents of file %s: %sRusenv=%ss ---- %s ----R0(RFRRJR R RmRRrRxR7R0R,R.R/(R1RoRQR{R0R}RpR-((s5/usr/lib/python2.7/site-packages/future/tests/base.pyRU`s   %(ii(ii(ii(R4R5R6RLtFalsetTrueRYRbRZRgRiRRRTRSR>R?RU(((s5/usr/lib/python2.7/site-packages/future/tests/base.pyR:qs" +   "  !   -s#this test is known to fail on Py2.6cC@sts |Stj|ƒS(N(RtunittesttexpectedFailure(tfunc((s5/usr/lib/python2.7/site-packages/future/tests/base.pytexpectedFailurePY3scC@sts |Stj|ƒS(N(RRR‚(Rƒ((s5/usr/lib/python2.7/site-packages/future/tests/base.pytexpectedFailurePY26„scC@sts |Stj|ƒS(N(RRR‚(Rƒ((s5/usr/lib/python2.7/site-packages/future/tests/base.pytexpectedFailurePY27ŠscC@sts |Stj|ƒS(N(RRR‚(Rƒ((s5/usr/lib/python2.7/site-packages/future/tests/base.pytexpectedFailurePY2stassertRaisesRegexcC@snt|ttfƒr'tj|ƒ}n|j|ƒsj|p?d}d||j|f}|j|ƒ‚ndS(s=Fail the test unless the text matches the regular expression.sRegex didn't matchs%s: %r not found in %rN(R[tstrtunicodetretcompiletsearchtpatterntfailureException(R1ttexttexpected_regexR-((s5/usr/lib/python2.7/site-packages/future/tests/base.pyt assertRegex›s  R’t_AssertRaisesBaseContextcB@s)eZddd„Zd„Zd„ZRS(cC@sš||_||_|dk rTy|j|_Wq]tk rPt|ƒ|_q]Xn d|_t|ttfƒr„t j |ƒ}n||_ d|_ dS(N( R`t test_caseR7R4tobj_nametAttributeErrorR‰R[R\R‹RŒR‘R-(R1R`R”t callable_objR‘((s5/usr/lib/python2.7/site-packages/future/tests/base.pyR2ªs      cC@s.|jj|j|ƒ}|jj|ƒ‚dS(N(R”t_formatMessageR-R(R1t standardMsgR-((s5/usr/lib/python2.7/site-packages/future/tests/base.pyt _raiseFailureºscC@sC|dkr%|jddƒ|_|S||||ŽWdQXdS(sÛ If callable_obj is None, assertRaises/Warns is being used as a context manager, so check for a 'msg' kwarg and return self. If callable_obj is not None, call it passing args and kwargs. R-N(R7tpopR-(R1tnameR—targsRh((s5/usr/lib/python2.7/site-packages/future/tests/base.pythandle¾s  N(R4R5R7R2RšRž(((s5/usr/lib/python2.7/site-packages/future/tests/base.pyR“¨s  t_AssertWarnsContextcB@s eZdZd„Zd„ZRS(sBA context manager used to implement TestCase.assertWarns* methods.cC@svx5tjjƒD]$}t|ddƒri|_qqWtjdtƒ|_ |j j ƒ|_tj d|j ƒ|S(Nt__warningregistry__trecordtalways( R>tmodulestvaluestgetattrR7R twarningstcatch_warningsR€twarnings_managert __enter__t simplefilterR`(R1tv((s5/usr/lib/python2.7/site-packages/future/tests/base.pyR©ÍscC@so|jj|||ƒ|dk r&dSy|jj}Wn tk rXt|jƒ}nXd}x—|jD]Œ}|j}t ||jƒsqin|dkr¥|}n|j dk rÓ|j j t|ƒƒ rÓqin||_ |j |_ |j|_dSW|dk r-|jdj|j jt|ƒƒƒn|jrU|jdj||jƒƒn|jdj|ƒƒdS(Ns"{}" does not match "{}"s{} not triggered by {}s{} not triggered(R¨t__exit__R7R`R4R–R‰R¦tmessageR[R‘RtwarningRotlinenoRštformatRŽR•(R1texc_typet exc_valuettbtexc_nametfirst_matchingtmtw((s5/usr/lib/python2.7/site-packages/future/tests/base.pyR¬Øs8           (R4R5R6R©R¬(((s5/usr/lib/python2.7/site-packages/future/tests/base.pyRŸÊs cO@s(t|||ƒ}|jd|||ƒS(s•Fail unless a warning of class warnClass is triggered by callable_obj when invoked with arguments args and keyword arguments kwargs. If a different type of warning is triggered, it will not be handled: depending on the other warning filtering rules in effect, it might be silenced, printed out, or raised as an exception. If called with callable_obj omitted or None, will return a context object used like this:: with self.assertWarns(SomeWarning): do_something() An optional keyword argument 'msg' can be provided when assertWarns is used as a context object. The context manager keeps a reference to the first matching warning as the 'warning' attribute; similarly, the 'filename' and 'lineno' attributes give you information about the line of Python code from which the warning was triggered. This allows you to inspect the warning after the assertion:: with self.assertWarns(SomeWarning) as cm: do_something() the_warning = cm.warning self.assertEqual(the_warning.some_attribute, 147) t assertWarns(RŸRž(R1texpected_warningR—RRhtcontext((s5/usr/lib/python2.7/site-packages/future/tests/base.pyR¸ûsR¸(-t __future__RRRCRARR>R‹R¦RlttextwrapRt future.utilsRRRRRtfuture.moves.subprocessRR R t unittest2R R+R,R8R9tTestCaseR:tskipIftskip26R„R…R†R‡RxtassertRaisesRegexpRˆR7R’tobjectR“RŸR¸(((s5/usr/lib/python2.7/site-packages/future/tests/base.pytsD       ( @ÿ      "1