^Yc@`sdZddlmZmZmZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlZddlmZee dre jdZn`eedrdZnejd krd Znd ZejZd eeded fZ[[derSdpVdZdeeZejd dJkoejdKkZejddkZejdLkrejZ n dZ de!fdYZ"dZ#dj$dZ%dj$dZ&e j'dZ(dZ)dZ*d Z+d!d"Z,ddd#Z.d$Z/dd%l0m1Z2d&Z3d'Z4d(Z5d)Z6id*ej76d+ej86d,ej96d-ej:6Z;id.ej<6d/ej=6d0ej>6d1ej?6d2ej@6d3ejA6d4ejB6d5ejC6d6ejD6d7ejE6d8ejF6d9ejG6d:ejH6d;ejI6d<ejJ6d=ejK6d>ejL6d?ejM6d@ejN6dAejO6dBejP6dCejQ6ZRydDeRejS*stAssertionRewritingHookcB`sqeZdZdZdZd dZdZdZdZ dZ dZ e d Z d ZRS( s*PEP302 Import hook which rewrites asserts.cC`sS||_|jd|_d|_i|_t|_|jt|_ dS(Nt python_files( tconfigtgetinitfnpatsRtsessiontmodulestsett_rewritten_namest_register_with_pkg_resourcest _must_rewrite(tselfR((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyt__init__0s     cC`s ||_dS(N(R(RR((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyt set_session9scC`sS|jj}|jd||jdd}|d}d}|dk ryt|}t|dkry|d}qyn|dkrQytj||\}}} Wnt k rdSX|dk r|j n| d} | tj kr;t tdr.ytj |}Wq8tk r*|d }q8XqN|d }qw| tjkrwdSn&tjj||jddd}tjj|} |j|| |sdS|jj|tj } tjj| jd } | rytj| Wqtk rtjdj}|tj kr q|tj!tj"gkrAt#} q|tj$tj%tj&gkr||jd | jt#} qqXn| j'd t(}tjj| |}t)| ||j}|dkr(|jd |ft*|j| \}}|dkr dS| r<t+||||q<n|jd |f||f|j,|<|S(Nsfind_module called for: %sR iiiitsource_from_caches.pyt __pycache__sread only directory: %ris rewriting %rs!found cached rewritten pyc for %r(-Rt _assertstatettracetrsplitRtlisttlentimpt find_modulet ImportErrortcloset PY_COMPILEDthasattrR!t ValueErrort PY_SOURCEtostpathtjoint rpartitiontpytlocalt_should_rewriteRtaddtsystdont_write_bytecodetdirnametmkdirtOSErrortexc_infoterrnotEEXISTtENOENTtENOTDIRtFalsetEACCEStEROFStEPERMtbasenametPYC_TAILt _read_pyct _rewrite_testt_make_rewritten_pycR(RtnameR1tstatetnamestlastnametpthtfdtfntdescttpt fn_pypathtwritet cache_dirtet cache_nametpyctcot source_stat((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyR)<sv           &       cC`st|}|jdkr3|jd|ftS|jdk ro|jj|ro|jd|ftSnx8|jD]-}|j|ry|jd|ftSqyWx;|j D]0}|j |r|jd||ftSqWt S(Ns conftest.pysrewriting conftest file: %rs0matched test file (was specified on cmdline): %rsmatched test file %rs matched marked file %r (from %r)( tstrRFR$tTrueRRt isinitpathRtfnmatchRt startswithRB(RRKRTRLRQtpattmarked((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyR6s$   cG`skt|jttj}|rWx0|D]%}||jkr+|j|q+q+Wn|jj|dS(sMark import names as needing to be re-written. The named module or package as well as any nested modules will be re-written on import. N(Rt intersectionR8RRt_warn_already_importedRtupdate(RRMtalready_importedRK((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyt mark_rewrites  cC`s|jjdd|dS(NtP1s4Module already imported so can not be re-written: %s(Rtwarn(RRK((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyRds cC`s|tjkrtj|S|jj|\}}tj|}tj|R$RBRUR(t get_magictinttmtimetsizetstructtpacktmarshaltdumpR+R](RLRZR[RYtfpterrRR((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyt _write_pycs   s sutf-8s s#^[ \t\f]*#.*coding[:=][ \t]*[-\w.]+sc C`s|j}y|j}|jd}Wntk r<d SXtr |jd}|jd|d}|jt r t j |d|!dkr t j ||d|!dkr t |drd St |_ z*y|jdWntk rdSXWd|` Xq ntr)|jttt}nytj|}Wn&tk rd|jd|fdSXt|||yt||jd d t }Wn&tk r|jd |fdSX||fS(s8Try to read and rewrite *fn* and return the code object.R|s iit _indecodetasciiNsfailed to parse: %rtexect dont_inheritsfailed to compile: %r(NN(NN(NN(NN(NN(R#tstatR~tEnvironmentErrorRtASCII_IS_DEFAULT_ENCODINGtfindR`tBOM_UTF8t cookie_retmatchR-R]RtdecodetUnicodeDecodeErrortREWRITE_NEWLINEStreplacetRNtNR tparset SyntaxErrorR$trewrite_assertstcompiletstrpath( RRQRLRtsourcetend1tend2ttreeRZ((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyRIsF           cC`sntjjdr(t||||nB|dttj}t||||rjtj||ndS(s$Try to dump rewritten code to *pyc*.twinR N(R8tplatformR`RR\R0tgetpidtrename(RLR[RYRZtproc_pyc((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyRJWs cC`sdS(N(R(tx((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyRdsc C`sWyt|d}Wntk r'd SX|#y1t|j}|j}|jd}Wn(tk r}|d||fd SXt|dks|d t j kst j d|d||fkr|d|d Syt j|}Wn(tk r$}|d||fd SXt|tjsI|d|d S|SWd QXd S( ssPossibly read a pytest pyc containing rewritten code. Return rewritten code if successful or None if not. R|i s"_read_pyc(%s): EnvironmentError %sis\}}}y | }Wntk rRt}nX|rPqqWtjdk rtj||||||d}|dk r|Sn|S(Ni(tziptrangeR'RR]Rt _reprcompareR( topstresultstexplsteach_objtitrestexpltdonetcustom((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyt_call_reprcompares.   % snot %ss~%ss-%ss+%st|t^t&s<>t+t-t*t/s//s%%s==s!=ts>=s**tissis nottinsnot int@t NameConstantcC`stjt|tjS(N(R tNameR\tLoad(R ((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyt _NameConstantsc`s#fd||||S(s*Set node location information recursively.c`sdd|jkr||_nd|jkr6||_nx'tj|D]}|||qFWdS(Ntlinenot col_offset(t _attributesRRR titer_child_nodes(tnodeRRtchild(t_fix(s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyRs   ((RRR((Rs9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyt set_locationsRcB`seZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZejdkreZneZdZdZRS(s2 Assertion rewriting implementation. The main entrypoint is to call .run() with an ast.Module instance, this will then find all the assert statements and re-write them to provide intermediate values and a detailed assertion error. See http://pybites.blogspot.be/2011/07/behind-scenes-of-pytests-new-assertion.html for an overview of how this works. The entry point here is .run() which will iterate over all the statements in an ast.Module and for each ast.Assert statement it finds call .visit() with it. Then .visit_Assert() takes over and is responsible for creating new ast statements to replace the original assert statement: it re-writes the test of an assertion to provide intermediate values and replace it with an if statement which raises an assertion error with a detailed explanation in case the expression is false. For this .visit_Assert() uses the visitor pattern to visit all the AST nodes of the ast.Assert.test field, each visit call returning an AST node and the corresponding explanation string. During this state is kept in several instance attributes: :statements: All the AST statements which will replace the assert statement. :variables: This is populated by .variable() with each variable used by the statements so that they can all be set to None at the end of the statements. :variable_counter: Counter to create new unique variables needed by statements. Variables are created using .variable() and have the form of "@py_assert0". :on_failure: The AST statements which will be executed if the assertion test fails. This is the code which will construct the failure message and raises the AssertionError. :explanation_specifiers: A dict filled by .explanation_param() with %-formatting placeholders and their corresponding expressions to use in the building of an assertion message. This is used by .pop_format_context() to build a message. :stack: A stack of the explanation_specifiers dicts maintained by .push_format_context() and .pop_format_context() which allows to build another %-formatted string while already building one. This state is reset on every new assert statement visited and used by the other visitors. cC`s)tt|j||_||_dS(N(tsuperRRRR(RRR((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyRBs cC`si|js dStjtjjjdtjddg}t}d}d}x|jD]}|rt|tj rt|j tj r|j j }d|krdS|t |d7}t}n>t|tj s|jdks|jdkr|j}Pn|d7}qVWg|D]$}tj|gd |d d^q} | |j||+|g} x | rd| j} xtj| D]\} } t| tr(g}xtt| D]f\}}t|tjr|j|j|q|j|t|tjr| j|qqWt| | |q}t| tjr}t| tj r}| j| q}q}Wq[WdS( s5Find all assert statements in *mod* and rewrite them.Ns @py_builtinss_pytest.assertion.rewrites @pytest_aritPYTEST_DONT_REWRITEit __future__RR(tbodyR taliasR4RptbuiltinsRxR]RtExprtvaluetStrRR'RBt ImportFromtleveltmoduleRtImportRjt iter_fieldsR&t enumeratetAsserttextendtvisittappendtASTtsetattrtexpr(RRstaliasestexpect_docstringtposRtitemtdocRtimportstnodesRRKtfieldtnewRR((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyRGsN    " .    cC`s-dtt|j}|jj||S(sGet a new variable.s @py_assert(R\tnexttvariable_countert variablesR (RRK((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pytvariablezscC`sS|j}|jjtjtj|tjg|tj|tjS(sGive *expr* a name.(Rt statementsR R tAssignRtStoreR(RR RK((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pytassigns 1cC`s|jd|S(s&Call py.io.saferepr on the expression.R(thelper(RR ((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pytdisplayscG`sMtjdtj}tj|d|tj}t|t|gS(sCall a helper in this module.s @pytest_art_(R RRt Attributetast_CallR&(RRKtargstpy_nametattr((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyRscC`s1tjdtj}tj||tjS(s!Return the builtin called *name*.s @py_builtins(R RRR"(RRKt builtin_name((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyRpscC`s2dtt|j}||j|assert tsassert RR9iiN('RttestR tTupleRRRRRiRRt itertoolstcountRR+R2R,R tUnaryOptNotR tIftmsgRRR0tAddR8RRR#R8t version_infotRaiseRRRRR(Rtassert_R<t top_conditionRRtnegationt assertmsgttemplateRFtfmtterr_nametexctraise_RKRtcleartstmt((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyt visit_AssertsH $          $  +cC`st|jdgg}tjtj|jtjg|g}|jd|}tjtj ||g}tj ||j |tj|j}||j |fS(Ntlocalstshould_repr_global_name( R#RpR tCompareRtidtInRtBoolOptOrtIfExpR R*(RRKtlocstinlocstdoreprR?R ((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyt visit_Names -*cC`sL|j}|jtjgtj}tj|dtj}tt|jtj }|j }}|j }t |j d} |jxOt|j D]>\} } | rg} |j jtj| | g| |_ n|j|j| \}}|jtjtj|tjg||jtj|}t||gg}|j jtj|| | kr|} |rtjtj| } ng}|j jtj| |g||_ }qqW||_ ||_ |jd|tj|}|j|}tj|tj|j|fS(NR it format_boolop(RRR tListRR"RRtopR\RR2R'R/R,RR RER RRRR8RR#RRCRDRtNumR*(Rtbooloptres_vart expl_listtappRRtsavet fail_savetlevelsRtvt fail_innertcondRRt expl_formattcalltinnert expl_template((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyt visit_BoolOps> !     .   cC`sWt|jj}|j|j\}}|jtj|j|}|||ffS(N(t unary_mapRdt __class__R toperandRR RC(Rtunarytpatternt operand_rest operand_explR((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyt visit_UnaryOp+sc C`s~t|jj}|j|j\}}|j|j\}}d|||f}|jtj||j|}||fS(Ns (%s %s %s)( t binop_mapRdRvR tlefttrightRR R0( Rtbinoptsymbolt left_exprt left_explt right_exprt right_explRR((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyt visit_BinOp1s !cC`s\|j|j\}}g}g}g}x@|jD]5}|j|\}} |j| |j|q4Wxz|jD]o} |j| j\}} |jtj| j|| jr|j| jd| qw|jd| qwWd|dj |f} tj |||} |j | }|j |j |} d| | | f} || fS(s? visit `ast.Call` nodes on Python3.5 and after t=s**s%s(%s)s, s %s {%s = %s }(R tfuncR$R tkeywordsRR tkeywordtargR2RRR*R (RRqtnew_funct func_explt arg_explstnew_argst new_kwargsRRRRtnew_calltres_explt outer_expl((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyt visit_Call_359s(  cC`s&|j|j\}}|d|fS(NR(R R(RtstarredRR((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyt visit_StarredTscC`s|j|j\}}g}g}g}d}}x@|jD]5} |j| \} } |j| |j| q>Wx]|jD]R} |j| j\} } |jtj| j | |j| j d| qW|j r |j|j \}} |jd| n|j rA|j|j \}} |jd| nd|dj |f} tj |||||} |j| } |j|j| }d||| f}| |fS(s9 visit `ast.Call nodes on 3.4 and below` RRs**s%s(%s)s, s %s {%s = %s }N(R RRR$R RRR RRtstarargstkwargsR2RRR*R (RRqRRRRRtnew_start new_kwargRRRRRRR((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pytvisit_Call_legacyYs4     iicC`st|jtjs"|j|S|j|j\}}|jtj||j tj}|j |j |}d}|||||j f}||fS(Ns%s {%s = %s.%s }( RtctxR RR:R RRR"R&R*R (RR&Rt value_explRRRaR((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pytvisit_Attributes 'cC`s|j|j|j\}}t|jtjtjfrRdj|}ngtt |j D]}|j ^qh}g|D]}t j |t j^q}g|D]}t j |t j^q}ttt |j |j |j} g} g} |g} x| D]\}} }|j|\}}t|tjtjfrjdj|}n| j|t| j}| jt j|d|||f}| jt j|t j|| g|g}|jjt j||g|||}}qW|jdt j| t jt j|t jt j| t jt j| t j}t |j dkrt jt j|}n |d}||j|j|fS(Ns({0})s%s %s %stcall_reprcompareii(R,R R~Rt_astRXR[tformatRR'RRR RRRRt comparatorsR R}RvRRRRR@tAndR*R8(Rtcomptleft_resRRt res_variablesRmt load_namest store_namestitRtsymsRRdt next_operandtnext_rest next_expltsymRtres_exprt expl_callR((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyt visit_Compares@ .++'   #  (ii(RxRRRRRRR RRpR*R,R8R:RURaRtR|RRRRR8RHt visit_CallRR(((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyRs22  3         4 #     "  (ii(ii(ii(ZRRRRRR RR>RAR(RR0treRR8RR4t_pytest.assertionRR-Rt PYTEST_TAGtimplRRHtvert __debug__tPYC_EXTRGRRRR#tobjectRRtencodeRRRRRRIRJRHRRRt_pytest.assertion.utilRt_format_explanationRRRRRDtInverttUSubtUAddRutBitOrtBitXortBitAndtLShifttRShiftRGtSubtMulttDivtFloorDivR1tEqtNotEqtLttLtEtGttGtEtPowtIstIsNotRZtNotInR}tMatMulttAttributeErrorRRRt NodeVisitorR(((s9/tmp/pip-build-hU8Cw8/pytest/_pytest/assertion/rewrite.pyts               "    9 !