ó \ K]c@súddlmZddlZddlZddljZddlmZm Z m Z m Z ddl m Z mZmZddlmZmZddlmZdefd „ƒYZd efd „ƒYZd efd „ƒYZedkröejƒndS(iÿÿÿÿ(tprint_functionN(tnjittjitttestingtutils(t TypingErrort LoweringErrortUnsupportedErrori(tTestCasettag(texec_t TestClosurecBs†eZd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z d „Z d „Z d „Z ejd „ƒZRS( c s`d‰‡fd†}td||ƒ}|j|dƒdƒd‰|j|dƒdƒdS(Ni cs|ˆS(N((tx(tY(s7lib/python2.7/site-packages/numba/tests/test_closure.pytadd_Yssi4(i4)ii i (Rt assertEqual(tselftjitargsRtc_add_Y((R s7lib/python2.7/site-packages/numba/tests/test_closure.pytrun_jit_closure_variables cCs|jdtƒdS(Ntforceobj(RtTrue(R((s7lib/python2.7/site-packages/numba/tests/test_closure.pyttest_jit_closure_variablescCs|jdtƒdS(Ntnopython(RR(R((s7lib/python2.7/site-packages/numba/tests/test_closure.pyttest_jit_closure_variable_npm!sc s§d‰‡fd†}td||ƒ}|j|dƒdƒd‰td||ƒ}|j|dƒdƒd‰|j|dƒdƒ|j|dƒdƒdS(Ni cs|ˆS(N((R (R (s7lib/python2.7/site-packages/numba/tests/test_closure.pyR'ssi4(i4)ii i i (RR(RRRRt c_add_Y_2((R s7lib/python2.7/site-packages/numba/tests/test_closure.pytrun_rejitting_closure$scCs|jdtƒdS(NR(RR(R((s7lib/python2.7/site-packages/numba/tests/test_closure.pyttest_rejitting_closure6scCs|jdtƒdS(NR(RR(R((s7lib/python2.7/site-packages/numba/tests/test_closure.pyttest_rejitting_closure_npm9sc sMd‰d‰‡‡fd†}td||ƒ}|j|dƒdƒdS(Ni ics |ˆˆS(N((R (R tZ(s7lib/python2.7/site-packages/numba/tests/test_closure.pyt add_Y_mult_Z@ssi4(i4)ii(RR(RRRtc_add_Y_mult_Z((R Rs7lib/python2.7/site-packages/numba/tests/test_closure.pyt"run_jit_multiple_closure_variables<s cCs|jdtƒdS(NR(R R(R((s7lib/python2.7/site-packages/numba/tests/test_closure.pyt#test_jit_multiple_closure_variablesFscCs|jdtƒdS(NR(R R(R((s7lib/python2.7/site-packages/numba/tests/test_closure.pyt'test_jit_multiple_closure_variables_npmIsc s‰d„}td||ƒ‰ˆjƒ‡fd†}td||ƒ}|jƒ|j|ˆƒ|j|dƒdƒWdQXdS(NcSs|dS(Ni ((ta((s7lib/python2.7/site-packages/numba/tests/test_closure.pytmult_10Mss intp(intp)csˆ|dƒS(Ni((R (t c_mult_10(s7lib/python2.7/site-packages/numba/tests/test_closure.pytdo_mathSsii2(Rtdisable_compiletassertRefCountR(RRR$R&t c_do_math((R%s7lib/python2.7/site-packages/numba/tests/test_closure.pytrun_jit_inner_functionLs   cCs|jdtƒdS(NR(R*R(R((s7lib/python2.7/site-packages/numba/tests/test_closure.pyttest_jit_inner_function\scCs|jdtƒdS(NR(R*R(R((s7lib/python2.7/site-packages/numba/tests/test_closure.pyttest_jit_inner_function_npm_scCs;d„}t|ƒ}|j|dƒƒ|dƒƒƒdS(Ncs‡fd†}|S(NcsˆdS(Ni(((R (s7lib/python2.7/site-packages/numba/tests/test_closure.pytinnergs((R R-((R s7lib/python2.7/site-packages/numba/tests/test_closure.pytouteresi (RR(RR.tcfunc((s7lib/python2.7/site-packages/numba/tests/test_closure.pyttest_return_closurebs  (t__name__t __module__RRRRRRR R!R"R*R+R,Rtallow_interpreter_modeR0(((s7lib/python2.7/site-packages/numba/tests/test_closure.pyR s           tTestInlinedClosurecBs›eZdZedƒd„ƒZedƒd„ƒZedƒd„ƒZeje j d kdƒd„ƒZ edƒd „ƒZ edƒd „ƒZ RS( sª Tests for (partial) closure support in njit. The support is partial because it only works for closures that can be successfully inlined at compile time. t importantcCs5d„}t|ƒ}|j|dƒ|dƒƒdS(NcSsd„}||ƒ||ƒS(NcSs||S(N((R ((s7lib/python2.7/site-packages/numba/tests/test_closure.pyR-|s((R R-((s7lib/python2.7/site-packages/numba/tests/test_closure.pyR.zs i (RR(RR.R/((s7lib/python2.7/site-packages/numba/tests/test_closure.pyttest_inner_functionws  cCs5d„}t|ƒ}|j|dƒ|dƒƒdS(Ncs-|d‰‡fd†}||ƒ||ƒS(Nics ||ˆS(N((R (ty(s7lib/python2.7/site-packages/numba/tests/test_closure.pyR-Šs((R R-((R7s7lib/python2.7/site-packages/numba/tests/test_closure.pyR.‡s i (RR(RR.R/((s7lib/python2.7/site-packages/numba/tests/test_closure.pyt test_inner_function_with_closure„s  cCs5d„}t|ƒ}|j|dƒ|dƒƒdS(Ncs3|d‰‡fd†}||ƒ‰ˆ||ƒS(Nics|ˆS(N((R (R7(s7lib/python2.7/site-packages/numba/tests/test_closure.pyR-˜s((R R-((R7s7lib/python2.7/site-packages/numba/tests/test_closure.pyR.•s  i (RR(RR.R/((s7lib/python2.7/site-packages/numba/tests/test_closure.pyt"test_inner_function_with_closure_2’s iisneeds Python 3cCsSd}i}t|jƒ|ƒt|dƒ}|j|dƒ|ddƒƒdS(Ns def outer(x): y = x + 1 z = 0 def inner(x): nonlocal z z += x * x return z + y return inner(x) + inner(x) + z R.i (R tstripRR(RtcodetnsR/((s7lib/python2.7/site-packages/numba/tests/test_closure.pyt"test_inner_function_with_closure_3¡s cCs5d„}t|ƒ}|j|dƒ|dƒƒdS(Ncs‡fd†}|ˆˆƒS(NcsC‡‡fd†}d}x$tˆƒD]}|||ƒ7}q%W|S(Ncs ˆˆ|S(N((tz(R R7(s7lib/python2.7/site-packages/numba/tests/test_closure.pyt innermost½si(trange(R7R?tsti(R (R7s7lib/python2.7/site-packages/numba/tests/test_closure.pyR-»s ((R R-((R s7lib/python2.7/site-packages/numba/tests/test_closure.pyR.¹s i (RR(RR.R/((s7lib/python2.7/site-packages/numba/tests/test_closure.pyttest_inner_function_nested¶s  csyt‡fd†ƒ‰‡fd†}‡fd†}d„}d„}d„}d„}d‰‡fd †}d‰‡fd †}d „} d „} d „} d„} dd„} dd„}dd„}d„}d„}d„}d„}d„}d„}||||||| | | | |||||g}x<|D]4}t|ƒ}d}|j||ƒ||ƒƒq9W|jtƒ&}tdtƒ|ƒ}||ƒWdQXd}|j|t|jƒƒ|jtƒ&}tdtƒ|ƒ}||ƒWdQXd}|j|t|jƒƒ|jt ƒ&}tdtƒ| ƒ}||ƒWdQXd}|j|t|jƒƒ|jt ƒ&}tdtƒ|ƒ}||ƒWdQXd}|j|t|jƒƒ|jt ƒ&}tdtƒ|ƒ}||ƒWdQXd}|j|t|jƒƒ|jt ƒ&}tdtƒ|ƒ}||ƒWdQXd }|j|t|jƒƒdS(!s3 Tests the large number of use cases defined below cs,|dkr|Sˆ|dƒˆ|dƒS(Nii((tn(tfib3(s7lib/python2.7/site-packages/numba/tests/test_closure.pyREÏs cs‡fd†}||ƒS(s, Test calling recursive function from inner cs ˆ|ƒS(N((R (RE(s7lib/python2.7/site-packages/numba/tests/test_closure.pyR-×s((R R-(RE(s7lib/python2.7/site-packages/numba/tests/test_closure.pytouter1Õscs&|d‰‡‡fd†}||ƒS(s. Test calling recursive function from closure ics|ˆˆƒS(N((R (RER>(s7lib/python2.7/site-packages/numba/tests/test_closure.pyR-ßs((R R-(RE(R>s7lib/python2.7/site-packages/numba/tests/test_closure.pytouter2Ûs cs‡fd†‰ˆ|ƒS(s Test recursive inner cs"|dkrdSˆ|dƒdS(Nii i((R (R-(s7lib/python2.7/site-packages/numba/tests/test_closure.pyR-ås ((R ((R-s7lib/python2.7/site-packages/numba/tests/test_closure.pytouter3ãscs&|d‰‡‡fd†‰ˆ|ƒS(s Test recursive closure ics&|ˆdkrdSˆ|dƒdS(Nii i((R (R-R7(s7lib/python2.7/site-packages/numba/tests/test_closure.pyR-ðs((R ((R-R7s7lib/python2.7/site-packages/numba/tests/test_closure.pytouter4ìs cs#|d‰‡fd†}||ƒS(s Test nested closure ics+ˆ|d‰‡fd†}||ƒˆS(Nics|ˆS(N((R (R>(s7lib/python2.7/site-packages/numba/tests/test_closure.pytinner2þs((R RJ(R7(R>s7lib/python2.7/site-packages/numba/tests/test_closure.pytinner1ûs((R RK((R7s7lib/python2.7/site-packages/numba/tests/test_closure.pytouter5÷s cs#|d‰‡fd†}||ƒS(s. Test closure with list comprehension in body ics+ˆ|d}gt|ƒD] }|^qS(Ni(R@(R R>tt(R7(s7lib/python2.7/site-packages/numba/tests/test_closure.pyRK s((R RK((R7s7lib/python2.7/site-packages/numba/tests/test_closure.pytouter6s i cs|d}||ˆS(s) Test use of outer scope var, no closure i((R R>(t_OUTER_SCOPE_VAR(s7lib/python2.7/site-packages/numba/tests/test_closure.pytouter7s cs&|d‰‡‡fd†}||ƒS(s+ Test use of outer scope var, with closure ics |ˆˆS(N((R (ROR>(s7lib/python2.7/site-packages/numba/tests/test_closure.pyR-s((R R-(RO(R>s7lib/python2.7/site-packages/numba/tests/test_closure.pytouter8s cs)|d‰‡fd†}|}||ƒS(s Test closure assignmentics|ˆS(N((R (R>(s7lib/python2.7/site-packages/numba/tests/test_closure.pyR-#s((R R-tf((R>s7lib/python2.7/site-packages/numba/tests/test_closure.pytouter9s cs2|d‰‡fd†‰‡fd†}||ƒS(s! Test two inner, one calls other ics|ˆS(N((R (R>(s7lib/python2.7/site-packages/numba/tests/test_closure.pyR-,scs ˆ|ƒS(N((R (R-(s7lib/python2.7/site-packages/numba/tests/test_closure.pyRJ/s((R RJ((R-R>s7lib/python2.7/site-packages/numba/tests/test_closure.pytouter10(s cs|d‰‡fd†}|S(s return the closure ics|ˆS(N((R (R>(s7lib/python2.7/site-packages/numba/tests/test_closure.pyR-8s((R R-((R>s7lib/python2.7/site-packages/numba/tests/test_closure.pytouter114s cs&|d‰d‡fd†}||ƒS(s closure with kwargiics |ˆ|S(N((R tkw(R>(s7lib/python2.7/site-packages/numba/tests/test_closure.pyR-@s((R R-((R>s7lib/python2.7/site-packages/numba/tests/test_closure.pytouter12<s icSs|d|}|S(s outer with kwarg no closurei((R RVR>((s7lib/python2.7/site-packages/numba/tests/test_closure.pytouter13Dscs&|d‰‡‡fd†}||ƒS(s! outer with kwarg used in closureics |ˆˆS(N((R (RVR>(s7lib/python2.7/site-packages/numba/tests/test_closure.pyR-Ms((R RVR-((RVR>s7lib/python2.7/site-packages/numba/tests/test_closure.pytouter14Is cs&|d‰‡fd†}|||ƒS(s# outer with kwarg as arg to closureics |ˆ|S(N((R RV(R>(s7lib/python2.7/site-packages/numba/tests/test_closure.pyR-Us((R RVR-((R>s7lib/python2.7/site-packages/numba/tests/test_closure.pytouter15Qs cs)|d‰‡fd†}t||ƒƒS(s( closure is generator, consumed locally ic3s |ˆVdS(N((R (R>(s7lib/python2.7/site-packages/numba/tests/test_closure.pyR-]s(tlist(R R-((R>s7lib/python2.7/site-packages/numba/tests/test_closure.pytouter16Ys cs#|d‰‡fd†}||ƒS(s closure is generator, returned ic3s |ˆVdS(N((R (R>(s7lib/python2.7/site-packages/numba/tests/test_closure.pyR-fs((R R-((R>s7lib/python2.7/site-packages/numba/tests/test_closure.pytouter17bs cs:|d‰‡fd†}x||ƒD] }|}q&W|S(s( closure is generator, consumed in loop ic3s |ˆVdS(N((R (R>(s7lib/python2.7/site-packages/numba/tests/test_closure.pyR-os((R R-RBRM((R>s7lib/python2.7/site-packages/numba/tests/test_closure.pytouter18ks   cs?|d‰|d‰‡fd†}‡fd†}|||ƒS(s# closure as arg to another closure iics|ˆS(N((R (tz1(s7lib/python2.7/site-packages/numba/tests/test_closure.pyR-|scs||ƒˆS(N((RRR (tz2(s7lib/python2.7/site-packages/numba/tests/test_closure.pyRJs((R R-RJ((R_R`s7lib/python2.7/site-packages/numba/tests/test_closure.pytouter19ws   cs#|d‰‡fd†}||ƒS(Nics|tjˆƒS(N(tnumpytcos(R (R>(s7lib/python2.7/site-packages/numba/tests/test_closure.pyR-ˆs((R R-((R>s7lib/python2.7/site-packages/numba/tests/test_closure.pytouter20„s cs#|d‰‡fd†}||ƒS(Nics|tjˆƒS(N(tnpRc(R (R>(s7lib/python2.7/site-packages/numba/tests/test_closure.pyR-s((R R-((R>s7lib/python2.7/site-packages/numba/tests/test_closure.pytouter21Œs i RNs.Unsupported use of op_LOAD_CLOSURE encounteredt make_functions#with argument(s) of type(s): (none)s1'NoneType' object has no attribute 'yield_points's-Invalid use of getiter with parameters (none)( RRt assertRaisestNotImplementedErrorRRtassertIntstrt exceptionRRR(RRFRGRHRIRLRNRPRQRSRTRURWRXRYRZR\R]R^RaRdRfRRtrefR/tvartraisestmsg((ROREs7lib/python2.7/site-packages/numba/tests/test_closure.pyttest_bulk_use_casesÊsz            (ii(R1R2t__doc__R R6R8R9tunittesttskipIfRt PYVERSIONR=RCRq(((s7lib/python2.7/site-packages/numba/tests/test_closure.pyR4ps $tTestObjmodeFallbackcBsSeZeedeƒgZd„Zd„Zd„Zd„Zd„Z d„Z RS(RcCstd„}tjjdƒ}tjtjjddƒdƒjtjƒ}x$|jD]}||ƒ||ƒqSWdS(NcSs~|j\}}xht|ƒD]Z}||}tt|ƒdtƒ}g|D]'}ttj||kƒjƒƒ^qKqWdS(Ntreverse(tshapeR@tsortedtsetRRetargwheretflatten(tscorestcooctrowstcolsRBtcoxvtgroupsR ((s7lib/python2.7/site-packages/numba/tests/test_closure.pyt numbaFailureÍs  i gƒÀÊ¡E¶û?(i i (Retrandomtabstrandntastypetintt decorators(RRƒR R7td((s7lib/python2.7/site-packages/numba/tests/test_closure.pyttest_issue2955Ës  .cCs¬d„}xœ|jD]‘}||ƒtjdƒjddƒtjdƒjddƒƒ}|tjdƒjddƒtjdƒjddƒƒ}tjj||ƒqWdS(NcSsÖt|ƒtjk r'tj|ƒ}nt|ƒtjk rNtj|ƒ}n|j\}}tjtjgt|ƒD]}dg^qvƒ|fƒ}tjtj||jƒ|ƒ}|d}|d}||fS(Nii( ttypeRetndarraytarrayRxthstackR@tdottT(tXR7tmt_trest intercepttcoefs((s7lib/python2.7/site-packages/numba/tests/test_closure.pytfitÛs( !  i i(R‰RetarangetreshapeRt assert_equal(RR˜RŠR•texp((s7lib/python2.7/site-packages/numba/tests/test_closure.pyttest_issue3239Ùs !cs@ddg‰‡fd†}x|jD]}||ƒƒq%WdS(Nii|i4cs<gddgD]}ˆ|^q xtdƒD]}q.WdS(Niii(R@(tindexR (tb(s7lib/python2.7/site-packages/numba/tests/test_closure.pyR#÷s!(ii|(i4i(R‰(RR#RŠ((RŸs7lib/python2.7/site-packages/numba/tests/test_closure.pyttest_issue3289ôs cCsxd„}itjdƒd6dgddggd6}x<|jD]1}||ƒ|ƒ}tjj|||ƒƒq?WdS(NcSsXtg|dD]}t|ƒ^qƒ}|ddk}t|ƒrNd}n||fS(NR7R ii(tmaxtlentany(tdataR“RMtmaskR>((s7lib/python2.7/site-packages/numba/tests/test_closure.pytfoos )  iR iiiR7(ReR™R‰Rtassert_allclose(RR¦R¤RŠR•((s7lib/python2.7/site-packages/numba/tests/test_closure.pyttest_issue3413þs  ,cCsFd„}x6|jD]+}||ƒƒ}tjj||ƒƒqWdS(NcSs5tjddfƒ}tjg|D] }|^q"ƒS(Niiii(ii(ii(ReRŽ(R#R ((s7lib/python2.7/site-packages/numba/tests/test_closure.pytmains(R‰ReRR§(RR©RŠR•((s7lib/python2.7/site-packages/numba/tests/test_closure.pyttest_issue3659s cCs[d„}tjdƒ}x<|jD]1}||ƒ|ƒ}tjj|||ƒƒq"WdS(NcSs\tjg|jD]}tj|ƒ^qƒtjg|jD]}tjdƒ^q<ƒ|S(Ni(ReRŽR‘tfloat_(R’R R”((s7lib/python2.7/site-packages/numba/tests/test_closure.pytcenters,,i (i (RetzerosR‰RR§(RR¬R’RŠR•((s7lib/python2.7/site-packages/numba/tests/test_closure.pyttest_issue3803s  ( R1R2RRR‰R‹RR R¨RªR®(((s7lib/python2.7/site-packages/numba/tests/test_closure.pyRvÃs    t__main__(t __future__RRbRetnumba.unittest_supporttunittest_supportRstnumbaRRRRt numba.errorsRRRtsupportRR t numba.sixR R R4RvR1R©(((s7lib/python2.7/site-packages/numba/tests/test_closure.pyts  "bÿTa