\ K]c@`sddlmZmZmZddlZddlZddlmZm Z ddlm Z ddlm Z m Z mZmZmZmZddlmZmZmZddlmZdd lmZmZd efd YZd Zd ZdZdZ dZ!e eZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)defdYZ*deefdYZ+deefdYZ,dee jfdYZ-d ee jfd!YZ.d"ee jfd#YZ/e0d$kr e j1ndS(%i(tprint_functiontdivisiontabsolute_importN(tnjitt vectorize(tunittest_support(tcompilerttypingttypeoftirtutilsttypes(tPipelinet_PipelineManagertFlags(tcpui(tMemoryLeakMixintTestCaset NamespacecB`seZdZRS(cC`s*||kr||Stt|j|S(N(tsuperRt __getattr__(tstk((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyRs(t__name__t __module__R(((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyRscC`s |||S(N((tatxty((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pytaxyscC`s |||S(N((RRR((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pytax2scC`s%| |dd||dd|S(Ng@g@g?((tAstBstCs((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pytpos_rootscC`s7d|}d||}|d|}| |d|S(Ng@g?((RRR t_2Ast_4AsCst _Bs2_4AsCs((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pytneg_root_common_subexprs cC`s;d|}d||}|d|d}| |d|S(Ng@yg?((RRR R"R#R$((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pytneg_root_complex_subexpr"s cC`s&tjt|tj|ddS(Ng?(tnptcostvaxytsin(ta0ta1((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyt call_stuff*scC`s|dd||dkS(Niii((RRR ((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pytare_roots_imaginary-scC`s |||S(N((RRR ((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pytdiv_add0scC`s|dS(Ni((R((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pytcube3scC`s#tj||tj|||S(N(R'R(tadd(Rtbtout((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pytexplicit_output6scC`s*||}|||}|||}|S(N((RR2tctdtu((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pytvariable_name_reuse:s cC`s|jd}tj||fdtj}xit|D][}xRt||D]A}tj||||dd|||f<|||f>sis9Common subexpressions detected in array expressions ({0})N(RR^RhRiRaRjRkRlRmRRtzipt intersectiontfailR( R[R\RRzR{tarray_expr_instrst array_setst expr_set_0t expr_set_1t intersections((R[s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyttest_common_subexpressions+s*, $ cC`s |jtS(N(RR&(R[((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyttest_complex_subexpressionFscC`s:tjjd}tjjd}g||fD]}t|^q1}tjj}tj|d|jd|j }|j t }|j }tj |d|jd|j } | j t } | j } t ||} |||} | ||}tjj| | tjj| ||j|jj| jjdS(sn Verify that ufunc and DUFunc calls are being properly included in array expressions. i RQRRN(R'RRR)t _dispatchert targetdescrRERVRQRRRYR-RZRSRfRRRhRi(R[RnRRqR]t vaxy_descrR^R_tnb_call_stuff_0RaRbtnb_call_stuff_1RwRyRx((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyttest_ufunc_and_dufunc_callsIs.%       cC`s5|jt}|j|jjj|jjjdS(sQ Verify that comparison operators are supported by the rewriter. N(RR.RR^RhRiRa(R[R((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyt test_cmp_opjscC`s5|jt}|j|jjj|jjjdS(sQ Check that ufunc calls with explicit outputs are not rewritten. N(RR4RR^RhRiRa(R[R((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyttest_explicit_outputrsN(RRRdR}RRHRR!RR0RRRtFalseRRRRR%RRRRR(((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyRXes"            ! tTestRewriteIssuescB`s>eZdZdZdZdZdZdZRS(c`sddlm}ddl}|dtd|dtfd}|jd}||\}}|j|||j||dS(Ni(tjittnopythoncS`s|S(N((tarr((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pytfoosc`s"|}|}||fS(N((RR5R6(R(s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pytbars  i (tnumbaRtnumpyRFRtassertIs(R[RR'RRtout_ctout_d((Rs;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyttest_issue_1184}s cC`smd}tjjd|dj|df}t|}tt|}tjj||tj dS(Nidtsizei( R'RtuniformtreshapeRDRRfRtgctcollect(R[tnRRwRx((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyttest_issue_1264s ( cC`s{ddlm}|d}tjjd}tjjd}|j||}|||}tjj||dS(s*Test array expression with duplicated termi(RcS`stj|}|||S(N(R'R*(RR2((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyRsi N(RRR'RRtpy_funcRftassert_allclose(R[RRRR2texpecttgot((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyttest_issue_1372scC`sbtd}d}tjddtj}|j||}|||}|j||dS(sP Typing of unary array expression (np.negate) can be incorrect. cS`s ||| S(N((RR2((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyRsg?i R9N(RR'Rtint32RR(R[RR2RRR((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyttest_unary_arrayexprs cC`sztd}tjttttg}tjttttg}|j||}|||}|j||dS(sd Typing of bitwise boolean array expression can be incorrect (issue #1813). cS`s ||@S(N((RR2((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyRsN(RR'tarrayRFRRR(R[RRR2RR((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyttest_bitwise_arrayexprs cC`s}tt}tjddd}|||||tj}|j||j}|jd||jd|dS(so Type annotation of array expressions with disambiguated variable names (issue #1466). iii s # u.1 = s # u.2 = N( RR8R'ReR tStringIOt inspect_typestgetvaluetassertIn(R[tcfuncRtbuftres((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyttest_annotationss    (RRRRRRRR(((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyR{s     t TestSemanticscB`seZdZRS(cC`st}t|}tjddtdg}tjdddg}tj|}||||}||||}tjj||dS(Ngg?tinf(R/RR'R<tfloatt ones_likeRfRg(R[tpyfuncRRR2R5RR((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyttest_division_by_zeros (RRR(((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyRst TestOptionalscB`s eZdZdZdZRS(se Tests the arrival and correct lowering of Optional types at a arrayexpr derived ufunc, see #3972c`stdtfd}tjddf}||}|j|}tjj||j}|dd}|jt|t j |jt|j t j dS(NcS`s||S(N((RR((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pytarr_exprsc`s(|dkrd}n|}||S(Ni(RH(RRtz(R(s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pytdo_calls  ig333333ii( RR'RRRfRt signaturest assertTrueRR tOptionalttypetFloat(R[RRLRRwRtoty((Rs;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyttest_optional_scalar_types  c`stdtfd}tjdtjdf}||}|j|}tjj||j}|dd}|jt|t j |jt|j t j |jt|j j t jdS(NcS`s||S(N((RR((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyR sc`s,|ddkrd}n|}||S(Ni(RH(RRR(R(s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyRs ig@ii(RR'RRRfRRRRR RRtArrayR9R(R[RRLRRwRR((Rs;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyttest_optional_array_type s  (RRt__doc__RR(((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyRs tTestOptionalsExceptionscB`seZdZdZRS(c`s|jtdtfd}tjddf}|jt}||WdQX|jdt|jj }|dd}|j t |t j |j t |jt jdS(NcS`s||S(N((RR((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyR-sc`s(|dkrd}n|}||S(Ni(RH(RRR(R(s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyR1s  ig?sexpected float64, got Noneii(tdisable_leak_checkRR'Rt assertRaisest TypeErrorRtstrt exceptionRRRR RRR(R[RRLtraisesRR((Rs;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyt+test_optional_scalar_type_exception_on_none)s  c`s|jtdtfd}tjdtjddf}|jt}||WdQXt|j}|jd||jd|j }|dd }|j t |t j |j t |jt j|j t |jjt jdS( NcS`s||S(N((RR((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyRKsc`s,|ddkrd}n|}||S(Ni(RH(RRR(R(s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyROs ig?g@sexpected array(float64,sgot Noneii(RRR'RRRRRRRRRR RRRR9R(R[RRLRtexcstrRR((Rs;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyt*test_optional_array_type_exception_on_noneGs ! (RRR R (((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyR&s t__main__(2t __future__RRRRRR'RRRRtunittestRRRR R R tnumba.compilerR R Rt numba.targetsRtsupportRRtdictRRRR!R%R&R)R-R.R/R0R4R8RDRERXRRRRRtmain(((s;lib/python2.7/site-packages/numba/tests/test_array_exprs.pyts>  .             ]=C