2@j<ddlZddlZddlZddlmZmZddlmZmZddl m Z ddl m Z m Z Gdde ZGdd e Zed krejdSdS) N)njitjit) TypingErrorUnsupportedError)ir)TestCaseIRPreservingTestPipelinecVeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd S) TestClosurec dfd}tdi||}||ddd||dddS)N c|zSNxYs IC:\Projects\Historical\venv\Lib\site-packages\numba/tests/test_closure.pyadd_Yz3TestClosure.run_jit_closure_variable..add_Y q5Li4(i4) rr assertEqual)selfjitargsrc_add_Yrs @rrun_jit_closure_variablez$TestClosure.run_jit_closure_variables      +#**'**511 R(((  R(((((rc2|ddSNT)forceobjr"rs rtest_jit_closure_variablez%TestClosure.test_jit_closure_variable %%t%44444rc2|ddSNTnopythonr&r's rtest_jit_closure_variable_npmz)TestClosure.test_jit_closure_variable_npmr)rc jdfd}tdi||}||dddtdi||}||ddd||dd||dddS) Nr c|zSrrrs rrz0TestClosure.run_rejitting_closure..add_Y%rrrrrr rr)rr rr! c_add_Y_2rs @rrun_rejitting_closurez!TestClosure.run_rejitting_closure"s      +#**'**511 R((( ,C,,G,,U33  1r***  1r*** R(((((rc2|ddSr$r3r's rtest_rejitting_closurez"TestClosure.test_rejitting_closure4 ""D"11111rc2|ddSr+r5r's rtest_rejitting_closure_npmz&TestClosure.test_rejitting_closure_npm7r7rc ddfd}tdi||}||dddS)Nr c|zzSrr)rrZs r add_Y_mult_ZzDTestClosure.run_jit_multiple_closure_variables..add_Y_mult_Z>sEQ; rrrrr)rr r>c_add_Y_mult_Zrr=s @@r"run_jit_multiple_closure_variablesz.TestClosure.run_jit_multiple_closure_variables:sq        21111,?? **B/////rc2|ddSr$rAr's r#test_jit_multiple_closure_variablesz/TestClosure.test_jit_multiple_closure_variablesD ///>>>>>rc2|ddSr+rCr's r'test_jit_multiple_closure_variables_npmz3TestClosure.test_jit_multiple_closure_variables_npmGrErc Xd}tdi||fd}tdi||}|||5||ddddddS#1swxYwYdS)Nc |dzS)Nr r)as rmult_10z3TestClosure.run_jit_inner_function..mult_10Ks r6Mr intp(intp)c |dzSNr)r c_mult_10s rdo_mathz3TestClosure.run_jit_inner_function..do_mathQs9QU## #rr2)rL)rdisable_compileassertRefCountr)rr rKrQ c_do_mathrPs @rrun_jit_inner_functionz"TestClosure.run_jit_inner_functionJs%   1C000099 !!### $ $ $ $ $1C000099 !!###  I 6 6 / /   YYq\\2 . . . / / / / / / / / / / / / / / / / / /s2 BB#&B#c2|ddSr$rVr's rtest_jit_inner_functionz#TestClosure.test_jit_inner_functionZ ##T#22222rc2|ddSr+rXr's rtest_jit_inner_function_npmz'TestClosure.test_jit_inner_function_npm]rZrN)__name__ __module__ __qualname__r"r(r.r3r6r9rArDrGrVrYr\rrrr r s ) ) )555555)))$222222000??????/// 33333333rr cheZdZdZdZdZdZdZdZdZ dZ d Z e j d Zd Zd S) TestInlinedClosurez 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. czd}t|}||d|ddS)Nc6d}||||zS)Nc ||zSrrrs rinnerzDTestInlinedClosure.test_inner_function..outer..innerl 1u rrrrfs routerz5TestInlinedClosure.test_inner_function..outerjs.   588eeAhh& &rr rrrricfuncs rtest_inner_functionz&TestInlinedClosure.test_inner_functionhsJ ' ' 'U  rEE"II.....rczd}t|}||d|ddS)NcF|dzfd}||||zS)Nrc||zzSrrrys rrfzQTestInlinedClosure.test_inner_function_with_closure..outer..innerys1uqy rrrrfrrs @rrizBTestInlinedClosure.test_inner_function_with_closure..outervsBAA ! ! ! ! !588eeAhh& &rr rjrks r test_inner_function_with_closurez3TestInlinedClosure.test_inner_function_with_closuretsJ ' ' 'U  rEE"II.....rczd}t|}||d|ddS)NcJ|dzfd}||||zS)Nrc|zSrrrqs rrfzSTestInlinedClosure.test_inner_function_with_closure_2..outer..inner 1u rrrss @rrizDTestInlinedClosure.test_inner_function_with_closure_2..outersFAA     aAuuQxx< rr rjrks r"test_inner_function_with_closure_2z5TestInlinedClosure.test_inner_function_with_closure_2sJ   U  rEE"II.....rcd}i}t||t|d}||d|dddS)Na 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 rir )execstriprr)rcodensrls r"test_inner_function_with_closure_3z5TestInlinedClosure.test_inner_function_with_closure_3si   TZZ\\2R[!! rKBwKOO44444rczd}t|}||d|ddS)Nc*fd}|zS)NcZfd}d}tD]}|||z }|S)Ncz|zSrrzrrrs r innermostz^TestInlinedClosure.test_inner_function_nested..outer..inner..innermostsq519$rrrange)rrrsirs` rrfzKTestInlinedClosure.test_inner_function_nested..outer..innersW%%%%%%q&&A1%AArrrhs` rriz.outers0     5Q<< rr rjrks rtest_inner_function_nestedz-TestInlinedClosure.test_inner_function_nestedsJ U  rEE"II.....rcltfdfd}fd}d}d}d}d}dfd }dfd }d } d } d } d} d"d} d"d}d"d}d}d}d}d}d}d}d}||||||| | | | |||||g}|D];}t|}d}|||||<|t5}t d|}||dddn #1swxYwYd}||t |j|t5}t d|}||dddn #1swxYwYd}||t |j|t5}t d| }||dddn #1swxYwYd}||t |j|t5}t d|}||dddn #1swxYwYd }||t |j|t5}t d|}||dddn #1swxYwYd }||t |j|t5}t d|}||dddn #1swxYwYd }||t |j|t5}t d|}|dddn #1swxYwYd!}||t |jdS)#z3 Tests the large number of use cases defined below cN|dkr|S|dz |dz zS)Nr;rr)nfib3s rrz4TestInlinedClosure.test_bulk_use_cases..fib3s41uu4A;;a!e, ,rc$fd}||S)z, Test calling recursive function from inner c|Srr)rrs rrfzETestInlinedClosure.test_bulk_use_cases..outer1..innerstAwwrr)rrfrs router1z6TestInlinedClosure.test_bulk_use_cases..outer1s)     588Orc2|dzfd}||S)z. Test calling recursive function from closure rc |zSrr)rrrs rrfzETestInlinedClosure.test_bulk_use_cases..outer2..inners4477{"rr)rrfrrs @router2z6TestInlinedClosure.test_bulk_use_cases..outer2s8AA # # # # # #588Orc$fd|S)z Test recursive inner c4|dkrdS|dz dSNr;r rrrhs rrfzETestInlinedClosure.test_bulk_use_cases..outer3..inners&q552E!a%LLLLLrrrhs @router3z6TestInlinedClosure.test_bulk_use_cases..outer3s+ ! ! ! ! ! 588Orc2|dzfd|S)z Test recursive closure rc:|zdkrdS|dz dSrrrss rrfzETestInlinedClosure.test_bulk_use_cases..outer4..inners*q51992E!a%LLLLLrrrss @@router4z6TestInlinedClosure.test_bulk_use_cases..outer4s:AA ! ! ! ! ! ! 588Orc.|dzfd}||S)z Test nested closure rc<|zdzfd}||zS)Nr;c|zSrrrrs rinner2zVTestInlinedClosure.test_bulk_use_cases..outer5..inner1..inner2s q5Lrr)rrrrrs @rinner1zFTestInlinedClosure.test_bulk_use_cases..outer5..inner1s>EAI!!!!!vayy1}$rrrrrrs @router5z6TestInlinedClosure.test_bulk_use_cases..outer5s5AA % % % % %6!99 rc.|dzfd}||S)z. Test closure with list comprehension in body rcF|zdz}dt|DS)Nr;cg|]}|Srr).0ts r zZTestInlinedClosure.test_bulk_use_cases..outer6..inner1..s,,,a,,,rr)rrrrs rrzFTestInlinedClosure.test_bulk_use_cases..outer6..inner1s*EAI,,588,,,,rrrs @router6z6TestInlinedClosure.test_bulk_use_cases..outer6s5AA - - - - -6!99 r c|dz}||zzS)z) Test use of outer scope var, no closure rr)rr_OUTER_SCOPE_VARs router7z6TestInlinedClosure.test_bulk_use_cases..outer7sAAq5++ +rc2|dzfd}||S)z+ Test use of outer scope var, with closure rc|zzSrr)rrrs rrfzETestInlinedClosure.test_bulk_use_cases..outer8..inners1u///rr)rrfrrs @router8z6TestInlinedClosure.test_bulk_use_cases..outer8s8AA 0 0 0 0 0 0588Orc2|dzfd}|}||S)z Test closure assignmentrc|zSrrrs rrfzETestInlinedClosure.test_bulk_use_cases..outer9..innerrxrr)rrffrs @router9z6TestInlinedClosure.test_bulk_use_cases..outer9 s7AA     A1Q44Krc:|dzfdfd}||S)z! Test two inner, one calls other rc|zSrrrs rrfzFTestInlinedClosure.test_bulk_use_cases..outer10..innerrxrc|Srrrhs rrzGTestInlinedClosure.test_bulk_use_cases..outer10..inner2suQxxrr)rrrfrs @@router10z7TestInlinedClosure.test_bulk_use_cases..outer10sOAA          6!99 rc|dzfd}|S)z return the closure rc|zSrrrs rrfzFTestInlinedClosure.test_bulk_use_cases..outer11..inner#rxrrrrfrs @router11z7TestInlinedClosure.test_bulk_use_cases..outer11s*AA     Lrc0|dzdfd }||S)z closure with kwargrc|z|zSrrrkwrs rrfzFTestInlinedClosure.test_bulk_use_cases..outer12..inner+1urz!rrrrs @router12z7TestInlinedClosure.test_bulk_use_cases..outer12's7AA " " " " " "588Orrc|dz|z}|S)z outer with kwarg no closurerrrs router13z7TestInlinedClosure.test_bulk_use_cases..outer13/sA AHrc2|dzfd}||S)z! outer with kwarg used in closurerc|zzSrrrs rrfzFTestInlinedClosure.test_bulk_use_cases..outer14..inner8rrrrrrfrs ` @router14z7TestInlinedClosure.test_bulk_use_cases..outer144s8AA " " " " " "588Orc0|dzfd}|||S)z# outer with kwarg as arg to closurerc|z|zSrrrs rrfzFTestInlinedClosure.test_bulk_use_cases..outer15..inner@rrrrs @router15z7TestInlinedClosure.test_bulk_use_cases..outer15<s5AA " " " " "5B<< rcH|dzfd}t||S)z( closure is generator, consumed locally rc3K|zVdSrrrs rrfzFTestInlinedClosure.test_bulk_use_cases..outer16..innerH!e r)listrs @router16z7TestInlinedClosure.test_bulk_use_cases..outer16Ds;AA     a>> !rc.|dzfd}||S)z closure is generator, returned rc3K|zVdSrrrs rrfzFTestInlinedClosure.test_bulk_use_cases..outer17..innerQrrrrs @router17z7TestInlinedClosure.test_bulk_use_cases..outer17Ms4AA     588Orc<|dzfd}||D]}|}|S)z( closure is generator, consumed in loop rc3K|zVdSrrrs rrfzFTestInlinedClosure.test_bulk_use_cases..outer18..innerZrrr)rrfrrrs @router18z7TestInlinedClosure.test_bulk_use_cases..outer18VsJAA     U1XX  HrcF|dz|dzfd}fd}|||S)z# closure as arg to another closure rr;c|zSrr)rz1s rrfzFTestInlinedClosure.test_bulk_use_cases..outer19..innergs 2v rc ||zSrr)rrz2s rrzGTestInlinedClosure.test_bulk_use_cases..outer19..inner2jsqttby rr)rrfrrrs @@router19z7TestInlinedClosure.test_bulk_use_cases..outer19bs\QBQB      ! ! ! ! !6%## #rc.|dzfd}||S)z Test calling numpy in closure rc2|tjzSr)numpycosrs rrfzFTestInlinedClosure.test_bulk_use_cases..outer20..innerss59Q<<''rrrs @router20z7TestInlinedClosure.test_bulk_use_cases..outer20os2AA ( ( ( ( (588Orc.|dzfd}||S)z) Test calling numpy import as in closure rc2|tjzSr)nprrs rrfzFTestInlinedClosure.test_bulk_use_cases..outer21..inner{s26!99}$rrrs @router21z7TestInlinedClosure.test_bulk_use_cases..outer21ws2AA % % % % %588Orcd}d}||dS)z6Test to ensure that unsupported *args raises correctlycdSrr)rJbs rbarzDTestInlinedClosure.test_bulk_use_cases..outer22..barsr)rr;Nrrrs router22z7TestInlinedClosure.test_bulk_use_cases..outer22s%   A CGGGGrr Tr,Nz,Unsupported use of cell variable encounteredz%Cannot capture the non-constant valuez-The use of yield in a closure is unsupported.z,Calling a closure with *args is unsupported.r) rr assertRaisesNotImplementedErrorrassertInstr exceptionrr)rrrrrrrrrrrrrrrrrrrrrrrrrefrlvarraisesmsgrrs @@rtest_bulk_use_casesz&TestInlinedClosure.test_bulk_use_casess  - - - -  -                         , , , , ,                                " " "       $ $ $         VVVVV Wgw gw 1 3 3CIIEC   UU3ZZS 2 2 2 2  2 3 3 v&C&&&v..E E#JJJ               = c3v/00111   2 3 3 v&C&&&v..E E#JJJ               = c3v/00111   { + + v&C&&&w//E E#JJJ               6 c3v/00111   / 0 0 F&C&&&w//E E#JJJ               > c3v/00111   / 0 0 F&C&&&w//E E#JJJ               > c3v/00111   / 0 0 F&C&&&w//E E#JJJ               > c3v/00111   / 0 0 F&C&&&w//E EGGG               = c3v/0011111s~%DDD%FFF%HHH%JJJ%LLL%NNN$O??PPclttd}||ddd|j|jdjd}g}|jD]j}|jD]`}t|tj rDt|j tj r%|j j dkr||ak|t|d |D]&}|jj}d }|||'dS) N)pipeline_classc4d}||||fS)Ncd}d}||z|zS)Nr rrs rrzITestInlinedClosure.test_closure_renaming_scheme..foo..bars1uqy rr)rJrrs rfooz.foos- ! ! !3q6633q66> !rr )#r preserved_irrr;zclosure__locals__bar_v[0-9]+.x)rr r overloads signaturesmetadatablocksvaluesbody isinstancerAssignvalueConstappendlentargetname assertRegex) rrfunc_irstoreblkstmtrrregexs rtest_closure_renaming_schemez/TestInlinedClosure.test_closure_renaming_schemesL 5 6 6 6 " " 7 6 " Rh///-q 12;NK>((** / /C / /dBI../!$*bh77/:+q00!LL...  / UQ''' * *A8=D5E   T5 ) ) ) ) * *rctdtfd}tj|ddS)Ncdd}|S)N皙?c ||zSrrrqs rrz;TestInlinedClosure.test_issue9222..foo..barrgr)rr)rs rrz.TestInlinedClosure.test_issue9222..foos    Jrc*dSrNr)rsrconsumez2TestInlinedClosure.test_issue9222..consumes335588Orgffffff@)rrtestingassert_allclose)rrrs @rtest_issue9222z!TestInlinedClosure.test_issue9222si             ""7799g66666rc:td}|dS)Ncd}tdD]0}tjdt|dDd}1dS)Nrrcg|]}dS)rr)r_s rrzFTestInlinedClosure.test_issue_9577.._inner..s;;;!;;;rr)rrarray) range_startr%s r_innerz2TestInlinedClosure.test_issue_9577.._innersVK1XX  ;;U;%:%:;;;<<<   r)r)rr(s rtest_issue_9577z"TestInlinedClosure.test_issue_9577s,     rc8td}|t5}|dddn #1swxYwYd}||t |jtd}|t5}|ddddn #1swxYwY||t |jtd}||ddS)Nc*dfd}|S)NrcdddSNrr;rresrrzKTestInlinedClosure.test_issue_10379..reassign_nonlocal..barsrrrs @rreassign_nonlocalz>TestInlinedClosure.test_issue_10379..reassign_nonlocals3A      CEEEHrz!Reassigning the nonlocal variablec.dfd}|S)NrcdrddSdSr-r)condrsrrzLTestInlinedClosure.test_issue_10379..reassign_in_branch..bars#AAArr)r1rrs` @rreassign_in_branchz?TestInlinedClosure.test_issue_10379..reassign_in_branchs9A       CEEEHrFc*dfd}|S)Nrc ddS)NrrresrrzNTestInlinedClosure.test_issue_10379..assign_nonlocal_once..bar&s rrrs @rassign_nonlocal_oncezATestInlinedClosure.test_issue_10379..assign_nonlocal_once"s3A      CEEEHrr)rrrrrrr)rr.rrr2r5s rtest_issue_10379z#TestInlinedClosure.test_issue_10379s       / 0 0 F                    1 c3v/00111       / 0 0 &F  u % % % & & & & & & & & & & & & & & & c3v/00111      --//33333s# AA A! B99B=B=N)r]r^r___doc__rmrtryrrrrr!rrun_test_in_subprocessr)r6rrrraraas / / / / / / / / /555(///&B2B2B2H * * *D777$$%$4444444444rra__main__)rrunittestnumbarrnumba.core.errorsrr numba.corernumba.tests.supportrr r rar]mainrrrr@s  ;;;;;;;;BBBBBBBBR3R3R3R3R3(R3R3R3jL4L4L4L4L4L4L4L4^ zHMOOOOOr