ó Q˜[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((s0lib/python2.7/site-packages/future/tests/base.pyt reformat_codes cC@sW|jdƒ}gt|ƒD]!\}}|jdƒr|^q}gt|ƒD]0\}}|jdƒsz|jdƒrP|^qP}gt|ƒD]!\}}|jdƒr“|^q“}|jƒ|ksØtdƒ‚d„}d„}||ƒ||ƒkstd ƒ‚tg|D]}||^qƒ} tt|| ƒƒ} tg|D]}||^qPƒ} tt|| ƒƒ} tg|D]}||^qˆƒ} tt|| ƒƒ}g}x‹tt |ƒƒD]w}||krõ|j | |ƒqÏ||kr|j | |ƒqÏ||kr5|j ||ƒqÏ|j ||ƒqÏWdj |ƒ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 builtinssIinternal usage error: dedent the code before calling order_future_lines()cS@s t|ƒdkrt|ƒSdS(Ni(tlentmax(tnumbers((s0lib/python2.7/site-packages/future/tests/base.pytmymax7scS@s&t|ƒdkrt|ƒStdƒS(Nitinf(Rtmintfloat(R((s0lib/python2.7/site-packages/future/tests/base.pytmymin:ss2the __future__ and future imports are out of order( tsplitt enumerateR tlstriptAssertionErrortsortedtdicttziptrangeRtappendtjoin(R tlinestitlinetuufuture_line_numberstfuture_line_numberstbuiltins_line_numbersRRtuultsorted_uufuture_linestfltsorted_future_linestbltsorted_builtins_linest new_lines((s0lib/python2.7/site-packages/future/tests/base.pytorder_future_liness:    ###   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/R0R1R2((s0lib/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(R1R0R/R2(R3((s0lib/python2.7/site-packages/future/tests/base.pyt__str__fsN(t__name__t __module__t__doc__tNoneR4R5(((s0lib/python2.7/site-packages/future/tests/base.pyR.[s t FuturizeErrorcB@seZRS((R6R7(((s0lib/python2.7/site-packages/future/tests/base.pyR:jstPasteurizeErrorcB@seZRS((R6R7(((s0lib/python2.7/site-packages/future/tests/base.pyR;mst 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(R3tpypath((s0lib/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_scriptRBt_run_test_script( R3R RORPRQtreformattrunRRR2RSt_((s0lib/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(R3R2texpectedtignore_imports((s0lib/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(R3R R2R"((s0lib/python2.7/site-packages/future/tests/base.pyR\Õs  c C@s~|j|d|d|d|d|d|ƒ} |rTd|krH|jn|j} nd} |j| | t|ƒ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. RORPRQRYRRitRcN(R[R?R>RdR ( R3tbeforeRbRORPRcRQRYRRR2theaders((s0lib/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(Rh(R3R tkwargs((s0lib/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_tiotopenRHtwriteR(R3R tfilenametf((s0lib/python2.7/site-packages/future/tests/base.pyRTs"cC@s8tj|j|dddƒ}|jƒ}WdQX|S(NtrtRlsutf-8(RmRnRHtread(R3RpRqt newsource((s0lib/python2.7/site-packages/future/tests/base.pyRV(s"c C@sŠg}t|ƒ}|r(|jdƒn|r7d}nrd}|dgkr\|jdƒn7|dgkr{|jdƒn|ddgks“t‚|r©|jdƒn|j|}tj|g|d |g} yt| d td |jƒ} Wn‘t k r…} t |ƒ7} d d j | ƒd|j|d| j ƒf} WdQXd|kr[t nt}|| | j| jd| jƒ‚nX| S(Ns --all-importss pasteurize.pys futurize.pyis--stage1is--stage2s--conservatives-wtstderrRLs8Error running the command %s %s Contents of file %s: %st senv=%ss ---- %s ----tfuturizeR2(tlistRRRHR@RARR RLR RnRRsR:R;R0R1R2(R3RpRORPRQRRtparamstscripttfnt call_argsR2teRqR/t ErrorClass((s0lib/python2.7/site-packages/future/tests/base.pyRU-s8     %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(NRLRus8Error running the command %s %s Contents of file %s: %sRvsenv=%ss ---- %s ----R2(RHRRLR R RnRRsthasattrR9R2R.R0R1(R3RpRSR{R2R}RqR/((s0lib/python2.7/site-packages/future/tests/base.pyRWXs   %(ii(ii(ii(R6R7R8RNtFalsetTrueR[RdR\RhRjRTRVRUR@RARW(((s0lib/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((s0lib/python2.7/site-packages/future/tests/base.pytexpectedFailurePY3wscC@sts |Stj|ƒS(N(RR‚Rƒ(R„((s0lib/python2.7/site-packages/future/tests/base.pytexpectedFailurePY26|scC@sts |Stj|ƒS(N(RR‚Rƒ(R„((s0lib/python2.7/site-packages/future/tests/base.pytexpectedFailurePY27‚scC@sts |Stj|ƒS(N(RR‚Rƒ(R„((s0lib/python2.7/site-packages/future/tests/base.pytexpectedFailurePY2ˆstassertRaisesRegexcC@s€t|ttfƒr9|s'tdƒ‚tj|ƒ}n|j|ƒs||pQd}d||j|f}|j|ƒ‚ndS(s=Fail the test unless the text matches the regular expression.s!expected_regex must not be empty.sRegex didn't matchs%s: %r not found in %rN( R]tstrtunicodeRtretcompiletsearchtpatterntfailureException(R3ttexttexpected_regexR/((s0lib/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( Rbt test_caseR9R6tobj_nametAttributeErrorRŠR]R^RŒRR’R/(R3RbR•t callable_objR’((s0lib/python2.7/site-packages/future/tests/base.pyR4¢s      cC@s.|jj|j|ƒ}|jj|ƒ‚dS(N(R•t_formatMessageR/R(R3t standardMsgR/((s0lib/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(R9tpopR/(R3tnameR˜targsRi((s0lib/python2.7/site-packages/future/tests/base.pythandle¶s  N(R6R7R9R4R›RŸ(((s0lib/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@tmodulestvaluestgetattrR9R¡twarningstcatch_warningsRtwarnings_managert __enter__t simplefilterRb(R3tv((s0lib/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__R9RbR6R—RŠR§tmessageR]R’RŽtwarningRptlinenoR›tformatRR–(R3texc_typet exc_valuettbtexc_nametfirst_matchingtmtw((s0lib/python2.7/site-packages/future/tests/base.pyR­Ðs8           (R6R7R8RªR­(((s0lib/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Ÿ(R3texpected_warningR˜RžRitcontext((s0lib/python2.7/site-packages/future/tests/base.pyR¹ósR¹(-t __future__RRRERCR‚R@RŒR§RmttextwrapRt future.utilsRRRRRtfuture.moves.subprocessRR R t unittest2R R-R.R:R;tTestCaseR<tskipIftskip26R…R†R‡RˆRtassertRaisesRegexpR‰R9R“tobjectR”R R¹(((s0lib/python2.7/site-packages/future/tests/base.pytsD       ( @ÿ     "1