o i@s<ddlZddlZddlZddlZddlZddlZddlmZm Z ddl m Z m Z ddl mZmZmZddlZddlmZddlmZddZd d Zd d Zd dZddZGddde ej ZGddde ZGdddeZGddde ZGdddeZGdddeZ Gddde ej Z!e"dkre#dSdS) N)njit vectorize)MemoryLeakMixinTestCase) TypingErrorNumbaNotImplementedErrorNumbaExperimentalFeatureWarning)dufunc) from_dtypecCs||SNa0a1r r ^/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/numba/tests/npyufunc/test_dufunc.pypyuaddrcCs||Sr r r r r rpysubrrcCs||Sr r r r r rpymultrrcCs||Sr r r r r rpydivrrcCs||kr|S|Sr r r r r rpymin"rc@sDeZdZddZddZddZddZd d Zd d Zd dZ dS) TestDUFunccCstj|tdddS)NT)nopython) targetoptions)r DUFuncdict)selfpyfuncr r rnopython_dufunc(szTestDUFunc.nopython_dufunccCs|t}||jd|_||j|t d|_Wdn1s'wY|t|t dddt dddWddS1sLwYdS)NTFr ) rr assertFalse_frozen assertTrue assertRaises ValueError TypeErrornplinspacerduaddr r r test_frozen+s      "zTestDUFunc.test_frozencCs&|t}|tdd|dddS)Nr r")rr assertEqualr+r r r test_scalar5s zTestDUFunc.test_scalarc s|ttfdd}tddd}|dd}|dd}td}||||tj||||d}|d}td}||||tj||||dd} td} ||| | tj|| | dS) Ncs|||dSr r )rro0r,r rnpmadd<z(TestDUFunc.test_npm_call..npmaddrffffff?r!r"r7) rrrr)r*zerostestingassert_array_equalreshape) rr2XX0X1out0Y0Y1out1Y2out2r r1r test_npm_call9s$            zTestDUFunc.test_npm_callc s|ttfdd}tddd}|dd}|dd}|||}tj||||d}|d}|||}tj||||dd} ||| } tj|| | |d d } || d dS) Nc ||Sr r r r1r rr2R z8TestDUFunc.test_npm_call_implicit_output..npmaddrr4r5r!r6r7?@g@) rrrr)r*r9r:r;r.) rr2r<r=r>r?r@rArBrCrDout3r r1rtest_npm_call_implicit_outputOs"          z(TestDUFunc.test_npm_call_implicit_outputcCs|t}||jd||jd||j|j|j||jd||jg||jd|dd||jd||jt |j| |j dS)Nr"r r) rrr.ninnoutnargsntypestypesidentitylen assertIsNone signaturer+r r rtest_ufunc_propsds  zTestDUFunc.test_ufunc_propscsr|tddjjjjjd}fdd}|D]\}}t||}| ||d|q!dS)Nr r")rLrMrNrQrTcs.d|d}i}tt|di||dS)Nz> def impl(): return duadd.z r,impl)exectextwrapdedent)attrfnlr1r r get_attr_fn}s z4TestDUFunc.test_ufunc_props_jit..get_attr_fnz!Attribute differs from original: ) rrrLrMrNrQrTitemsrr.)r attributesr]rZvalcfuncr r1rtest_ufunc_props_jitqs     zTestDUFunc.test_ufunc_props_jitN) __name__ __module__ __qualname__rr-r/rErKrUrbr r r rr&s  rc@s4eZdZejd ddZddZddZdd ZdS) TestDUFuncMethodsBaseNcs|dvsJ|dkr1jdkrt|dfddn t|dfddtdfd d }|S|d krTjdkr>td t|dfd dtdfdd }|Sjdkret|dfddn t|dfddtfdd }|S)N)reducereduceatatrgr"rQcrFr r abufuncr r z5TestDUFuncMethodsBase._generate_jit..c|Sr r rlrnr rrprcj|||dSNaxisinitialrgarrayrxryvecr rr[r3z/TestDUFuncMethodsBase._generate_jit..fnrh,reduceat only supported for binary functionscrFr r rkrnr rrprqcs|||||Sr rh)r|indicesrxdtypeoutr}r rr[scrFr r rkrnr rrprqcrrr r rsrnr rrprtcs j|Sr ri)argsr}r rr[rGrN)rNN)rLrrr')rrokindrQr[r ror~r _generate_jits*    z#TestDUFuncMethodsBase._generate_jitcC|j|d|dS)NrgrjrrrorQr r r_reducerzTestDUFuncMethodsBase._reducecCr)Nrhrjrrr r r _reduceatrzTestDUFuncMethodsBase._reduceatcCs ||dS)Nrir)rror r r_ats zTestDUFuncMethodsBase._atr ) rcrdre functoolscacherrrrr r r rrfs    rfc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZejddZdd Zd!d"Zd#d$Zd%d&Zejd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3S)4 TestDUFuncAtcGsB|}|}|j|g|R||g|R|||dSr )copyriassertPreciseEqual)rr[rorlrexpectedgotr r r_compare_outputs zTestDUFuncAt._compare_outputcsxtjdtd}|tj}|tj}tddtfdd}||tj|gddd }| t ||| gd dWdn1sKwY||tj| gd t gd }||tj| gd|d }| t ||| gd gd Wdn1swY| t || gd ddgdgWddS1swYdS)Nr!rcS t|Sr r)negativersr r rrprqz8TestDUFuncAt.test_numpy_ufunc_at_basic..cs|||Sr r)rlrrm negative_vecr rnegative_jit_2z>TestDUFuncAt.test_numpy_ufunc_at_basic..negative_jit_2r"r7r"r zsecond operand needed for ufunc)r"r7drrz+second operand provided when ufunc is unaryr r"rr")r)arangeintraddrrrrassertRaisesRegexrrr|r&)rrladd_at negative_atrerr_msgrmr rrtest_numpy_ufunc_at_basics*   "z&TestDUFuncAt.test_numpy_ufunc_at_basicc Cs*tjd}tjtjtjf}|D]}}z tt|Wn ty#Yqw|D]e}tjd|d}t tjdtj dtj ddtj dg}|j d}| |}||||t|} |tjurgd| d <| d<q&| d d|tjurs|n| 7<| dd|tjur|n| 7<q&q||| dS) NComplexr!ry?r )r) typecodesrsubtractmultiplyr rrones concatenateintpfulltyper ones_liker) rrufuncstypecoderorlindxvalueufunc_atrr r rtest_ufunc_at_inner_loops.       $&z%TestDUFuncAt.test_ufunc_at_inner_loopc Cs@tjdtd}|tj}||tj|tdtjdtddS)Nr7r)r)r8rrrrslicerrarrrr r rtest_ufunc_at_ellipsiss   z#TestDUFuncAt.test_ufunc_at_ellipsiscCsJtjdtjd}td}|tj}|||t|gdks#JdS)Nr7r)rrrrr)r)rint32rrrall)rrrrir r rtest_ufunc_at_negatives    z#TestDUFuncAt.test_ufunc_at_negativecCsxtjdtjd}tjdtd}d|d<d|d<d|dd<tjdtd}|tj}|||||d|ks:JdS) Ni rr!rr7r ri )r)r8int16floatrrsum)rrrmrlrr r rtest_ufunc_at_larges   z TestDUFuncAt.test_ufunc_at_largecCsXtd}td}|tj}tjt|tjd}|||||dt|ks*JdS)Nr!irr)r)r8rrrrRuint8)rrvaluesrindexr r rtest_cast_index_fastpath!s    z%TestDUFuncAt.test_cast_index_fastpathc Csztdtdtddf}|D])}td}ttdd}|tj}||||tj |t |d|qdS)Nr r rHir") r)rfloat64r8repeatrrrr9r: full_like)rrrrrrr r r#test_ufunc_at_scalar_value_fastpath*s   z0TestDUFuncAt.test_ufunc_at_scalar_value_fastpathc Cstddd}tgdgdgdg}|tj}||tdtgdf|||tgdgdgd gtd ddd}tgd }||tdtdtgdf|||tgd gdgd ggdgdgdggdgdgdggtddd}tgdgdgdg}||tgdtdf|||tgdgdgdgtd ddd}tgd }||tdtgdtdf|||tgdgdgdggdgdgdggdgdgdggtddd}tgd }||dtgdf|||tgd gd gd!gtd ddd}tgd }||tgddtdf|||tgdgd gd!ggd"gd#gd$ggd%gd&gd'ggtd ddd}tgd }|tj}||tdtdtdf|||tgd(gd)gdggd*gd+gdggd%gd,gdggdS)-N rr)rr),rrr r"r )rf)r)ri_4)rrr)ri)ri)r) )i)ri))irr r")iri)r)ri])jrr)rr! )rif)si=)rr5)rio)|iFrrrr7)r)ric)r )r)vi@)r)r)rri.)gi1)mi7)pi:)yiC) r)rr;r|rrrasarrayr)rrlrmrr r rtest_ufunc_at_multiD4s  $   $         z!TestDUFuncAt.test_ufunc_at_multiDcCstd}|tj}||dd||td|t||ddWdn1s0wYtd}||dd||tgd|t||gdWddS1sbwYdS)Nrr r r)r r r") r)r|rrrr&rrr')rrlrrmr r rtest_ufunc_at_0Ds       "zTestDUFuncAt.test_ufunc_at_0DcCs@td}|tj}||gdd||tgddS)Nr!)r r"rr"g @) rr i>.rr7rrrr)r)rrpowerrr|)rrlpower_atr r rtest_ufunc_at_dtypess  z!TestDUFuncAt.test_ufunc_at_dtypescCsLtd}|ddk}|tj}|||gd||tgddS)Nr!r"r)rr"rrr) r r r rr r7r rr r)r)rrequalrr|)rrlrequal_atr r rtest_ufunc_at_booleans   z"TestDUFuncAt.test_ufunc_at_booleancCsHtjddd}|tj}||gd||tjgdtjddS)Nr!u4rr) rr r"rrlrrrr)r)rrinvertrr|uint32)rrl invert_atr r rtest_ufunc_at_boolean2s   z#TestDUFuncAt.test_ufunc_at_boolean2cCs\td}|dddfddddf}|tj}||ddgd||tddS)Nrrr r)r)rrrr)rorigrlrr r rtest_ufunc_at_advanceds   z#TestDUFuncAt.test_ufunc_at_advancedcCs`tgdtd}tgdtd}|tj}|||d||gddS)Nrir r"rrfr)r rrr)r)r|r newbyteorderrrr)rrrrr r rtest_ufunc_at_advanced_2s   z%TestDUFuncAt.test_ufunc_at_advanced_2cCsvtjddgtd}|tj}|t||ddgdWdn1s'wY||tjddgtddS)Nrlr rr)r)r|objectrrr&rr)rrrr r rtest_ufunc_at_advanced_3s   z%TestDUFuncAt.test_ufunc_at_advanced_3cCsN|tj}|t|tddgWddS1s wYdS)Nr!r )rr)modfr&rr)rmodf_atr r rtest_ufunc_at_advanced_4s  "z%TestDUFuncAt.test_ufunc_at_advanced_4cCsB|tj}tgd}||dgd||tgddS)Nrr)rr)maximumr|r)r maximum_atrlr r rtest_ufunc_at_advanced_5s z%TestDUFuncAt.test_ufunc_at_advanced_5c Cstjdtjd}tjtjtjtjtjf}|D]q}|dvrqz tt|Wn t y1Yqw|D]T}t dd |}t gd tj }tj gd|jd}|}t||D] \} } ||| | || <q]||} | |||tj||t|gdksJq4qdS) N AllIntegerFloat)err!)rr rr")r r7r"r!r)r)rrrdivideminimumr+r rrrastyper|rrziprr9r:r) rdtypesrrrorlindxsvalsrr!vrr r rtest_ufunc_at_negative_indexess.   z+TestDUFuncAt.test_ufunc_at_negative_indexescCsdtd}td}|tj}d}|t|||dg|WddS1s+wYdS)N)r"r"r")r r"r"z1does not support ufunc with non-trivial signaturer)r)rrmatmulrr)rrlrm matmul_atrr r r test_ufunc_at_not_none_signatures   "z-TestDUFuncAt.test_ufunc_at_not_none_signaturecCs^tjdtd}|tj}|t||ddgddgWddS1s(wYdS)Nr!rrr )r)rstrrrr&rrr r rtest_ufunc_at_no_loop_for_ops   "z)TestDUFuncAt.test_ufunc_at_no_loop_for_opcCs<tdg}|tj}||dgdg|ddksJdS)Nrr)r)r|rr)rrrr r rtest_ufunc_at_output_casting s  z)TestDUFuncAt.test_ufunc_at_output_castingcCs`td}|tj}d}|t|||ddggdWddS1s)wYdS)Nr7z=operands could not be broadcast together with remapped shapesrr r)r)rrrrr')rrrmsgr r rtest_ufunc_at_broadcast_failures  "z,TestDUFuncAt.test_ufunc_at_broadcast_failurecCsNtd}tdd}|t|jd||ddg|t|jddS)Nr7cSs|dS)Nr r xr r rincz/TestDUFuncAt.test_ufunc_at_dynamic..incrr r)r)rrr.rRrPri assertGreater)rrrDr r rtest_ufunc_at_dynamics  z"TestDUFuncAt.test_ufunc_at_dynamiccCstd}|tj}tjdd}tdt||ddgdWdn1s*wY|t |d| dt |dj dS) Nr7T)recordalwaysrrr!z ufunc.at feature is experimental) r)rrrwarningscatch_warnings simplefilterrrFrRassertInr=message)rrrwr r r"test_ufunc_at_experimental_warning)s   z/TestDUFuncAt.test_ufunc_at_experimental_warningN)rcrdrerrrrrrrrrrrrrr unittestexpectedFailurer%r'r*r-r9r<r>r?rArGrPr r r rrs8$   U     rc@steZdZdddZejddZddZdd Zd d Z d d Z ddZ ddZ ddZ ddZddZddZdS)TestDUFuncReduceNumPyTestsNcsNjdkrt|dfddn t|dfddtd fdd }|S) Nr"rjcrFr r rkrnr rrp<rqz:TestDUFuncReduceNumPyTests._generate_jit..crrr r rsrnr rrp>rtrcrurvrzr{r}r rr[@r3z4TestDUFuncReduceNumPyTests._generate_jit..fnr)rLrr)rrorQr[r rrr:s z(TestDUFuncReduceNumPyTests._generate_jitcCs0|tj}tjd}|d}|||dS)Nr)rr)rrgr)r power_reducerrr r rtest_numpy_scalar_reductionEs  z6TestDUFuncReduceNumPyTests.test_numpy_scalar_reductioncCsdd}|jtjdd}d|d<d|d<|||dd d|||d d gd |||d d gd |||dd ddg|||dd gd gdgdg|||dd gdgd g|||dd gdgd g|||dd |d|d<d|d<|||dd d|||d d gd |||d d gd|||dd ddg|||dd gdgd gdg|||dd gd gdg|||dd gdgdg|||dd |d|d<d|d<|||dd d|||d d gd|||d d gd |||dd ddg|||dd gdgdgdg|||dd gdgdg|||dd gd gdg|||dd |dS)NcSstj||dSr )r)r9 assert_equalrkr r rcompare_outputNszOTestDUFuncReduceNumPyTests.check_identityless_reduction..compare_output reorderablerjr .r)r rrrxrr )rr r r rr")rr r r r")r r r r r")r r r r rr r)r rr )rrr )r rr r )rr)r2)rrlrWminimum_reducer r rcheck_identityless_reductionMsd         z7TestDUFuncReduceNumPyTests.check_identityless_reductioncCtjddd}||dS)Nr"rrCorderr)emptyr_rrlr r r(test_numpy_identityless_reduction_corderzCTestDUFuncReduceNumPyTests.test_numpy_identityless_reduction_cordercCr`)NraFrcrergr r r(test_numpy_identityless_reduction_forderrizCTestDUFuncReduceNumPyTests.test_numpy_identityless_reduction_fordercCs$tjddddd}||dS)N)r"rrrbrcr r"r)rfswapaxesr_rgr r r,test_numpy_identityless_reduction_otherorderszGTestDUFuncReduceNumPyTests.test_numpy_identityless_reduction_otherordercCs>tjddddd}|ddddddf}||dS)N)rr7rrbrcr r"rlrgr r r+test_numpy_identityless_reduction_noncontigszFTestDUFuncReduceNumPyTests.test_numpy_identityless_reduction_noncontigcCsTtjddd}|ddjdd}|d}|ddddddf}||dS)N)ii1rr f8r)r)rfviewr;r_rgr r r5test_numpy_identityless_reduction_noncontig_unaligneds  zPTestDUFuncReduceNumPyTests.test_numpy_identityless_reduction_noncontig_unalignedc CsB|tj}|tj}|tj}||tgddd||tgtjdtj||tgtj dtj ||tdgddd||tdgddd||tdgddd||tdgdddd}||fD]!}|t ||tgddWdn1swYq}dS)Nrrygr7r&zero-size array to reduction operation) rr)rr2r+rrinfrr')r add_reduce min_reduce max_reducer@funcr r rtest_numpy_initial_reductions*    z7TestDUFuncReduceNumPyTests.test_numpy_initial_reductioncCstd}|tj}tjj|dd}||dd}|||||jdd}|t|||ddWdn1s?wYtd}|t|||ddWdn1s`wYtjj|ddd}||ddd}|||dS) N)rr7r rY)rrur)rrr7rw) r)r8r true_dividergrshaperr')rrtrue_divide_reducerrr@r r r'test_numpy_empty_reduction_and_identitys"     zBTestDUFuncReduceNumPyTests.test_numpy_empty_reduction_and_identitycCstgdgdg}|tj}||dd}||tgd||dd}||tddg||d d}|||dS) N) @rIrI)rHg?g?rrY)rg@rr rIrr )r)r|rr1rr)rrl divide_reduceresr r r*test_identityless_reduction_nonreorderables    zETestDUFuncReduceNumPyTests.test_identityless_reduction_nonreorderablecsdd}fdd}dd}jtjdd}tjjd|||d d |||d d |||d d |||d d ||g|td d d jtjd d}|||d d |||d d |||d d |||d d |tjgtjd|td d d dS)Nc_s||i|dSr r r#rkwargsr r roksz.okcs>t||i|WddS1swYdSr )r&r'rrr rerrs "z=TestDUFuncReduceNumPyTests.test_reduce_zero_axis..errc Ssp||t||fdd||t||fdd||t|d|d|fdd||t|d||dfdd||t||d|dfdd||t|d||dfdd||t|d|d|d|d|dfdd||td ||fd d||td ||fdd||t|d |fdd||td ||fdd||td ||fdddS) Nr rYrr"r\r[rrr!rZ)r)r8)expectrznmr r rts """" z;TestDUFuncReduceNumPyTests.test_reduce_zero_axis..trXrjrrrr rYr) rr)r+r.rQr8rr|int64)rrrrmaximum_reducerwr rrtest_reduce_zero_axiss$  z0TestDUFuncReduceNumPyTests.test_reduce_zero_axisr )rcrdrerrQrRrUr_rhrkrnrorsr{rrrr r r rrS6s  7 rSc@speZdZdddZdddZddZdd Zd d Zd d ZddZ ddZ ddZ ddZ ddZ ddZdS)TestDUFuncReduceNc stfdd}tdtdddtddddg}|D]}t|jD]}|||||}|||||} ||| q(q!dS)Ncsj||||dS)N)rxrryrz)rlrxrryrnr rfoo s z+TestDUFuncReduce._check_reduce..foor7rr"()rr)rr;rangendimpy_funcr) rrorryrinputsr|rxrrr rnr _check_reduceszTestDUFuncReduce._check_reducecstfddfdd}djdtjd|dd d d tjd |dg}|D])td jd D]}tjtj|dD]|qEq9dD]|qPq/dS)Ncsj||dSrvrzrlrx)ryror rr4r3z0TestDUFuncReduce._check_reduce_axis..fooc sz}Wn>tyF}z2|jdt }Wdn1s/wYWYd}~dSWYd}~dSd}~ww}||dS)Nr)rr'r.rrrr)rrr0r)r|rxexc_msgrrr r_check8s 8 z3TestDUFuncReduce._check_reduce_axis.._checkzreduction operation 'z:' is not reorderable, so at most one axis may be specifiedrrr7rr"r!r )r)r N) rrcr)rr;rr itertools combinations)rrorryrrr!r )r|rxrrryrror_check_reduce_axis2s     z#TestDUFuncReduce._check_reduce_axiscCs.tdddt}|||j|tjddS)Nint64(int64, int64)rrjr)rrrrr)rr+r r rtest_add_reduceRs z TestDUFuncReduce.test_add_reducecCstdddt}||dS)Nrr rj)rrr)rdumulr r rtest_mul_reduceWsz TestDUFuncReduce.test_mul_reducecCsXtdddt}tdddt}|||j|tjd|||j|tjddS)Nrrjr)rrrrrr)r)rdusubdudivr r rtest_non_associative_reduce[s   z,TestDUFuncReduce.test_non_associative_reducecCs$tdddt}|j|tjddS)Nfloat64(float64, int64)rrjr)rrrr)rr+r r rtest_reduce_dtypecsz"TestDUFuncReduce.test_reduce_dtypecCs2tdddt}|j|dd|j|tjddS)NrrXrjr!rtr)rrrrr)r)rduminr r rtest_min_reducegsz TestDUFuncReduce.test_min_reducecCs&tdddt}|j|tjdddS)Nrrrjr)rryrrrr)rr+r r rtest_add_reduce_initiallsz(TestDUFuncReduce.test_add_reduce_initialcCs tdt}|j|tjddS)Nrrrr+r r r&test_add_reduce_no_initial_or_identityqs z7TestDUFuncReduce.test_add_reduce_no_initial_or_identitycs^tdddttfdd}d}|t| |dWddS1s(wYdS)Nrrrjcs |Sr rzrsr1r rryrGz0TestDUFuncReduce.test_invalid_input..fooz-The first argument "array" must be array-likerl)rrrrr)rrrr r1rtest_invalid_inputvs "z#TestDUFuncReduce.test_invalid_inputcshtdddttfdd}tdddd }d }|D]}||}|||}|||qdS) Nrrrjcj||dSNrYrzrr1r rrrz7TestDUFuncReduce.test_dufunc_negative_axis..foorr7rr")rr)rr)r)r r)rrrr)rr;rgr)rrrlcasesrxrrr r1rtest_dufunc_negative_axiss  z*TestDUFuncReduce.test_dufunc_negative_axisc stdddttfdd}tdddd }d }|D]}d }|t| |||Wdn1s8wYqd }|D]}|td  |||Wdn1sZwYqBdS)Nrrrjcrrrzrr1r rrrz6TestDUFuncReduce.test_dufunc_invalid_axis..foorr7rr")rr])rr)rr)rr"zduplicate value in 'axis')r)rrz Invalid axis)rrrr)rr;rr')rrrlrrxr@r r1rtest_dufunc_invalid_axiss$  z)TestDUFuncReduce.test_dufunc_invalid_axis)NNr )rcrdrerrrrrrrrrrrrr r r rrs    rc@seZdZddZddZejddZddZd d Z d d Z d dZ ddZ ddZ ddZddZddZddZddZdS)TestDUFuncReduceAtc Ks\t|d}|||}|}|}|j||fi||||fi||||dS)NrQ)getattrrrrhr) rrorlidxrrQr[rrr r rrs  z"TestDUFuncReduceAt._compare_outputcCsrtd}tgd}|tjd}||||d|tgd|||||d|tgd|dS)Nrrrr r"rr)rrrr)rrrr)r)rrrrr)rrrrwr r rtest_reduceat_out_kws z'TestDUFuncReduceAt.test_reduceat_out_kwc Cstdddtddddtdf}tgdtgdf}d }|D]}|D]}|D] }|jtj|||d q1q-q)dS) Nrrrr7r"rrrrr r"rr)r rrrY)r)rr;rrrr)rarraysrrxr|raxr r rtest_reduceat_axis_kws  z(TestDUFuncReduceAt.test_reduceat_axis_kwc Cs~td}tgd}|tjd}dD]'}d|d|j}|t|||||Wdn1s7wYqdS)Nrrr)r"rzaxis z) is out of bounds for array of dimension )r)rrrrrrr')rrr add_reduceatrr@r r rtest_reduceat_invalid_axiss z-TestDUFuncReduceAt.test_reduceat_invalid_axiscCsr|tjd}gd}gd}tj||}|||}|||d}d}tj||}|||}|||dS)Nrr"r)rr)rrhr)rrrlrrrr r r test_reduceat_cast_args_to_arrays   z3TestDUFuncReduceAt.test_reduceat_cast_args_to_arraycCs&td}gd}|tj||dS)Nr)rrr r7r"rrr)r)rrrrrCrr r rtest_reduceat_basics z&TestDUFuncReduceAt.test_reduceat_basiccCs2tddddd}gd}|tj||dS)Nrrrrr)r)r*r;rrrr r rtest_reduceat_basic_2dsz)TestDUFuncReduceAt.test_reduceat_basic_2dc CsRd}t|}tttt|dt|dd}|tj||dS)Nrr")r)rr|listr4ravelrr)rLrCrr r rtest_reduceat_shifting_sums 0z-TestDUFuncReduceAt.test_reduceat_shifting_sumcCstjdtjd}tjdtjd}|jtj|td|d|jtj|td|d|||tjdtjd}ddtdD|dddf<tjdtjd}ddtdD|dddf<|jtj|td|d d |jtj|td|d d |||dS) Nrrr)r"rcSg|]}dqSr"r .0r!r r r zOTestDUFuncReduceAt.test_reduceat_int_array_reduceat_inplace..rcSrrr rr r rrrr)rrx)r)rrrrrrr)rrrr r r(test_reduceat_int_array_reduceat_inplaces z;TestDUFuncReduceAt.test_reduceat_int_array_reduceat_inplacec Cs||tjd}dD]2}td}td}|r|tj}|t||ddg|dWdn1s6wYq dS)Nr)TFr7rr)rr)rrr3rr&r')rr with_castrrr r r test_reduceat_out_shape_mismatchs    z3TestDUFuncReduceAt.test_reduceat_out_shape_mismatchcCstgd}tgd}|tjd}tj||}|||}|||||j|jtd}tj gtj d}|j tj||dd|j tj||dddS) z&Reduceat should work with empty arraysi4rqr)r7r"rrYr N) r)r|rrrhrr.rrrrr)rrrCrrrrr r rtest_reduceat_emptys     z&TestDUFuncReduceAt.test_reduceat_emptycCsf|tjd}td}tddd}|td|||WddS1s,wYdS)Nrr7r!r"zhave at most 1 dimension)rr)rrr;rr)rrrlrr r rtest_reduceat_error_ndim_2(s   "z-TestDUFuncReduceAt.test_reduceat_error_ndim_2csjtddtfdd}td}d}|t|||gdWddS1s.wYdS)NcSrr rrsr r rneg3rGzGTestDUFuncReduceAt.test_reduceat_error_non_binary_function..negcs ||Sr r)rlrrr r neg_reduceat7s zPTestDUFuncReduceAt.test_reduceat_error_non_binary_function..neg_reduceatr7rr)rrr)rrr)rrrlr@r rr'test_reduceat_error_non_binary_function1s  "z:TestDUFuncReduceAt.test_reduceat_error_non_binary_functioncCs|tjd}|td|dgdWdn1swY|td|tddWdn1s=wY|td|tdgdd d Wdn1s_wY|td |tdgdd d WddS1swYdS)Nrz"array" must be array-likeabcrz"indices" must be array-liker7abcdzoutput must be an arrayr rz"axis" must be an integer)r rY)rr)rrrr)rrr r r"test_reduceat_error_argument_types@s"z5TestDUFuncReduceAt.test_reduceat_error_argument_typesN)rcrdrerrrQrRrrrrrrrrrrrrr r r rrs     rc@s$eZdZddZddZddZdS)TestDUFuncPicklingcsht|}t|d}|d||||tfdd}|d}|d||||dS)N{crrr r rBrebuiltr rr[rEz%TestDUFuncPickling.check..fooiA)pickledumpsloadsr.assertIsInstancer)rident result_typebufrrr rrcheckQs     zTestDUFuncPickling.checkcCs$tdd}|j|ttjfddS)NcS|Sr r x1r r rrdz3TestDUFuncPickling.test_unrestricted..identr)rrrr)integerrrr r rtest_unrestrictedcs z$TestDUFuncPickling.test_unrestrictedcCs$tdgdd}|j|tddS)Nzfloat64(float64)cSrr r rr r rrkrz1TestDUFuncPickling.test_restricted..identr)rrrrr r rtest_restrictedjs z"TestDUFuncPickling.test_restrictedN)rcrdrerrrr r r rrPs r__main__)$rrrrXrJnumpyr)numbarrnumba.tests.supportrrnumba.core.errorsrrrrQnumba.np.ufuncr numba.np.numpy_supportr rrrrrrrfrrSrrrrcmainr r r rs@  m-xh."