i!ddlmZddlZddlmZmZmZddlZddlZddl m Z m Z m Z m Z mZddlmZmZddlmZmZddlmZddlZdZd Zd Zd ZGd d eeZedk(rej<yy))partialN)chainproductstarmap)jit literallynjittypeof TypingError)utilstypes)TestCaseMemoryLeakMixin) _header_leadcH|j|j|jfSN)startstopstep)sls l/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/numba/tests/test_slices.py slice_passingrs 88RWWbgg %%cXt|}|j|j|jfSr)slicerrr)argsrs rslice_constructorrs$ B 88RWWbgg %%rct|}||Sr)r)rlrs rslice_construct_and_user s B R5Lrc |j|Sr)indices)sindargss r slice_indicesr%s 199g rcHeZdZdZdZdZdZdZdZdZ dZ d Z d Z y ) TestSlicesc  fd}tj}| dz }tdt gd}d|fddg}d d g}t j |||D]\\}}\} } \} } ||| | || | d|fd d g}d|fddg}d d g}t j |||D]\\}}\} } \} } ||| | || | j t5 tddddddy#1swYyxYw)zC Check passing a slice object to a Numba function. cZt|||}|} j||||fyr)rassertPreciseEqual) abcdefrgotcfuncselfs rcheckz,TestSlices.test_slice_passing..check$s/q!QB)C  # #C!Q 3rTnopython))Nr*r9r;N) r<)r=Nr5) r?r8r:)r@g?) sysmaxsizerr itertoolsr assertRaises TypeErrorr)r3r4 maxposint maxnegint start_cases stop_cases step_casesr+r.r,r/r-r0r2s` @rtest_slice_passingzTestSlices.test_slice_passings8  4 KK JN "T"=16 Y'< * &/&7&7 8B8B'D "FQFQFQ !Q1a #'D i((H= Y'<  + &/&7&7 8B8B'D "FQFQFQ !Q1a #'D   y ) %Q" #* ) )s C44C=ctj}| dz }tjd}t dt }t dt }dd|dffddd|dffd d|dffd d d d|d ffddd|d ffdd||dffdd|dffd|ddffdfD]C\}}||}|j||t ||} |||} |j| | Ey)zQ Test the 'happy path' for slice() constructor in nopython mode. r5 Tr6rr)))rrNr5)NN)r5N))N)rrOr5))r5rO)r5rOr5)NNrP))NrOrP)rrOrP)r5NrP)r5rOrPrQ)NNr;r;)rMNr;)NrNr;rN)rMrNr;rRN)rArBnparangerrr r*) r3rFrGr+r2 cfunc_userexpectedr1usagecusages rtest_slice_constructorz!TestSlices.test_slice_constructorCs&KK JN IIbM"T"#45&C&'>? q)Q' (  Ay!, - Iq) * "  q)Q/ 0 % Ay!, - "  9b9 : b)R0 1 Y2. / & ND( ,C  # #C 2+D!4EtQ'F  # #E6 2) rc d tj fdtdD}tjd}t dt }|j|D](} t ||}|j||||*y#t$r}t|}|jt|dj||jt5}|||dddn #1swYnxYw|j!t"tj$|j!d j'd |Dt|j$Yd}~d}~wt($rr} |jt+| 5}|||dddn #1swYnxYw|j!t| tj$Yd} ~ d} ~ wwxYw) zW Test that slice constructor behaves same in python and compiled code. )Nr;rr5c38K|]}t|yw))repeatN)r).0noptionss r z:TestSlices.test_slice_constructor_cases..gs( 081GGA & &srNrMTr6z7slice expected at (most|least) (3|1) arguments?, got {}Nz, c3DK|]}tt|ywr)strr )r]args rr`z:TestSlices.test_slice_constructor_cases..s?$3c&+.$s )r from_iterablerangerSrTrr disable_leak_checkr*rElen assertRegexrbformatrDr assertInr exceptionjoin Exceptiontype) r3 arg_casesarrayr2rrV py_type_en_argsnumba_epy_er_s @rtest_slice_constructor_casesz'TestSlices.test_slice_constructor_casesbs#''( 05a(   " "T"#:; !D F24?8''%e2DE= T   NNVF^ &&{3w$&433  ))* II?$??))* &&tDz2g$&322 I))* sV, B G.A E0' C:1 E0:D ?A+E00 G.<G) F) G))F2 .5G))G.c dttttt dddtt dddtt ddd}t dd}t dt }t||D]+\}} t ||}|j||||-y#t$rq}|jt|5}|||dddn #1swYnxYw|jt|tjYd}~d}~wwxYw) zITest that a numba slice returns same result for .indices as a python one.rNrrPTr6N)rrrrrerr%r*rmrDrnrjrbrk) r3sliceslengthsr2r#rrVrtrss rtest_slice_indiceszTestSlices.test_slice_indicess  eBlG,eBlG,eBlG,  A,"T"=1FG,DAq ?(A.''%1+>- &&tDz2g!QK322 I))* s0 B55 D/>D* C+" D*+C4 05D**D/ctdt}|jt5}|t ddddddd|j dt j|jt5}|t dddd ddd|j d td zt |jy#1swYxYw#1swYGxYw) zTests for specific error cases.Tr6Nr5rOrPz.indices() takes exactly one argument (3 given)rg333333?z/'%s' object cannot be interpreted as an integer) rr%rDr rrjrbrkr )r3cslice_indicesr/s rtest_slice_indices_examplesz&TestSlices.test_slice_indices_exampless+d+M:   { +q 5;1a 0, =     { +q 5tQ/ 5, =s K   , +, +sCCCC'cdtjgdgdgdD]Dttdfd}|j ||j Fy)N)r5rOrP)Nr5rOrP)Nr5rOr;rxTr6cSr) test_tupletssrtest_fnz4TestSlices.test_slice_from_constant..test_fns !"~%r)rCrrr assertEqualpy_func)r3rrrs @@rtest_slice_from_constantz#TestSlices.test_slice_from_constantsb! ## _.B BB $  &  &   WY(9 : rcHtjjtddd}tjjtddd}tjjtddd}|j |||j ||yr>)r misc SliceLiteralrassertNotEqualr)r3sl1sl2sl3s rtest_literal_slice_distinctz&TestSlices.test_literal_slice_distinctsyjj%%eAtT&:;jj%%eD$&=>jj%%eAtT&:; C% c"rc td}tdddtddtdtdddtdddtdddtdtdddf}|D]}|j|||y)Nct|Srr)xs rr0z/TestSlices.test_literal_slice_boxing..f Q< rr5rrO)r rr)r3r0ryrs rtest_literal_slice_boxingz$TestSlices.test_literal_slice_boxings     !QN !QK !H $1  !T1  $a $K $d #  B   R2 'rcjtdddtfd}|j|y)Nr5rOrPcSrrzsrfooz2TestSlices.test_literal_slice_freevar..foosHr)rr r)r3rrs @rtest_literal_slice_freevarz%TestSlices.test_literal_slice_freevars8 !QN     CE"rc td}td}|jt5}|t dd| dz ddd|j dt jy#1swY/xYw)Nct|Srrrs rrz1TestSlices.test_literal_slice_maxint..foorrlr5zInt value is too large)r intrD ValueErrorrrjrbrk)r3rmaxvalr/s rtest_literal_slice_maxintz$TestSlices.test_literal_slice_maxintss    U   z *a dD6'!), -+ $   + *s A77BN) __name__ __module__ __qualname__rKrYrur{r~rrrrrrrrr'r's7"$H3>+FZ?6 $ ;#(*# rr'__main__) functoolsrrCrrrrAnumpyrSnumbarrr r r numba.corer r numba.tests.supportrrnumba.core.types.functionsrunittestrrr r%r'rmainrrrrsh-- ;;#93&&b (b J zHMMOr