g<ddlmZmZddlZddlmZmZddlm Z ddl m Z ddl m Z mZmZddlZdZdZd Zd Zd Zd Zd ZdZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)d Z*d!Z+d"Z,d#Z-d$Z.d%Z/d&Z0d'Z1d(Z2d)Z3d*Z4d+Z5d,Z6d-Z7d.Z8d/Z9d0Z:d1Z;d2Zd5Z?d6Z@Gd7d8ee ZAeABGd9d:ee ZCeCDeEd;krejFdSdS)<)productcombinations_with_replacementN)jittypeof)compile_isolated) numpy_version)TestCaseMemoryLeakMixintagc*|SN)allarrs W/granary/edev/quantum/lib/python3.11/site-packages/numba/tests/test_array_reductions.py array_allr  7799c*tj|Sr )nprrs rarray_all_globalr 6#;;rc*|Sr )anyrs r array_anyrrrc*tj|Sr )rrrs rarray_any_globalrrrc*|Sr )cumprodrs r array_cumprodr s ;;==rc*tj|Sr )rrrs rarray_cumprod_globalr" :c??rc*tj|Sr )r nancumprodrs rarray_nancumprodr&s =  rc*|Sr )cumsumrs r array_cumsumr)! ::<<rc*tj|Sr )rr(rs rarray_cumsum_globalr,$ 9S>>rc*tj|Sr )r nancumsumrs rarray_nancumsumr0' <  rc*|Sr )sumrs r array_sumr4*rrc*tj|Sr )rr3rs rarray_sum_globalr6-rrc*|Sr )prodrs r array_prodr90 88::rc*tj|Sr )rr8rs rarray_prod_globalr<3 73<<rc*|Sr )meanrs r array_meanr@6r:rc*tj|Sr )rr?rs rarray_mean_globalrB9r=rc*|Sr )varrs r array_varrE<rrc*tj|Sr )rrDrs rarray_var_globalrG?rrc*|Sr )stdrs r array_stdrJBrrc*tj|Sr )rrIrs rarray_std_globalrLErrc*|Sr )minrs r array_minrOHrrc*tj|Sr )rrNrs rarray_min_globalrQKrrc*tj|Sr )raminrs r array_aminrTNr=rc*|Sr )maxrs r array_maxrWQrrc*tj|Sr )rrVrs rarray_max_globalrYTrrc*tj|Sr )ramaxrs r array_amaxr\Wr=rc*|Sr argminrs r array_argminr`Zr*rc*tj|Sr rr_rs rarray_argmin_globalrc]r-rc*|Sr argmaxrs r array_argmaxrg`r*rc*tj|Sr rrfrs rarray_argmax_globalrjcr-rc*tj|Sr )rmedianrs rarray_median_globalrmfr-rc*tj|Sr )rnanminrs r array_nanminrpir-rc*tj|Sr )rnanmaxrs r array_nanmaxrslr-rc*tj|Sr )rnanmeanrs r array_nanmeanrvor#rc*tj|Sr )rnansumrs r array_nansumryrr-rc*tj|Sr )rnanprodrs r array_nanprodr|ur#rc*tj|Sr )rnanstdrs r array_nanstdrxr-rc*tj|Sr )rnanvarrs r array_nanvarr{r-rc*tj|Sr )r nanmedianrs rarray_nanmedian_globalr~r1rc,tj||Sr )r percentilerqs rarray_percentile_globalrs =a  rc,tj||Sr )r nanpercentilers rarray_nanpercentile_globalrs C # ##rc*tj|Sr )rptpas rarray_ptp_globalrs 6!99rc*|Sr )rrs r array_ptprs 5577Nrc,tj||Sr )rquantilers rarray_quantile_globalrs ;sA  rc,tj||Sr )r nanquantilers rarray_nanquantile_globalrs >#q ! !!rctjkrd}nfd}|d}|ddd}|dddddd }|jjs |jjrJ|||gS) NcP|dzdksJtjddg|dzzS)Nr)rbool_)ns rfactoryz!base_test_arrays..factorys0q5A::::8QFa1f-.. .rc6tj|dzS)Ndtyper)rarange)rrs rrz!base_test_arrays..factorys9Qe,,,q0 0r r )rrA)order)rrreshapeflags c_contiguous f_contiguous)rra1a2a3s` rbase_test_arraysrs  / / / / 1 1 1 1 1 B   Q " "B '"++ttt  $ $Yc $ : :B%?)>??? B<rct|}|tjkr|d|Dz }|tjkrbg}|D]H}|dz d|zdz z}tj|ddd|ddd<||I|||D]!}|jtj|ksJ"|S)Ncg|]}|dz S)r).0rs r z$full_test_arrays..s222!q2v222rr? r)rrfloat32 complex64conjappendextendr)r array_listaccrtmps rfull_test_arraysrs!%((J  22z2222     Ab&26B;&Cws33Q3x((C!H JJsOOOO# **w"(5//))))) rct|}t||g}||}||}||fSr )rr entry_point) compare_func test_arrayarrtycres numpy_result numba_results rrun_comparativersM :  E L5' 2 2D< ++L##J//L  %%rc4eZdZdZfdZdZefdZefdZ dZ dZ dZ d Z d Zd Zd Zd ZdZdZdZdZdZdZdZdZedZdZdZdVdZdZdZ dZ!dZ"dZ#d Z$d!Z%d"Z&d#Z'd$Z(d%Z)d&Z*d'Z+d(Z,d)Z-d*Z.dWd,Z/d-Z0d.Z1d/Z2d0Z3d1Z4d2Z5d3Z6d4Z7d5Z8d6Z9d7Z:d8Z;d9Zd<Z?d=Z@d>ZAd?ZBd@ZCdAZDdBZEdCZFdDZGdEZHdFZIdGZJdHZKdIZLdJZMdKZNdLZOdMZPdNZQdOZRdPZSdQZTdRZUdSZVdTZWeXdUZYxZZS)XTestArrayReductionszQ Test array reduction methods and functions such as .sum(), .max(), etc. ctt|tjddS)N*)superrsetUprrandomseed)self __class__s rrzTestArrayReductions.setUps: !4((..000 rrc tdfd}tjgd}||tjddg}||tjgd}||tjgd}||tjgd }||tjgd }||tjgd }||tjgd }||tjd d g}||dS)NTnopythoncJj||fidSr assertPreciseEqual)rcfunckwargspyfuncrs rcheckz8TestArrayReductions.check_reduction_basic..checks6 #D #FF3KKs F Fv F F F F Fr)?@rrr)r@inf)rr-inf)rrrr)nanrrr@)rrrrrrr)g@rrrr)rrfloat64)rrrrrrs``` @rcheck_reduction_basicz)TestArrayReductions.check_reduction_basics"T"""6** G G G G G G G Gj999:: c j$&& c j+++,, c j,,,-- c j33344 c j77788 c jFFFGG c j22233 c j%(( c rcntdfd}tjddtdtdg}||d|d <||d |d <|||d }||||ddd dS) NTrcX||dSr rrrrrs rrz1TestArrayReductions.test_all_basic..check/  # #FF3KKs < < < < .checkrr)rrrrrrrrrrrrs`` @rtest_any_basicz"TestArrayReductions.test_any_basics"T"""6** = = = = = = =j...// c uA c uA c A c kk&!! c  c$$B$irc:|tdSr )rr4rs rtest_sum_basicz"TestArrayReductions.test_sum_basic ""9-----rc:|tdSr )rr@rs rtest_mean_basicz#TestArrayReductions.test_mean_basics "":.....rc>|tddSNdouble)prec)rrErs rtest_var_basicz"TestArrayReductions.test_var_basic s! ""98"<<<<|tddSr)rrrs rtest_nanvar_basicz%TestArrayReductions.test_nanvar_basic.s! "".check3s9vc{{H%**C  # #C 2 2 2 2 2rc~||d}||jdS)N) rTrrs r check_oddz9TestArrayReductions.check_median_basic..check_odd9sA E!HHH &!!A E!HHH E!#JJJJJr?g%@c~||d}||jdS)N)r'r)s r check_evenz:TestArrayReductions.check_median_basic..check_evenBsA E!HHH '""A E!HHH E!#JJJJJr@)rrr)rrarray_variationsr*rr/rrs`` @@rcheck_median_basicz&TestArrayReductions.check_median_basic1s"T"""6** 3 3 3 3 3 3 3       "!")B--$"677  A IaLLLL      "!")B--$"677  A JqMMMM  rc#K|V|ddd}|Vtj||Vd||dzdk<|Vtj||dzdk<|Vtj|dd<|VdS)Nr @r-rr)copyrrshufflerrs r_array_variationsz%TestArrayReductions._array_variationsJs dddGLLNN !!a%1* !a%1* v!!!rcFt}d}|||dS)Nc3K|V|ddd}|Vtj||Vd||dzdk<|VdS)Nrr4r-r)r5rrr6rs r variationsz9TestArrayReductions.test_median_basic..variations\slGGG$$B$ AGGG I  a GGGAa!eqjMGGGGGr)rmr2)rrr:s rtest_median_basicz%TestArrayReductions.test_median_basicYs4$     33333rc l tdd!fd }jdddd}t jd|dddd }|||||d|||d z |||tjtj tjg}j|d |j dd <j |j ||||| }| }||||t|t|jgd d }t jd|d }|||t jddz}tj|ddtj||dtj||d z dt jddd}t jd|z}tj||gdt jdddd d}t jd|z}tj||jdt jd|z}tj||jdt jddd}tj||d z dt j||d z tjgdddt jgd }||d z gtj|t jgd dS)"NTr-q=c||}||}tj|}tj|r|||dS|||||dSN)abs_tolrisfiniterr rrr@r"r#finiterrrs rrz@TestArrayReductions.check_percentile_and_quantile..checkkva||H%1++C[**Fvf~~ 9''Xw'GGGGG''F Xf5E07(99999rrrrrr)rrrr-r?rr4g?rr-)?rIr)g@@g&@ih)rJrIr)rJrIg?)rrK)rrrr)rrr-r)r=)rrrandnrrlinspacerrchoiceflatr6flattentolisttuplertesting assert_equalarrayshape assertTrueisscalar) rr q_upper_boundrrr not_finitexrs `` @rcheck_percentile_and_quantilez1TestArrayReductions.check_percentile_and_quantilehsI"T"""6** 9 9 9 9 9 9 9 9 K  b ! ! ) )!Q 2 2 K=" - -ddd 3 a  a  a"### afrvgrv. k((R88ss  A A a IIKK    IIKK    a  eAhha!!! K  |||R 0 0 K=! , , a  IaLL3  a R000 a 7 7=== a): ; ;TBBB IbMM ! !!Q ' ' H% & & 6 a ->->->??? Im $ $ , ,Q1a 8 8 H\ " "] 2 a 14888 H' ( (= 8 a 14888 IbMM ! !!Q ' ' a): ; ;SAAA  EE!]Q->$?$?@@AAA iii 3 3Q777 H\\\ " " a-!#$%%% 28LLL#9#9:::::rdc td d fd }d fd }d}dd|zd |z|f}d d tjtjtj f}||D]}||||d kr|} n|} tjd }tjd }| ||d }|d z }| ||dS)NTr+=c||}||}tj|}tj|r|||dS|||||dSr?rArCs rrz>TestArrayReductions.check_percentile_edge_cases..checkrErc||tj}||}|||dSr?)astyperrr)rrr@r"r#rrrs rconvert_to_float_and_checkzSTestArrayReductions.check_percentile_edge_cases..convert_to_float_and_checksRva||**2:66H%1++C  # #C7 # C C C C Crc3KtddD]+}t||D]}tj|V,dS)Nrr)rangerrrW)elementsicombs r_array_combinationszLTestArrayReductions.check_percentile_edge_cases.._array_combinationss\1b\\ ) )9(AFF))D(4..(((() ) )rr皙?g?rrrr)ra)rrrrrW) rrr[rrerkr element_poolr_checkrs `` @rcheck_percentile_edge_casesz/TestArrayReductions.check_percentile_edge_casessG"T"""6** 9 9 9 9 9 9 9 9 D D D D D D D D  ) ) ) m#S=%8- H2rvrvw7 $$\22  A E!QKKKK A  /FFF HQKK HQKKq!  A q! rctd|fd}tjd}||d||d||ddtjf5}tjdd z}d }||dddn #1swxYwYd t|jdS) NTrct5}||dddn #1swxYwYdt|jdS)Nz)Percentiles must be in the range [0, 100] assertRaises ValueError assertEqualstr exceptionrrraisesrrs r check_errzBTestArrayReductions.check_percentile_exceptions..check_errs"":.. &a                   ;F$%%      599r)rrirrrrlNot supported for complex dtype rdisable_leak_checkrrrassertTypingErrorassertInrvrwrrrzrerrs` @rcheck_percentile_exceptionsz/TestArrayReductions.check_percentile_exceptionssL"T"""6**       !!! IaLL !R !\""" !aRV_%%%  # # % %  ! r!AA E!QKKK                7Q[9I9IJJJJJ&CCCctd|fd}tjd}||d||d||ddtjf5}tjddz}d}||dddn #1swxYwYd t|jdS) NTrct5}||dddn #1swxYwYdt|jdS)Nz%Quantiles must be in the range [0, 1]rrrxs rrzz@TestArrayReductions.check_quantile_exceptions..check_errs"":.. &a                   7F$%%     r{rg)rlrlg?rlrr~rrs` @rcheck_quantile_exceptionsz-TestArrayReductions.check_quantile_exceptionssN"T"""6**       !!! IaLL !T !&''' !c4()))  # # % %  ! r!AA E!QKKK                7Q[9I9IJJJJJrct}||d||d||dSNr_r[)rr^rorrrs rtest_percentile_basicz)TestArrayReductions.test_percentile_basic sS( **6*EEE ((s(CCC ((00000rct}||d||d||dSr)rr^rorrs rtest_nanpercentile_basicz,TestArrayReductions.test_nanpercentile_basicsS+ **6*EEE ((s(CCC ((00000rct}||d||d||dSNrr)rr^rorrs rtest_quantile_basicz'TestArrayReductions.test_quantile_basicsS& **6*CCC ((q(AAA &&v.....rct}||d||d||dSr)rr^rorrs rtest_nanquantile_basicz*TestArrayReductions.test_nanquantile_basicsS) **6*CCC ((q(AAA &&v.....rcJt}|||jdSr )rr2r7rs rtest_nanmedian_basicz(TestArrayReductions.test_nanmedian_basic#s&' (>?????rcltjdtj}t|}||jd||jdtt|g}|j }|tj |||dSNrrrC) rrint32rrundimlayoutrr6rr3rrrrrs rtest_array_sum_globalz)TestArrayReductions.test_array_sum_global'si"(+++s  Q''' s+++ 05'::  eeCjj11111rcrtjdtjdz}t|}||jd||jdtt|g}|j }|| ||dSr) rrrrrurrrr9rr8rs rtest_array_prod_int_1dz*TestArrayReductions.test_array_prod_int_1d2si"(+++a/s  Q''' s+++ UG44  UU3ZZ00000rctjdtjdz}t|}||jd||jdtt|g}|j }tj | ||dS)Nrrrlrr) rrrrrurrrr9rrUassert_allcloser8rs rtest_array_prod_float_1dz,TestArrayReductions.test_array_prod_float_1d=si"*---6s  Q''' s+++ UG44  ""388::uuSzz:::::rctjdtj}t|}||jd||jdtt|g}|j }tj tj |||dSr) rrrrrurrrr<rrUrr8rs rtest_array_prod_globalz*TestArrayReductions.test_array_prod_globalHsi"(+++s  Q''' s+++ 1E7;;  ""273<<s<<<<|tddSNT)r)rr0rs rtest_nancumsum_magnitudez,TestArrayReductions.test_nancumsum_magnitudes! (($(GGGGGrcv|td|tddSr)rr9r<rs rtest_prod_magnitudez'TestArrayReductions.test_prod_magnitudes; ((T(BBB (():D(IIIIIrcv|td|tddSr)rr r"rs rtest_cumprod_magnitudez*TestArrayReductions.test_cumprod_magnitudes; (((EEE (()=t(LLLLLrc>|tddSr)rr&rs rtest_nancumprod_magnitudez-TestArrayReductions.test_nancumprod_magnitudes" (()94(HHHHHrcn|t|tdSr )rr@rBrs rtest_mean_magnitudez'TestArrayReductions.test_mean_magnitudes1 ((444 (():;;;;;rcn|t|tdSr )rrErGrs rtest_var_magnitudez&TestArrayReductions.test_var_magnituderrcn|t|tdSr )rrJrLrs rtest_std_magnitudez&TestArrayReductions.test_std_magnituderrc|t|}td|}||||||||dd||dd|ddd}|||||tj||||||d|jvrNtdt|dD]}d||<|||||| | d|||||dS)NTrrrlrrNaT) rrrr5rrr6__name__rglenfillr)rrrrrr]s r_do_check_nptimedeltaz)TestArrayReductions._do_check_nptimedeltass "T"""6** c FF3KK888 c#2#hCRC1A1ABBB$$B$inn c FF3KK888 # c FF3KK888 6? * * 1c#hh**  A  # #EE#JJs < < < U##$$$ c FF3KK88888rctjdd}|||dS)NrzM8[Y]rrrrdrrrrs rcheck_npdatetimez$TestArrayReductions.check_npdatetime<imm"""11 ""63/////rctjdd}|||dS)Nrzm8[s]rrrs rcheck_nptimedeltaz%TestArrayReductions.check_nptimedeltarrcn|t|tdSr )rrOrrs rtest_min_npdatetimez'TestArrayReductions.test_min_npdatetime0 i((( y)))))rcn|t|tdSr )rrWrrs rtest_max_npdatetimez'TestArrayReductions.test_max_npdatetimerrcn|t|tdSr )rr`rrs rtest_argmin_npdatetimez*TestArrayReductions.test_argmin_npdatetime0 l+++ |,,,,,rcn|t|tdSr )rrgrrs rtest_argmax_npdatetimez*TestArrayReductions.test_argmax_npdatetimerrc:|tdSr )rrmrs rtest_median_npdatetimez*TestArrayReductions.test_median_npdatetimes 233333rc:|tdSr )rr4rs rtest_sum_npdatetimez'TestArrayReductions.test_sum_npdatetimes y)))))rc:|tdSr )rr)rs rtest_cumsum_npdatetimez*TestArrayReductions.test_cumsum_npdatetimes |,,,,,rc:|tdSr )rr@rs rtest_mean_npdatetimez(TestArrayReductions.test_mean_npdatetimes z*****rc :tdfd}d}d}|D]>}|||||tj?|tjg|tjdtjtjtjdtjdd d d d d g }|d|dddzz}|dd}||dS)NTrc`|}|}||dSr rrr"r#rrrs rrz7TestArrayReductions.check_nan_cumulative..check9vayyH%((C  # #Hc 2 2 2 2 2rc|jdz}tj|tjt |j|dtj|S)NrF)replace)sizerputrrPrgr)rps r_set_some_values_to_nanzITestArrayReductions.check_nan_cumulative.._set_some_values_to_nansG! A F1bi&&uQV}}a&GG P P PHrc3KtjddddddVtjtjddgVtjgdVtjddVtjtjdd d z Vtjddtj dddVdS) Nrr<r-r)TTTFrrr0gfffff@@r)rrOrrWrrasfortranarrayrrrr a_variationsz>TestArrayReductions.check_nan_cumulative..a_variationss+b!R((00Aq99 9 9 9(BFAq>** * * *(44455 5 5 5)Ar"" " " "#BIa$4$4t$;<< < < <)Ar444TTrT: : : : : :rrrr-rrLr&rHr%rrr)rrdrrrWfullrr)rrrrrrpartsrs`` @rcheck_nan_cumulativez(TestArrayReductions.check_nan_cumulativesL"T"""6** 3 3 3 3 3 3 3      ; ; ; A AA E!HHH E))!((2:*>*>?? @ @ @ @ bhrll bgb"&!!""""&!RVQ1aA>?? Bttt$ $ IIaOO arcn|t|tdSr )rr&rrs rtest_nancumprod_basicz)TestArrayReductions.test_nancumprod_basics2 ./// !!"233333rcn|t|tdSr )rr0rrs rtest_nancumsum_basicz(TestArrayReductions.test_nancumsum_basics0 o... !!/22222rcttdfd}d}|D] }||dS)NTrc`|}|}||dSr rrs rrz1TestArrayReductions.test_ptp_basic..check rrc3\KtjdVtjdtjdgVtjtj dgVdVdVtjddVgdVtjddd d d d }|Vtj|V|ddd Vtj d  ||VdVdVtj VdVdtjgVdtjgVdVtjdtjVdtjdzzVtjtjdzzVtjVdS)Nrg皙@r)r-rrr)rgffffffrlr.r-rrrrLg@y?@))gffffff@rgffffff)g333333@g@g@rr) rrrWrrrrOrrr RandomStater6rs rrz8TestArrayReductions.test_ptp_basic..a_variationss)B--   (D"&#.// / / /(RVGQ<(( ( ( (OOOJJJ'!Q--   """ " " " CR((00Aq99AGGG#A&& & & &DDbD'MMM I ! !! $ $ , ,Q / / /GGGGGGIII6'MMMLLL-   -   5 5 5 5'!RV$$ $ $ $bfrk/ ! ! !&26B;& & & &&LLLLLr)rr)rrrrrrs` @@rtest_ptp_basicz"TestArrayReductions.test_ptp_basics!"T"""6** 3 3 3 3 3 3 3    4  A E!HHHH  rct}td|}tjd}||}||}|||dS)NTrr)rrrrr)rrrrr"r#s rtest_ptp_methodz#TestArrayReductions.test_ptp_method.sf"T"""6** IbMM6!99eAhh #.....rc t td  fd}dd}tddgdD]\}}|||}||tjd}tjd d }|d |zz}|||d |zz }||tjd d }||dS)NTrc`|}|}||dSr rrs rrz3TestArrayReductions.test_ptp_complex..check=rrFctjddd}|rtj|dd<tjddd}|rtj|dd<|d |zzddS) Nr-r%r|rr&rr)rrOrr)real_nanimag_nanrealimags r make_arrayz8TestArrayReductions.test_ptp_complex..make_arrayBsu;r1b))D #FQqS ;r1b))D $VQrT 29$--a33 3rr)repeatrHrr-r)r-r-y?) fill_value)FF)rrrronesrr) rrrrrcomprrrrs ` @@rtest_ptp_complexz$TestArrayReductions.test_ptp_complex9s!"T"""6** 3 3 3 3 3 3 3  4 4 4 4#*4-"B"B"B   Hh:h11D E$KKKKwqzzyQb4i d b4i d wv6333 d rcFt}td|}||5}|t jddddn #1swxYwYd}||t|j| t5}|t jgdddn #1swxYwYd}||t|jdS)NTr)TTFz+Boolean dtype is unsupported (as per NumPy)z&zero-size array reduction not possible) rrrrrrWrrvrwrsrt)rrrrmsgs rtest_ptp_exceptionsz'TestArrayReductions.test_ptp_exceptionsYs!"T"""6** !!!  # # % % 1 E"(.// 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1< c3q{++,,,   z * * a E"(2,,                  7 c3q{++,,,,,s$ A33A7:A7C,,C03C0c> ttf}|D] td  fd}tjddd}|d|dd<|d z}||d zz }||t dD]}j|j|jtj tj g}||d zz }|d|dd<|| | dS) NTrc`|}|}||dSr rrs rrz=TestArrayReductions.test_min_max_complex_basic..checkr9!6!99eAhh''#66666rrr(rr-rr) rQrYrrrOrgrr6rPr complex128rd) rpyfuncsrrrr_rrrs ` @@rtest_min_max_complex_basicz.TestArrayReductions.test_min_max_complex_basiclsP"$44 ' 'F&C&&&v..E 7 7 7 7 7 7 7 ;sB++DBxD!H!8Dtby A E!HHH2YY ' ' ##D))) ##D))) **BL"-+HII4"9$""1"ahhuoo&&&&  ' ' 'rcttf}|D]tdfd}tjddd}|d|dd<tj|d d <|d z}tj|d d <||dzz }||t dD]V}j|j|||dzz }|d|dd<||WdS)NTrc`|}|}||dSr rrs rrzCTestArrayReductions.test_nanmin_nanmax_complex_basic..checkr rrrr!rr-rr%rr&rr) rprsrrrOrrgrr6) rr#rrrrr$rrs ` @@r test_nanmin_nanmax_complex_basicz4TestArrayReductions.test_nanmin_nanmax_complex_basicsA ,  F&C&&&v..E 7 7 7 7 7 7 7 ;sB++DBxD!HD1I!8DD2Jtby A E!HHH2YY   ##D))) ##D)))4"9$""1"a  !  rcttf}fd}d}|D]3td|D] }||4dS)Nc`|}|}||dSr rrs rrzFTestArrayReductions.test_nanmin_nanmax_non_array_inputs..checkrrc3KgdVdVtjdVdVdVdVtjdddfVtjtjtj gVtjd fd gVdS) N)rrLr-r))rr-i)rirrMgA`" @Fgrg?)ggffffff!@)rrWrrrrrrzMTestArrayReductions.test_nanmin_nanmax_non_array_inputs..a_variationss,,   0 0 0 0(1++   JJJKKKKKK65$, , , ,626BF7+ + + +FC=+. . . . . .rTr)rprsr)rr#rrrrrs` @@r#test_nanmin_nanmax_non_array_inputsz7TestArrayReductions.test_nanmin_nanmax_non_array_inputss , 3 3 3 3 3 3 3 / / /  F&C&&&v..E!\^^  a   rcjtjgd}tjddd}|dxxdz cc<tjddddd d z}|d xxdz cc<|d xxd zcc<|||fD]}t t |jdt |jDz}d|D}d|D}|D]4}|||||5dS)Nrrr-rLrrrrr_xr-rrrrrrrrrr3cg|]}|dz  Srrrris rrzATestArrayReductions.test_argmax_axis_1d_2d_4d..+,,,!A#,,,rcg|]}|fdS)c.tj||SNaxisrir_axiss rzJTestArrayReductions.test_argmax_axis_1d_2d_4d...bi&>&>&>rrrr=s rrzATestArrayReductions.test_argmax_axis_1d_2d_4d..1!%>>>rc@g|]}td|STrrrrs rrzATestArrayReductions.test_argmax_axis_1d_2d_4d..;/5"T"""6**r rrWrrlistrgrrpy_func rarr1darr2darr4draxes py_functions c_functionsrs rtest_argmax_axis_1d_2d_4dz-TestArrayReductions.test_argmax_axis_1d_2d_4d'' ! $$Q** c c  #&&q!Q22R7 jS  jR5%( H HCch((,, %ch,,,D L9EK% H H'' c(:(:EE#JJGGGG H H HrcHtjd}tjddd}tddfd}||d||d ||d ||ddS) NrLrrTrc,tj||Sr rirr=s r jitargmaxzDTestArrayReductions.test_argmax_axis_out_of_range..jitargmax9S$'' 'rctd5||dddn #1swxYwYtd5||ddddS#1swxYwYdSNzaxis.*out of boundsassertRaisesRegexrtrK)rr=rXrs r assert_raiseszHTestArrayReductions.test_argmax_axis_out_of_range..assert_raises'' 4IJJ - -!!#t,,, - - - - - - - - - - - - - - -'' 4IJJ % % #t$$$ % % % % % % % % % % % % % % % % % %#AAA& BBBrrrrrr)rrMrNr^rXs` @rtest_argmax_axis_out_of_rangez1TestArrayReductions.test_argmax_axis_out_of_ranges !  ! $$Q** d    ( (   ( % % % % % %  eQ eR    eR    eQ !!!!!rc tjd}tdd}|5}||ddddn #1swxYwY|dt |jdS)NrLTrc,tj||Sr rirWs rrXzGTestArrayReductions.test_argmax_axis_must_be_integer..jitargmaxrYrfooaxis must be an integerrrrrrrvrw)rrrXrs r test_argmax_axis_must_be_integerz4TestArrayReductions.test_argmax_axis_must_be_integerill d    ( (   ( # # % % " Ic5 ! ! ! " " " " " " " " " " " " " " " /Q[1A1ABBBBB AA"Actjdddfd}||t d|dS)NrLrrc0dSNrr<rerrNs rrfz;TestArrayReductions.test_argmax_method_axis..argmax<.argmaxrqrTrrruint8rrrrss @rtest_argmax_return_typez+TestArrayReductions.test_argmax_return_type !28,,,44Q:: ( ( ( ( ( u : 2T 2 2 26 : :5 A A C C C C Crcjtjgd}tjddd}|dxxdz cc<tjddddd d z}|d xxdz cc<|d xxd zcc<|||fD]}t t |jdt |jDz}d|D}d|D}|D]4}|||||5dS)Nr/rLrrr1r_r2r-rrr3r4r5cg|]}|dz  Srrr7s rrzATestArrayReductions.test_argmin_axis_1d_2d_4d.. r8rcg|]}|fdS)c.tj||Sr;rbr>s rr@zJTestArrayReductions.test_argmin_axis_1d_2d_4d...rArrrBs rrzATestArrayReductions.test_argmin_axis_1d_2d_4d..rCrc@g|]}td|SrErFrGs rrzATestArrayReductions.test_argmin_axis_1d_2d_4d..rHrrIrLs rtest_argmin_axis_1d_2d_4dz-TestArrayReductions.test_argmin_axis_1d_2d_4drTrcHtjd}tjddd}tddfd}||d||d ||d ||ddS) NrLrrTrc,tj||Sr rbrWs r jitargminzDTestArrayReductions.test_argmin_axis_out_of_range..jitargminrYrctd5||dddn #1swxYwYtd5||ddddS#1swxYwYdSr[r\)rr=rrs rr^zHTestArrayReductions.test_argmin_axis_out_of_range..assert_raises!r_r`rrarbrc)rrMrNr^rs` @rtest_argmin_axis_out_of_rangez1TestArrayReductions.test_argmin_axis_out_of_ranges !  ! $$Q** d    ( (   ( % % % % % %  eQ eR    eR    eQ !!!!!rc tjd}tdd}|5}||ddddn #1swxYwY|dt |jdS)NrLTrc,tj||Sr rbrWs rrzGTestArrayReductions.test_argmin_axis_must_be_integer..jitargmin2rYrrgrhri)rrrrs r test_argmin_axis_must_be_integerz4TestArrayReductions.test_argmin_axis_must_be_integer/rkrlctjdddfd}||t d|dS)NrLrrc0dSror^rps rr_z;TestArrayReductions.test_argmin_method_axis..argmin=rqrTrrrrr_rNs @rtest_argmin_method_axisz+TestArrayReductions.test_argmin_method_axis:rurctjdtjddfd}||t d|dS)NrLrrrc0dSror^rps rr_z;TestArrayReductions.test_argmin_return_type..argminGrqrTrrxrs @rtest_argmin_return_typez+TestArrayReductions.test_argmin_return_typeCr{rc>tttttt t tttttttttt t"t$t&t(t*t,g}t.t0t2t4g}|t6t8t:gz }|t<gz }t>j t>j!t>j"t>j#g}fd}||dd||||dS)Nc|D]n}t|}t||D]L\}d}|j|jj|j}||ffd }t||ModS)Nztest_{0}_{1}_{2}dcd}djvrtj|rd}t||\}}||||d|dS)Nrr8rsingle)rrulps)rr iscomplexobjrr)rredFunc testArraytestNamerrrred_funcs rnew_test_functionz]TestArrayReductions.install_generated_tests..install_tests..new_test_functionus{ !!X%666OI667$%D#27I#F#FS//Sh5=D0JJJJJr)rrformatrrnamersetattr) dtypesfuncsdt test_arraysr test_namerrclss @r install_testszBTestArrayReductions.install_generated_tests..install_testsjs ? ?.r22 ,3E;,G,G??(Hj 3I ) 0 01B1;1A1F1;!B!BI9A4>3< J J J J J JC,=>>>>+? ? ?rr)$r4r6r9r<r@rBrErGrJrLrrrrrOrQr\rTrWrYrsrpryr`rcrgrjrvrrr|rrrrr)rreduction_funcsreduction_funcs_rspacedtypes_to_testrs` rinstall_generated_testsz+TestArrayReductions.install_generated_testsMs %&6%'8%'8$&6$&6$&6$&6$&6%z$&6''  #/0C".0C"E M<FFM?*(BJ",G ? ? ? ? ?8  nSbS)+ABBB  no66666r)r_)F)[r __module__ __qualname____doc__rrrrrrrrrrr r r rrrrrrrrr2 staticmethodr7r;r^rorrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr r rrr%r(r-rSrdrjrtrzrrrrr classmethodr __classcell__)rs@rrrsl4%.    %."...///===.........111111111111222111222111@@@2  \  4 4 4?;?;?;B,,,,\KKK4KKK4111 111 /// /// @@@ 2 2 2 1 1 1 ; ; ; = = = / / /,,,333---444****&;;;>>>HHHJJJMMMIII<<<;;;;;;9994000000******------444***---+++!!!F444333$$$L / / /@---&'''242HHH."""* C C CCCCCCCHHH.""", C C CCCCCCC;7;7[;7;7;7;7;7rrcJeZdZejdZdZedZdS)TestArrayReductionsExceptionsrctd|}|t5||jdddn #1swxYwY|t5}||jdddn #1swxYwY||t |jdS)NTr)rrs BaseException zero_sizertrrvrw)rrrrrs rcheck_exceptionz-TestArrayReductionsExceptions.check_exceptions5"T"""6**   } - - # # F4> " " " # # # # # # # # # # # # # # #  z * * "a E$. ! ! ! " " " " " " " " " " " " " " " c3q{++,,,,,s#AAA6BBBct}d}d}ttttfD]}|||<t t t t fD]}|||<d}|D]z\}}||j }||j } | dd dd} || fd} t||| {dS)Nz'attempt to get {0} of an empty sequencez*zero-size array to reduction operation {0}ztest_zero_size_array_{0}array__globalc2|||dSr )r)rfuncmessages rtest_fnz6TestArrayReductionsExceptions.install..test_fns$$T733333r) dictrgrjr`rcrWrOitemsrrrr) r fn_to_msg empty_seq op_no_identr] name_templatefnrrlmsgrs rinstallz%TestArrayReductionsExceptions.installs FF =   3\%' % %A$IaLLY 9= ' 'A&IaLL2  (( - -GB%,,R[99I::bk**D<<,,44YrBBD#%t 4 4 4 4 CG , , , , - -rN) rrrrrrrrrrrrrrsO ! I-----[---rr__main__)G itertoolsrrnumpyrnumbarrnumba.core.compilerrnumba.np.numpy_supportrnumba.tests.supportr r r unittestrrrrr r"r&r)r,r0r4r6r9r<r@rBrErGrJrLrOrQrTrWrYr\r`rcrgrjrmrprsrvryr|rrrrrrrrrrrrrrrrrmainrrrrs <<<<<<<<000000000000>>>>>>>>>>!!!$$$""""(&&&H7H7H7H7H7/8H7H7H7V++---%-%-%-%-%-OX%-%-%-N%%''' zHMOOOOOr