U ,d@s<ddlZddlZddlZddlZddlZddlZddlZddlmZddl Z ddl Z ddl m Z mZmZmZmZddlmZmZmZmZmZddlmZddlmZddlmZddlmZdd l m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&dd l'm(Z(ddl)Z)dd l*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?m@Z@mAZAmBZBd dlCmDZDmEZEz4ddlFZFeeFjGdkrdZHnddlImJmKZHWneLk rdZHYnXGdddeMZNGdddejOZPePdZQeNZRe+SeNddZTe.ePejUddZVddZWe3ePe2jXe5ePddZYddZZeZd\Z[Z\Z]e5e[d d!Z^dd"d#Z_d$d%Z`e,e_e`Zae-e_e-e_ejbd&d'Zce-e_ejdd(d)Zee@Gd*d+d+e>Zfe-d,ePd-d.Zgd/d0Zhd1d2Zid3d4Zje/ehd5d6Zke/ehd7d8Zle/emd9d:Zne/ejod;d<Zpe/ejqd=d>Zre/ejsd?d@Zte/ejudAdBZvdCdDZwdEdFZxe/ejydGdHZzdIdJZ{dKdLZ|dMdNZ}dOdPZ~dQdRZe1ePdSdTdUZdVdWZdXdYZdZd[Zd\d]Zd^d_Zd`daZdbdcZdddeZdfdgZdhdiZdjdkZdldmZdndoZe/edpdqZdrdsZdtduZdvdwZeAeGdxdydyeBZdzd{ZGd|d}d}eMZGd~ddejZe3eGddde2jXZeAeeeeeddZe/e jddZGddde!ZGddde!ZGddde!ZddZe&Gddde!ZddZGddde!ZGddde)j!ZGddde)j!Ze1ePdddidddZe1ePdddidddZe0ePdddidddZe0ePdddidddZGddde!ZGddde!ZddZe&Gddde!ZGddde!ZGddde!Zedkr8e)dS)N) LooseVersion)njitjit vectorize guvectorizeobjmode)typeserrorstypingcompilercgutils)type_inference_stage) cpu_target)compile_isolated)lower_constant)TestCasecaptured_stdouttemp_directoryoverride_configrun_in_new_process_in_cache_dirskip_if_typeguard) LoweringError) typeof_impl type_callable lower_builtin lower_castoverloadoverload_attributeoverload_methodmodelsregister_modelboxunbox NativeValue intrinsic _Intrinsicregister_jitableget_cython_function_address is_jittedoverload_classmethod)ConcreteTemplate signatureinfer infer_globalAbstractTemplate)IndexSeriesz0.19c@s eZdZdS)MyDummyN__name__ __module__ __qualname__r7r7>/tmp/pip-unpacked-wheel-eu7e0c37/numba/tests/test_extending.pyr2Nsr2c@seZdZddZdS) MyDummyTypecCs"t|tjrddlm}|jSdS)Nr) Conversion) isinstancerNumberZnumba.core.typeconvr:safe)selfcontexttotyr:r7r7r8can_convert_toSs  zMyDummyType.can_convert_toN)r4r5r6rAr7r7r7r8r9Rsr9mydummycCstSN) mydummy_typevalcr7r7r8typeof_mydummy^srHcCs ||dS)z2 Implicit conversion from MyDummy to int. *) get_constant)r?builderZfromtyr@rFr7r7r8mydummy_to_numbercsrLcCstSrC)rBr7r7r7r8 get_dummyksrMcCst|jSrCr#r?get_dummy_valuetypobjrGr7r7r8 unbox_indexrsrScsXGdddt}Gdddtj}||t|fdd}t|tj||fS)Nc@s eZdZdS)z(base_dummy_type_factory..DynTypeNr3r7r7r7r8DynType|srTc@s eZdZdS)z,base_dummy_type_factory..DynTypeTypeNr3r7r7r7r8 DynTypeTypesrUcsSrCr7rE dyn_type_typer7r8rHsz/base_dummy_type_factory..typeof_mydummy)objectrOpaquerregisterr r OpaqueModel)namerTrUrHr7rVr8base_dummy_type_factory{sr]mydummy2cCst|jSrCrNrPr7r7r8 unbox_index2sr_cCstdSrCNotImplementedErrorxr7r7r8func1srdcCsddd}|S)NcSs(|dtjfkrtjSt|tjr$|SdSrC)rnoneint32r;Floatrbr7r7r8typers ztype_func1_..typer)Nr7)r?rhr7r7r8 type_func1_s ricCs||jdSNrI)rJ return_type)r?rKsigargsr7r7r8 func1_nullarysrncCsdd}|||||S)NcSstd|S)N)mathsqrtrbr7r7r8 func1_implszfunc1_unary..func1_impl)Zcompile_internal)r?rKrlrmrrr7r7r8 func1_unarysrsc@seZdZdZeejegZdS) PrintDummy print_itemN) r4r5r6keyr+rrerDZcasesr7r7r7r8rtsrtrucCs<|\}||}||d}|||||S)Nzhello!)Zget_python_apiZ unserializeZserialize_objectZ print_objectZdecrefrO)r?rKrlrmrcpyapiZstrobjr7r7r8 print_dummys    rxcCstdSrCr`condrcyr7r7r8wheresr|cCst|||S)z8 Wrap np.where() to allow for keyword arguments )npr|ryr7r7r8np_wheresr~cCst|||dS)N)r{rc)r|ryr7r7r8 call_wheresrcCsTt|tjrP|j|jkr"tdtdd|||fDrDdd}ndd}|SdS)z' Implement where() for arrays. "x and y should have the same dtypecss|]}|jdkVqdS)CN)Zlayout).0tyr7r7r8 sz(overload_where_arrays..c Sst|j}|j|ks|j|kr"tdt|}|j}|j}|j}|j}t|jD] } || rb|| n|| || <qN|S)zM Fast implementation for C-contiguous arrays %all inputs should have the same shape)shape ValueErrorr} empty_likeZflatrangesize) rzrcr{rrescfZxfZyfrfir7r7r8 where_impls z)overload_where_arrays..where_implcSs\|j}|j|ks|j|kr"tdt|}t|D] \}}|rJ||n||||<q6|S)zI Generic implementation for other arrays r)rrr}rZ ndenumerate)rzrcr{rridxrGr7r7r8rs N)r;rArraydtyper TypingErrorallrzrcr{rr7r7r8overload_where_arrayss     rcCs.t|tjs*||krtddd}|SdS)z( Implement where() for scalars. z!x and y should have the same typecSs"|r|n|}t|}||d<|S)z@ Scalar where() => return a 0-dim array r7)r}r)rzrcr{Zscalarrr7r7r8rs  z*overload_where_scalars..where_implN)r;rrr rrr7r7r8overload_where_scalarss    rcCst|trdd}|SdS)NcSsdS)N r7argr7r7r8len_impl1sz$overload_len_dummy..len_implr;r9)rrr7r7r8overload_len_dummy-s rcCs,t|ttfr(t|ttfr(dd}|SdS)NcSsdSrjr7arg1arg2r7r7r8dummy_add_impl=sz*overload_add_dummy..dummy_add_implr;r9 MyDummyType2)rrrr7r7r8overload_add_dummy7s rcCs&t|tr"t|tjr"dd}|SdS)NcSstd||dS)NdelprintrRrr7r7r8dummy_delitem_implGsz2overload_dummy_delitem..dummy_delitem_implr;r9rInteger)rRrrr7r7r8overload_dummy_delitemCsrcCs&t|tr"t|tjr"dd}|SdS)NcSs|dS)N{r7rr7r7r8dummy_getitem_implQsz2overload_dummy_getitem..dummy_getitem_implr)rRrrr7r7r8overload_dummy_getitemMsrcCs4tt|tt|tjt|tjgr0dd}|SdS)NcSst||dSrCrrRrrFr7r7r8dummy_setitem_implasz2overload_dummy_setitem..dummy_setitem_impl)rr;r9rr)rRrrFrr7r7r8overload_dummy_setitemWs  rcCs t||SrCoperatoraddrr7r7r8call_add_operatorgsrcCs||SrCr7rr7r7r8call_add_binopksrcCs,t|ttfr(t|ttfr(dd}|SdS)NcSsdSrjr7rr7r7r8dummy_iadd_implusz,overload_iadd_dummy..dummy_iadd_implr)rrrr7r7r8overload_iadd_dummyos rcCs t||SrCrrr7r7r8call_iadd_operator{srcCs ||7}|SrCr7rr7r7r8call_iadd_binopsrcCs ||=dSrCr7rr7r7r8 call_delitemsrcCs||SrCr7rr7r7r8 call_getitemsrcCs |||<dSrCr7rr7r7r8 call_setitemsrlengthcCs dd}|S)NcSst|SrClenrr7r7r8impsz#overload_method_length..impr7)rrr7r7r8overload_method_lengthsrcCs|SrC)rrbr7r7r8cache_overload_method_usecasesrcCstSrCrdr7r7r7r8call_func1_nullarysrcCst|SrCrrbr7r7r8call_func1_unarysrcCst|SrCrrbr7r7r8 len_usecasesrcCs t|dSrCrrbr7r7r8 print_usecasesrcCs||SrCr7)rcrvr7r7r8getitem_usecasesrcCstt|SrC)r}cossinrbr7r7r8npyufunc_usecasesrcCs|jSrC)_datarbr7r7r8get_data_usecasesrcCs|jSrC)_indexrbr7r7r8get_index_usecasesrcCs|jSrC)Zis_monotonic_increasingrbr7r7r8is_monotonic_usecasesrcCs t||SrC)r1)dataindexr7r7r8make_series_usecasesrcCs |||SrC)Zclip)rclohir7r7r8 clip_usecasesrcCstSrCr}r7r7r7r8return_non_boxablesrcCs dd}|S)NcSstSrCrr7r7r7r8rsz(overload_return_non_boxable..impr7)rr7r7r8overload_return_non_boxablesrcCst}||SrC)rarange)szmodr7r7r8non_boxable_ok_usecasesrcCstSrC)rr7r7r7r8non_boxable_bad_usecasesrcCsdSrCr7)fr7r7r8 mk_func_inputsrc@seZdZddZdS) MkFuncTypingcCs$t|dtjstttjf|SNr)r;rMakeFunctionLiteralAssertionErrorr+re)r>rmkwsr7r7r8genericszMkFuncTyping.genericN)r4r5r6rr7r7r7r8rsrcCstdddS)NcSs|SrCr7ar7r7r8z#mk_func_test_impl..)rr7r7r7r8mk_func_test_implsrc@s eZdZdS)MyClassNr3r7r7r7r8rsrcs4eZdZfddZddZddZddZZS) CallableTypeRefcs.||_i|_g|_tt|d|jdS)Nzcallable_type_ref[{}]) instance_typesig_to_impl_keycompiled_templatessuperr__init__format)r>r __class__r7r8rs zCallableTypeRef.__init__c Cstd}|jt|D]\}z|||}Wntk r:YqXt|di}||kr|||j|<|j|qpq|S)NZ_compiled_overloads)Z _functionstypeapply Exceptiongetattrrrappend)r>r?rmrZres_sigtemplateZcompiled_ovldsr7r7r8 get_call_types  zCallableTypeRef.get_call_typecCst|j}|dfS)NT)listrkeys)r>Zsigsr7r7r8get_call_signaturessz#CallableTypeRef.get_call_signaturescCs |j|SrC)r)r>rlr7r7r8 get_impl_keyszCallableTypeRef.get_impl_key)r4r5r6rrrr __classcell__r7r7rr8rs rc@seZdZddZdS)CallableTypeModelcCstj|||dSrC)rr[r)r>ZdmmZfe_typer7r7r8rszCallableTypeModel.__init__N)r4r5r6rr7r7r7r8rsrcCs|SrC)rO)r?rKrZpyvalr7r7r8constant_callable_typeref%srcCst|trdd}|SdS)NcSsdS)N>[=r7rRr7r7r8r1szoverload_np_exp..imprrRrr7r7r8overload_np_exp-s r c@s8eZdZdZddZddZddZdd Zd d Zd S) TestLowLevelExtendingz4 Test the low-level two-tier extension API. cCsVt}tdd|}||dt}tdd|}||dd||dddS)NTnopythonrI2@@)rrassertPreciseEqualrr>pyfunccfuncr7r7r8 test_func1@sz TestLowLevelExtending.test_func1cCsFt}t|d}||dt}t|tjf}||dddS)Nr7rIrr)rrr entry_pointrrfloat64r>rZcrr7r7r8test_func1_isolatedIs  z)TestLowLevelExtending.test_func1_isolatedcCs|tt|tdSrC)assertIs type_func1riassertIsNotNoner>r7r7r8!test_type_callable_keeps_functionQs z7TestLowLevelExtending.test_type_callable_keeps_functioncCs&t}t|dtj}||ddS)Nr7gE@)rMrrrrrrr7r7r8test_cast_mydummyUsz'TestLowLevelExtending.test_cast_mydummycCsXtt}tj}tj}||t|||dd}|t dd|j DdS)zFmake sure make_function is passed to typer class as a literal r7Ncss|]}t|tjVqdSrC)r;rr)rrr7r7r8resz=TestLowLevelExtending.test_mk_func_literal..) r Z run_frontendrrZtyping_contextZtarget_contextZrefreshr assertTrueanytypemapvalues)r>Ztest_ir typingctxZ targetctxZ typing_resr7r7r8test_mk_func_literalZs  z*TestLowLevelExtending.test_mk_func_literalN) r4r5r6__doc__rrrrr$r7r7r7r8r 8s  r c@s`eZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ dS)TestPandasLikeze Test implementing a pandas-like Index object. Also stresses most of the high-level API. cCs0ttd}tddt}|||ddS)NTr )r0r}rrrr)r>rrr7r7r8test_index_lenrszTestPandasLike.test_index_lencCshttdddg}tddt}|||dd||tdd}||t|t |ddgdS)NrITr r/) r0r}rfrrrsliceassertIsInstance assertEqualrr>rriir7r7r8test_index_getitemws  z!TestPandasLike.test_index_getitemcCsVttdddg}tddt}||}||t||jtt |jdS)z7 Check Numpy ufunc on an Index object. rIr)r*Tr N) r0r}rfrrr,rrrrr.r7r7r8test_index_ufuncs  zTestPandasLike.test_index_ufunccCs<ttdddg}tddt}||}|||jdS)NrIr)r*Tr )r0r}rfrrrr)r>rrrr7r7r8test_index_get_datasz"TestPandasLike.test_index_get_datacCs`tddt}dddgdfdddgdfgdffD]*\}}tt|}||}|||q0dS)NTr r)rIF)rrr0r}rfr-)r>rr"expectedrgotr7r7r8test_index_is_monotonics   z&TestPandasLike.test_index_is_monotoniccCsLttdddg}ttdddg|}tddt}|||ddS Nror'?@@Tr )r0r}rfr1rrrr)r>rsrr7r7r8test_series_lenszTestPandasLike.test_series_lencCs`ttdddg}ttdddg|}tddt}||}||t||j |j dSr7) r0r}rfr1rrrr,rr)r>rr<rr5r7r7r8test_series_get_indexs  z$TestPandasLike.test_series_get_indexcCsttdddg}ttdddg|}tddt}||}||t||jt| |jj |j | |j t t|j d S) z8 Check Numpy ufunc on an Series object. rIr)r*r/ror'Tr N)r0r}rfr1int64rrr,rrrr_valuesrrr>rr<rssr7r7r8test_series_ufuncs z TestPandasLike.test_series_ufunccCszttdddg}tdddg}tddt}|||}||t||jt| |jj |j | |j |dS) NrIr)r*r9r:r;Tr ) r0r}rfrrrr,r1rrrr@)r>rdrr5r7r7r8test_series_constructors  z&TestPandasLike.test_series_constructorcCsttdddg}ttdddg|}tddt}||d d }||t||jt| |jj |j | |j td d dgdS) NrIr)r*r9r:r;Tr g?g@) r0r}rfr1rrrr,rrrrr@rAr7r7r8test_series_clips  zTestPandasLike.test_series_clipN)r4r5r6r%r(r0r1r2r6r=r>rCrErFr7r7r7r8r&ls    r&c@seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*S)+TestHighLevelExtendingz+ Test the high-level combined API. c st}tdd|fdd}|dddd|ddd|tdddgtd d dgtd d d gtj$}tgtgtgW5QRX d t |j dS)z> Test implementing a function with @overload. Tr cs$t||}||}||dSrC)r~r)rmkwargsr4r5rr>r7r8checks  z0TestHighLevelExtending.test_where..checkr'r))rcrzr{Fr/ror8r3rN) rrr}Zbool_rf assertRaisesr rr?assertInstr exception)r>rrJraisesr7rIr8 test_wheres  (z!TestHighLevelExtending.test_wherecCs8tddt}||td||ddgddS)zN Test re-implementing len() for a custom type with @overload. Tr rr8r3roN)rrrr2r>rr7r7r8test_lenszTestHighLevelExtending.test_lenc Cs@tddt}t"|t|tjdW5QRXdS)zP Test re-implementing print() for a custom type with @overload. Tr zhello! N)rrrr2r-sysstdoutgetvaluerQr7r7r8 test_prints z!TestHighLevelExtending.test_printcCsTt}tdd|}||ddd||ttd||ttddS zW Test re-implementing operator.add() for a custom type with @overload. Tr r/ror'rITN)rrrMyDummy2r2rr7r7r8test_add_operators z(TestHighLevelExtending.test_add_operatorcCsTt}tdd|}||ddd||ttd||ttddS zY Test re-implementing '+' for a custom type via @overload(operator.add). Tr r/ror'rIrXN)rrrrYr2rr7r7r8test_add_binop s z%TestHighLevelExtending.test_add_binopcCsTt}tdd|}||ddd||ttd||ttddSrW)rrrrYr2rr7r7r8test_iadd_operators z)TestHighLevelExtending.test_iadd_operatorcCsTt}tdd|}||ddd||ttd||ttddSr[)rrrrYr2rr7r7r8test_iadd_binop(s z&TestHighLevelExtending.test_iadd_binopc Cst}tdd|}t}d}t<}z||dWn&tk rX}z|}W5d}~XYnXW5QRX|dk rp|||ddS)NTr Azdel hello! 321 )rrr2rrr-rUr>rrrReoutexcr7r7r8 test_delitem6s z#TestHighLevelExtending.test_delitemcCs*t}tdd|}||tdddS)NTr r_i)rrrr2rr7r7r8 test_getitemFsz#TestHighLevelExtending.test_getitemc Cst}tdd|}t}d}t>}z||ddWn&tk rZ}z|}W5d}~XYnXW5QRX|dk rr|||ddS)NTr r_rz321 123 )rrr2rrr-rUr`r7r7r8 test_setitemKs z#TestHighLevelExtending.test_setitemc Csxtddt}d}||}t|}tj||tddt}|t }|W5QRXt|j }d}| ||dS)zW Test overloading whose return value cannot be represented in CPython. Tr  zcannot convert native ModuleN) rrr}ZtestingZ assert_equalrrK TypeErrorrMrNrL) r>Zok_cfuncnr5expectZ bad_cfuncrOerrmsgZ expectmsgr7r7r8test_no_cpython_wrapper[s  z.TestHighLevelExtending.test_no_cpython_wrapperc Csd'dd}d}d(dd}|tj}||dd d d W5QRXt|j}||||d ||d ||d|d)dd}|tj}||dd d d W5QRXt|j}||||d||d||d|d*dd}|tj}||dd d d W5QRXt|j}||||d||d|k|d||d|ddlm}m } |tj}||dd d d W5QRXt|j}||||d||d|k|d||tj}|| dd d d W5QRXt|j}||||d||d|k|d||d|d+dd} |tj}|| dd d d W5QRXt|j}||||d||d|k|d||d|d,dd } |tj}|| dd d d W5QRXt|j}||||d||d|k|d!|d-d"d#} |tj}|| dd d d W5QRXt|j}||||d||d$|d%d&} |tj}|| dd d d W5QRXt|j}||||d||d|dS).z} Tests that an overload which has a differing typing and implementing signature raises an exception. Ncs:d ddtd fdd }tddfdd}|S) NcSsdSrCr7rbrGkwr7r7r8 myoverloaduszjTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling..gen_ol..myoverloadcsSrCr7rm)implr7r8_myoverload_implxszpTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling..gen_ol.._myoverload_implTr cs||||ddS)Nror7)rrnrGrDrpr7r8foo|szcTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling..gen_ol..foo)N)Nrr)rqrrrur7)rqrpr8gen_olts  zVTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling..gen_olz-Typing and implementation arguments differ in cSs|dkr dSdSdSNrgr/r7rmr7r7r8impl1szUTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling..impl1r/ror'r8zkeyword argument default valueszzcSs|dkr dSdSdSryr7)rrnrGkwargr7r7r8impl2szUTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling..impl2zkeyword argument nameszcSstdkr dSdSdSryr)zrnrGror7r7r8impl3szUTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling..impl3zargument nameskeywordzz)impl4impl5zFirst difference: 'z'cSs|dkr dSdSdSryr7)rrnrGrDraror7r7r8impl6szUTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling..impl6zzcSs|dkr dSdSdSryr7)rrnror7r7r8impl7szUTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling..impl7zcSs|dkr dSdSdSryr7)rrnrGroZ extra_kwargr7r7r8impl8szUTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling..impl8zcSs|dkr dSdSdSryr7)rrnrGr7r7r8impl9szUTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling..impl9)N)rx)N)N)N)N)NN) rKr rrMrNrL assertFalseoverload_usecasesrr)r>rwsentinelr{ramsgr}rrrrrrrr7r7r8/test_typing_vs_impl_signature_mismatch_handlingns                                               zFTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handlingc sd ddddlm}t|tddfdd }|tj}|dd W5QRXt|j}| d || d |dS)z Tests that an overload which has a differing typing and implementing signature raises an exception and uses VAR_POSITIONAL (*args) in typing NcSsdSrCr7rror7r7r8rpsziTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_positional..myoverloadr/)var_positional_implTr cs||dddS)N rsr7rrnrtr7r8ru szbTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_positional..foor3z)VAR_POSITIONAL (e.g. *args) argument kindz-offending argument name is '*star_args_token')N) rrrrrKr rrMrNrL)r>rrurarr7rtr8>test_typing_vs_impl_signature_mismatch_handling_var_positionals     zUTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_positionalc Csddd}dd}||ddd|tj}||ddW5QRXt|j}|d ||d |dd d }|tj}||ddW5QRXt|j}|d ||d |dS)zJ Tests that an overload which uses **kwargs (VAR_KEYWORD) Tcs4dddt|d|tddfdd}|S) NcSsdSrCr7rr7r7r8rpszvTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_keyword..gen_ol..myoverloadstrictTr cs |ddS)Nrrsr7rrtr7r8ru"szoTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_keyword..gen_ol..foo)Nrv)rqrrur7rtr8rws  zbTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_keyword..gen_olc[sddd}|S)NrgcSs|SrCr7rr7r7r8rq*szmTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_keyword..ol1..impl)rgr7)rrrqr7r7r8ol1)s z_TestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_keyword..ol1Fr/roz1use of VAR_KEYWORD (e.g. **kwargs) is unsupportedz"offending argument name is '**kws'rcSs dd}|S)Nc[s|SrCr7)rrr7r7r8rq8szmTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_keyword..ol2..implr7)rrorqr7r7r8ol27sz_TestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_keyword..ol2N)T)r)rKr rrMrNrL)r>rwrrarrr7r7r8;test_typing_vs_impl_signature_mismatch_handling_var_keywords       zRTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_keywordcCs@ttjdd dd}tdd}td}|||ddS) NrurgcSsddd}|S)NrgcSs|SrCr7)ra_kwargr7r7r8rqGszQTestHighLevelExtending.test_overload_method_kwargs..fooimpl..impl)rgr7)rrrqr7r7r8fooimplEs zCTestHighLevelExtending.test_overload_method_kwargs..fooimplcSs||d|jddfS)N)rruAr7r7r8barLsz?TestHighLevelExtending.test_overload_method_kwargs..barr3)rgrr)rg)rrrrr}rr-)r>rrZr7r7r8test_overload_method_kwargsCs     z2TestHighLevelExtending.test_overload_method_kwargscCsFttjddd}tdd}td}|||||ddS)NlitfoocSs(t|tjr$t|tjs$dd}|SdS)NcSs|SrCr7)rrFr7r7r8rq]szXTestHighLevelExtending.test_overload_method_literal_unpack..litfoo..impl)r;rrLiteral)rrFrqr7r7r8rVs  zJTestHighLevelExtending.test_overload_method_literal_unpack..litfoocSs |dSN)rrr7r7r8rbszGTestHighLevelExtending.test_overload_method_literal_unpack..barr/r)rrrrr}zerosr-)r>rrrr7r7r8#test_overload_method_literal_unpackTs    z:TestHighLevelExtending.test_overload_method_literal_unpackcCstdd}||ddS)NcSs ttSrC)r}exprBr7r7r7r8testnsz8TestHighLevelExtending.test_overload_ufunc..testr)rr-)r>rr7r7r8test_overload_ufuncjs z*TestHighLevelExtending.test_overload_ufunccCsttddd}tdd}t}|||ddd|||ddd d |||ddd d d td d}|||d|||dddd|||dddd|||dddddS)Nmethod_starargcWs dd}|S)NcWs |||fSrCr7)rRrFval2rmr7r7r8getwsz\TestHighLevelExtending.test_overload_method_stararg.._ov_method_stararg..getr7)rRrFrrmrr7r7r8_ov_method_stararguszOTestHighLevelExtending.test_overload_method_stararg.._ov_method_starargcWs |j|SrCr)rRrmr7r7r8ru|sz@TestHighLevelExtending.test_overload_method_stararg..foor/ror/ror7r'r/ror'r8r/ror'r8cSs(|dd|ddd|ddddfS)Nr/ror'r8rrr7r7r8rs  z@TestHighLevelExtending.test_overload_method_stararg..bar)rrrr)r/ro)rr)r/ro)rr')r8r3)r/ro)rrr9rr2r-)r>rrurRrr7r7r8test_overload_method_starargts4      z3TestHighLevelExtending.test_overload_method_starargc sGdddtjtddd}tfdd}d}|||t|td d }|tj }||W5QRX| d t |j dS) Nc@s eZdZdS)zATestHighLevelExtending.test_overload_classmethod..MyArrayNr3r7r7r7r8MyArraysr array_alloccSs dd}|S)NcSst|}|SrCr}r)clsnitemsrr7r7r8rqs zVTestHighLevelExtending.test_overload_classmethod..ol_array_alloc..implr7)rrrqr7r7r8ol_array_allocszHTestHighLevelExtending.test_overload_classmethod..ol_array_alloccs |SrC)rrrr7r8rusz=TestHighLevelExtending.test_overload_classmethod..foorcSs tj|SrC)rrrrr7r7r8no_classmethod_in_baseszPTestHighLevelExtending.test_overload_classmethod..no_classmethod_in_basez"Unknown attribute 'array_alloc' of) rrr)rrr}rrKr rrLrMrN)r>rrurrrOr7rr8test_overload_classmethods  z0TestHighLevelExtending.test_overload_classmethodcCs\ttdd}ttdd}tdd}d}d|d d |d f}||||dS) NcSst|tjrdd}|SdS)NcSsd|S)N@E@r7rbr7r7r8rqszeTestHighLevelExtending.test_overload_callable_typeref..callable_type_call_ovld1..impl)r;rrrcrqr7r7r8callable_type_call_ovld1s zWTestHighLevelExtending.test_overload_callable_typeref..callable_type_call_ovld1cSst|tjrdd}|SdS)NcSsd|S)N42.5r7rbr7r7r8rqszeTestHighLevelExtending.test_overload_callable_typeref..callable_type_call_ovld2..impl)r;rZ UnicodeTyperr7r7r8callable_type_call_ovld2s zWTestHighLevelExtending.test_overload_callable_typeref..callable_type_call_ovld2cSst|t|fSrC)rrr7r7r8ruszBTestHighLevelExtending.test_overload_callable_typeref..foo)r84rrrr/)rrrr)r>rrrurmr4r7r7r8test_overload_callable_typerefs   z5TestHighLevelExtending.test_overload_callable_typerefN)r4r5r6r%rPrRrVrZr\r]r^rdrerfrlrrrrrrrrrr7r7r7r8rGs, , + rGcCsD|j|jd}||kr td|j|jd}||kr@tddS)Nrzcache not used)Z _cache_hitsZ signaturesrZ _cache_misses)rZ expect_hitZ expect_misseshitZmissr7r7r8_assert_cache_statss rc@s eZdZdZddZddZdS)TestOverloadMethodCachingFc Cs2t|jj|_td|j|W5QRXdS)N CACHE_DIR)rrr4 _cache_dirrrun_caching_overload_methodrr7r7r8test_caching_overload_methodsz6TestOverloadMethodCaching.test_caching_overload_methodcCstdddt}||tdt|dd|tf}dd|D}|t |dzt d}Wnt k rt }YnX| }|jt||jfd }||t|||jd|jdd }||ddS) NTr cacherrr/cSs"g|]}|drd|kr|qS)Zdeclarer) startswith)rlnr7r7r8 s zITestOverloadMethodCaching.run_caching_overload_method..spawn)targetrm)timeout)rrrr2rZ inspect_llvmrD splitlinesr-rmultiprocessingZ get_contextAttributeErrorQueueProcessrrstartputjoinexitcoder)r>rZllvmirZdeclsctxqprr7r7r8rs.     z5TestOverloadMethodCaching.run_caching_overload_methodN)r4r5r6_numba_parallel_test_rrr7r7r7r8rsrc CsPtd|<|}tdddt}||}||t|ddW5QRXdS)zH Used by TestOverloadMethodCaching.test_caching_overload_method rTrr/rN)rrrrrr)r cache_dirrrrr7r7r8r s   rc@s4eZdZddZddZddZddZd d Zd S) TestIntrinsicc stddtddtddfdd}tddfd d }||d |t }|W5QRX|d |jjd S)z Verify that returning a None from codegen function is handled automatically for void functions, otherwise raise exception. cSsttj}dd}||fS)NcSsdSrCr7r?rKr+rmr7r7r8codegen#szBTestIntrinsic.test_void_return..void_func..codegenrvoidrfr#rrlrr7r7r8 void_funcs z1TestIntrinsic.test_void_return..void_funccSsttj}dd}||fS)NcSsdSrCr7rr7r7r8r-szFTestIntrinsic.test_void_return..non_void_func..codegen)rrfrr7r7r8 non_void_func)s z5TestIntrinsic.test_void_return..non_void_funcTr cs ddSNr/rr7r7)rr7r8call_void_func2sz6TestIntrinsic.test_void_return..call_void_funccs ddSrr7r7)rr7r8call_non_void_func7sz:TestIntrinsic.test_void_return..call_non_void_funcrznon-void function returns NoneN)r$rr-rKrrLrNr)r>rrrar7)rrr8test_void_returns   zTestIntrinsic.test_void_returnc sddlm}m}m}m}fddddtddfd d }d d }|d |||}||tddfdd}tjdtj d} t } || | } W5QRX| ttt| } | | d S)zV Usecase test: custom reinterpret cast to turn int values to pointers r) CFUNCTYPEPOINTERc_floatc_intcs&ttjsttfdd}|S)Ncs8|tjt|tjr4tj}dd}||fSdS)NcSs"|\}|j}||}|||SrC)rkZget_value_typeZinttoptr)r?rKr+rmsrcZrtypeZllrtyper7r7r8rTs z_TestIntrinsic.test_ll_pointer_cast..unsafe_caster..unsafe_cast..codegen)r,r Contextr;rrZuintp)r#rrlr) result_typer>r7r8 unsafe_castMs   zNTestIntrinsic.test_ll_pointer_cast..unsafe_caster..unsafe_cast)r;rCPointerrr$)rrr)rr8 unsafe_casterJsz9TestIntrinsic.test_ll_pointer_cast..unsafe_castercSs tddS)Nznot callable from pythonr`rr7r7r8unsafe_get_ctypes_pointer`szETestIntrinsic.test_ll_pointer_cast..unsafe_get_ctypes_pointerFrcs0t|tjr,t|jfdd}|SdS)Ncs|jjdS)Nr)ctypesr)rrr7r8 array_implhszdTestIntrinsic.test_ll_pointer_cast..array_impl_unsafe_get_ctypes_pointer..array_impl)r;rrrr)Zarrtyper)rrr8$array_impl_unsafe_get_ctypes_pointercs  zPTestIntrinsic.test_ll_pointer_cast..array_impl_unsafe_get_ctypes_pointercSst|D]}t||qdSrC)rr)ptrrirr7r7r8 my_c_fun_rawns z8TestIntrinsic.test_ll_pointer_cast..my_c_fun_rawNTr cs|}||jdSrC)r)rr)my_c_funrr7r8ruwsz/TestIntrinsic.test_ll_pointer_cast..foorg)r)rrrrrrrr}rZfloat32rrUrclosermaprMr-) r>rrrrrrZ prototyperurbufr5rjr7)rr>rrr8test_ll_pointer_castCs"   z"TestIntrinsic.test_ll_pointer_castcstddtddfdd}||ddtj}t|}t|}|d7}||t|t|}||t|||d|dt}||t|t|}| |||t|dS) z9 Test serialization of intrinsic objects cSsdd}||}||fS)NcSs|dSrr7rr7r7r8rszCTestIntrinsic.test_serialization..identity..codegenr7)r?rcrrlr7r7r8identitysz2TestIntrinsic.test_serialization..identityTr cs|SrCr7rbrr7r8rusz-TestIntrinsic.test_serialization..foor/N) r$rr-r%_memorpickledumpsloadsr)r>rumemo memo_sizeZserialized_fooZ foo_rebuiltZserialized_identityZidentity_rebuiltr7r r8test_serializations$      z TestIntrinsic.test_serializationcCsdd}tj}t|}td|}||j|t|}|d7}||t|~||t|tj |d8}||t|t |}| |j|t |}||j|jdS)z3 Test deserialization of intrinsic cSsdd}|||fS)NcSs|dSrr7rr7r7r8rszATestIntrinsic.test_deserialization..defn..codegenr7)r?rcrr7r7r8defnsz0TestIntrinsic.test_deserialization..defnrur/N) r%r rrZ_defnr r r-Z_recentclearr Z assertIsNot)r>rrroriginalZpickledZrebuiltsecondr7r7r8test_deserializations"     z"TestIntrinsic.test_deserializationcCs`ttddd}|d|j|d|j|d|j|dti|j|d|jdS)NrcSsttj}dd}||fS)void_func docstringcSsdSrCr7rr7r7r8rsz@TestIntrinsic.test_docstring..void_func..codegenrrr7r7r8rs z/TestIntrinsic.test_docstring..void_funcznumba.tests.test_extendingrrr) r$intr-r5r4r6ZassertDictEqual__annotations__r%)r>rr7r7r8test_docstrings zTestIntrinsic.test_docstringN)r4r5r6rrrrrr7r7r7r8rs *B.(rc@seZdZddZddZdS)TestRegisterJitablecsJtddfdd}tdd|}|dd}|dd}|||dS) NcSs||SrCr7rcr{r7r7r8rusz.TestRegisterJitable.test_no_flags..foocs ||SrCr7rrr7r8rsz.TestRegisterJitable.test_no_flags..barTr r/ro)r&rr-)r>rcbarrjr5r7rr8 test_no_flagss    z!TestRegisterJitable.test_no_flagsc stddddfdd}||ddd d gtd d |}|tj}|d W5QRXd }||t|j dS)NF)_nrtcSs t|SrCrrir7r7r8rusz2TestRegisterJitable.test_flags_no_nrt..foocs|SrCr7rrr7r8rsz2TestRegisterJitable.test_flags_no_nrt..barr'rr/roTr COnly accept returning of array passed into the function as argument) r&r-tolistrrKr rrLrMrN)r>rrrOrr7rr8test_flags_no_nrts  z%TestRegisterJitable.test_flags_no_nrtN)r4r5r6rr"r7r7r7r8rsrc@sDeZdZeedkdddZddZeedkdddZdS) TestImportCythonFunctionNz&Only run if SciPy >= 0.19 is installedcsLtdd}ttjtj}||tddfdd}||dddS) Nscipy.special.cython_specialj0Tr cs|SrCr7rbZ_j0r7r8rrz@TestImportCythonFunction.test_getting_function..rr/)r'rrc_doublerr-)r>addrZfunctyper%r7r&r8test_getting_functionsz.TestImportCythonFunction.test_getting_functionc CsD|t}tddW5QRXd}t|t|j}||dS)NZ fakemoduleZ fakefunctionzNo module named '?fakemodule'?)rK ImportErrorr'rematchrMrNr)r>rOrr,r7r7r8test_missing_modules  z,TestImportCythonFunction.test_missing_modulec Cs:|t}tddW5QRXd}||t|jdS)Nr$ruzINo function 'foo' found in __pyx_capi__ of 'scipy.special.cython_special')rKrr'r-rMrN)r>rOrr7r7r8test_missing_function&s z.TestImportCythonFunction.test_missing_function) r4r5r6unittestZskipIfscr)r-r.r7r7r7r8r#s  r#method_jit_option_check_nrtrTZ jit_optionscCs dd}|S)NcSs tdSNrgrrr7r7r8r7sz+ov_method_jit_option_check_nrt..impr7r r7r7r8ov_method_jit_option_check_nrt3sr4method_jit_option_check_no_nrtFcCs dd}|S)NcSs tdSr3rrr7r7r8rAsz.ov_method_jit_option_check_no_nrt..impr7r r7r7r8!ov_method_jit_option_check_no_nrt=sr6attr_jit_option_check_nrtcCs dd}|S)NcSs tdSr3rrr7r7r8rKsz)ov_attr_jit_option_check_nrt..impr7r r7r7r8ov_attr_jit_option_check_nrtGsr8attr_jit_option_check_no_nrtcCs dd}|S)NcSs tdSr3rrr7r7r8rUsz,ov_attr_jit_option_check_no_nrt..impr7r r7r7r8ov_attr_jit_option_check_no_nrtQsr:c@s4eZdZddZddZddZddZd d Zd S) TestJitOptionsNoNRTc Os<d}|tj}|||W5QRX||t|jdS)Nr )rKr rrLrMrN)r>funcrmrHrrOr7r7r8check_error_no_nrt^s z&TestJitOptionsNoNRT.check_error_no_nrtcsXddtd|iddd}tfdd}|rJ||td n ||dS) NcSs tdSr3rr7r7r7r8dummyisz8TestJitOptionsNoNRT.no_nrt_overload_check..dummyrr2cSs dd}|S)NcSs tdSr3rr7r7r7r8r>nszJTestJitOptionsNoNRT.no_nrt_overload_check..ov_dummy..dummyr7r>r7r7r8ov_dummylsz;TestJitOptionsNoNRT.no_nrt_overload_check..ov_dummycsSrCr7r7r?r7r8russz6TestJitOptionsNoNRT.no_nrt_overload_check..foorg)rrrr}rr=)r>flagr@rur7r?r8no_nrt_overload_checkhs z)TestJitOptionsNoNRT.no_nrt_overload_checkcCs|d|ddS)NTF)rBrr7r7r8test_overload_no_nrt|s z(TestJitOptionsNoNRT.test_overload_no_nrtcCs>tdd}||ttdtdd}||tdS)NcSs|SrC)r1rbr7r7r8udtsz.udtrgcSs|SrC)r5rbr7r7r8rDsrrrBr}rr=r>rDr7r7r8test_overload_method_no_nrts   z/TestJitOptionsNoNRT.test_overload_method_no_nrtcCs>tdd}||ttdtdd}||tdS)NcSs|jSrC)r7rbr7r7r8rDsz?TestJitOptionsNoNRT.test_overload_attribute_no_nrt..udtrgcSs|jSrC)r9rbr7r7r8rDsrErFr7r7r8test_overload_attribute_no_nrts   z2TestJitOptionsNoNRT.test_overload_attribute_no_nrtN)r4r5r6r=rBrCrGrHr7r7r7r8r;[s   r;cs<eZdZfddZddZddZddZd d ZZS) TestBoxingCallingJITcs0ttd}|\|_|_|_||_dS)Nr^)rsetUpr]rUrTrWdyn_type)r>manyrr7r8rJs zTestBoxingCallingJIT.setUpcslddtfddt|jfdd}t|jdd}td d }||j}||dS) Nrrcsfdd}t||}||fS)Ncs|\}|||SrCrrr?rKrlrmrF magic_offsetr7r8rqszKTestBoxingCallingJIT.test_unboxer_basic..my_intrinsic..implr+r#rFrqrlrOr7r8 my_intrinsics  z=TestBoxingCallingJIT.test_unboxer_basic..my_intrinsiccsLfdd}|jtjg}ttjtj}|j|||\}}t||dS)Ncs|SrCr7rbrSr7r8bridgeszHTestBoxingCallingJIT.test_unboxer_basic..unboxer..bridgeis_error r?rJrintpr+Zvoidptrrw call_jit_coder#rQrRrGrUrmrlrWr magic_tokenrSr7r8unboxers  z8TestBoxingCallingJIT.test_unboxer_basic..unboxercSs|j|tj}|j|SrCrKZptrtointr Zintp_trwlong_from_ssize_trQrFrGrr7r7r8boxersz6TestBoxingCallingJIT.test_unboxer_basic..boxercSs|SrCr7rbr7r7r8passthrusz9TestBoxingCallingJIT.test_unboxer_basic..passthrur$r"rUr!rrKr-)r>r^rbrcrbr7rPr]rSr8test_unboxer_basics     z'TestBoxingCallingJIT.test_unboxer_basicc Csft|jdd}t|jdd}tdd}|t}||jW5QRX|dt|j dS)NcSsHdd}|jtjdg}ttjtj}|j|||\}}t||dS)NcSs|dkrtd|S)Nrcannot be x > 0rrbr7r7r8rUszHTestBoxingCallingJIT.test_unboxer_raise..unboxer..bridger/rVrXr[r7r7r8r^s z8TestBoxingCallingJIT.test_unboxer_raise..unboxercSs|j|tj}|j|SrCr_rar7r7r8rbsz6TestBoxingCallingJIT.test_unboxer_raise..boxercSs|SrCr7rbr7r7r8rcsz9TestBoxingCallingJIT.test_unboxer_raise..passthrurg r"rUr!rrKrrKrLrMrNr>r^rbrcrOr7r7r8test_unboxer_raises    z'TestBoxingCallingJIT.test_unboxer_raisecslddtfddt|jdd}t|jfdd}td d }||j}||dS) Nri8csfdd}t||}||fS)Ncs|\}|||SrCrMrNrOr7r8rqszCTestBoxingCallingJIT.test_boxer..my_intrinsic..implrQrRrOr7r8rSs  z5TestBoxingCallingJIT.test_boxer..my_intrinsiccSst|jSrCrNrPr7r7r8r^sz0TestBoxingCallingJIT.test_boxer..unboxercsLfdd}|jtjg}ttjtj}|j|||\}}|j|S)Ncs|SrCr7rbrTr7r8rUsz>TestBoxingCallingJIT.test_boxer..boxer..bridge)r?rJrrYr+rwrZr`)rQrFrGrUrmrlrWrr\r7r8rbs  z.TestBoxingCallingJIT.test_boxer..boxercSs|SrCr7rbr7r7r8rc sz1TestBoxingCallingJIT.test_boxer..passthrurd)r>r^rbrcrr7rer8 test_boxers     zTestBoxingCallingJIT.test_boxerc Csft|jdd}t|jdd}tdd}|t}||jW5QRX|dt|j dS)NcSst|jSrCrNrPr7r7r8r^sz6TestBoxingCallingJIT.test_boxer_raise..unboxerc Ssdd}|jtjdg}ttjtj}|j|||\}}tj|j |jj dd}|j |j | |j |} |j | |W5QRX|j |S)NcSs|dkrtd|S)Nrcannot do x > 0rhrbr7r7r8rUszDTestBoxingCallingJIT.test_boxer_raise..boxer..bridger/T)zfill)r?rJrrYr+rwrZr Z alloca_oncerKZpyobjZif_thennot_r`storeload) rQrFrGrUrmrlrWrretvalrRr7r7r8rbs z4TestBoxingCallingJIT.test_boxer_raise..boxercSs|SrCr7rbr7r7r8rc'sz7TestBoxingCallingJIT.test_boxer_raise..passthrurnrirjr7r7r8test_boxer_raises    z%TestBoxingCallingJIT.test_boxer_raise) r4r5r6rJrfrkrmrtrr7r7rr8rIs  *%rIcCsdSrCr7rbr7r7r8with_objmode_cache_ov_example2sruc@sXeZdZdZdZddZddZddZed d Z ed d Z ed dZ ddZ dS)TestCachingOverloadObjmodezTTest caching of the use of overload implementations that use `with objmode` FcCstdtjdS)Nerror)warnings simplefilterr Z NumbaWarningrr7r7r8rJ?sz TestCachingOverloadObjmode.setUpcCs tdSrC)rx resetwarningsrr7r7r8tearDownBsz#TestCachingOverloadObjmode.tearDownc st|jj}td|ddfddttfdd}tdd d d }d }|d }|||tdd |j}|d }|||W5QRXdS) NrcSst||}tj|SrC)r}rZlinalgZnorm)rcrr7r7r8realworkIszJTestCachingOverloadObjmode.test_caching_overload_objmode..realworkcs|SrCr7rb)r|r7r8 python_codeNszMTestCachingOverloadObjmode.test_caching_overload_objmode..python_codecsfdd}|S)Nc s"tdd|}W5QRX|SNr)r{rrr}r7r8rqTs zqTestCachingOverloadObjmode.test_caching_overload_objmode.._ov_with_objmode_cache_ov_example..implr7rrr7r8!_ov_with_objmode_cache_ov_exampleRs zcTestCachingOverloadObjmode.test_caching_overload_objmode.._ov_with_objmode_cache_ov_exampleTrcSst|SrC)rurbr7r7r8testcase[szJTestCachingOverloadObjmode.test_caching_overload_objmode..testcaser) rrr4rrrurr-py_func)r>rrrrjr5Ztestcase_cachedr7)r}r|r8test_caching_overload_objmodeEs     z8TestCachingOverloadObjmode.test_caching_overload_objmodecs\ddddtfdd}tddfd d }|}|}||krXtd |S) NcSst||SrCr}sumrr7r7r8do_thisiszGTestCachingOverloadObjmode.check_objmode_cache_ndarray..do_thiscSst||SrCrrr7r7r8 do_somethinglszLTestCachingOverloadObjmode.check_objmode_cache_ndarray..do_somethingcsfdd}|S)Nc s$tdd||}W5QRX|Sr~r)rrnr{rr7r8_do_something_implqs zqTestCachingOverloadObjmode.check_objmode_cache_ndarray..overload_do_something.._do_something_implr7)rrnrrr7r8overload_do_somethingos zUTestCachingOverloadObjmode.check_objmode_cache_ndarray..overload_do_somethingTrcstd}td}||S)Nrrr)rr7r8 test_cachingws  zLTestCachingOverloadObjmode.check_objmode_cache_ndarray..test_cachingzincorrect result)rrrr)rrrr5rjr7)rrr8check_objmode_cache_ndarraygsz6TestCachingOverloadObjmode.check_objmode_cache_ndarraycCs |dSrC)r)rr7r7r8*populate_objmode_cache_ndarray_check_cacheszETestCachingOverloadObjmode.populate_objmode_cache_ndarray_check_cachecCs<|}t|jjdkr tdt|jjdkr8tddS)Nrzunexpected cache misszunexpected missing cache hit)rrstatsZ cache_missesrZ cache_hits)rZdispr7r7r8'check_objmode_cache_ndarray_check_caches zBTestCachingOverloadObjmode.check_objmode_cache_ndarray_check_cachec CsNt|jj}td|t|j|t|j|}W5QRX||dddS)Nrrr)rrr4rrrrr-)r>rrr7r7r8 test_check_objmode_cache_ndarrays  z;TestCachingOverloadObjmode.test_check_objmode_cache_ndarrayN) r4r5r6r%rrJr{r classmethodrrrrr7r7r7r8rv8s"   rvc@seZdZddZddZdS)TestMisccCsndd}|t||tt||tt||ttdd||ttdd|dS)NcSsdSrCr7rbr7r7r8rusz$TestMisc.test_is_jitted..fooT)parallelzvoid(float64[:])z(m))rr(rrrr)r>rur7r7r8test_is_jittedszTestMisc.test_is_jittedcCs@tdd}tdd}||||||dS)NcSstjdddjS)Nr)r fill_valuer}fullrr7r7r7r8standard_ordersz?TestMisc.test_overload_glue_arg_binding..standard_ordercSstjdddjS)Nrr)rrrr7r7r7r8reversed_ordersz?TestMisc.test_overload_glue_arg_binding..reversed_order)rrr)r>rrr7r7r8test_overload_glue_arg_bindings   z'TestMisc.test_overload_glue_arg_bindingN)r4r5r6rrr7r7r7r8rs rc@seZdZddZddZdS)TestOverloadPreferLiteralcsdddddd}tdd|t|tfd d }|d \}}}||d ||d ||dtfdd}|d \}}}||d||d ||ddS)NcSsdSrCr7rbr7r7r8 prefer_litsz;TestOverloadPreferLiteral.test_overload..prefer_litcSsdSrCr7rbr7r7r8non_litsz8TestOverloadPreferLiteral.test_overload..non_litcSs>t|tjr.|jdkr"dd}|Stdn dd}|SdS)Nr/cSsdSrr7rbr7r7r8rqszATestOverloadPreferLiteral.test_overload..ov..impl literal valuecSs|dSNdr7rbr7r7r8rqsr;rZIntegerLiteralZ literal_valuer rrr7r7r8ovs   z3TestOverloadPreferLiteral.test_overload..ovTZprefer_literalcsdd|fSNr/ror7rb)rr7r8check_prefer_litszATestOverloadPreferLiteral.test_overload..check_prefer_litr'r,csdd|fSrr7rb)rr7r8 check_non_litsz>TestOverloadPreferLiteral.test_overload..check_non_litr)rrr-r>rrrrnrGrr7)rrr8 test_overloads"      z'TestOverloadPreferLiteral.test_overloadcCsdd}ttddd|ttddd|tdd }|td \}}}||d ||d ||d tdd}|td \}}}||d||d ||d dS)NcSs>t|tjr.|jdkr"dd}|Stdn dd}|SdS)Nr/cSsdSrr7r>rcr7r7r8rqszHTestOverloadPreferLiteral.test_overload_method..ov..implrcSs|dSrr7rr7r7r8rqsr)r>rcrqr7r7r8rs   z:TestOverloadPreferLiteral.test_overload_method..ovmethod_prefer_literalTrmethod_non_literalFcSs|d|d||fSr)rr>rcr7r7r8rszHTestOverloadPreferLiteral.test_overload_method..check_prefer_litr'rrrcSs|d|d||fSr)rrr7r7r8rszETestOverloadPreferLiteral.test_overload_method..check_non_litrrrr7r7r8test_overload_methods6       z.TestOverloadPreferLiteral.test_overload_methodN)r4r5r6rrr7r7r7r8rs*r__main__)N)rprrSr rrrxZdistutils.versionrr+Znumpyr}ZnumbarrrrrZ numba.corerr r r r Znumba.core.typed_passesr Znumba.core.registryrZnumba.core.compilerrZnumba.core.imputilsrZnumba.tests.supportrrrrrrZnumba.core.errorsrr/Znumba.extendingrrrrrrrrr r!r"r#r$r%r&r'r(r)Znumba.core.typing.templatesr*r+r,r-r.Zpdlike_usecaser0r1Zscipy __version__r0Zscipy.special.cython_specialspecialZcython_specialr*rXr2rYr9rDrBrZrHr<rLrMr[rSr]rrYZmydummy_type_2r_rdrirrernrgrsrtrxr|r~rrrrrrrdelitemrgetitemrsetitemrrriaddrrrrrrrrrrrrrrrrrrrrrrrrrrrCallablerrrrr r r&rGrrrrrr#r4r6r8r:r;rIrurvrrr4mainr7r7r7r8sF       P              1         "  4a ( Y#     @h_