\ K]c@`sddlmZmZmZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddlmZddlZddlmZddlmZmZm Z ddlmZmZddlmZdd lmZd d lmZdd lmZm Z ej!d d dfkZ"ej#j$doFe"Z%ej&d dkZ'dZ(e'pke%Z)ej*e)e(Z+edZ,edddffdZ-edddZ.edZ/edZ0eddfdZ1eddfdZ2edZ3e) rMed e4d!Z5nd"ej6fd#YZ7d$e7fd%YZ8d&fd'YZ9d(d)Z:e+d*e7fd+YZ;e<d,krej=ndS(-i(tprint_functiontdivisiontabsolute_importN(tcontextmanager(tdeepcopy(tunittest_support(tnjittstencilttypes(t compile_extratFlags(tregistry(tParallelOptionsi(ttag(t LoweringErrort TypingErroriitwin32i sparfors not supportedcC`s$d|d|d|d|dS( Ng?iii(ii(ii(ii(ii((ta((s8lib/python2.7/site-packages/numba/tests/test_stencils.pytstencil1_kernel$st neighborhoodicC`s:|d}x%tddD]}|||7}qWd|S(Niiig333333?(trange(Rtcumti((s8lib/python2.7/site-packages/numba/tests/test_stencils.pytstencil2_kernel)s tcvalg?cC`s d|dS(Ng?ii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pytstencil3_kernel1scC`sDd|d|d|d|d|d |d |d |d S( Ng?iii(ii(ii(ii(ii(ii(ii(ii(ii((Rtb((s8lib/python2.7/site-packages/numba/tests/test_stencils.pytstencil_multiple_input_kernel6scC`sD||d|d|d|d|d|d |d |d S( Niii(ii(ii(ii(ii(ii(ii(ii(ii((RRtw((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt!stencil_multiple_input_kernel_var<ststandard_indexingRcC`s |d|d|d|dS(Niii((RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt!stencil_with_standard_indexing_1dBscC`s@|d|d|d|d|d|d |d |d S( Niii(ii(ii(ii(ii(ii(ii(ii(ii((RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt!stencil_with_standard_indexing_2dGscC`s|dS(Ni((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt addone_njitMstparallelcC`s|dS(Ni((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt addone_pjitTstTestStencilBasecB`sDeZeZdZdZdZdZdZdZ RS(cG`s6t|_|jjdtt|j|dS(Ntnrt(R tcflagstsettsuperR$t__init__(tselftargs((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR)]s cC`s(ttjjtjj||d|iS(N(R R t cpu_targetttyping_contextttarget_contexttNone(R*tfunctsigtflags((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt _compile_thisds cK`sQt}|jd|s"tn|}|jdt||j|||S(NR%t auto_parallel(R R'tTrueR R3(R*R0R1tkwsR2toptions((s8lib/python2.7/site-packages/numba/tests/test_stencils.pytcompile_parallelis   cC`s|j||d|jS(NR2(R3R&(R*R0R1((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt compile_njitpscO`sVtg|D]}tj|^q }|j||}|j||}||fS(N(ttupletnumbattypeofR8R9(R*tpyfuncR+tkwargstxR1tcpfunctcfunc((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt compile_allss(c G`s|j||\}}||}||}|j|}|j|} tjj||ddtjj||ddtjj| |dd|jd|jjdS(Ntdecimalis@do_scheduling(RBt entry_pointtnpttestingtassert_almost_equaltassertIntlibraryt get_llvm_str( R*tno_stencil_funcR=R+RAR@texpectedt py_outputt njit_outputt parfor_output((s8lib/python2.7/site-packages/numba/tests/test_stencils.pytcheck{s  ( t__name__t __module__tFalset_numba_parallel_test_R)R3R8R9RBRP(((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR$Ys     t TestStencilcB`s.eZdZeeddZeeddZeeddZeeddZeeddZ eeddZ eeddZ eedd Z eedd Z eedd Zed ZRS( cO`stt|j||dS(N(R(RUR)(R*R+R>((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR)st importantcC`sKd}d}d}d}|j||||j|||dS(sHTests whether the optional out argument to stencil calls works. cS`sZtj|dj||f}tj|dj||f}t|d|}|S(Nitout(REtarangetreshapetzerosR(tntAtB((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_with_outs""cS`s2tj|dj||f}t|}|S(Ni(RERXRYR(R[R\R]((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyttest_without_outs" cS`stj|dj||f}tj|dj||f}xtd|dD]|}xstd|dD]^}d|||df||d|f|||df||d|f|||fsRW(RERXRYRZR;R(R[R\R]((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_impl1~s "" cS`sftj|dj||f}tj|dj||f}d}tj||}|S(NicS`s$d|d|d|d|dS( Ng?iii(ii(ii(ii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pytsfs(RERXRYRZR;R(R[R\R]Ry((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_impl2s "" cS`stj|dj||f}tj|dj||f}xtd|dD]|}xstd|dD]^}d|||df||d|f|||df||d|f|||ftj|}tj|}tjd|d||S(NcS`sd|d|d|dS(Ng333333?iii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRwtRW(RERXRZR;R(R[R\R]((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_implscS`sjtj|}tj|}xEtd|dD]0}d||d||||d||N( tasttAssigntNametStoretCallt AttributetLoadtgen_numtkeywordtgen_callRRR/(R*torigtvart dtype_vartinit_valtnew((s8lib/python2.7/site-packages/numba/tests/test_stencils.pytgen_alloc_return`s.        ! c C`sg|D]$}tjd|dtj^q}tjdtjdtjd|dtjdtjdtjd|dtjdtjgd|}|S(s Generates an AST equivalent to: `retvar[(*index_names,)] = value[]` RRRRtslicetelts(RRRRt SubscripttIndextTupleR(R*RRt index_namesR?t elts_infoR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt gen_assigns1      cC`st|tr$tjd|}n|}t|trNtjd|}n|}tjdtjd|dtjdtjdtjdddtjd||gd gd dd dd |d gS(s} Generates an AST equivalent to a loop in `var` from `start` to `stop` with body `body`. R[ttargetRRtiterR0RR+RRR>tbodytorelseN( t isinstancetintRtNumtForRRRRR/(R*RtstarttstopRt start_valtstop_val((s8lib/python2.7/site-packages/numba/tests/test_stencils.pytgen_loops   cC`s(tjdtjd|dtjS(sE Generates an AST equivalent to `return var` RRR(RtReturnRR(R*R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt gen_returnscC`stjdtjd|S(s'Generates an Index with the given valueRR[(RRR(R*R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt gen_slicescC`s:tjdtjd|dtjd|dtjS(sA Generates AST equivalent to `name.attr` RRRR(RRRR(R*tnameR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pytgen_attrs c C`s|j||}|j|}tjd|d|dtj}|r|dkrctjn tj}tjd|d|dtjdt |}n|S( s Generates an AST equivalent to a subscript, something like: name.attr[slice(index) +/- offset] RRRitlefttoptrightR[( RRRRRtAddtSubtBinOpRtabs( R*RRtindexRgt attributetslisetsstpm((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt gen_subscripts$$3cC`set|tr"tjt|St|dkrAtj|Stjtjtj| SdS(s? Generates an ast.Num of value `value` iN(RtboolRRRRtUnaryOptUSub(R*R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRs  c C`sg|D](}tjdd|dtj^q}|d k rg|D];}tjdd|dtjt|jdj^qH}ng}tjd|dtj}tj dtj d|d|d |d d d d dtjS( s| Generates an AST equivalent to a call, something like: `call_name(*args, **kwargs) Rs%sRRRiR0R+RRR>N( RRRR/RtparseRRRtExprR(R*t call_nameR+R>R?t fixed_argsRR0((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRs5 H N(RQRRt__doc__R)tvRR?tchrtordtidsRRRR/RRRRRRR(((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRLs B     t FoldConstcB`sDeZdZiejej6ejej6ej ej 6Z dZ RS(sn Folds const expr, this is so const expressions in the relidx are more easily handled cC`s|j|}|jj|jj}t|dd}t|dd}|oZ|oZ|sa|St|tj rt|tj rtj ||j j |j j S|SdS(NRR( t generic_visitt supported_opstgetRt __class__tgetattrR/RRRRR[R(R*tnodeRtlhstrhs((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt visit_BinOps"( RQRRRtoperatortaddRRtsubRtmultMultRR(((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRs   t FixRelIndexcB`seeZdZdZdZdZedZedZedZ edZ RS(sa Fixes the relative indexes to be written in as induction index + relative index cO`stjj|||tjj|||||_||_d|_d|_ d|_ t j t j|_t j t j|_|r|ng|_||_|rdnd|_dS(Nis__%sns__%s(RtNodeTransformerR)RRt _argnamest_const_assignst_idx_lenR/t_minst_maxesREtiinfoRtmint_imintmaxt_imaxt_standard_indexingt _neighborhoodt_id_pat(R*targnamest const_assignsRRR+R>((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR) s      cC`sFt|tjr|jSt|tjr6|jj StddS(sF Gets the literal value from a Num or UnaryOp s,get_val_from_num: Unknown indexing operationN(RRRR[RtoperandR(R*R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pytget_val_from_nums  c`s2fd}fd}fdfdfd}fd}j|}|jjjkr*|jjjkr*t|jtjr||St|jtj r||St|jjtj r||St|jtj r||St dn|SdS( s Transforms subscripts of the form `a[x]` and `a[x, y, z, ...]` where `x, y, z` are relative indexes, to forms such as: `a[x + i]` and `a[x + i, y + j, z + k]` for use in loop induced indexing. c `sg}xt|jjjD]~\}}jj||}|jtjdtj dj j |dtj dtj d|dtj qWjdkrt|_n$jt|krtdnt|jtjr d}t|ntj }tjd |jd tjd tjd |dtj dtj d|}tj||tj|jdkrjdkrjgj_jgj_njsx2t|jjjD]\}}t|tjs#t|tj r5j!|} n@t"|dri|j#jkrij|j#} n td | j|kr| j|7 K0    cC`s)|jdkrtdn|jSdS(Nis+Transform has not been run/no indexes found(R R(R*((s8lib/python2.7/site-packages/numba/tests/test_stencils.pytidx_lenWs cC`s|jS(N(R (R*((s8lib/python2.7/site-packages/numba/tests/test_stencils.pytmaxes_scC`s|jS(N(R (R*((s8lib/python2.7/site-packages/numba/tests/test_stencils.pytminscscC`s|jS(N(R(R*((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt id_patterngs( RQRRRR)RR4tpropertyR5R6R7R8(((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRs  0tTransformReturnscB`s/eZdZdZdZedZRS(s; Transforms return nodes into assignments. cO`s^tjj|||tjj|||||_|j|_d|j}||_dS(Ns__b%s( RRR)RRt _relidx_infoRt _ret_var_idxt _retvarname(R*t relidx_infoR+R>tretvar((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR)ps   cC`sb|j||jj}|jj}|j|j|jgt|D]}||j|^qDS(N( RR;R5R8RR=RRR(R*Rtnloopst var_patterntl((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt visit_Returnxs     cC`s|jS(N(R=(R*((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt ret_var_names(RQRRRR)RCR9RD(((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR:ks  tFixFunccB`s eZdZdZdZRS(s The main function rewriter, takes the body of the kernel and generates: * checking function calls * return value allocation * loop nests * return site * Function definition as an entry point c O`stjj|||tjj||||j|_|j|_|j |_ ||_ ||_ |rn|ng|_ |r|nt|_g|jD]} | |j kr| ^q|_t|dr|j|_n ||_|jd|_dS(NRi(RRR)RRtoriginal_kernelt_original_kernelRR trettyt_rettyR;t _ret_infoRR:Rt _relidx_argsRttolistRt stencil_arr( R*tkpropsR>tret_infoRRRR+R>R?((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR)s     . c C`s|j|d}tj|}tj|jjd}tj||jd|j ddg}|j j }d}|j j }|j} d} xt |D]} ||j j| |j j| \} } | }|j|jd| | }|j||j| ||d| }|g} | d 7} qWtj|||jjd}|jj}|j|j|||j}tj|||j|}tj||trtjdtjg}ntjddg}tjd d d tj!g}tj"d |j#j#|d|dddddgdg}tj$dd|j%d |d|||j&|j|||gdg}tj|||S(s Transforms the kernel function into a function that will perform the stencil like behaviour on the kernel. c_`s|jd}|dj}|dk rRt|t|krRtdqRnxB|dD]6}t|dr]||jkrtdqq]q]WdS(NRisInvalid neighborhood suppliediRs#Input stencil arrays do not commute(RRR/RdRR(R+R>Rt init_shapeR?((s8lib/python2.7/site-packages/numba/tests/test_stencils.pytcheck_stencil_arrayss  iRQR>RcS`s<|dkrdn| }|dkr,| nd}||fS(Ni((R7tmaxstminlimtmaxlim((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt computeboundsRRiRR/RR+tdefaultstvarargtkwargt kwonlyargst kw_defaultsRs__%stdecorator_listN('Rtinspectt getsourceRRtstripRRRRKR;R5R8RR7R6RRMRRRRIRRJRDRRRt_py27RtParamRR/Rt argumentsR+t FunctionDefRRG(R*RRQtchecksrct check_implt checker_callR@RURAt loop_bodyt loop_countRBRSRTtminboundtmaxboundtloopst _rettynameR?tallocatetreturnert add_kwargRVtnewargsR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pytvisit_FunctionDefsz            !       (RQRRRR)Rp(((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyREs tGetKernelPropscB`seeZdZdZdZdZedZedZedZ edZ RS(sV Gets the argument names and other properties of the original kernel. cO`s]tjj|||tjj|||d|_d|_d|_d|_ i|_ dS(N( Rt NodeVisitorR)RRR/R t _kwargnamesRIRGR (R*R+R>((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR) s    cC`s|jdk s|jdk r-tdntr<d}nd}g|jjD]}t||^qO|_|jjrg|jjD]}|j^q|_n|j |j |j}t j dt j dddt jgd|j|_t jt||_|j|dS(Ns%multiple definition of function/args?RRRt __retdtypeRR(R R/Rst RuntimeErrorR_R+RRXRRRRRRRRRIRRRGR(R*RRR?tcompute_retdtype((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRps + (!cC`s|j||j}t|dkr|d}t|tjrt|jtjrr|jj|j |j scC`s|jS(s) The return type (RI(R*((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRHFscC`s|jS(s7 The original unmutated kernel (RG(R*((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRFMs( RQRRRR)RpRyR9RRRHRF(((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRqs  tFixCallscB`seZdZdZRS(s3 Fixes call sites for astor (in case it is in use) c C`sD|j|tjd|jd|jd|jdddd}|S(NR0R+RRR>(RRRR0R+RR/(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt visit_CallWs      (RQRRRR{(((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRzTsc `stj|}tj|j}t}|rPtdttj|nfd}|||rtdttj|dty#ddl } t| j |Wqt k rqXn|S(s Generates the AST tree for a stencil from: func - a python stencil kernel cval, standard_indexing and neighborhood as per the @stencil decorator tORIGINALc `sj}|j||j}|j}rfx2D]'}||kr8d}t|q8q8Wnj}|j|j||}|j|j|}|j|j|||} | j|j } | j|t j |j ddS(s the pipeline of manipulations s4Non-existent variable specified in standard_indexingiN( RqtvisitRRRRRR:RERzRRR( ttreet kernel_propstargnmt const_asgnR?Rt fold_constt relidx_fixertreturn_transformertfixert callFixer(RRR*R(s8lib/python2.7/site-packages/numba/tests/test_stencils.pytpipeliness6              s NEWtinclude_attributesiN( R\R]RRR^RStprinttdumpR5tastort to_sourcet ImportError( R*R0RRRtsrcR~tDEBUGRR((RRR*Rs8lib/python2.7/site-packages/numba/tests/test_stencils.pytgenerate_stencil_treebs" -    (RQRRRRRRRRR:RERrRqRzR(((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRFseLtconstantc K`s)t|tsd}|}n't|ts9td|}d }x-|D]%}|dkrLtd|qLqLW|dkrtd|n|jdd}|jdd }|jdd }t}|j||||} t| d d d d } | j d} t j | t } | S(ss A pure python implementation of (a large subset of) stencil functionality, equivalent to StencilFunc. RsWstencil mode should be a stringRRRsUnknown stencil option sUnsupported mode style itfilenamestmodetexecN(RRR( RRRR/RRRRtcompilet co_conststpytypest FunctionTypetglobals( t func_or_modeR7RR0toptionRRRtgenR~tmod_codet func_codet full_func((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt pyStencils,        tTestManyStencilscB`seZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!Z#d"Z$d#Z%d$Z&e'j(d%d&Z)d'Z*d(Z+d)Z,d*Z-d+Z.d,Z/d-Z0d.Z1d/Z2d0Z3d1Z4d2Z5d3Z6d4Z7d5Z8d6Z9d7Z:d8Z;d9Z<e'j(d:d;Z=d<Z>d=Z?d>Z@d?ZAd@ZBdAZCdBZDdCZEdDZFdEZGdFZHdGZIdHZJdIZKdJZLdKZMdLZNdMZOdNZPdOZQdPZRdQZSdRZTdSZUdTZVdUZWdVZXdWZYdXZZdYZ[dZZ\d[Z]d\Z^d]Z_d^Z`d_Zad`ZbdaZcdbZddcZeddZfdeZgdfZhdgZidhZjdiZkdjZle'j(dkdlZmdmZne'j(dkdnZoRS(ocO`stt|j||dS(N(R(RR)(R*R+R>((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR)sc `s|jdt}|jd}t}ggtd d fd}t|tr|d}|d} |d} |d} n|}|} |} |} i|d6} | j|t} yGtd||}|d |jd |}|r t d |nWn;t k rJ}||d |Wd QX|}t} nXd }|| d t | |}Wd QXt |d krfd}n^t |dkrfd}n:t |dkrfd}nt dt |tg|D]}tj|^q }d }|| d&|j||}|j|}Wd QXd }|| d&|j||}|j|}Wd QX|rt d|t d|t d|n| ry<| stjj||dd |j|j|jnWnPt k rg}jd dt|t|fft dt|nXy<| stjj||dd |j|j|jnWnPt k r}jddt|t|fft dt|nXy| sutjj||dd |j|j|jy|jd|jjWqutk rqd}t|quXnWqt k r}jddt|t|fft dt|qXn|rt dnrgD]}d |^q}t d!|nrgD]}d |d"^q"}d#j!gD]}d$|^qI}d%|}d&|}t ||n| s|d krt d'|qnd S((s For a given kernel: The expected result is computed from a pyStencil version of the stencil. The following results are then computed: * from a pure @stencil decoration of the kernel. * from the njit of a trivial wrapper function around the pure @stencil decorated function. * from the njit(parallel=True) of a trivial wrapper function around the pure @stencil decorated function. The results are then compared. R7texpected_exceptionc3`sy dVWntk r}|dk r~t|dr<|n|g}t}x!|D]}|t||O}qRW|sqqj|dt|t|ffnX|dk rj|ndS(Nt__iter__s%s: %s(t ExceptionR/RRSRRRR(textytusecaseRtlextytfoundtex(t should_failtshould_not_fail(s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt errorhandlers    ! RRRtparforRRs Expected: Ns@stencilic`s |S(N((targ0(tstencil_func_impl(s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt wrap_stencil0sic`s ||S(N((Rtarg1(R(s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR3sic`s|||S(N((RRtarg2(R(s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR6ss+Up to 3 arguments can be provided, found %stparforss @stencil_output: s njit_output: s parfor_output: RCs%s: %ss@stencil failed: %ss@njit failed: %ss@do_schedulings*Could not find `@do_scheduling` in LLVM IRs@njit(parallel=True) failed: %ss s%ssMThe following implementations should have raised an exception but did not: %siR|s%s: Message: %s sNThe following implementations should not have raised an exception but did: %s sErrors were: %ss)pyStencil failed, was not caught/expected("RtdictRSRR/RtupdateR5RRRRRdRR:R;R<R9RDR8RERFRGt assertEqualRRRRRHRIRJRRutjoin( R*R=R+R>R7Rt DEBUG_OUTPUTRt pystencil_ext stencil_extnjit_ext parfor_ext stencil_argstexpected_presenttast_implRLRtpyStencil_unhandled_extstencilfunc_outputRR?R1RNt wrapped_cfuncROtwrapped_cpfuncRRRterrststr1tstr2((RRRs8lib/python2.7/site-packages/numba/tests/test_stencils.pyRPs         (    )    !   cK`s\t}d|dW|S(NRRRR(RR/titems(R*R>RtkR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pytexception_dicts     cC`s8d}tjdjdd}|j||dS(s rel indexcS`s|dS(Ni(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsi iiN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic00s cC`s8d}tjdjdd}|j||dS(srel index add constcS`s|dS(Nii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsg(@iiN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic01s cC`s8tjdjdd}d}|j||dS(srel index add constg(@iicS`s|dS(Nii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic02s cC`s8tjdjdd}d}|j||dS(srel index add constg(@iicS`s|dS(Nii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic03s cC`s8tjdjdd}d}|j||dS(srel index add constg(@iicS`s|dS(Nii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic04s cC`s8tjdjdd}d}|j||dS(srel index add constg(@iicS`s|dS(Nii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic05s cC`s8tjdjdd}d}|j||dS(srel index add constg(@iicS`s|dS(Nii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic06s cC`s8tjdjdd}d}|j||dS(srel index add constg(@iicS`s|dS(Ni(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic07s cC`s8tjdjdd}d}|j||dS(srel index add constg(@iicS`s|dS(Ni(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic08s cC`s8tjdjdd}d}|j||dS(srel index add constg(@iicS`s|dS(Nii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic09s cC`s8tjdjdd}d}|j||dS(srel index add constg(@iicS`s|d|dS(Nii(ii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic10s cC`s8tjdjdd}d}|j||dS(srel index add constg(@iicS`s|d|dS(Niii(ii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic11s cC`s8tjdjdd}d}|j||dS(srel index add constg(@iicS`s|d|dS(Nii(ii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic12s cC`s8tjdjdd}d}|j||dS(srel index add constg(@iicS`s|d|dS(Nii(ii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic13s cC`s8tjdjdd}d}|j||dS(s!rel index add domain change consti iicS`s |ddS(Niy?(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR sN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic14s cC`s8tjdjdd}d}|j||dS(s!rel index add domain change consti iicS`sd}|d|S(Ny?i(ii((Rtt((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic14bs cC`s8tjdjdd}d}|j||dS(stwo rel index, add consti iicS`s|d|ddS(Niig?(ii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic15s cC`sPtjdjdd}d}|jdt}|j||d|dS(stwo rel index OOB, add consti iicS`s|d|ddS(Nii g?(ii(i i((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR#sRRN(RERXRYRt IndexErrorRP(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic16s cC`s8tjdjdd}d}|j||dS(s&two rel index boundary test, add consti iicS`s|d|ddS(Niig?(ii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR.sN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic17*s cC`s8tjdjdd}d}|j||dS(s&two rel index boundary test, add consti iicS`s|d|ddS(Niig?(ii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR6sN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic182s cC`s8tjdjdd}d}|j||dS(s&two rel index boundary test, add consti iicS`s|d|ddS(Niig?(ii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR>sN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic19:s cC`s8tjdjdd}d}|j||dS(s&two rel index boundary test, add consti iicS`s|d|ddS(Niig?(ii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRFsN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic20Bs cC`s8tjdjdd}d}|j||dS(ssame rel, add consti iicS`s|d|ddS(Nig?(ii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRNsN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic21Js cC`s8tjdjdd}d}|j||dS(s%rel idx const expr folding, add constg(@iicS`s|d|ddS(Niig?i(ii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRVsN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic22Rs cC`s8tjdjdd}d}|j||dS(srel idx, work in bodyg(@iicS`s+tjd|d}|d|d|S( Ni iii(iii(ii(ii(REtsin(RR?((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR^sN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic23Zs cC`s8tjdjdd}d}|j||dS(s,rel idx, dead code should not impact rel idxg(@iicS`s'tjd|d}|d|dS( Ni iii(iii(ii(ii(RER(RR?((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRgsN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic23acs cC`sDtjdjdd}d}|j||dttgdS(s1d idx on 2d arri iicS`s |ddS(Nig?((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRpsRN(RERXRYRPRR(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic24ls cC`sDtjdjdd}d}|j||dttgdS(sno idx on 2d arri iicS`sdS(Ng?((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRxsRN(RERXRYRPRR(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic25ts cC`s;tjdjddd}d}|j||dS(s3d arri@iiicS`s|d|ddS(Niig?(iii(iii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic26|s cC`s>tjdjdddd}d}|j||dS(s4d arriiiicS`s|d|ddS(Niiig?(iiii(iiii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic27s! cC`sDtjdjddjtj}d}|j||dS(s type widen i iicS`s|dtjdS(Nig$@(ii(RER(R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsN(RERXRYtastypetfloat32RP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic28s' cC`sDtjdjdd}d}|j||dttgdS(sconst index from func g(@iicS`s|dttjdfS(Ni(RREtcos(R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsRN(RERXRYRPRR(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic29s cC`s8tjdjdd}d}|j||dS(s signed zerosg(@iicS`s|dS(Ni(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic30s cC`s8tjdjdd}d}|j||dS(sdoes a const propagate? 2Dg(@iicS`sd}||dfS(Nii((RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic31s s constant folding not implementedcC`s8tjdjdd}d}|j||dS(sdoes a const propagate?g(@iicS`sd}d|}||dfS(Nii((RtsR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRs N(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic31bs cC`s,tjd}d}|j||dS(sdoes a const propagate? 1Dg(@cS`sd}||S(Ni((RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsN(RERXRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic31cs cC`sDtjdjdd}d}|j||dttgdS(styped int indexg(@iicS`s|tjddfS(Nii(REtint8(R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsRN(RERXRYRPRR(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic32s cC`s8tjdjdd}d}|j||dS(s add 0d arrayg(@iicS`s|dtjdS(Nii(ii(REtarray(R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic33s cC`s8d}tjdjdd}|j||dS(s<More complex rel index with dependency on addition rel indexcS`s;d|d}||d|d|d tj|d S( Ng@iiii(ii(ii(ii(ii(ii(RER(Rtg((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsii N(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic34s cC`sid}tjdjdd}|jdtdtdt}|j||did d 6d |d S( s simple cval cS`s|dS(Nii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsg(@iiRRRR7iRRN(RERXRYRRRRP(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic35s   cC`sEd}tjdjdd}|j||didd6dS( smore complex with cvalcS`s |d|d|d|dS(Niii(ii(ii(ii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsg(@iiR7g@RN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic36s cC`sEd}tjdjdd}|j||did d6d S( s cval is exprcS`s |d|d|d|dS(Niii(ii(ii(ii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsg(@iiR7igO@RNgQ@(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic37s cC`sid}tjdjdd}|jdtdtdt}|j||did d 6d |d S( scval is complexcS`s |d|d|d|dS(Niii(ii(ii(ii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsg(@iiRRRR7y?RRN(RERXRYRRRRP(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic38s   cC`s[d}tjdjdd}|j||ditjdtjdd6d S( scval is func exprcS`s |d|d|d|dS(Niii(ii(ii(ii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsg(@iiR7g@iRN(RERXRYRPRR(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic39s cC`sVd}tjdjdd}tjdjdd}|j|||dS(s2 args!cS`s|d|dS(Niii(ii(ii((RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsg(@iiN(RERXRYRP(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic40s cC`sbd}tjdjdd}tjdjdd}|j|||dttgdS( s(2 args! rel arrays wildly not same size!cS`s|d|dS(Niii(ii(ii((RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsg(@iig?iRN(RERXRYRPRR(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic41 s   cC`sbd}tjdjdd}tjdjdd}|j|||dttgdS(s%2 args! rel arrays very close in sizecS`s|d|dS(Niii(ii(ii((RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsg(@iig"@RN(RERXRYRPRR(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic42s   cC`sVd}tjdjdd}tjdjdd}|j|||dS(s2 args more complexitycS`s |d|d|d|d S( Niiiii(ii(ii(ii(ii((RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR"sg>@iiN(RERXRYRP(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic43 s cC`sbd}tjdjdd}tjdjdd}|j|||dttgdS(s!2 args, has assignment before usecS`sd|d<|dS(Ni ii(ii(ii((RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR*s g(@iiRN(RERXRYRPRR(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic44(s   cC`sbd}tjdjdd}tjdjdd}|j|||dttgdS(s02 args, has assignment and then cross dependencycS`sd|d<|d|dS(Ni ii(ii(ii(ii((RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR5s g(@iiRN(RERXRYRPRR(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic453s   cC`sbd}tjdjdd}tjdjdd}|j|||dttgdS(s#2 args, has cross relidx assignmentcS`s|d|d<|d|dS(Niii(ii(ii(ii(ii((RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR@sg(@iiRN(RERXRYRPRR(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic46>s   cC`std}tjdjdd}tjdjdd}tjdjdd}|j||||dS(s3 argscS`s|d|d|dS(Niii(ii(ii(ii((RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRKsg(@iiN(RERXRYRP(R*RRRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic47Is  cC`s8d}tjdjdd}|j||dS(s22 args, has assignment before use via memory aliascS`s-|j}d|ddddf<|dS(Ni ii(ii(tT(RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRVs g(@iiN(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic48Ts cC`scd}tjdjdd}tjdjdd}|j|||didd6dS( s#2 args, standard_indexing on secondcS`s|d|dS(Niii(ii(ii((RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR_sg(@iiR7RRN(RERXRYRP(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic49]s s&dynamic range checking not implementedcC`sid}tjdjdd}tjdjdd}|j|||didd6dtd S( s2 args, standard_indexing OOBcS`s|d|dS(Niii(ii(ii((RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRhsg(@iiR7RRRN(RERXRYRPR(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic50es  c C`sud}tjdjdd}tjdjdd}|j|||diddgd6d ttgd S( s$2 args, standard_indexing, no relidxcS`s|d|dS(Niii(ii(ii((RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRvsg(@iiR7RRRRN(RERXRYRPRR(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic51ts c C`sd}tjdjdd}tjdjdd}tjdjdd}|j||||didd 6d S( s(3 args, standard_indexing on middle arg cS`s|d|d|dS(Niii(ii(ii(ii((RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsg(@iig@iR7RRN(RERXRYRP(R*RRRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic52s  c C`sd}tjdjdd}tjdjdd}|jdtdtdtdt}|j|||d id d 6d |d S(s92 args, standard_indexing on variable that does not existcS`s|d|dS(Niii(ii(ii((RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsg(@iiRRRRR7RRRN(RERXRYRRRRP(R*RRRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic53s    cC`scd}tjdjdd}tjdjdd}|j|||didd6dS( s)2 args, standard_indexing, index from varcS`sd}|d|d|fS(Niii(ii((RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsg(@iiR7RRN(RERXRYRP(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic54s cC`scd}tjdjdd}tjdjdd}|j|||didd6dS( s62 args, standard_indexing, index from more complex varcS`s&d}d|}|d|d|fS(Niii(ii((RRRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRs g(@iiR7RRN(RERXRYRP(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic55s cC`scd}tjdjdd}tjdjdd}|j|||didd6dS( s,2 args, standard_indexing, added complexity cS`s_d}d}x(|dddfD]}||7}q#Wd|d}|d|d|f|S(Niii(ii((RRRtaccRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRs g(@iiR7RRN(RERXRYRP(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic56s cC`scd}tjdjdd}tjdjdd}|j|||didd6dS( s12 args, standard_indexing, split index operation cS`s|d}|d|dS(Nii(ii((RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRs g(@iiR7RRN(RERXRYRP(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic57s cC`scd}tjdjdd}tjdjdd}|j|||didd6dS( s?2 args, standard_indexing, split index with broadcast mutation cS`s|dd}|d|dS(Nii(ii((RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsg(@iiR7RRN(RERXRYRP(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic58s c C`srd}tjdjdd}tjdjdd}d}|j||||diddgd 6d S( s>3 args, mix of array, relative and standard indexing and constcS`s|d|d|S(Nii(ii(ii((RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsg(@iii R7RRRN(RERXRYRP(R*RRRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic59s  c C`sd}tjdjdd}tjdjdd}d }|jdt}|j||||didd gd 6d |d S(sP3 args, mix of array, relative and standard indexing, tuple pass throughcS`s|d|d|dS(Nii(ii(ii((RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsg(@iii RR7RRRRN(i (RERXRYRRRP(R*RRRRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic60s cC`sid}tjdjdd}tjdjdd}|j|||didd6dtd S( s"2 args, standard_indexing on firstcS`s|d|dS(Nii(ii(ii((RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsg(@iiR7RRRN(RERXRYRPR(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic61s  cC`sjd}tjdjdd}tjdjdd}|j|||didd6dd 6d S( s"2 args, standard_indexing and cvalcS`s|d|dS(Nii(ii(ii((RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRsg(@iiR7RRg$@RN(RERXRYRP(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic62s c C`sd}tjdjdd}tjdjdd}|jdtdtdtd t}|j|||d id d 6d |dS(sQ2 args, standard_indexing applied to relative, should fail, non-const idxcS`s|d|dfS(Nii(ii((RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR sg(@iii RRRRR7RRRN(RERXRYRRRRP(R*RRRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic63 s    cC`sQd}tjdjdd}|j||didd6dttgd S( s!1 arg that uses standard_indexingcS`s|dS(Ni(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR sg(@iiR7RRRN(RERXRYRPRR(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic64 s  cC`s9d}tjd}|j||did d6dS( sbasic induced neighborhood testcS`s6d}x%tddD]}|||7}qW|dS(Niiii(R(RtcumulR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR' sgN@R7iiRN(ii((ii(RERXRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic65% s cC`s9d}tjd}|j||did d6dS( sbasic const neighborhood testcS`s6d}x%tddD]}||d7}qW|dS(Niiii(R(RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR2 sgN@R7iiRN(ii((ii(RERXRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic660 s cC`sKd}tjd jdd}|j||did dfd 6d S(s"basic 2d induced neighborhood testcS`sVd}xEtddD]4}x+tddD]}||||f7}q,WqW|dS(Niiiii ii2(R(RRRR`((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR< s g$@g4@i iR7iiiRNgi@(ii(ii(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic67: s cC`sQd}tjd jdd}|j||didd 6d ttgd S(s basic 2d induced 1D neighborhoodcS`s<d}x+tddD]}||d|f7}qW|dS(Niiii ii2(R(RRR`((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRG sg$@g4@i iR7iiRRNgi@(ii((ii(RERXRYRPRR(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic67bE s  cC`sKd}tjd jdd}|j||did dfd 6d S(s0basic 2d one induced, one cost neighborhood testcS`sVd}xEtddD]4}x+tddD]}|||df7}q,WqW|dS(Niiiii ii2(R(RRRR`((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR\ s g$@g4@i iR7iiiRNgi@(ii(ii(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic68Z s cC`sKd}tjd jdd}|j||did dfd 6d S(s#basic 2d two cost neighborhood testcS`sPd}x?tddD].}x%tddD]}||d7}q,WqW|dS( Niiiii i(iii2(R(RRRR`((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRh s g$@g4@i iR7iiiRNgi@(ii(ii(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic69f s cC`sKd}tjd jdd}|j||did dfd 6d S(sneighborhood adding complexitycS`sjd}d}xStddD]B}||}x/tddD]}||||f|7}q<WqW|dS( Nig(@iiii ii2(R(RRtzzRRR`((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRs s  g$@g4@i iR7iiiRNgi@(ii(ii(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic70q s cC`s9d}tjd}|j||did d6dS( sneighborhood, type changecS`sUd}xDtddD]3}d}|dkr7d}n||||7}qW|dS(Niiigiy?i(R(RRRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR s  gN@R7iiRN(ii((ii(RERXRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic71~ s cC`s9d}tjd}|j||did d6dS( s+neighborhood, narrower range than specifiedcS`s6d}x%tddD]}|||7}qW|dS(Niiii(R(RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR sgN@R7iiRN(ii((ii(RERXRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic72 s cC`s9d}tjd}|j||did d6dS( sneighborhood, +ve rangecS`s6d}x%tddD]}|||7}qW|dS(Niii i(R(RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR sgN@R7ii RN(ii ((ii (RERXRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic73 s cC`s9d}tjd}|j||did d6dS( sneighborhood, -ve rangecS`s6d}x%tddD]}|||7}qW|dS(Niiii(R(RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR sgN@R7iiRN(ii((ii(RERXRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic73b s cC`s9d}tjd}|j||did d6dS( s!neighborhood, -ve->+ve range spancS`s6d}x%tddD]}|||7}qW|dS(Niii i(R(RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR sgN@R7ii RN(ii ((ii (RERXRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic74 s cC`s9d}tjd}|j||did d6dS( s!neighborhood, -ve->-ve range spancS`s6d}x%tddD]}|||7}qW|dS(Niiii(R(RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR sgN@R7iiRN(ii((ii(RERXRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic75 s cC`sKd}tjd jdd}|j||did dfd 6d S(sneighborhood, mixed range spancS`sjd}d}xStddD]B}||}x/tddD]}||||f|7}q<WqW|dS(Nig(@iii ii2(R(RRR"RRR`((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR s  g$@g4@i iR7iiiRNgi@(ii(ii(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic76 s c C`sid}tjd jdd}tjd jdd}|j|||did dfd 6d S(s neighborhood, two args cS`sdd}xStddD]B}x9tddD](}||||f|||f7}q,WqW|dS(Niiig"@(R(RRRRR`((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR s *g$@g4@i iR7iiRNgi@gi@(ii(ii(RERXRYRP(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic77 s c C`sid}tjd jdd}tjdjdd}|j|||diddfd 6d S(s. neighborhood, two args, -ve range, -ve range cS`sdd}xStddD]B}x9tddD](}||||f|||f7}q,WqW|dS(Niiiiig"@(R(RRRRR`((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR s *g.@g4@iiR7iiiiRNgr@gr@(ii(ii(RERXRYRP(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic78 s  c C`sid}tjd jdd}tjdjdd}|j|||diddfd 6d S(s. neighborhood, two args, -ve range, +ve range cS`sdd}xStddD]B}x9tddD](}||||f|||f7}q,WqW|dS(Niiiii g"@(R(RRRRR`((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR s *g.@g4@iiR7iiii RNgr@gr@(ii(ii (RERXRYRP(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic78b s c C`sd}tjdjdd}tjdjddd}|jdtdtd td t}|j|||d iddfd6d|dS(s% neighborhood, two incompatible args cS`sdd}xStddD]B}x9tddD](}||||f|||f7}q,WqW|dS(Niiig"@(R(RRRRR`((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR s *g$@g4@i iiRRRRR7iiRRNgi@gi@(ii(ii(RERXRYRRRRP(R*RRRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic79 s   c C`sTd}tjd jdd}d}|j|||did dfd 6d S(s neighborhood, type change cS`sZd}xItddD]8}x/tddD]}||||f|7}q,WqW|dS(Niiig"@(R(RRRRR`((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR s  g$@g4@i iy(@R7iiRNgi@(ii(ii(RERXRYRP(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic80 s c C`sd}tjdjdd}|dj}|jdtdtd td t}|j|||d iddfd 6d|dS(s1 neighborhood, dimensionally incompatible arrays cS`s^d}xMtddD]<}x3tddD]"}||||f||7}q,WqW|dS(Niiig"@(R(RRRRR`((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR s $g$@g4@i iiRRRRR7iRRNgi@(ii(ii( RERXRYtcopyRRRRRP(R*RRRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic81 s   c C`sad}tjd jdd}|j}|j|||diddfd 6d d 6d S(s% neighborhood, with standard_indexingcS`s^d}xMtddD]<}x3tddD]"}||||f|d7}q,WqW|dS(Niiiig"@(ii(R(RRRRR`((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR' s $g$@g4@i iR7iiRRRNgi@(ii(ii(RERXRYR0RP(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic82% s   c C`shd}tjdjdd}|j}|j|||diddfd 6d d 6d d 6dS(s. neighborhood, with standard_indexing and cvalcS`s^d}xMtddD]<}x3tddD]"}||||f|d7}q,WqW|dS(Niiiig"@(ii(R(RRRRR`((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR6 s $g$@g4@i iR7iiRRRg?RNgi@(ii(ii(RERXRYR0RP(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic834 s   cC`s8d}tjdjdd}|j||dS(s kernel calls njit cS`s|dt|dS(Nii(ii(ii(R!(R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRE sg$@g4@i iNgi@(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic84C s cC`s8d}tjdjdd}|j||dS(s! kernel calls njit(parallel=True)cS`s|dt|dS(Nii(ii(ii(R#(R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRM sg$@g4@i iNgi@(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic85K s cC`sQd}tjd jdd}|j||didd6dttgd S( s bad kwarg cS`s|dS(Ni(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRV sg$@g4@i iR7tbadRNgi@(RERXRYRPRR(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic86T s cC`s8d}tjdjdd}|j||dS(s reserved arg name in use cS`s|dS(Ni(ii((t __sentinel__((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR_ sg$@g4@i iNgi@(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic87] s c C`skd}tjdjdd}|jdtdtdtdt}|j||d d id |d S( s use of reserved word cS`s ||dS(Nii(ii((RRW((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRg sg(@iiRRRRg?R7RN(RERXRYRRRRP(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic88e s   cC`s8d}tjdjdd}|j||dS(s basic multiple returncS`s4|ddkrdS|d dkr,|d SdSdS( Niii g$@iig@(ii(ii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRx s g$@g4@i iNgi@(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic89v s c C`shd}tjdjdd}|j}|j|||diddfd 6d d 6d d 6dS(s@ neighborhood, with standard_indexing and cval, multiple returnscS`s|d}xMtddD]<}x3tddD]"}||||f|d7}q,WqW|d}|dkrt|dS|SdS( Niiiig"@gi@g?(ii(R(RRRRR`tres((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR s$  g$@g4@i iR7iiRRRg?RNgi@(ii(ii(RERXRYR0RP(R*RRR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic90 s   cC`s8d}tjdjdd}|j||dS(s? Issue #3454, const(int) == const(int) evaluating incorrectly. cS`s'd}ddkrd}n|d|S(Nii(ii((RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR s  g$@g4@i iNgi@(RERXRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic91 s cC`sQd}tjtjddjddjtj}|j||dS(s7 Issue #3497, bool return type evaluating incorrectly. cS`sH|d|dA|dA|dA|dA|d A|d A|d A|d AS( Niii(ii(ii(ii(ii(ii(ii(ii(ii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR siiiiN(RERRXRYRtbool_RP(R*RR\((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic92 s 4cC`s^d}tjtjddjddjtj}|j||ditd6dS( s7 Issue #3497, bool return type evaluating incorrectly. cS`sH|d|dA|dA|dA|dA|d A|d A|d A|d AS( Niii(ii(ii(ii(ii(ii(ii(ii(ii(ii((R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR siiiiR7RN(RERRXRYRR?RPR5(R*RR\((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic93 s 4cC`sTd}tjddtjjdd}|j||did d fd 6d S( s" Issue #3528. Support for slices. cS`s#tj|ddddfS(Nii(REtmedian(R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR siRiiR7iiRN(ii(ii(RERXtuint32RYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic94 s $snot yet supportedcC`sAd}tjddtjjdd}|j||dS(s Slice, calculate neighborhood. cS`s#tj|ddddfS(Niiii(RERB(R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR siRiiN(RERXRCRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic95 s $cC`sBd}tjddtj}|j||did d6dS( s 1D slice. cS`stj|dd!S(Nii(RERB(R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR siRR7iiRN(ii((ii(RERXRCRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic96 s cC`sAd}tjddtjjdd}|j||dS(s 2D slice and index. cS`stj|dddfS(Niii(RERB(R((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyR siRiiN(RERXRCRYRP(R*RR((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyt test_basic97 s $(pRQRRR)RPRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRtunittesttskipRRRRRRRRRRRRRRRRRRR R R R R RRRRRRRRRRRRRRRRR R!R#R$R%R&R'R(R)R*R+R,R-R.R/R1R2R3R4R5R7R9R:R;R=R>R@RARDRERFRG(((s8lib/python2.7/site-packages/numba/tests/test_stencils.pyRs                                                          t__main__(>t __future__RRRtsystnumpyRERR\RRRt contextlibRR0RR;RRHRRtnumba.compilerR R t numba.targetsR tnumba.targets.cpuR tsupportR t numba.errorsRRt version_infoR_tplatformt startswitht _windows_py27tmaxsizet_32bitt_reasont _unsupportedtskipIfRRRRRRRR R!R5R#tTestCaseR$RURRRRQtmain(((s8lib/python2.7/site-packages/numba/tests/test_stencils.pytsb        !7k '