\ K]c@`sddlmZmZmZddlZddlZddlZddlZddlZddl Z ddl m Z ddl Z ddl ZddlmZddlmZmZmZmZmZmZddlmZddlmZdd lmZmZmZm Z m!Z!dd l"m#Z#dd l$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5dd l6m7Z7m8Z8m9Z9m:Z:m;Z;ej<d fkZ=ddl>m?Z?m@Z@y@ddlAZAe eAjBdkreCZDnddlEjFjGZDWneHk r;eCZDnXdeIfdYZJdejKfdYZLeLdZMeJZNe%jOeJdZPe(eLejQdZRdZSe-eLe,jTe/eLdZUdeIfdYZVdejKfdYZWeWdZXeVZYe%jOeVdZPdZZe-eWe,jTe/eWd ZUeCd!Z[e&e[d"Z\e'e[e'e[ej]d#Z^e'e[ej_d$Z`e9d%e7fd&YZae'd'eLd(Zbd)Zcd*Zdd+Zee)ecd,Zfe)ecd-Zge)ehd.Zie)ejjd/Zke)ejld0Zme)ejnd1Zoe)ejpd2Zqd3Zrd4Zse)ejtd5Zud6Zvd7Zwd8Zxd9Zyd:Zze+eLd;d<Z{d=Z|d>Z}d?Z~d@ZdAZdBZdCZdDZdEZdFZdGZdHZdIZe)edJZdKZdLZdMZe:edNe;fdOYZdPZe)ejdQZdRefdSYZdTefdUYZdVefdWYZdXZdYefdZYZd[Zd\efd]YZd^ejfd_YZd`ejfdaYZedbkrejndS(ci(tprint_functiontdivisiontabsolute_importN(t LooseVersion(tunittest_support(tnjittjitttypesterrorsttypingtcompiler(t cpu_target(tcompile_isolatedi(tTestCasetcaptured_stdoutttagttemp_directorytoverride_config(t LoweringError(t typeof_implt type_callablet lower_builtint lower_casttoverloadtoverload_attributetoverload_methodtmodelstregister_modeltboxtunboxt NativeValuetmake_attribute_wrappert intrinsict _Intrinsictregister_jitabletget_cython_function_address(tConcreteTemplatet signaturetinfert infer_globaltAbstractTemplatei(tIndextSeriess0.19tMyDummycB`seZRS((t__name__t __module__(((s9lib/python2.7/site-packages/numba/tests/test_extending.pyR+6st MyDummyTypecB`seZdZRS(cC`s-t|tjr)ddlm}|jSdS(Ni(t Conversion(t isinstanceRtNumbertnumba.typeconvR/tsafe(tselftcontextttotyR/((s9lib/python2.7/site-packages/numba/tests/test_extending.pytcan_convert_to:s(R,R-R7(((s9lib/python2.7/site-packages/numba/tests/test_extending.pyR.9stmydummycC`stS(N(t mydummy_type(tvaltc((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttypeof_mydummyBscC`s|j|dS(s2 Implicit conversion from MyDummy to int. i*(t get_constant(R5tbuildertfromtyR6R:((s9lib/python2.7/site-packages/numba/tests/test_extending.pytmydummy_to_numberFscC`stS(N(R8(((s9lib/python2.7/site-packages/numba/tests/test_extending.pyt get_dummyMscC`st|jjS(N(RR5tget_dummy_value(ttyptobjR;((s9lib/python2.7/site-packages/numba/tests/test_extending.pyt unbox_indexRstMyDummy2cB`seZRS((R,R-(((s9lib/python2.7/site-packages/numba/tests/test_extending.pyRFZst MyDummyType2cB`seZRS((R,R-(((s9lib/python2.7/site-packages/numba/tests/test_extending.pyRG]st mydummy_2cC`stS(N(tmydummy_type_2(R:R;((s9lib/python2.7/site-packages/numba/tests/test_extending.pyR<cscC`stS(N(RH(((s9lib/python2.7/site-packages/numba/tests/test_extending.pyt get_dummy_2hscC`st|jjS(N(RR5RB(RCRDR;((s9lib/python2.7/site-packages/numba/tests/test_extending.pyREmscC`s tdS(N(tNotImplementedError(tx((s9lib/python2.7/site-packages/numba/tests/test_extending.pytfunc1vscC`sdd}|S(NcS`s6|dtjfkrtjSt|tjr2|SdS(N(tNoneRtnonetint32R0tFloat(RL((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttyper{s(RN(R5RR((s9lib/python2.7/site-packages/numba/tests/test_extending.pyt type_func1ys cC`s|j|jdS(Ni*(R=t return_type(R5R>tsigtargs((s9lib/python2.7/site-packages/numba/tests/test_extending.pyt func1_nullaryscC`sd}|j||||S(NcS`stjd|S(Ni(tmathtsqrt(RL((s9lib/python2.7/site-packages/numba/tests/test_extending.pyt func1_impls(tcompile_internal(R5R>RURVRZ((s9lib/python2.7/site-packages/numba/tests/test_extending.pyt func1_unarys t PrintDummycB`s#eZdZeejegZRS(t print_item(R,R-tkeyR%RROR9tcases(((s9lib/python2.7/site-packages/numba/tests/test_extending.pyR]sR^cC`sT|\}|j|}|j|jd}|j||j||jS(Nshello!(tget_python_apit unserializetserialize_objectt print_objecttdecrefRB(R5R>RURVRLtpyapitstrobj((s9lib/python2.7/site-packages/numba/tests/test_extending.pyt print_dummys    cC`s tdS(N(RK(tcondRLty((s9lib/python2.7/site-packages/numba/tests/test_extending.pytwherescC`stj|||S(s8 Wrap np.where() to allow for keyword arguments (tnpRk(RiRLRj((s9lib/python2.7/site-packages/numba/tests/test_extending.pytnp_wherescC`st|d|d|S(NRjRL(Rk(RiRLRj((s9lib/python2.7/site-packages/numba/tests/test_extending.pyt call_wherescC`srt|tjrn|j|jkr6tjdntd|||fDrad}n d}|SdS(s' Implement where() for arrays. s"x and y should have the same dtypecs`s|]}|jdkVqdS(tCN(tlayout(t.0tty((s9lib/python2.7/site-packages/numba/tests/test_extending.pys sc S`s|j}|j|ks'|j|kr6tdntj|}|j}|j}|j}|j}x9t|jD](} || r|| n|| || return a 0-dim array ((RlRu(RiRLRjtscaltarr((s9lib/python2.7/site-packages/numba/tests/test_extending.pyRs N(R0RRRR(RiRLRjR((s9lib/python2.7/site-packages/numba/tests/test_extending.pytoverload_where_scalarss   cC`s t|trd}|SdS(NcS`sdS(Ni ((targ((s9lib/python2.7/site-packages/numba/tests/test_extending.pytlen_impls(R0R.(RR((s9lib/python2.7/site-packages/numba/tests/test_extending.pytoverload_len_dummys cC`s;t|ttfr7t|ttfr7d}|SdS(NcS`sdS(Ni*((targ1targ2((s9lib/python2.7/site-packages/numba/tests/test_extending.pytdummy_add_impls(R0R.RG(RRR((s9lib/python2.7/site-packages/numba/tests/test_extending.pytoverload_add_dummys* cC`s2t|tr.t|tjr.d}|SdS(NcS`std||dS(Ntdel(tprint(RDR((s9lib/python2.7/site-packages/numba/tests/test_extending.pytdummy_delitem_impl s(R0R.RtInteger(RDRR((s9lib/python2.7/site-packages/numba/tests/test_extending.pytoverload_dummy_delitem s! cC`s2t|tr.t|tjr.d}|SdS(NcS`s|dS(Ni{((RDR((s9lib/python2.7/site-packages/numba/tests/test_extending.pytdummy_getitem_impls(R0R.RR(RDRR((s9lib/python2.7/site-packages/numba/tests/test_extending.pytoverload_dummy_getitems! cC`sGtt|tt|tjt|tjgrCd}|SdS(NcS`st||dS(N(R(RDRR:((s9lib/python2.7/site-packages/numba/tests/test_extending.pytdummy_setitem_impl s(RR0R.RR(RDRR:R((s9lib/python2.7/site-packages/numba/tests/test_extending.pytoverload_dummy_setitems   cC`stj||S(N(toperatortadd(RR((s9lib/python2.7/site-packages/numba/tests/test_extending.pytcall_add_operator%scC`s||S(N((RR((s9lib/python2.7/site-packages/numba/tests/test_extending.pytcall_add_binop)scC`s;t|ttfr7t|ttfr7d}|SdS(NcS`sdS(Ni*((RR((s9lib/python2.7/site-packages/numba/tests/test_extending.pytdummy_iadd_impl0s(R0R.RG(RRR((s9lib/python2.7/site-packages/numba/tests/test_extending.pytoverload_iadd_dummy-s* cC`stj||S(N(RR(RR((s9lib/python2.7/site-packages/numba/tests/test_extending.pytcall_iadd_operator6scC`s||7}|S(N((RR((s9lib/python2.7/site-packages/numba/tests/test_extending.pytcall_iadd_binop:s cC`s ||=dS(N((RDR((s9lib/python2.7/site-packages/numba/tests/test_extending.pyt call_delitem@scC`s||S(N((RDR((s9lib/python2.7/site-packages/numba/tests/test_extending.pyt call_getitemDscC`s|||t(R(((s9lib/python2.7/site-packages/numba/tests/test_extending.pytmk_func_test_implscC`s t|trd}|SdS(NcS`sdS(NIﾭ((RD((s9lib/python2.7/site-packages/numba/tests/test_extending.pyRs(R0R.(RDR((s9lib/python2.7/site-packages/numba/tests/test_extending.pytoverload_np_exps tTestLowLevelExtendingcB`s2eZdZdZdZdZdZRS(s4 Test the low-level two-tier extension API. cC`syt}tdt|}|j|dt}tdt|}|j|dd|j|dddS(Ntnopythoni*g2@g@(RRtTruetassertPreciseEqualRRN(R4tpyfunctcfunc((s9lib/python2.7/site-packages/numba/tests/test_extending.pyt test_func1scC`sct}t|d}|j|jdt}t|tjf}|j|jdddS(Ni*g2@g@((RR Rt entry_pointRRtfloat64(R4Rtcr((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_func1_isolateds cC`s5t}t|dtj}|j|jddS(NgE@((RAR RRRR(R4RR((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_cast_mydummyscC`sjtjt}tj}|jtj||dd\}}}|jt d|j DdS(sFmake sure make_function is passed to typer class as a literal cs`s!|]}t|tjVqdS(N(R0RR(RqR((s9lib/python2.7/site-packages/numba/tests/test_extending.pys sN(( R t run_frontendRR ttyping_contexttrefreshttype_inference_stageRNt assertTruetanytvalues(R4ttest_irt typingctxttypemapt_((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_mk_func_literals  (R,R-t__doc__RRRR(((s9lib/python2.7/site-packages/numba/tests/test_extending.pyRs   tTestPandasLikecB`seZdZdZdZdZdZdZeddZ eddZ d Z edd Z edd Z RS( se Test implementing a pandas-like Index object. Also stresses most of the high-level API. cC`sDttjd}tdtt}|j||ddS(NiR(R)RlRRRRR(R4R~R((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_index_lenscC`sttjdddg}tdtt}|j||dd||tdd}|j |t|j t |ddgdS(Ni*iiRi( R)RlRPRRRRtsliceRNtassertIsInstancet assertEqualtlist(R4R~Rtii((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_index_getitems cC`s{ttjdddg}tdtt}||}|j|t|j|jtj tj |jdS(s7 Check Numpy ufunc on an Index object. i*iiRN( R)RlRPRRRRRRRR(R4R~RR((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_index_ufuncs  cC`sVttjdddg}tdtt}||}|j||jdS(Ni*iiR(R)RlRPRRRtassertIsR(R4R~RR((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_index_get_datas cC`stdtt}xrdddgtfdddgtfgtfgD]=\}}ttj|}||}|j||qIWdS(NRii*i(RRRtFalseR)RlRPR(R4RRtexpectedR~tgot((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_index_is_monotonics t importantcC`snttjdddg}ttjdddg|}tdtt}|j||ddS(Niiig?g@g@R( R)RlRPR*RRRRR(R4R~tsR((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_series_lens!cC`sttjdddg}ttjdddg|}tdtt}||}|j|t|j |j |j dS(Niiig?g@g@R( R)RlRPR*RRRRRRR(R4R~RRR((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_series_get_index s ! cC`sttjdddg}ttjdddg|}tdtt}||}|j|t|j|j t|j |j j |j |j |j tjtj|j dS( s8 Check Numpy ufunc on an Series object. i*iiiiiRN(R)RlRPR*tint64RRRRRRRRt_valuesRR(R4R~RRtss((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_series_ufuncs! cC`sttjdddg}tjdddg}tdtt}|||}|j|t|j|j t|j |j j |j |j |j |dS(Ni*iig?g@g@R( R)RlRPRRRRRR*RRRR(R4R~tdRR((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_series_constructor scC`sttjdddg}ttjdddg|}tdtt}||dd }|j|t|j|j t|j |j j |j |j |j tjdd dgdS( Ni*iig?g@g@Rg?g@(R)RlRPR*RRRRRRRRRR(R4R~RRR((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_series_clip+s!(R,R-RRRRRRRRRRRR(((s9lib/python2.7/site-packages/numba/tests/test_extending.pyRs      tTestHighLevelExtendingcB`seZdZeddZeddZdZdZdZdZ dZ d Z d Z d Z d Zd ZejeddZdZdZdZdZRS(s+ Test the high-level combined API. Rc `st}tdt|fd}|dddtdd|tdd|tjtttgtjdd dgtjd d d gjtj 2}tjgtjgtj gWd QXj d t |j d S(s> Test implementing a function with @overload. Rc`s)t||}||}jdS(N(RmR(RVtkwargsRR(RR4(s9lib/python2.7/site-packages/numba/tests/test_extending.pytcheckDsRLiRiRjiiiiiNs"x and y should have the same dtype(RnRRRltbool_RRPt assertRaisesRRRtassertIntstrt exception(R4RRtraises((RR4s9lib/python2.7/site-packages/numba/tests/test_extending.pyt test_where<s-1 cC`sNtdtt}|j|td|j|ddgddS(sN Test re-implementing len() for a custom type with @overload. Ri iiiN(RRRRR+(R4R((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_lenTscC`sOtdtt}t+|t|jtjjdWdQXdS(sP Test re-implementing print() for a custom type with @overload. Rshello! N( RRRRR+Rtsyststdouttgetvalue(R4R((s9lib/python2.7/site-packages/numba/tests/test_extending.pyt test_print]s  cC`svt}tdt|}|j|ddd|j|ttd|j|ttddS(sW Test re-implementing operator.add() for a custom type with @overload. Riiii*iTN(RRRRRFR+(R4RR((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_add_operatorfs cC`svt}tdt|}|j|ddd|j|ttd|j|ttddS(sY Test re-implementing '+' for a custom type via @overload(operator.add). Riiii*iTN(RRRRRFR+(R4RR((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_add_binopss cC`svt}tdt|}|j|ddd|j|ttd|j|ttddS(sW Test re-implementing operator.add() for a custom type with @overload. Riiii*iTN(RRRRRFR+(R4RR((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_iadd_operators cC`svt}tdt|}|j|ddd|j|ttd|j|ttddS(sY Test re-implementing '+' for a custom type via @overload(operator.add). Riiii*iTN(RRRRRFR+(R4RR((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_iadd_binops c C`st}tdt|}t}d}t4}y||dWntk rb}|}nXWdQX|dk r~|n|j|jddS(NRiAsdel hello! 321 ( RRRR+RNRt ExceptionRR(R4RRRDtetouttexc((s9lib/python2.7/site-packages/numba/tests/test_extending.pyt test_delitems    cC`s;t}tdt|}|j|tdddS(NRiAi{i(RRRRR+(R4RR((s9lib/python2.7/site-packages/numba/tests/test_extending.pyt test_getitemsc C`st}tdt|}t}d}t7}y||ddWntk re}|}nXWdQX|dk r|n|j|jddS(NRiAi{s321 123 ( RRRR+RNRRRR(R4RRRDRRR((s9lib/python2.7/site-packages/numba/tests/test_extending.pyt test_setitems    c C`stdtt}d}||}t|}tjj||tdtt}|jt}|WdQXt |j }d}|j ||dS(sW Test overloading whose return value cannot be represented in CPython. Ri Nscannot convert native Module( RRRRlttestingt assert_equalRR t TypeErrorR R R ( R4tok_cfunctnRtexpectt bad_cfuncR terrmsgt expectmsg((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_no_cpython_wrappers   cC`sd d}d}dd}|jtj }||ddddWd QXt|j}|j|||jd |tr|jd ||jd |nd d }|jtj }||ddddWd QXt|j}|j|||jd|trB|jd||jd |nd d}|jtj }||ddddWd QXt|j}|j|||jd||jd|ktr|jd||jd|ntr(ddl m }m } |jtj }||ddddWd QXt|j}|j|||jd||jd|k|jd||jtj }|| ddddWd QXt|j}|j|||jd||jd|k|jd||jd|nd d} |jtj }|| ddddWd QXt|j}|j|||jd||jd|ktr|jd||jd|nd d} |jtj }|| ddddWd QXt|j}|j|||jd||jd|ktrn|jd|nd d d} |jtj }|| ddddWd QXt|j}|j|||jd|tr|jd|nd} |jtj }|| ddddWd QXt|j}|j|||jd|tr~|jd |nd S(s} Tests that an overload which has a differing typing and implementing signature raises an exception. c`sLddtdfd}tdtfd}|S(NcS`sdS(N((RtbR;tkw((s9lib/python2.7/site-packages/numba/tests/test_extending.pyt myoverloadsc`sS(N((RR)R;R*(timpl(s9lib/python2.7/site-packages/numba/tests/test_extending.pyt_myoverload_implsRc`s|||d|dS(NR*((RR)R;R(R+(s9lib/python2.7/site-packages/numba/tests/test_extending.pytfoos(RNRRR(R,R-R.((R,R+s9lib/python2.7/site-packages/numba/tests/test_extending.pytgen_ols   s-Typing and implementation arguments differ ini cS`s|dkrdSdSdS(Ni ii((RR)R;R*((s9lib/python2.7/site-packages/numba/tests/test_extending.pytimpl1s iiiiNskeyword argument default valuessscS`s|dkrdSdSdS(Ni ii((RR)R;tkwarg((s9lib/python2.7/site-packages/numba/tests/test_extending.pytimpl2s skeyword argument namesscS`stdkrdSdSdS(Ni ii(R(tzR)R;R*((s9lib/python2.7/site-packages/numba/tests/test_extending.pytimpl3 s sargument namestkeywordss(timpl4timpl5sFirst difference: 'z'cS`s|dkrdSdSdS(Ni ii((RR)R;RRR*((s9lib/python2.7/site-packages/numba/tests/test_extending.pytimpl6.s sscS`s|dkrdSdSdS(Ni ii((RR)R*((s9lib/python2.7/site-packages/numba/tests/test_extending.pytimpl7?s scS`s|dkrdSdSdS(Ni ii((RR)R;R*t extra_kwarg((s9lib/python2.7/site-packages/numba/tests/test_extending.pytimpl8Os scS`s|dkrdSdSdS(Ni ii((RR)R;((s9lib/python2.7/site-packages/numba/tests/test_extending.pytimpl9^s ( RNR RRR R R t_IS_PY3t assertFalsetoverload_usecasesR6R7(R4R/tsentinelR0RtmsgR2R4R6R7R8R9R;R<((s9lib/python2.7/site-packages/numba/tests/test_extending.pyt/test_typing_vs_impl_signature_mismatch_handlings       sPython 3+ only syntaxc`sddddlm}t|tdtfd}|jtj}|ddWdQXt |j }|j d||j d |dS( s Tests that an overload which has a differing typing and implementing signature raises an exception and uses VAR_POSITIONAL (*args) in typing cS`sdS(N((RR*((s9lib/python2.7/site-packages/numba/tests/test_extending.pyR+rsi(tvar_positional_implRc`s||dddS(Ni R*i ((RR)(R+(s9lib/python2.7/site-packages/numba/tests/test_extending.pyR.ysiNs)VAR_POSITIONAL (e.g. *args) argument kinds-offending argument name is '*star_args_token'( RNR?RCRRRR RRR R R (R4RCR.RRA((R+s9lib/python2.7/site-packages/numba/tests/test_extending.pyt>test_typing_vs_impl_signature_mismatch_handling_var_positionalls cC`std}d}||tdd|jtj}||ddWdQXt|j}|jd||jd|dd }|jtj}||ddWdQXt|j}|jd||jd|dS( sJ Tests that an overload which uses **kwargs (VAR_KEYWORD) c`sDddtd||tdtfd}|S(NcS`sdS(N((RR*((s9lib/python2.7/site-packages/numba/tests/test_extending.pyR+ststrictRc`s|ddS(NR*i ((RR)(R+(s9lib/python2.7/site-packages/numba/tests/test_extending.pyR.s(RNRRR(R,RER.((R+s9lib/python2.7/site-packages/numba/tests/test_extending.pyR/s c[`sdd}|S(Ni cS`s|S(N((RR*((s9lib/python2.7/site-packages/numba/tests/test_extending.pyR,s((RRR,((s9lib/python2.7/site-packages/numba/tests/test_extending.pytol1s iiNs1use of VAR_KEYWORD (e.g. **kwargs) is unsupporteds"offending argument name is '**kws'icS`s d}|S(Nc[`s|S(N((RR((s9lib/python2.7/site-packages/numba/tests/test_extending.pyR,s((RR*R,((s9lib/python2.7/site-packages/numba/tests/test_extending.pytol2s (RRR RRR R R (R4R/RFRRARG((s9lib/python2.7/site-packages/numba/tests/test_extending.pyt;test_typing_vs_impl_signature_mismatch_handling_var_keywords   cC`sVttjddd}td}tjd}|j||ddS( NR.i cS`sdd}|S(Ni cS`s|S(N((Rta_kwarg((s9lib/python2.7/site-packages/numba/tests/test_extending.pyR,s((RRIR,((s9lib/python2.7/site-packages/numba/tests/test_extending.pytfooimpls cS`s(|j|jd|jddfS(NiRIi(R.(tA((s9lib/python2.7/site-packages/numba/tests/test_extending.pytbarsiii(i ii(RRRRRlRR(R4RJRLtZ((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_overload_method_kwargss cC`s]ttjdd}td}tjd}|||j||ddS(NtlitfoocS`s8t|tjr4t|tjs4d}|SndS(NcS`s|S(N((RR:((s9lib/python2.7/site-packages/numba/tests/test_extending.pyR,s(R0RRtLiteral(RR:R,((s9lib/python2.7/site-packages/numba/tests/test_extending.pyROs cS`s |jdS(Ni(RO(RK((s9lib/python2.7/site-packages/numba/tests/test_extending.pyRLsii(RRRRRltzerosR(R4RORLRK((s9lib/python2.7/site-packages/numba/tests/test_extending.pyt#test_overload_method_literal_unpacks   cC`s&td}|j|ddS(NcS`s tjtS(N(RltexpR8(((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttestsIﾭ(RR(R4RT((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_overload_ufuncs(R,R-RRRRRRRRRRRRR(RBtunittestt skipUnlessR=RDRHRNRRRU(((s9lib/python2.7/site-packages/numba/tests/test_extending.pyR7s$      +  cC`sb|j|jd}||kr/tdn|j|jd}||kr^tdndS(Niscache not used(t _cache_hitst signaturesRt _cache_misses(Rt expect_hitt expect_missesthittmiss((s9lib/python2.7/site-packages/numba/tests/test_extending.pyt_assert_cache_statss   tTestOverloadMethodCachingcB`s eZeZdZdZRS(cC`s<t|jj|_td|j|jWdQXdS(Nt CACHE_DIR(Rt __class__R,t _cache_dirRtrun_caching_overload_method(R4((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_caching_overload_methodsc C`s]tdtdtt}|j|tdt|dd|jtf}g|jD]'}|j drcd|krc|^qc}|j t |dyt j d}Wntk rt }nX|j}|jd td ||jf}|j|jt|j|j |jd|jd d}|j |ddS( NRtcachei iitdeclareRtspawnttargetRVttimeout(RRRRR+R_t inspect_llvmR9t splitlinest startswithRRtmultiprocessingt get_contexttAttributeErrortQueuetProcessRdRctstarttputtjointexitcodetget( R4RtllvmirtlntdeclstctxtqtpRy((s9lib/python2.7/site-packages/numba/tests/test_extending.pyRds('     (R,R-Rt_numba_parallel_test_ReRd(((s9lib/python2.7/site-packages/numba/tests/test_extending.pyR`s c C`sjtd|U|j}tdtdtt}||}|j|t|ddWdQXdS(sH Used by TestOverloadMethodCaching.test_caching_overload_method RaRRfiiN(RRwRRRRtR_(R|t cache_dirRRRy((s9lib/python2.7/site-packages/numba/tests/test_extending.pyRd s    t TestIntrinsiccB`s,eZdZdZdZdZRS(c`stdtdtdtfd}tdtfd}|j|d|jt}|WdQX|jd|jjdS( s Verify that returning a None from codegen function is handled automatically for void functions, otherwise raise exception. cS`s%tjtj}d}||fS(NcS`sdS(N((R5R>R%RV((s9lib/python2.7/site-packages/numba/tests/test_extending.pytcodegen s(RtvoidRP(RRRUR((s9lib/python2.7/site-packages/numba/tests/test_extending.pyt void_funcs cS`s%tjtj}d}||fS(NcS`sdS(N((R5R>R%RV((s9lib/python2.7/site-packages/numba/tests/test_extending.pyR)s(RRP(RRRUR((s9lib/python2.7/site-packages/numba/tests/test_extending.pyt non_void_func&s Rc`sddS(Nii(((R(s9lib/python2.7/site-packages/numba/tests/test_extending.pytcall_void_func-s c`sddS(Nii(((R(s9lib/python2.7/site-packages/numba/tests/test_extending.pytcall_non_void_func2s iNsnon-void function returns None( R RRRR RR R RA(R4RRR((RRs9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_void_returns  c `s"ddlm}m}m}m}fddtdtfd}d}|d |||}||tdt fd }t j d d t j } t #} || | jj} Wd QX| jttt| } j| | d S( sV Usecase test: custom reinterpret cast to turn int values to pointers i(t CFUNCTYPEtPOINTERtc_floattc_intc`s4ttjsttfd}|S(Nc`sKj|tjt|tjrGtj}d}||fSdS(NcS`s1|\}|j}|j|}|j||S(N(RTtget_value_typetinttoptr(R5R>R%RVtsrctrtypetllrtype((s9lib/python2.7/site-packages/numba/tests/test_extending.pyROs  (RR tContextR0RRtuintp(RRRUR(t result_typeR4(s9lib/python2.7/site-packages/numba/tests/test_extending.pyt unsafe_castHs  (R0RtCPointerRR (RR(R4(Rs9lib/python2.7/site-packages/numba/tests/test_extending.pyt unsafe_casterEscS`stddS(Nsnot callable from python(RK(R((s9lib/python2.7/site-packages/numba/tests/test_extending.pytunsafe_get_ctypes_pointer[sREc`sAt|tjr=tj|jfd}|SdS(Nc`sd|jjS(NR(tctypesR(R(R(s9lib/python2.7/site-packages/numba/tests/test_extending.pyt array_implcs(R0RRRR(tarrtypeR(R(Rs9lib/python2.7/site-packages/numba/tests/test_extending.pyt$array_impl_unsafe_get_ctypes_pointer^scS`s)x"t|D]}t||q WdS(N(RwR(tptrR#R~((s9lib/python2.7/site-packages/numba/tests/test_extending.pyt my_c_fun_rawhsRc`s |}||jdS(N(Rx(RR(tmy_c_funR(s9lib/python2.7/site-packages/numba/tests/test_extending.pyR.qs i RN(RRRRRRRRNRRRlRtfloat32RRRltcloseRtmapR R( R4RRRRRRt prototypeR.RtbufRR$((RR4RRs9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_ll_pointer_cast>s" !  !   c`s&tdtdtfd}|j|ddtj}t|}tj|}|d7}|j|t|tj |}|j|t||j|d|dtj}|j|t|tj |}|j ||j|t|dS(s9 Test serialization of intrinsic objects cS`sd}||}||fS(NcS`s|dS(Ni((R5R>R%RV((s9lib/python2.7/site-packages/numba/tests/test_extending.pyRs((R5RLRRU((s9lib/python2.7/site-packages/numba/tests/test_extending.pytidentitys  Rc`s |S(N((RL(R(s9lib/python2.7/site-packages/numba/tests/test_extending.pyR.siN( R RRRR!t_memoRtpickletdumpstloadsR(R4R.tmemot memo_sizetserialized_foot foo_rebuilttserialized_identitytidentity_rebuilt((Rs9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_serializations     cC`sd}tj}t|}td|}|j|j|tj|}|d7}|j|t|~|j|t|tjj |d8}|j|t|tj |}|j |j|tj |}|j|j|jdS(s3 Test deserialization of intrinsic cS`sd}|||fS(NcS`s|dS(Ni((R5R>R%RV((s9lib/python2.7/site-packages/numba/tests/test_extending.pyRs((R5RLR((s9lib/python2.7/site-packages/numba/tests/test_extending.pytdefns R.iN( R!RRRt_defnRRRt_recenttclearRt assertIsNot(R4RRRtoriginaltpickledtrebuilttsecond((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_deserializations"      (R,R-RRRR(((s9lib/python2.7/site-packages/numba/tests/test_extending.pyRs ' A .tTestRegisterJitablecB`seZdZdZRS(c`setdfd}tdt|}|dd}|dd}|j||dS(NcS`s||S(N((RLRj((s9lib/python2.7/site-packages/numba/tests/test_extending.pyR.sc`s ||S(N((RLRj(R.(s9lib/python2.7/site-packages/numba/tests/test_extending.pyRLsRii(R"RRR(R4RLtcbarR$R((R.s9lib/python2.7/site-packages/numba/tests/test_extending.pyt test_no_flagss c`stdtdfd}|j|djdddgtdt|}|jtj}|dWdQXd }|j |t |j dS( Nt_nrtcS`s tj|S(N(RlR(R#((s9lib/python2.7/site-packages/numba/tests/test_extending.pyR.sc`s |S(N((R#(R.(s9lib/python2.7/site-packages/numba/tests/test_extending.pyRLsiiiiRsCOnly accept returning of array passed into the function as argument( R"RRttolistRRR RRR R R (R4RLRR RA((R.s9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_flags_no_nrts%(R,R-RR(((s9lib/python2.7/site-packages/numba/tests/test_extending.pyRs tTestImportCythonFunctioncB`sSeZejedkddZdZejedkddZRS(s&Only run if SciPy >= 0.19 is installedc`sktdd}tjtjtj}||tdtfd}|j|dddS(Nsscipy.special.cython_specialtj0Rc`s |S(N((RL(t_j0(s9lib/python2.7/site-packages/numba/tests/test_extending.pyRRii(R#RRtc_doubleRRR(R4taddrtfunctypeR((Rs9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_getting_functions  cC`sY|jt}tdd}WdQXd}tj|t|j}|j|dS(Nt fakemodulet fakefunctionsNo module named '?fakemodule'?(R t ImportErrorR#tretmatchR R tassertIsNotNone(R4R RRAR((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_missing_modules cC`sJ|jt}tdd}WdQXd}|j|t|jdS(Nsscipy.special.cython_specialR.sINo function 'foo' found in __pyx_capi__ of 'scipy.special.cython_special'(R RtR#RR R (R4R RRA((s9lib/python2.7/site-packages/numba/tests/test_extending.pyttest_missing_functionsN( R,R-RVtskipIftscRNRRR(((s9lib/python2.7/site-packages/numba/tests/test_extending.pyRs! t__main__(t __future__RRRRXRRRRnRtdistutils.versionRRtnumpyRltnumbaRRVRRRRR R tnumba.targets.registryR tnumba.compilerR tsupportR RRRRt numba.errorsRtnumba.extendingRRRRRRRRRRRRRR R!R"R#tnumba.typing.templatesR$R%R&R'R(t version_infoR=tpdlike_usecaseR)R*tscipyt __version__RNRtscipy.special.cython_specialtspecialtcython_specialRtobjectR+tOpaqueR.R9R8tregisterR<R1R@RAt OpaqueModelRERFRGRIRHRJRMRSRORWRQR\R]RhRkRmRnRRRRRRtdelitemRtgetitemRtsetitemRRRtiaddRRRRRRRRRRRRRRRRRRRRRRRRRRRSRRRRR_R`RdRRRR,tmain(((s9lib/python2.7/site-packages/numba/tests/test_extending.pyts        .(p (                -                            *c $