ó \ K]c@`siddlmZmZmZddlZddlmZddlm Z m Z ddl m Z ddl mZddlmZmZej Zejd d eƒjd d ƒZejƒZejd dƒZejdd eƒjddƒddd…ddd…fZd„Zd„Zd„Zd„Ze gZdefd„ƒYZe dkreej!ƒndS(i(tabsolute_importtprint_functiontdivisionN(tunittest_support(tfloat32tjit(t Vectorize(t TypingErrori(ttagtTestCaseiPtdtypeii tordertFiicC`s||S(N((tatb((s>lib/python2.7/site-packages/numba/tests/npyufunc/test_ufunc.pytaddscC`s||||S(N((R Rtctd((s>lib/python2.7/site-packages/numba/tests/npyufunc/test_ufunc.pytadd_multiple_argsscC`s>d}x1t|jdƒD]}|||||7}qW|S(Ngi(trangetshape(R Rtresultti((s>lib/python2.7/site-packages/numba/tests/npyufunc/test_ufunc.pyt gufunc_addscC`s-x&t|jƒD]}|j|ƒ}qW|S(N(Rtndimtreduce(tufunctargR((s>lib/python2.7/site-packages/numba/tests/npyufunc/test_ufunc.pyt ufunc_reduce#st TestUFuncscB`sbeZd„Zd„Zedƒd„ƒZedƒd„ƒZedƒd„ƒZd„ZRS(cG`së|t|Œ}|jtttƒƒ|jƒ}||jf}|j|||ƒ||d|ƒ|jt||ƒtj|ƒd|ƒ|j|j|ƒtjj|ƒd|ƒ|j|j ||ƒtjj ||ƒd|ƒdS(sTest ufunc attributestmsgN( RRt build_ufuncRtassertPreciseEqualRtnptsumt accumulatetouter(tselftclsR Rtargst vectorizerRtinfo((s>lib/python2.7/site-packages/numba/tests/npyufunc/test_ufunc.pyt_test_ufunc_attributes4s #(+c C`sx|tƒ}|jtttttƒƒ|jƒ}||jf}|j|||||ƒ||||d|ƒdS(sTest multiple argsRN(RRRRRR ( R%R&R RRRR(RR)((s>lib/python2.7/site-packages/numba/tests/npyufunc/test_ufunc.pyt_test_broadcasting@s   t importantc C`s°x)tD]!}|j|tdtdƒqWx!tD]}|j|ttƒq3WxYtD]Q}|j|tdd…tjdd…fttjdd…dd…fƒqWWdS(Ni(t vectorizersR*R RR!tnewaxis(R%tv((s>lib/python2.7/site-packages/numba/tests/npyufunc/test_ufunc.pyttest_ufunc_attributesIs   (c C`sx7tD]/}|j|tdtdtdtdƒqWx'tD]}|j|ttttƒqAWx—tD]}|j|tdd…tjdd…fttjdd…dd…ftdd…tjdd…fttjdd…dd…fƒqkWdS(Ni(R-R+R RRRR!R.(R%R/((s>lib/python2.7/site-packages/numba/tests/npyufunc/test_ufunc.pyttest_broadcastingSs -  GcC`sŽx‡tD]}|tƒ}|jtttƒƒ|jƒ}ttjdd…tjtjdd…f}|j|t|ƒt|ƒqWdS(N( R-RRRRR!R.R R (R%R/R(Rtbroadcasting_b((s>lib/python2.7/site-packages/numba/tests/npyufunc/test_ufunc.pyttest_implicit_broadcasting]s   .c C`sÍtjdƒ}t|j_g}d}|jtdtƒt|fƒ|jtdtƒt dfƒxa|D]Y\}}}d„}|j |ƒ}||ƒ|ƒWdQX|j |t |j ƒƒqlWdS(Ni s<ufunc 'sin' called with an explicit output that is read-onlytnopythonsoutput array is read-onlycS`s,tj|j|jƒ}tj||ƒdS(N(R!tonesRR tsin(txR ((s>lib/python2.7/site-packages/numba/tests/npyufunc/test_ufunc.pyttestss(R!R5tFalsetflagst writeabletappendRtTrueRt ValueErrort assertRaisestassertIntstrt exception( R%tztteststexpecttdectexcRR8traises((s>lib/python2.7/site-packages/numba/tests/npyufunc/test_ufunc.pyt)test_ufunc_exception_on_write_to_readonlyhs   ( t__name__t __module__R*R+RR0R1R3RI(((s>lib/python2.7/site-packages/numba/tests/npyufunc/test_ufunc.pyR2s     t__main__i@("t __future__RRRtnumpyR!tnumbaRtunittestRRtnumba.npyufuncRt numba.errorsRtsupportRR R tarangetreshapeR tcopyRRRRRRRR-RRJtmain(((s>lib/python2.7/site-packages/numba/tests/npyufunc/test_ufunc.pyts&  ! =     K