idhddlZddlZddlZddlZddlmZddlZddlZddlZddl m Z ddl m Z ddlmZddlZddlmZddlmZmZmZmZmZmZmZmZmZmZmZmZm Z ddl!m"cm#Z$ddl%m&Z&ddl'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGdd lHmIZId ZJd ZKGd d ZLGddZMGddZNGddZOGddZPGddZQGddZRGddZSGddZTGddZUGd d!ZVGd"d#ZWGd$d%ZXGd&d'ZYGd(d)ZZGd*d+Z[Gd,d-Z\Gd.d/Z]Gd0d1Z^Gd2d3Z_dsd5Z`dtd7ZaGd8d9ZbGd:d;ZcGd<d=ZdGd>d?Zeejfgd@dAejhdBzejhdCzejfgdDgdEGdFdGZiGdHdIZjGdJdKZkGdLdMZlGdNdOZmGdPdQZnGdRdSZoGdTdUZpGdVdWZqGdXdYZrGdZd[ZsGd\d]ZtGd^d_ZuGd`daZvGdbdcZwgddZxGdedfZyGdgdhZzGdidjZ{GdkdlZ|GdmdnZ}GdodpZ~GdqdrZdS)uNFraction)arrays)partial)ma) assert_ assert_equalassert_array_equalassert_almost_equalassert_array_almost_equal assert_raisesassert_allcloseIS_PYPY assert_warnsassert_raises_regexsuppress_warnings HAS_REFCOUNTIS_WASM)rand) add_newdoc_ufuncangleaveragebartlettblackmancorrcoefcovdeletediffdigitizeextractflipudgradienthamminghanningi0insertinterpkaisermeshgridmsort piecewiseplacerot90selectsetxor1dsinctrapz trim_zerosunwrapunique vectorize)normalize_axis_tuplecntj|}tj||}|SN)nparangeaddouter)ndatas W/opt/cloudlinux/venv/lib/python3.11/site-packages/numpy/lib/tests/test_function_base.pyget_matr@s) 9Q<>> @ @ @ @ @ @rAN)__name__ __module__ __qualname__rnrvrrAr?rLrL0sMLLL@CCC @@@@@rArLcDeZdZdZdZdZdZdZdZdZ dZ d Z d S) TestFlipcttjtjtjddttjtjtjddttjtjtjddttjtjtjdddS)NrNrRaxisrNrNrPr^rrZ)r r9 AxisErrorfliprfrhs r?rvzTestFlip.test_axesvsblBGRWQZZa@@@@blBGRWV__1EEEEblBGRWV__2FFFFblBGRWV__6JJJJJJrActd}|dddddf}ttj|d|gdgdg}gdgdg}ttj|d|dS)NrNrarRrQrYr]r\r@r r9rrhribs r? test_basic_lrzTestFlip.test_basic_lr|s AJJ aaa2gJRWQ]]A&&& YY YY  YY YY RWQ]]A&&&&&rActd}|dddddf}ttj|d|gdgdg}gdgdg}ttj|d|dS)NrNrarrQrYrrs r? test_basic_udzTestFlip.test_basic_uds AJJ dddAAAgJRWQ]]A&&& YY YY  YY YY RWQ]]A&&&&&rActjddgddggddgddggg}tjddgddggddgddggg}ttj|d|dS NrrRrPrZrNr[ryrzr9arrayr rrs r?test_3d_swap_axis0zTestFlip.test_3d_swap_axis0 H1v1v1v1v  ! ! H1v1v1v1v  ! ! RWQ]]A&&&&&rActjddgddggddgddggg}tjddgddggddgddggg}ttj|d|dSrrrs r?test_3d_swap_axis1zTestFlip.test_3d_swap_axis1rrActjddgddggddgddggg}tjddgddggddgddggg}ttj|d|dSrrrs r?test_3d_swap_axis2zTestFlip.test_3d_swap_axis2rrAc Btjddddd}t|jD]_}t tj||tj|d||d`dS)NxrPrZrNr[r) r9r:r{rgndimr rr!swapaxes)rhriis r?test_4dzTestFlip.test_4ds Im $ $ , ,Q1a 8 8qv E EA A1::a#3#344==aCC E E E E E ErActjgdgdg}tjgdgdg}ttj||dS)NrRrPrZrNr[ry)ryr[rNrZrPrRrrs r?test_default_axiszTestFlip.test_default_axissf Hiiiii! " " Hiiiii! " "RWQZZ#####rActjddgddggddgddggg}ttj|d |tjddgddggddgddggg}ttj|d |tjddgddggddgddggg}ttj|d |dS) NrrRrPrZrNr[ryrzrrrVrsrrhrircs r?test_multiple_axeszTestFlip.test_multiple_axess H1v1v1v1v  ! ! RWQR(((!,,, H1v1v1v1v  ! ! RWQV,,,a000 H1v1v1v1v  ! ! RWQV,,,a00000rAN) rrrrvrrrrrrrrrrAr?rrtsKKK '''''' ' ' ' ' ' ' ' ' 'EEE $$$11111rArceZdZdZdZdS)TestAnycgd}gd}gd}ttj|ttj|ttj| dS)NrrrRrrrrr)rRrrRr)rr9anyrhy1y2y3s r?rnzTestAny.test_basicsg \\ \\ \\r r BF2JJrAcgdgdgdg}ttj|ttj|dgdttj|dgddS)NrrrrrRrrRrRrrrrRrrRrR)rr9rr rhrs r?test_ndzTestAny.test_nds|iiIII .r 26"1---yyy99926"1---yyy99999rANrrrrnrrrAr?rrs2   :::::rArceZdZdZdZdS)TestAllcPgd}gd}gd}ttj| ttj|ttj| ttjtj|dS)N)rrRrRrrrRrRrRrR)rr9allrrs r?rnzTestAll.test_basics \\ \\ \\BF2JJr BF2JJ }%%&&&&&rAcgdgdgdg}ttj| ttj|dgdttj|dgddS)NrrrRrrRrRrRrrrR)rr9rr rs r?rzTestAll.test_nds~iiIII .BF2JJ26"1---yyy99926"1---yyy99999rANrrrAr?rrs2''':::::rArc eZdZdZdZdZdS)TestCopyctjddgddgg}tj|}t||d|d<t |ddt |dddS)NrRrPrZrN )rr)r9rcopyr r )rhria_copys r?rnzTestCopy.test_basicsr Hq!fq!f% & &1f%%%t QtWa   VD\2&&&&&rAcZtjddgddgg}t|jjt|jj tjddgddggd}t|jj t|jjtj|}t|jjt|jj tj|}t|jj t|jjdS)NrRrPrZrNForder)r9rrflags c_contiguous f_contiguousr)rhria_fortr a_fort_copys r? test_orderzTestCopy.test_orders Hq!fq!f% & &$%%%AG(()))Aq6Aq6*#666FL--... )*** )***FL--...gfoo K%22333 !./////rAcltjd}ttjt j|d ttjt j|dttjt j| dS)Nr[F)subokT)rrfr isMaskedArrayr9r)rhmxs r? test_subokzTestCopy.test_suboks WQZZB$RWRu%=%=%=>>>??? 4!8!8!899:::B$RWR[[11122222rAN)rrrrnrrrrAr?rrsA'''000"33333rArc eZdZdZejdgdddggddgdgfgd gd gd gd gd dggdggdgfgdZdZdZ dZ dZ dZ dZ dS) TestAveragectjgd}tt|ddktjgd}tt|ddkgd}tt|ddktjd}d|d <d |d <t |dt|dt |d t|d td d }t |dt|dt |d t|d dS)Nrrr@?r@)rrrrrbrPrcrRr[)r9rrrrfr meanr)rhrrry4y5s r?rnzTestAverage.test_basic#sK Xiii ###r)*** Xlll # ####r)*** \\###r)*** WV__44BGGAJJA777BGGAJJA777 !QZZBGGAJJA777BGGAJJA77777rAz.subclassNrrrrrAr?subclassr DrArrRrPrZrNr)r9ndarrayrviewr typer)rhrrirs r?test_subclasseszTestAverage.test_subclassess     rz    HqeQqE] # # ( ( 2 2 HqeQqE] # # ( ( 2 2T"*Q--(((333T"*Q22233X>>>>>rAcgd}|D]y\}}}tjddgddgg|}tjddgddgg|}ttj||jtj|zdS)N))i4rf8)rf4r)r rr)r r r )r rrrRrPrZrNrr)r9rr rr)rhtypsatwtrtrirs r?test_upcastingzTestAverage.test_upcastings999 G GJBB1Q%1b111A1Q%1b111A Aq1117" F F F F G GrAcDtjdtdD}tjdtdD}||z}t |dt ||dS)Nc6g|]}tj|SrdecimalDecimal).0rs r? z1TestAverage.test_object_dtype.."<<.r.rArr)r9rrgrr rr)rhrirs r?test_object_dtypezTestAverage.test_object_dtypes H<<%))<<< = = H<<%))<<< = = QUUWW AFF1IIwq!'<'<'<=====rActjtddtddg}ttj|tdddS)NrRr[rZrP)r9rrr rrus r? test_average_class_without_dtypez,TestAverage.test_average_class_without_dtypesL Hhq!nnhq!nn5 6 6RZ]]HQNN33333rA)rrrrnpytestmark parametrizerrrrr&r2r4rrAr?rr!s888$ [F ))TC5)))dVcU ; 99jjj !1&7 a&???#iii[ 2 3 00  0"*,*,*,X4444???GGG>>> 44444rArceZdZejgdejgdejgdgZejgdejgdejgdgZddZd Zd Z d Z d Z d Z dZ dS) TestSelectrrrzrx )FFFFTFFFTrcg}tt|D]$|fdt||Dp|gz }%|S)Nc6g|]\}}| |Srr)r,VCms r?r-z&TestSelect._select..s*BBB1QqTBqtBBBrA)rglenzip)rhcondvaluesdefaultoutputrBs @r?_selectzTestSelect._selects^s4yy!! P PA BBBBFD(9(9BBBOwi OFF rAc|j}|j}tt||d|||dt t |dt t |ddS)NrGrZ)choices conditionsr r.rIr rCrhrMrNs r?rnzTestSelect.test_basics,_ 6*grBBB<< GR<HH J J J S\\1%%%S__a(((((rAc^tjdtjgdg}dtjdddg}t t ||tjdtt dgdgdg jd dS) NTr<rR rNrZrNrZrrLr0) r9rr:r{r r.rfr rErhrNrMs r?test_broadcastingzTestSelect.test_broadcastingshtnnbh/C/C/C&D&DE bimm++Aq1126*g66HHHVTFQC!555;TBBBBBrAc tt|j|jdjt jd|jD}tt|j|jt jt jdddt j ddg}t j |}tt|g|gdddt j ddgdS) N?cLg|]!}|tj"Sr)rr9int8)r,choices r?r-z0TestSelect.test_return_dtype..s&EEEf6==))EEErArRrPrZr[rzr) r r.rNrMrr9rFrXrnanisnan)rhrMdrBs r?test_return_dtypezTestSelect.test_return_dtypesVDOT\2>>D[ " " "FE EEEVDOW55;RWEEE HaArvq!, - - HQKKVQC!%%1aA'>?????rAcxtttggdtttggdS)N@)r rer.rs r?test_deprecated_emptyz TestSelect.test_deprecated_emptys2j&"b"555j&"b11111rAcz|j}|jdd}|dtj|d<t t t|||dtj|d<t t t||t t t||dSNr) rMrNrr9int_r TypeErrorr.uint8rOs r?test_non_bool_deprecationz$TestSelect.test_non_bool_deprecations,_QQQ' "1 ,,RW55 1 iW==="1 ,,RX66 1 iW===iW=====rActjdggdz}tjdggdz}t||dS)NFdrR)r9rr.rSs r?test_many_argumentszTestSelect.test_many_argumentssGhw''(3. 8QC==/C'z7#####rANr)rrrr9rrMrNrIrnrTr]r`rfrirrAr?r9r9srx ""rx ""rx ""$G"(00011"(///00"(///002J )))CCC @ @ @222>>>$$$$$rAr9ceZdZdZdZdZdZdZdZdZ e j dd d gd Z d S) TestInsertc $gd}tt|ddgdtt|ddgdtt|gdgdgdtt|dgdgdtt|gd d gd tt|td dd d gd tt|gdgdgdtjddgtj}tt|d|dgdtt|gg|t jd5}t jddttt|tjdgdzd gdt|dj tuddddS#1swxYwYdS)NrrrR)rRrRrPrZrZ)rRrPrZrRr)rRrRrPrZrPrZ)rRrarZr;)rRr;rPr;rZr;ra)r;rRr;rPr;rZrarRrZr:)rRrxrPrzrZr;r!rrrTrecordalwaysrN)rRr;r;r;r;rPrZ) r r&slicer9rrwarningscatch_warningsfilterwarnings FutureWarningrcategory)rhrirrs r?rnzTestInsert.test_basicsV IIVAq!__lll333VAq!__lll333VAyyy)))446H6H6HIIIVAq))),,.@.@.@AAAVAzzz1--/A/A/ABBBVAuRr22A668J8J8JKKKVAzzz999557I7I7IJJJ HaV2: . . .VAq!A$''666VAr2&&*** $D 1 1 1 4Q  #Hb- @ @ @ q"(D6A:..224I4I4I K K K AaDM]2 3 3 3  4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4sA4HH  H c gdg}gdgdg}tt|ddggdtt|dgdd|tt|ddd|tt|dddgdgtjddgddgd d gg}tjdd d d d }tj|ddddftjdd d d d j|ddddffd}tt|dgdgdgd ggd|tt|dggd d|tt|dgd d|tt|ddgdgd ggd|tjd dd}tt|ddddfd|dddfd|tt|ddddfd|dddfd|tjd d }tt|d|ddddd fdt|d|ddddd fdtt|d|dddddfdt|d|dddddfdttj t|d|dddddfd ttj t|d|dddddfdtjd d }tt|d|ddddd fdt|d|ddddd fdtt|d|dddddfdt|d|dddddfddS)NrrOrrRrrrP)rRrRrPrRrZrNrrrarX) r r&r9rr:repeatr{ concatenateTr r)rhrirrrs r? test_multidimzTestInsert.test_multidims YYK YY YY VAq1#&& 555VAq)))!444a888VAq!!,,,a000VAq!!,,,|||n=== Hq!fq!fq!f- . . IaOO " "1 % % - -a 3 3 N qqq!A#vY !Q..q1199!Q??A qqq!A#vY  ! ! ! VAsaS1#sO!<<>> IaLL A & &VAaaa!eHa111a4q9991===VAbqbdGQ!AAA#Q777;;; IbMM ! !) , ,VAq!AAAaaaF)"555Aq!AAAaaaF)!444 6 6 6VAq!AAAqF)"555Aq!AAAqF)!444 6 6 6 blFAq!AAAq!!!G*1EEEEblFAq!AAAq!!!G*2FFFF IbMM ! !) , ,VAq!AAAqqq!G*2666Aq!AAAqqq!G*1555 7 7 7VAq!AAAq!!!G*2666Aq!AAAq!!!G*1555 7 7 7 7 7rAcJtjd}tjtj5t |gdddddn #1swxYwYtjt 5t |gddddddS#1swxYwYdS)NrRrPrrnonsense)r9rr5raisesrr&rdrus r?test_0dzTestInsert.test_0d,s HQKK ]2< ( ( % % 1b!! $ $ $ $ % % % % % % % % % % % % % % % ]9 % % . . 1b!* - - - - . . . . . . . . . . . . . . . . . .s#AAA7BBBc hGddtj}tjd|}t t tj|ddg|t t tj|gg|t t tj|ddgddg|t t tj|tddddg|t t tj|tdddg|tjd|}t t tj|ddg|dS) NceZdZdS)*TestInsert.test_subclass..SubClassNrrrAr?SubClassr4rrArrrrRrPrXra) r9rr:rr isinstancer&rtrrhrris r? test_subclasszTestInsert.test_subclass3sr     rz    IbMM  x ( ( 29QA3//::;;; 29QB//::;;; 29QAA77BBCCC 29Qa aV<>>j1n Ia!Q % %1aV9bhs!'&B&B&BCCCCCrActjt5tjgdtjddgddgdddn #1swxYwYtjt5tjgdtjgt gddddS#1swxYwYdS)NrQrrrr!)r5r IndexErrorr9r&rfloatrs r?test_index_floatszTestInsert.test_index_floatsOsQ ]: & & A A Iiii3*!5!5Bx @ @ @ A A A A A A A A A A A A A A A ]: & & @ @ Iiii"E!:!:!:B ? ? ? @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @s#/AAA92B88B<?B<idxrNr|ctjtd5tjgd|gddgddddS#1swxYwYdS)Nz out of boundsmatchrQrZrN)r5rrr9r&)rhrs r?test_index_out_of_boundsz#TestInsert.test_index_out_of_boundsUs ]:_ = = = 0 0 Iiii#A / / / 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0sAA  A N)rrrrnrrrrrrr5r6r7rrrAr?rlrls444((7(7(7T... < < <--- DDD@@@  [UQG,,00-,000rArlceZdZdZdS)TestAmaxcgd}ttj|dgdgdgdg}ttj|dgdttj|d gd dS) NrZrNr[rr^r$@rZr"@rNrrrxrrrr)rrrrR)rrr)r r9amaxrs r?rnzTestAmax.test_basic]s & & &RWQZZ&&& ]] ^^ ]]  RWQQ''')9)9)9:::RWQQ''')9)9)9:::::rANrrrrnrrAr?rr[s#;;;;;rArceZdZdZdS)TestAmincgd}ttj|dgdgdgdg}ttj|dgdttj|d gd dS) Nrgrrrrr)rrrrR)rrr)r r9aminrs r?rnzTestAmin.test_basicis & & &RWQZZ&&& ]] ^^ ]]  RWQQ'''999RWQQ'''99999rANrrrAr?rrgs#:::::rArceZdZdZdS)TestPtpctjgd}t|ddtjgdgdgdg}t|dgdt|d gd t|dd gdgt|d d dggdS)Nrrr.@rrr)r@rra)rrrTrrbr)r9rr ptprs r?rnzTestPtp.test_basicus H/// 0 0QUUU]]D))) Hmmm$nn#mm% & & QUUU]]OOO444QUUU^^___555QUUDU11OOO3DEEEQUUU55w?????rANrrrAr?rrss( @ @ @ @ @rArceZdZdZdS) TestCumsumc gd}gdgdgdg}tjtjtjtjtjtjtjtjtj tj f D]}tj ||}tj ||}tj gd|}ttj |d|tj gdgdgd g|}ttj |d|tj gd gd gd g|}ttj |d |dS)NrRrPrrryr[rNrRrPrZrNr[ryrzr;rrZrNr[)rRrZr_r{#'rr)ryrxrr_)r)rRrZryr)r[rr)rr_rR)r9rXreint16uint16int32uint32rr complex64 complex128rr cumsum)rhbaba2ctyperia2tgts r?rnzTestCumsum.test_basicsc $ $ $||\\\===9grx29bhiRZm% ; ;EU##A#u%%B(555u==C ry333S 9 9 9(~~~/?/?/?@%IIC ry!444c : : :(1A1A1ABEKKC ry!444c : : : : ; ;rANrrrAr?rrs#;;;;;rArceZdZdZdS)TestProdcgd}gdgdgdg}tjtjtjtjtjtjtjtjfD] }tj ||}tj ||}|dvrBtttj |tttj |dst| dd t| dtj gd |t| d tj gd | dS) Nrrrr1rrRrr grp$Tra)r{bX)r9rrrrrrrrrr ArithmeticErrorprodr r rhrrrrirs r?rnzTestProd.test_basicsT $ $ $||\\\===9h 28RYj"*blBMK E EEU##A#u%%B ""orw:::orwA>>>>QVVV^^U333"2777??#%8,=,=,=u#E#EGGG"2777#3#3#%8OOOU#C#CEEEE E ErANrrrAr?rrs(EEEEErArceZdZdZdS) TestCumprodc tgd}gdgdgdg}tjtjtjtjtjtjtjtjfD]L}tj ||}tj ||}|dvrbtttj |tttj |dtttj |ttj |dtj gd |ttj |d tj gdgd gd g|ttj |dtj gd gdgdg|NdS)NrrrrrrRrar)rRrPri(irr)r[rQrr)rRrPryr{)r[rr)rrrr)r9rrrrrrrrrr rcumprodr rs r?rnzTestCumprod.test_basics $ $ $||\\\===9h 28RYj"*blBMK J JEU##A#u%%B ""orz1===orz2qAAAorz1===="2:ab#9#9#9#%8-@-@-@AF$H$HIII#2:bq#9#9#9#%8\\\-<__->->->-@AF$H$HIII#2:br#:#:#:#%8]]]-?-?-?-?-?-?-ABG$I$IJJJJ! J JrANrrrAr?rrs(JJJJJrArc>eZdZdZdZdZdZdZdZdZ dZ d S) TestDiffcgd}tjgd}tjgd}tjddg}tt||tt|d|tt|d|gd }tjgd }t t||gd }tjgd }tjd d g}tt||tt|d|dS)N)rRrNryrzrQ)rZrPrRr[)rararNrr[rPr=rZ)皙?皙@rgɿ皙)rg?g 皙?)TTFFr<T)r9rr rr )rhroutout2out3s r?rnzTestDiff.test_basics1   h|||$$x $$xA477C(((4Q<<<...4Q<<<... ' ' 'h,,,--DGGS))) & & &h+++,,xt %%477C(((4Q<<<.....rAc,tjd}d|dddddddf<tjd}d|dddddddf<tt |tjdtt |dtjdtt |dtjd tt |d|tt |d |t tjt|d t tjt|d tjd tj}t tt|dS)N)rrrrRrP)rrra)rrrr)r;rrrXrZr|gq?) r9rCrfr rr rrrre)rhrexps r? test_axiszTestDiff.test_axissY H\ " "!!!QTT111* gl##AAAqt!tQQQJ477BH\$:$:;;;4+++RXl-C-CDDD4???BH[,A,ABBB4???C0004+++S111blD!!4444blD!"5555 H]BJ / /j$*****rAc<dtdddz}|ddddddf|ddddddfz }|ddddddf|ddddddfz }|ddddddf|ddddddfz }|ddddddf|ddddddfz }tt||tt|d|tt|d |tt|dd |dS) NrrrrRrarPrrr)r=r)rr r)rhrout1rrout4s r?rzTestDiff.test_nds_ b"b!! !AAAqrr{Qqqq!!!SbSy\)AAAqqq!""H~QQQ3B3Y/QQQ{QssAAAqqqy\)ABB111H~SbS!!!QQQY/477D)))4Q<<<...4???D1114QQ///66666rAc xttdtttdfdtddD}ddgdgggg}t t dut t||dD]\}\}}t t|tj ut||t|j tjtt|t!dt|z dS) NrZrarc2g|]}t|S)r)r)r,r=rs r?r-z#TestDiff.test_n..s#4441$qA,,,444rArRr[rstart)listrgr rerr enumeraterDrr9rr r rrcrCmax)rhrHexpectedr=rrs @r?test_nzTestDiff.test_ns" qNNj$R00004444a 444FQCR(Q! !""""+C&,A,A"K"K"K 7 7 A# DII+ , , , sH - - - BG , , , S3q#a&&1*#5#5 6 6 6 6  7 7rActjddtj}tjddgdtjdgdg}|tjgdgdzt |dD]@\}}t || }t||t|j |j AdS) Nz 1066-10-13z 1066-10-16r!rRtimedelta64[D]rrZrr) r9r: datetime64rextendrrr r r)rhrrr=rrs r? test_timeszTestDiff.test_timess IlL F F F HaV#3 4 4 4 HaS 0 1 1 1  ",<===>BCCC222 / /FAsqA,,,C sC ( ( ( CI . . . . / /rAc >tjddgddgddgddgd d ggd d gd d gd d gd d gd d gg }t|}t|jdgdgdgdgdggt|jd gd gd gd gd ggt t|t|ut|d}t|jggggggt|jggggggt t|t|udS)NrRrPrZrNr[ryrzrxr;rFTmaskr)rrrr r>r rr)rhrrrs r?rzTestDiff.test_subclasssV Hq!fq!fq!fq!fq"g>!5>D%=!4=4,H I I I1gg38qcA3aS1#%>???38ugv'+ftfug&? @ @ @S T!WW$%%%A|||49r2r2r&:;;;49r2r2r&:;;;T d1gg%&&&&&rAc0tjddz}tt|dtjdtt|dgtjdttjtj|d|tt|ddgtjdtjddd}tj|dd }ddgddgg}t||tj|ddgdgg }t||tj|dd }ddgddgg}t||tj|dddgg }t||tttj|tj d ttj t|dd dS) Nr[rRr)prependraryrNrP)rr rZrZrZ)r r) r9r:r rrfrr{r rerCrrhrresultrs r? test_prependzTestDiff.test_prepends IaLL1 41---rwqzz:::4A3///<<<29RWQ%:%:%:;;Q???4B7333RWQZZ@@@ IaLL A & &A...FQF#68,,,aS1#J77768,,,A...FQF#68,,,aVH55568,,,j"'1bhuooFFFFblD!QQ??????rAcrtjd}t|d}gd}t||t|dg}t||t|ddg}|dgz}t||tjddd}tj|dd}dd gdd gg}t||tj|ddgdgg}t||tj|dd}ddgd d gg}t||tj|dddgg}t||t t tj|tjd t tjt|dd dS)Nr[r)append)rRrRrRrRr|rPrNrR)rrrar^rXrrZ)rr) r9r:rr r{r rerCrrs r? test_appendzTestDiff.test_append7s IaLLa"""###68,,,a$$$68,,,aA'''qc>68,,, IaLL A & &1---GaW%68,,,QC!:66668,,,1---FRH%68,,,QF844468,,,j"'1RXe__EEEEblD!AA>>>>>>rAN) rrrrnrrrrrrrrrAr?rrs///&+++ 7 7 7 7 7 7 / / / ' ' '@@@0?????rArceZdZdZdZdZdZdZdZdZ dZ d Z e j d ejd gd ggd Zd ZdS) TestDeletectjd|_tjddddd|_dS)Nr[rPrR)r9r:rir}r{nd_ars r? setup_methodzTestDelete.setup_methodVsD1IaLL''**221a;; rAc`t|j|}t|j|d}d|z}tt ||j|f|j|t |ddddf|jd|df}t||jddddf|dS)NrRrzDelete failed for obj: %r)err_msgr)rrirr r/)rhindicesa_delnd_a_delmsgxors r?_check_inverse_of_slicingz$TestDelete._check_inverse_of_slicingZstvw''$)W1555)G38E46'(+<==tv#& ( ( ( (x!!!Q'1gq=)ABB3 !AAAq& 13??????rAcgd}gd}|D]2}|D]-}|D](}t|||}||).3dS)N)rXrrRrPrNr[)r^rarRrZ)rtr!)rhlimsstepsrstopstepss r? test_sliceszTestDelete.test_slicescs&&& 6 6E 6 6!66DeT400A22155556 6 6 6rAc|tjddgddggtjt 5t |jdgdddn #1swxYwYtjt 5t |jdgdddn #1swxYwY|gd|gdtjt5t |jddddn #1swxYwYtjt5t |jd dddn #1swxYwYtjt5t |jd gd zddddS#1swxYwYdS) NrrRrPrhi)rrarPrP)TFFTFTFrN) r!r9rr5rrrrirers r? test_fancyzTestDelete.test_fancyls &&rx!Q!Q0@'A'ABBB ]: & & " " 46C5 ! ! ! " " " " " " " " " " " " " " " ]: & & # # 46D6 " " " # # # # # # # # # # # # # # # &&}}}555 &&'H'H'HIII]: & & ! ! 464  ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ]: & & " " 465 ! ! ! " " " " " " " " " " " " " " "]: & & & & 46E719 % % % & & & & & & & & & & & & & & & & & &sZA**A.1A.B11B58B5D%%D),D) E++E/2E/F66F:=F:cZ|d|ddS)Nrr|)r!rs r? test_singlezTestDelete.test_singles0 &&q))) &&r*****rAcFtjd}tjtj5t |gddddn #1swxYwYtjt 5t |gdddddS#1swxYwYdS)NrRrrr)r9rr5rrrrdrus r?rzTestDelete.test_0ds HQKK ]2< ( ( " " 1bq ! ! ! ! " " " " " " " " " " " " " " " ]9 % % + + 1bz * * * * + + + + + + + + + + + + + + + + + +s#AAA6BBBc TGddtj}|j|}t t t |d|t t t |g|t t t |ddg|t t t |tdd|t t t |tdd|dS)NceZdZdS)*TestDelete.test_subclass..SubClassNrrrAr?rr1rrArrrRrPrX)r9rrirrrrrtrs r?rzTestDelete.test_subclasss     rz    FKK ! ! 6!Q<<22333 6!R==(33444 6!aV,,h77888 6!U1a[[118<<=== 6!U1b\\22H==>>>>>rAc*tjdddd}t|t ddd}t |jj|jjt |jj|jjdS) NrrPr[rr<rRr) r9r:r{rrtr rrr)rhr`rBs r?test_array_order_preservez$TestDelete.test_array_order_preserves| IbMM ! !!Qc ! 2 2 1eBooA . . . QW)17+?@@@QW)17+?@@@@@rActjt5tjgdtjddgdddn #1swxYwYtjt5tjgdtjgt ddddS#1swxYwYdS)NrQrrr!)r5rrr9rrrrs r?rzTestDelete.test_index_floatss$ ]: & & 7 7 Iiii3*!5!5 6 6 6 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 ]: & & < < Iiii"E!:!:!: ; ; ; < < < < < < < < < < < < < < < < <  > 271::rx8<<< = = = > > > > > > > > > > > > > > > > > >s$ =EEE:8F??GGN)rrrrr!r)r+r-rrr4rr5r6r7r9rr:rArrAr?rrTs<<<@@@666&&&*++++++???AAA<<<  [Y1#(<==-->=->>>>>rArc6eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z ejd ejejejejgd Zejd ejejejejgdZejdejejejejgdZejdejejejejgdZdZdS) TestGradientcddgddgg}tj|}tjddgddggtjddgddggg}tt||tt||dS)NrRrZrNrrrr)r9rr r"rhvrdxs r?rnzTestGradient.test_basicsVaV  HQKKhR2r(+,,hR2r(+,,.8A;;+++8A;;+++++rAc tjtjd}gd}tjddd}t tjddt tjdtjdt tjd|t |dt |tjdt |||t ||dt ||dttd t |tj |gdzd ddS) Nr[)rrrr&@r?rPrRrz.*scalars or 1dra) r9rrfr:r{r"rrrestack)rhrG dx_unevenf_2ds r? test_argszTestGradient.test_argss9 Yrwqzz " "))) y}}$$Q** 1r"""1rx||,,,1r"""srx}}%%%y),,, r1 r"""" J(9 dBHbT!V"555q : : : : :rAc Htjddd}tjtjd}t t t||tjdt t t|dtjdt t t|tjdtjdt tt||t tt||dt tt||||t tt|dddt tt|||dt tt|ddddS)NrJr[rPrRrbr) r9r:r{rrfr rer"rd)rhrNrs r? test_badargszTestGradient.test_badargss3y}}$$Q** Ibgajj ! ! j(D!RWQZZ@@@j(D!RWQZZ@@@j(D"'!**bgajjIIIi4333i4????i4Aq999i4Aq999i4AA>>>>i4AA>>>>>>rActjgdd}tjgdd}tt||t |jtjdkdS)N)z 1910-08-16z 1910-08-11z 1910-08-10z 1910-08-12z 1910-10-12z 1910-12-12z 1912-12-12z datetime64[D]r!)rr^r=iirr9rr r"rrrhrrGs r?test_datetime64zTestGradient.test_datetime64s H 7 7 7! # # #X ) ) )"$$$ 8A;;+++BH%566677777rActjddgddggddgddgg}t|d}t t |t |t |j|jutjd}tjj |d<tj|d t|j gd dS) NrRrZrNFr rr[rP edge_order)FFTFF) r9rrr"r rr_maskr:maskedr r )rhrrx2s r? test_maskedzTestGradient.test_maskeds EKK!Q!Q($enuen=  ? ?qkk!nT#YYQ((( sy())) U\\!__ 1 B1%%%%27$F$F$FGGGGGrActjddd}|d|dz }d|dzzd|dzzzd|zz}d|dzzd|zzdz}tjtj||d |z dz }t tj|d kd ktjdtjtjd}d|dzzd|dzzzd|zz}d|dzzd|zzdz}tjtj||d |z dz }t tj|d kd kdS) NrrRrrPrZrNryrxrYgQ?T) r9linspaceabsr"rrrandomseedsort)rhrrGr analytical num_errors r?test_second_order_accuratez'TestGradient.test_second_order_accuratesw K1b ! ! qTAaD[ QJQ!V #a!e +aZ!a%'!+ FBK2!<<">">FF'"A"A"A5II&' $6M8J#K . . JAr*EEEDAvv!&:???DAvv!%*>>>D tT * * * tT * * * Q 3 3 3 Q 1 1 1ArjAAADAvA*EEED DJ$*, - - - gi 0 0 0ArjAAADAvA*EEED DJ$*, - - - tW - - - -&'$81o:N#O / / JAx!&:???DAx!%*>>>D tT * * * Q 3 3 3 Q 1 1 1AxaJGGGD gi 0 0 0AxaJGGGD g . . . .68%AFFF8V%AFFF47DG,,,47DG,,,DG]_555DG_55568%AFFF8V%AFFF47DG,,,47DG,,,DG]_555DG_55555rAcddgddgg}tj|}tjddgddggtjddgddggg}tt|d |dtt|d |dtt|d |dtt|d |d|dgt t|d |d|dgt t|d t|tt|d dd |ddz |ddz gt t t|dd d t tjt|d t tjt|d dS)NrRrZrNrrrrrrrarcrPr^)r9rr r"r r rdrrEs r?test_specific_axeszTestGradient.test_specific_axesksVaV  HQKKhR2r(+,,hR2r(+,,.8AA...16668AA...16668AB///A7778AF333beRU^DDD HQT222RUBqENCCCHQT222HQKK@@@ 8Aq!&999qE#Ir!uSy1 3 3 3 i1a;;;;blHaa8888blHab999999rActjgdd}tjgdd}tt||t |jtjdkdS)N)rr^rrQrTiA,rr!)rPrzrzrJwirUrVs r?test_timedelta64zTestGradient.test_timedelta64s H * * *" $ $ $X ( ( ("$$$ 8A;;+++BH%566677777rActjtjtjfD]S}tjgd|}t t |jtj|jTdS)Nrr!) r9float16rrrr r"rr)rhdtrs r?test_inexact_dtypesz TestGradient.test_inexact_dtypessg:rz2:6 > >B"---A !*BGAJJ,< = = = = > >rAchttjddttjddttttjddttttjddttttjddttttjddttttjdddS)NrPrRrYrZr)r"r9r:r rers r? test_valueszTestGradient.test_valuess1!,,,,1!,,,,j(BIaLLQGGGGj(BIaLLQGGGGj(BIaLLQGGGGj(BIaLLQGGGGj(BIaLLQGGGGGGrAf_dtypectjgd|}t|}t|dgt |zdS)N)r[rNrZrPrRr!rar9rr"r rC)rhrrogs r?test_f_decreasing_unsigned_intz+TestGradient.test_f_decreasing_unsigned_intsI H___G 4 4 4 QKK1rd3q66k*****rActj|j}tjddg}tjd|g|}t ||}t ||dzdzgdzdS)NrRrZrar!rP)r9iinforrr"r )rhrmaxintrrodfdxs r?test_f_signed_int_big_jumpz'TestGradient.test_f_signed_int_big_jumpsu'""& HaV   Hb&\ 1 1 11~~46A:!"3!4Q!677777rAx_dtypectjgd|}tjgd}t||}t|dgt |zdS)Nrr!rrPrNrXr)rhrrrors r?test_x_decreasing_unsignedz'TestGradient.test_x_decreasing_unsigneds^ HYYYg . . . HYYY  1~~4"c!ff-----rActj|j}tj|j}tjd|g|}tj|dzdg}t ||}t |ddgdS)Nrar!rPrri)r9rminrrr"r )rhrminintrrrors r?test_x_signed_int_big_jumpz'TestGradient.test_x_signed_int_big_jumps'""&'""& Hb&\ 1 1 1 Hfk1% & &1~~4#s,,,,,rActjddgddgf}tjrt|tusJdSt|t usJdS)NrRrPrZ)r9r"_using_numpy2_behaviorrtuplerrhr>s r?test_return_typezTestGradient.test_return_typesikAq6Aq6*++ $ & & %99%%%%%%99$$$$$$rAN)rrrrnrOrQrWr^rgrzr|rrrr5r6r7r9rerruint64rrXrrint64rrrrrrAr?rCrCs,,,:::2??? 8 8 8 H H H222(>6>6>6@:::4 8 8 8>>> H H H [Y29)+BI)?@@++@@+  [Y"()+28)=>>88>>8 [Y29)+BI)?@@..@@.  [Y"()+28)=>>-->>-%%%%%rArCceZdZdZdZdS) TestAnglec Pdtjddz dtjdzdz zdddddd g}t|}tjd tjd d tjdz tjtj dz tjd  tjtjd z g}t|d }tj|dztjz }t ||dt ||ddS)N?@rPrrVrRra?y@rrrT)degrr)r9sqrtrarctanpirr )rhrryozzos r?rnzTestAngle.test_basics RWQZZ#%RWQZZ!(;; BVW . !HH Ii IcNNAruqy"%"%# Yy ! ! !2529Y+?+?#?A !    Xb\\C "% '!!R,,,!!R,,,,,rActjddtjddz dzg}tjj|d<tjtjddtjdg}tjj|d<t |}tt|t|t|j |j t||dS)NrrRrP??rrr) r9rrrr\rrr rr )rhrrr s r?rzTestAngle.test_subclasss EKKBGAJJqLF$;< = =u|!5;; ) 4 4a3HIIel qT&\\4>>222V[(-000VX&&&&&rAN)rrrrnrrrAr?rrs2 - - -'''''rArc eZdZejgdZeeZee Z ee Z dZ dZdZdZdZdZejdejgd ejgd ejgd gd Zd ZdZdS) TestTrimZeros)rrrRrrPrZrNrc$d}fd|DS)N)rirrr\c38K|]}t|VdSr8)getattr)r,namerhs r? z'TestTrimZeros.values..s-;;d##;;;;;;rAr)rh attr_namess` r?rFzTestTrimZeros.valuess") ;;;; ;;;;rActjdd}|D]'}t|}t |||(dS)NrPrar9s_rFr2r rhslcarrr>s r?rnzTestTrimZeros.test_basicsTeAbDk;;== . .CS//C sCH - - - - . .rActjdd}|D])}t|d}t |||*dS)Nrartrimrrs r?test_leading_skipzTestTrimZeros.test_leading_skips[eCRCj;;== . .CSs+++C sCH - - - - . .rActjdd}|D])}t|d}t |||*dS)NrPrrrrs r?test_trailing_skipz TestTrimZeros.test_trailing_skips[eABBi;;== . .CSs+++C sCH - - - - . .rAc|D]i}tj||j}t |d}t |dksJt |d}t |dksJjdS)Nr!Brrro)rFr9 zeros_likerr2rC)rh_arrrrwrxs r? test_all_zerozTestTrimZeros.test_all_zerosKKMM " "D-DJ777Cc,,,Dt99>>>>c,,,Dt99>>>>> " "rAcltjd}t|}t||dSrb)r9rCr2r rhrr>s r?test_size_zerozTestTrimZeros.test_size_zero s0hqkkoo3$$$$$rAr)rlr)rlr)rlrcxtjdd}t|}t|||dS)NrRrP)r9rr2r )rhrrr>s r? test_overflowzTestTrimZeros.test_overflows8eAaCjoo3C)))))rAcptjgd}t|}t||dS)N)NrRN)r9rr2r rs r? test_no_trimzTestTrimZeros.test_no_trims6h''oo3$$$$$rAct|j}t|tsJdSr8)r2ritolistrrrs r?test_list_to_listzTestTrimZeros.test_list_to_list"s5))#t$$$$$$$rAN)rrrr9rrirrrcomplexrr=r\rFrnrrrrr5r6r7rrrrrAr?rrsF)))**A A A A<<<... ... ... """%%%  [ --- --- ---  " **  * %%% %%%%%rArc eZdZdZdZdZdS) TestExtinsc~tjgd}t|dk|}t|gddS)N)rRrZrPrRrPrZrZrR)rZrPrPrZrZ)r9rr r rs r?rnzTestExtins.test_basic)sE H*** + + AE1  1ooo.....rAcJtttgdddgddgtjgdtgdgdt gd ttjd gt tjdd tgd d d gt gdttdfdtjddgtddgdt ddgdS)NrTFrrR)rRrNrZrPr[rxrz)rrRrrRrrRrrPrNry)rRrPrZrNr[ryrzrzrx)rRrrRrrRrrRr;)rxrPr;rNrxryr;z!Cannot insert from an empty arrayc*tgdgS)N)rrrrrrRr)r,risr?z'TestExtins.test_place..=sE!-B-B-BB$G$GrA12349) r rdr,r9rr rCr:rrerus @r? test_placezTestExtins.test_place.sD i D%=1a&III H*** + + a&&& 2221333444 a!b!!!1bi1oo... a&&&A///1333444J(KGGGG I I I HdD\ " " a!Q1tSk*****rActd}|dk}|}t||}t||dt|||t ||dS)Nrrir)rrr r,r )rhrir acrs r? test_bothzTestExtins.test_bothDsl HH3w VVXX D!   aq aq1b!!!!!rAN)rrrrnrrrrAr?rr'sA/// +++,"""""rArrc0|tj|zSr8mathfloorrrs r?_foo1rPs TZ]]?rArc6|tj|z|zSr8r)rrrs r?_foo2rTs TZ]]?Q rAceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!Z#d"Z$d#Z%d$Z&d%Z'd&Z(d'Z)d(Z*d)Z+d*Z,d+Z-d,S)- TestVectorizecnd}t|}|gdgd}t|gddS)Nc"||kr||z S||zSr8rrirs r? addsubtractz.TestVectorize.test_simple..addsubtract[1uu1u 1u rArrZryr;rRrZr[rzrRryrRrPr5r rhrrors r? test_simplezTestVectorize.test_simpleZsS    k " " AlllLLL ) )1lll+++++rAcjd}t|}|gdd}t|gddS)Nc"||kr||z S||zSr8rrs r?rz.TestVectorize.test_scalar..addsubtractfrrArr[)r[rxrRrNrrs r? test_scalarzTestVectorize.test_scalaresO    k " " AlllA  1lll+++++rActjddd}td}||}t||dS)Nr^rPi'c|Sr8rrs r?rz*TestVectorize.test_large..rrA)r9r`r5r )rhrrors r? test_largezTestVectorize.test_largepsG KAu % % kk " " AaDD1a     rAc*ttj}tjddtjztjdtjzdtjzg}||}tj|}t ||dS)NrrirKrP)r5rcosr9rrr )rhroargsr1r2s r? test_ufunczTestVectorize.test_ufuncvso dh  xC"%Kbe QYGHH QtWW VD\\!"b)))))rAcdd}t|}tjgd}||}tjgd}t||||d}tjgd}t||dS)NrRc ||zSr8rrs r?rz(TestVectorize.test_keywords..foo q5LrArrrPrYr0r5r9rr rhrrorrrs r? test_keywordszTestVectorize.test_keywords}s     cNNx "" QtWW Xiii 2r""" QtQZZ Xiii 2r"""""rActttg}|tjdd}|tjd}t ||dSNotypesrrr5rrr9r:r rhrorrs r? test_keywords_with_otypes_order1z.TestVectorize.test_keywords_with_otypes_order1s` eUG , , ,Qry~~s # # Qry~~  2r"""""rActttg}|tjd}|tjdd}t ||dSr rrs r? test_keywords_with_otypes_order2z.TestVectorize.test_keywords_with_otypes_order2s` eUG , , ,Qry~~   Qry~~s # #2r"""""rAc0tttg}|tjd}|tjdd}|tjd}t ||t ||dS)Nrrrrrrhrorrr3s r? test_keywords_with_otypes_order3z.TestVectorize.test_keywords_with_otypes_order3s eUG , , ,Qry~~   Qry~~ % % % Qry~~  2r"""2r"""""rAc>tttg}|dd}|dd}|d}t|tddt|tddt|tddS)Nr$@rh)rrarr5rrr rs r?+test_keywords_with_otypes_several_kwd_args1z9TestVectorize.test_keywords_with_otypes_several_kwd_args1s eUG , , ,Qts^^^ Qtr]]] QtWWRts+++,,,Rtr***+++Rt%%%%%rActttg}|ddd}|ddd}t|tdddt|tddddS) Nrrhrra)rrrrRrPrZrrs r?+test_keywords_with_otypes_several_kwd_args2z9TestVectorize.test_keywords_with_otypes_several_kwd_args2s eUG , , ,Q # # # Qq!QZZR333444Rq!Q(((((rAcpddl} t|jdS#t$rt wxYwrb)rbr5 randrange ExceptionAssertionError)rhrbs r?test_keywords_no_func_codez(TestVectorize.test_keywords_no_func_codesP   # f& ' ' ' ' ' # # # "" " #s5c\d d}t|}tjgd}||}tjgd}t|||d|}t||||d}tjgd }t||dS) NrRc ||zSr8rrs r?rz5TestVectorize.test_keywords2_ticket_2100..foorrArrr)rrirPrrYr0r r s r?test_keywords2_ticket_2100z(TestVectorize.test_keywords2_ticket_2100s     cNNx "" QYYY Xiii 2r""" Qd^^^2r""" Qtq\\\ Xiii 2r"""""rAcd}tj|ddg}ddg}t||ddggd t||ddggd t||ddggddS) Nct|}|d}|r||z|dz}||Srb)rpop)rp_pr>s r? mypolyvalz;TestVectorize.test_keywords3_ticket_2100..mypolyvalsKaB&&))C *Agq ) *JrAr,rR)excludedrZryrr)rr,)r,)r9r5r )rhr.vpolyvalanss r?test_keywords3_ticket_2100z(TestVectorize.test_keywords3_ticket_2100s   < S!H===!f3Aq6YYY ? ? ?@@@3!Q999 = = =>>>3!Q ; ;<<<<_ks r?roz3TestVectorize.test_keywords4_ticket_2100..fs*C  r"v JrArRrPrZrNrrxrrhros r?test_keywords4_ticket_2100z(TestVectorize.test_keywords4_ticket_2100sN      111v!Q0001a&99999rAcftd}t|ddgddgddgdS)Nc*tj|Sr8)r9r)rFs r?roz3TestVectorize.test_keywords5_ticket_2100..fs71:: rArRrPrZrNrxrr8s r?test_keywords5_ticket_2100z(TestVectorize.test_keywords5_ticket_2100sK      11aVaV,,q!f55555rAcZd}t|}t|ddS)NcdSNrRrrrAr?rz5TestVectorize.test_coverage1_ticket_2100..foo1rArRrrhrros r?test_coverage1_ticket_2100z(TestVectorize.test_coverage1_ticket_2100s:    cNN1133"""""rAcd}t|}t|j|jd}t||}t|j|dS)Nc|S)zOriginal documentationrrs r?rz3TestVectorize.test_assigning_docstring..foo HrAzProvided documentation)doc)r5r __doc__)rhrrorFs r?test_assigning_docstringz&TestVectorize.test_assigning_docstringsa    cNNQY ,,,& cs # # #QY$$$$$rAcGdd}tt|jtjdtjddztt|j|tjdtjddzdS)NceZdZdZdZdS)9TestVectorize.test_UnboundMethod_ticket_1156..FoorPc||jzSr8r'rus r?rz=TestVectorize.test_UnboundMethod_ticket_1156..Foo.barsDF{"rAN)rrrrrrrAr?FoorKs(A # # # # #rArMr;rP)r r5rr9r:)rhrMs r?test_UnboundMethod_ticket_1156z,TestVectorize.test_UnboundMethod_ticket_1156s # # # # # # # # /9SSUUY// ! ==9Q<<1, . . .-9SW--cceeRYq\\BB9Q<<1, . . . . .rAcrtd}|tjd}|tjdd}td}|tjdd}|tjd}t||t||dS)Nc|Sr8rrs r?rz@TestVectorize.test_execution_order_ticket_1487..#rArZrc|Sr8rrs r?rz@TestVectorize.test_execution_order_ticket_1487..&rQrA)r5r9r:r )rhf1res1ares1bf2res2bres2as r? test_execution_order_ticket_1487z.TestVectorize.test_execution_order_ticket_1487!s {{ # #29Q<<  29S!$$%% {{ # #29S!$$%%29Q<<  UE"""UE"""""rAcftjd}d}t|||dS)Nc|Sr8rrs r?rz7TestVectorize.test_string_ticket_1892...1rAd0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789)r9r5r )rhror(s r?test_string_ticket_1892z%TestVectorize.test_string_ticket_1892,s6 L % % Q!rAcdgtfd}d|_tjd}t ||||zt dt |dS)Nrc.dxxdz cc<|dzS)NrrRrPr)r_callss r?roz#TestVectorize.test_cache..f6s! 1IIINIII6MrATr[)r5cacher9r:r r rC)rhrorras @r? test_cachezTestVectorize.test_cache2s        IaLL11Q44Q'''VAYA'''''rActjd}d|_tjd}t |||dS)Nc|Sr8rrs r?rz+TestVectorize.test_otypes..Ar\rArr[)r9r5rr:r rhrors r? test_otypeszTestVectorize.test_otypes@sE L % % IaLL11Q44#####rAc$ttjddgdgfttjddgdgfttjdddgdgfttjddgdgfttjd dgddgfttjd gd d gfttjd dgdgfttjddgdgfttjdddgdgfttjddgdgfttjddgddgfttjdgd d gftt5tjddddn #1swxYwYtt5tjddddn #1swxYwYtt5tjdddddS#1swxYwYdS)Nz(x)->()rrz (x,y)->()rz (x),(y)->()rz(x)->(y)z (x)->(y),()z(),(a,b,c),(d)->(d,e))r)rirr)r\)r\ez(x )->()z( x , y )->( )z (x),( y) ->()z( x)-> (y ) z (x)->( y),( )z*( ), ( a, b,c ) ,( d) -> (d , e)z (x)(y)->()z (x),(y)->z ((x))->(x))r nfb_parse_gufunc_signaturer rers r?test_parse_gufunc_signaturez)TestVectorize.test_parse_gufunc_signatureFsS0;;vh=MNNNS0==!lRD) + + +S0??v&- / / /S0<<h) + + +S0??h - / / /S01HII333j\B D D D S0<<x">NOOOS01BCC!lRD) + + +S0AAv&- / / /S01ABBh) + + +S01ABBh - / / /S0ACC333j\B D D D: & & 6 6  ' 5 5 5 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 : & & 5 5  ' 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 : & & 6 6  ' 5 5 5 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6s6#HH H#II I#JJ  J crd}t|d}|gdgd}t|gddS)Nc"||kr||z S||zSr8rrs r?rz8TestVectorize.test_signature_simple..addsubtractirrAz (),()->() signaturerrrrrs r?test_signature_simplez#TestVectorize.test_signature_simplehsX    k[ 9 9 9 AlllLLL ) )1lll+++++rActd}t|d}|ddgddgg}t|ddgdS)Nc*|Sr8rrs r?rz4TestVectorize.test_signature_mean_last..meants6688OrAz(n)->()rorRrZrPrNr)rhrrors r?test_signature_mean_lastz&TestVectorize.test_signature_mean_lastss[    di 0 0 0 A1v1v  1q!f%%%%%rAc|d}t|d}|ddgddgg}t|ddgddggdS) Nc0||z Sr8rtrs r?centerz3TestVectorize.test_signature_center..center|sqvvxx< rA(n)->(n)rorRrZrPrNrar)rhrxrors r?test_signature_centerz#TestVectorize.test_signature_center{sd    f 3 3 3 A1v1v  1AwQ011111rActdd}|gd}tt|tot |dkt |dgdt |dgddS)Nc ||fSr8rrs r?rz:TestVectorize.test_signature_two_outputs.. ArA ()->(),()rorrPrrR)r5rrrrCr rhrors r?test_signature_two_outputsz(TestVectorize.test_signature_two_outputss &&+ > > > AiiiLL 1e$$4Q15551Q4+++1Q4+++++rActtjd}|ddggd}t|gdgdg|ddggggd}t|gdgdggg|ddgddgggd}t|gdgdggdgdgg|ddggdgdg}t|gdgdggdgdggdS) Nz(a),(b)->(a,b)rorRrPrrrr)r5r9r<r rs r?test_signature_outerz"TestVectorize.test_signature_outersM bh*: ; ; ; Aq!fiii 1yyy)))4555 AAxj))) $ $1III 678999 A1v1v * *1 9995 ) 99957 8 8 8 Aq!fyyy))), - -1 9995 ) 99957 8 8 8 8 8rActdd}|gd}t|ddg|gdgdg}t|ddgddggdS) Nc|ddSNrarrs r?rz.#2#rAz(n)->(m)rorrRrPrrZrrs r?test_signature_computed_sizez*TestVectorize.test_signature_computed_sizes| &&* = = = AiiiLL1q!f%%% Ayyy)))$ % %11v1v./////rAcd d}t|ddh}t|gdgdt|gdd gddS) NrRc ||zSr8rrs r?rz2TestVectorize.test_signature_excluded..foorrA()->()r)rpr/rrrr'r0rrAs r?test_signature_excludedz%TestVectorize.test_signature_excludeds{     cX > > >11YYY<<33311YYY!,,,iii88888rActdddg}|gd}t|jtjdt |gddS)Nc|Sr8rrs r?rz5TestVectorize.test_signature_otypes..rrAryrrprr)r5r rr9r rs r?test_signature_otypesz#TestVectorize.test_signature_otypessa kkZ L L L AiiiLLQWbhy112221iii(((((rAcFttjd}ttd5|ddgdddn #1swxYwYtt d5|dddddn #1swxYwYtt d5|ddggddddn #1swxYwYttjd }ttd5|ddddddS#1swxYwYdS) Nz (n),(n)->(n)rozwrong number of positionalrRrPzdoes not have enough dimensions$inconsistent size for core dimensionrr)r5operatorr;rrdrer8s r?test_signature_invalid_inputsz+TestVectorize.test_signature_invalid_inputss* hln = = = ,H I I   Aq!fIII                =??   AaGGG               !BDD ! ! Aq!fiii  ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! hlh 7 7 7 ,H I I   AaGGG                  sGA  AA+ BB B$CCC< DDDctdd}ttd5|gddddn #1swxYwYtdd}ttd5|d dddn #1swxYwYtd d }ttd5|d d gddddS#1swxYwYdS) Nc|ddSrrrs r?rz>TestVectorize.test_signature_invalid_outputs..rrAryrorrc|Sr8rrs r?rz>TestVectorize.test_signature_invalid_outputs..rrAr~zwrong number of outputsrRc ||fSr8rrs r?rz>TestVectorize.test_signature_invalid_outputs..r}rArrP)r5rrer8s r?test_signature_invalid_outputsz,TestVectorize.test_signature_invalid_outputss &&* = = = BDD   AiiiLLL                kk[ 9 9 9 -F G G   AaDDD                &&( ; ; ; -F G G   Aq!fIII                  s5AA A4 B  BB>CC Cctjd}tjddgt}t t d5||dddn #1swxYwYd|_t|||tjdd }t t d5||dddn #1swxYwYtjd dd }t|||tjd d d }t|||tjdd }t||j|jtjddd }t t d5||ddddS#1swxYwYdS)Nc|Sr8rrs r?rz5TestVectorize.test_size_zero_output..r\rArr[r!rrc|Sr8rrs r?rz5TestVectorize.test_size_zero_output..r\rArroc|Sr8rrs r?rz5TestVectorize.test_size_zero_output..r\rArc|Sr8rrs r?rz5TestVectorize.test_size_zero_output..r\rAryc|Sr8rrs r?rz5TestVectorize.test_size_zero_output..r\rAc|gSr8rrs r?rz5TestVectorize.test_size_zero_output..sA3rAz()->(n)znew output dimensions) r9r5rCintrrerr rrfs r?test_size_zero_outputz#TestVectorize.test_size_zero_outputst L % % HaV3 ' ' ' X 6 6   AaDDD               11Q44### L 9 9 9 X 6 6   AaDDD                L E E E11Q44### L 3 G G G11Q44### L ; ; ;11QS6613''' L)C H H H -D E E   AaDDD                  s6 A  A$'A$7 CCC# F<<GGc\Gddtj}tjgdgdgdg|}tjgdgdgdg|}tjtjd }|||}t t||t |gd gd gd gtjd}|||}t t||t |||zdS)NceZdZdS)/TestVectorize.test_subclasses..subclassNrrrAr?rrrrAr)rrrro)rrrr)rrr)rrrz(m,m),(m)->(m)ro)rrr)rrr)rrrc ||zSr8rrs r?rz/TestVectorize.test_subclasses..s 1rA)r9rrrr5matmulr r)rhrrBrFmatvecrmults r?rzTestVectorize.test_subclassessM     rz    Hlll"ll"ll$ % %%)T(^^  HlllLLL,,,? @ @ E Eh O Obi3CDDD F1aLLT!WWh'''Q|||\\\BCCC|,,-- DAJJT!WWh'''QArAcJtjd}|jdksJdS)Nc ||zSr8rrs r?rVz#TestVectorize.test_name..f2s q5LrArV)r9r5r)rhrVs r? test_namezTestVectorize.test_names8     {d""""""rAcjtd}|gdgd}t|gddS)Nc"||kr||z S||zSr8rrs r?rz1TestVectorize.test_decorator..addsubtracts1uu1u 1u rArrrr)rhrrs r?test_decoratorzTestVectorize.test_decoratorsP      K lll 3 31lll+++++rAcntd}tjjdkr |jdksJdSdS)Nc|S) Docstringrrs r?roz'TestVectorize.test_docstring..fs HrArPr)r5sysroptimizerGr8s r?test_docstringzTestVectorize.test_docstringsN      9  ! !9 ++++ " !++rAcvd}t|d}tj|}|ddksJdS)Nc ||zSr8rrs r?rz'TestVectorize.test_partial..foorrArZrRrN)rr9r5)rhrrvbars r? test_partialzTestVectorize.test_partialsH   c1oo|C  tAww!||||||rActddgd}|gd}t|jtjdt |gd|jdksJdS)Nryrrc|Sr8rrs r?roz8TestVectorize.test_signature_otypes_decorator..f#rErArro)r5r rr9r rrs r?test_signature_otypes_decoratorz-TestVectorize.test_signature_otypes_decorator"s Z < < <   = <  AiiiLLQWbhy112221iii(((zS      rActt5tjd}ddddS#1swxYwYdS)NrZ)pyfuncr rdr9r5rhAs r?test_bad_inputzTestVectorize.test_bad_input,s 9 % % ) ) a(((A ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )s 8<<ctt5tjdd}ddddS#1swxYwYdS)NstringcdS)NrrrrAr?rz+TestVectorize.test_no_keywords..foo2surAr)rhrs r?test_no_keywordszTestVectorize.test_no_keywords0s 9 % %   \( # #  $ #                   sAAActddg}|dg}t|jtjddS)Nc|Sr8rrs r?rz?TestVectorize.test_positional_regression_9477..:rQrArrP)r5r rr9rs r?test_positional_regression_9477z-TestVectorize.test_positional_regression_94776sH {{i[ 1 1 AqcFFQWbhy1122222rAN).rrrrrrrr rrrrrr$r(r2r9r<rBrHrNrYr^rcrgrlrqrurzrrrrrrrrrrrrrrrrrrrAr?rrXs , , , , , ,!!! *** # # #### # # # # # # & & & ) ) )######" = = = : : :666### % % % . . . # # # ( ( ($$$ 6 6 6D , , ,&&&222,,,888 000999)))      6(### , , ,,,,!!!))) 33333rArceZdZGddZeje dejddej fdgdZ d S) TestLeaksc.eZdZdZdZedZdS) TestLeaks.ArcdSrbrrhrs r?boundzTestLeaks.A.boundCr@rAcdSrbr)rs r?unboundzTestLeaks.A.unboundFs1rAN)rrritersr staticmethodrrrAr?rr@sC          rArPython lacks refcountsreasonz name, incrr)rrcddl}t|j|}| t j|}t |jjD]f}|}tj t||dd|_ | tj d}gd}tt j|||zt dD]}| tt j|||dS#|wxYw)NrrRrr[)gcrrdisabler getrefcountrgrr9 frompyfuncror:r collectenable) rhrincrrA_funcrefcountrrirs r?test_frompyfunc_leakszTestLeaks.test_frompyfunc_leaksJs%  &&  v..H46<(( ) )FFHHmGAt$4$4a;;cc")B--((A 00(T/ B B B1XX   00( ; ; ; IIKKKKKBIIKKKKs DEEN) rrrrr5r6skipifrr7rrrrAr?rr?s [L(1IJJ [\ ag  , KJ rArceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z ejd dZdS) TestDigitizectjdd}tjdd}tt||tjddS)Nr#r[rrr9r:r rrhrbinss r? test_forwardzTestDigitize.test_forwardjsL Ib!  yQ8At,,bimm<<<<.ANrrrAr?rrrrArr[rRrZFT)r9rr:rrrr)rhrrirs r?rzTestDigitize.test_return_types          IaLL  a  IaOO  # #Jx1e44a888999Jx1d33Q77788888rAc`d}ttj||dz |dzgddSNlrRr r9rrs r?test_large_integers_increasingz+TestDigitize.test_large_integers_increasings5 R[QUAEN33Q77777rAz8gh-11022: np.core.multiarray._monoticity loses precisionrc`d}ttj||dz|dz gddSrrrs r?test_large_integers_decreasingz+TestDigitize.test_large_integers_decreasings7 R[QUAEN33Q77777rAN)rrrrrrrrrrrrrrr5r6xfailrrrAr?rrhs=== === /// BBBCCC CCC 777 555 444999888  [IKK88KK888rArceZdZdZdZdS) TestUnwrapc ttdddtjzzgddgt tjt ttddztjkdS)NrRrPrrh)r r3r9rrrrrrs r?rzTestUnwrap.test_simplesl61a!be)m"4551v>>>tF488c>2233be;<<=====rAc ttddgdddgttjt tt ddzddktjgd}tj|d}tt|d|tjgd }tj|d }t|d }t|gd t|d d }t|gd |j |j ksJdS)NrRiperiodrPr)rKr~)rrrrr~i)rrrrr~r)rdiscont) r r3rr9rrrrmodr)rh simple_seqwrap_seq uneven_seq wrap_uneven no_discont sm_disconts r? test_periodzTestUnwrap.test_periods?61g,s;;;aVDDDtF488d?3???@@3FGGHHHX44455 6*c**6(3777DDDX999:: fZ-- K444 :'B'B'BCCCKSAAA :'B'B'BCCC;#4444444rAN)rrrrrrrAr?r r s2>>> 55555rAr rO AllIntegerFloatM)rrRrcreZdZdededdfdZdededdfdZdededdfdZdededdfdZdededdfd Z dS) TestFilterwindowsrr!returnNctj||d}t|}|dkr tj}n$tj|jtj}|j|ksJt |t||dkr$t|tjgdS|dkr$t|tj ddSttj |ddddS) Nr!rrrRrr@rN) r9rr$rrrr r!r rfr rrhrr!scalarr ref_dtypes r? test_hanningzTestFilterwindows.test_hannings!5)))"- FOO C<< IIv|RZ@@Iw)#### Qq """ A:: q"(2,, / / / / / q[[ q"'!** - - - - - qq 1 1 15! < < < < #>#>$>>>>*>3>3>4>>>>*>3>3>4>>>>*;;;;;;;;;rAr#c eZdZdZdZdZdS) TestTrapzctjddd}ttjd|dzztjdtjzz d}t |dddS) NrrrkrP)rGrRrz)r9r:r1rrrr )rhrrs r?rzTestTrapz.test_simpleJsf Ic2r " " "&qAv&&RU););; D D DAq!$$$$$rAc2tjddd}tjddd}tjddd}tj||d|dz z}|dxxdzcc<|dxxdzcc<tj||d|dz z}|dxxdzcc<|dxxdzcc<tj||d|dz z}|dxxdzcc<|dxxdzcc<|ddddf|ddddfz|ddddfz}||ddddfzd}||ddddfzd} ||ddddfzd} t ||ddddfd } t | |t ||ddddfd } t | | t ||ddddfd } t | | t ||d } t | |t ||d } t | | t ||d } t | | dS) NrrRrZrPrxr_rar)rr)r9r` ones_likerr1r ) rhrrrwxwywzqqxqyqzrs r? test_ndimzTestTrapz.test_ndimPs K1a  K1a  K1b ! ! \!__!qt , 1  2!  \!__!qt , 1  2!  \!__!qt , 1  2!  aaatm qaaa .4aaa< @"QQQd]## ( (a ( 0 0"T111d]## ( (a ( 0 0"T4]## ( (a ( 0 0 !qD$'a 0 0 0Ar""" !qaaaQ / / /Ar""" !qtAAAQ / / /Ar""" !qq ! ! !Ar""" !qq ! ! !Ar""" !qq ! ! !Ar"""""rActjd}||z}|dk}tj||}d}t t |||tj||}t t |||tj||}t t |||dS)Nr[rPr g*@)r9r:rrr r1)rhrrr ymrxms r?r^zTestTrapz.test_maskedus IaLL EAv U[[[ & & E"aLL!,,, U[[[ & &E"bMM1--- U[[[ & &E!RLL!,,,,,rAN)rrrrrBr^rrAr?r6r6HsB%%% ######J-----rAr6ceZdZdZdZdS)TestSinccttddkttjddd}t |t |ddS)NrrRrarhrz)rr0r9r`r r!rhrs r?rzTestSinc.test_simplesSQ1  RC(( ) )!!VAYY22222rAcddg}ttj|}tt|}tt |}t ||t ||dS)Nrri)r0r9rrrr )rhrrrrs r?test_array_likezTestSinc.test_array_likesh H "(1++   $q'']] %((^^2r"""2r"""""rAN)rrrrrKrrAr?rGrGs2333 #####rArGceZdZdZdS) TestUniquec,tjgd}ttjt |gdktt tjgdtjdgkgd}ttjt |gdktjgd}ttjt |gdkdS) N) rNrZrPrRrRrPrZrNr)rrRrPrZrN)rRrRrRrRrRrR)widgethamrrrrP)rrrPrO)@@r?$@rrQ)rrRrQr)r9rrrr4rs r?rzTestUnique.test_simples H000 1 1vayyOOO344555rx0011RXqc]]BCCC 9 9 9vayy$C$C$CCDDEEE H::: ; ;vayy$A$A$AABBCCCCCrANrrrrrrAr?rMrMs(DDDDDrArMceZdZdZdZdS)TestCheckFinitecgd}ddtjg}ddtjg}tj|t t tjj|t t tjj|dS)NrrRrP)r9infrZlibasarray_chkfiniter rers r?rzTestCheckFinite.test_simpleso II 26N 26N   ###j"&":A>>>j"&":A>>>>>rAcgd}tj|dtj}t |jtjkdS)Nrr)rr)r9rXrYrrrrus r?test_dtype_orderz TestCheckFinite.test_dtype_ordersF II F $ $Qc $ D D2:%&&&&&rAN)rrrrr[rrAr?rUrUs2???'''''rArUc eZdZejgdgdgdgZejgdgdgdgZejgdgdgd gZejgd gd gd gd gdgdgZdZ dZ dZ dZ dZ dZdZdZejdejejejejgdZdS) TestCorrCoef)g^?g.{<?g$,?)g4?gp)?gJG?)gY?g?gވ0v?)gة?gBFU?gG)t?)g3TB?gL\?gA?)g> ?gqFE`?g5?)rޛ?r@)r^rp]4?)r_r`r)rr^r_?'L?(8?4V}?)r^rr`w{,H5?jS, ?)r_r`r)h?5#A?)rardrgrtoB C,)rbrerhrjr@02v?)rcrfrirkrlrc`ttjgdgdddgddggdS)Nr)rRrrRr)r r9rrs r?test_non_arrayzTestCorrCoef.test_non_arraysDBK 999== #Yr 3 5 5 5 5 5rAct|j}t||jt t jt j|dkt|j|j}t||j t t jt j|dkdSr5) rrr rwrr9rrarrx)rhtgt1tgt2s r?rzTestCorrCoef.test_simplesD$),,,rvd||s*++,,,''D$),,,rvd||s*++,,,,,rAcTt5}tjdttt |jd|ttt |jd|j tt |j|j d|j tt |jd|j tt |j|j d|j ddddS#1swxYwYdS)NrrraddofrZ) rru simplefilterrDeprecationWarningrrfilterr rwrrxrhsups r? test_ddofzTestCorrCoef.test_ddofsL   MC  !( + + + +XtvB G G G G JJ) * * * b 9 9 949 E E E b A A A49 M M M a 8 8 8$) D D D a @ @ @$) L L L M M M M M M M M M M M M M M M M M MsDDD!$D!c t5}tjdttt |j|jddttt |jd|ttt |jd|j ddddS#1swxYwYdS)NrrrRr)bias) rrurvrrwrrrrxr rwrys r? test_biaszTestCorrCoef.test_biass   EC  !( + + + +Xtvtvq! L L L +XtvA F F F F JJ) * * * a 8 8 8$) D D D  E E E E E E E E E E E E E E E E E EsB"B>>CCctjgdgdg}t|}tjddgddgg}t||t tjtj|dkdSNrrVy@r_rrrV)r9rrrrrra)rhrr>rs r? test_complexzTestCorrCoef.test_complexs Hiii. / /qkkhT S"I.//S!!!rvc{{c)**+++++rActjgdg}tjgdg}ttj||tjddgddggdSr)r9rrrrhrrs r?test_xyzTestCorrCoef.test_xyse Hiii[ ! ! Hlll^ $ $ Aq))28b$Z#r4K+L+LMMMMMrAc tjd5tjdtt t t jgt jt t t jg ddt jg ddt t t jg ddt jt jt jgt jt jggddddS#1swxYwYdSNTrprrrrP) rurvrvRuntimeWarningr rr9rrZr{rs r? test_emptyzTestCorrCoef.test_emptysm  $D 1 1 1 O O  !(N ; ; ; x 55rv > > > x (<(> test_typecv|j|}t||}||jksJdSNr!)rrrr)rhrcast_Ar>s r?test_corrcoef_dtypez TestCorrCoef.test_corrcoef_dtype s@y))vY///CI%%%%%%rAN)rrrr9rrrrwrxrorr{r~rrrrr5r6r7halfsingledouble longdoublerrrAr?r]r]s - - - - - - , , , . / /A  , , , - - - , , , . / /A 28 % % % $ $ $ & & & ( ) )D 28 I I I I I I K K K M M M J J J J J J  L M MD555--- M M MEEE,,,NNN OOO*** [[27BIry"-*XYY&&ZY&&&rAr]ceZdZejddgddgddggjZejddgddggZejgddZejgdZ ejd gdgdgdgdgd ggjZ ejd d gd d ggZ ej d ej ZejgdZejddgddggZej d ZejgdZdZdZdZdZdZdZdZdZdZdZejdej ej!ej"ej#gdZ$dS) TestCovrrPrRrrn)rrr)ndmin)rRrNrRrr皙?gٿrZr!)rrrgUUUUUU?gUUUUUU)g&S:?go_?g#~j?gZӼ?gQ|?cTtt|j|jdSr8)rrx1rwrs r?rnzTestCov.test_basic s"DG di00000rAc tjgdgdg}tjddgddgg}tt||tt|tjd|dS)NrrrrrVrZaweights)r9rrrrf)rhrr>s r?rzTestCov.test_complex s{ Hiii. / /hT S"I.//A$$$A 333S99999rActjgdg}tjgdg}tt||tjddgddggdSr)r9rrrrs r?rzTestCov.test_xy% sa Hiii[ ! ! Hlll^ $ $Aq 28b$Z#r,C#D#DEEEEErAc tjd5tjdtt t t jgt jt t t jg ddt jg ddt t t jg ddt jt jt jgt jt jggddddS#1swxYwYdSr) rurvrvrr rr9rrZr{rs r?rzTestCov.test_empty* sm  $D 1 1 1 O O  !(N ; ; ; s28B<<00"& 9 9 9 s28B<<#7#71#=#=>>!x||33Aq99 ; ; ; s28B<<#7#71#=#=>>!x"&"&)9BFBF;K(LMM O O O  O O O O O O O O O O O O O O O O O Orc btjd5tjdtt t |jdtjtj tj gtj tj ggddddS#1swxYwYdS)NTrprrr[rt) rurvrvrr rrr9rrWrs r?test_wrong_ddofzTestCov.test_wrong_ddof3 s  $D 1 1 1 > >  !(N ; ; ; s47333!x"&26'):+-6'26):)< = = > > > > > > > > > > > > > > > > > > > > >sBB$$B(+B(ctt|jt|jdtjgd}tt|j|t|j|ddS)NF)rowvar)g+?g4@?g&S?go_?gJ4?)rrx3r9rrhrs r?test_1D_rowvarzTestCov.test_1D_rowvar: spDG c$'%&@&@&@AAA H=== > >DGQTWa)F)F)FGGGGGrActt|jdtj|jddS)NrRrt)rrrr9varrs r?test_1D_variancezTestCov.test_1D_variance? s8DG!,,,bfTW1.E.E.EFFFFFrActt|j|jt|jtt|j|j|jtt|j|j|j|jdz}ttt|j|tj dtj }ttt|j|tj dtj }ttt|j|dtj dtj z}ttt|j|dS)N)fweightsrirPrZr!rPrarZ)rrr] frequencies x2_repeatsrrxunit_frequenciesrwr rdr9rfrc RuntimeErrorre)rhnonintros r? test_fweightszTestCov.test_fweightsB sIDGd.>???DO,, . . .DGd.>???  # # #DGd.CDDD  # # #!C'idg???? GF"' * * *lC1==== GARW % % %lC1==== "'*** *j#tw;;;;;;rActt|j|j|jtt|jd|jzt|j|jtt|j|j|jtjd}ttt|j|tjd}ttt|j|dtjdz}ttt|j|dS)NrrrrPrnrZ) rrrrry unit_weightsrwr9rfr rrerIs r? test_aweightszTestCov.test_aweightsR sDGdl;;;TYGGGDGcDL.@AAADGdl;;; = = =DGd.?@@@$)LLL GFOOlC1==== GAJJlC1==== 271:: j#tw;;;;;;rActt|j|j|jt|jtt|j|j|j|jtt|j|j|j|j tt|j|j|j |j tt|j|jd|j zt|j|j tt|j|j|j|j dS)N)rrrr) rrr]rrrrrxrrwrryrs r?test_unit_fweights_and_aweightsz'TestCov.test_unit_fweights_and_aweights^ szDGd.>%)%6888DO,, . . . DGd.>%)%6888  # # # DGd.C%)%6888  # # # DGd.C%)\333  # # # DGd.C%(4<%7999DGdl;;; = = = DGd.C%)%6888  # # # # #rArcv|j|}t||}||jksJdSr)rrrr)rhrcast_x1r>s r?test_cov_dtypezTestCov.test_cov_dtyper s@'..++'+++CI%%%%%%rAN)%rrrr9rrrrwr]rrrxrfrcrrryrrrnrrrrrrrrrr5r6r7rrrrrrrAr?rr s* Aq6Aq6Aq6* + + -B 28b#Yb * + +D /// + + +B"(999%%KC53%#useDEEGJ 28c4[4+. / /Drwq000bh''G 28gx(8W*=> ? ?D271::L ::: ; ;B111::: FFF OOO>>>HHH GGG<<< < < <###( [[27BIry"-*XYY&&ZY&&&rArc eZdZdZdZdZdS)Test_I0c ttdtjdtjgd}tjgd}tt||tt| |tjddgddgd d gd d gd dgg}tt|tjddgddgddgddgddggtjdg}t |jdt tjdgtjdgdS)Nrigǒ#?)gC7?gmM?g3a,?g$WV?ۏ?r)gֿN^?gE ?g閡1?g%m??gDn@g>@v?guͥ?gVX˳?gua&?gB-sJ?g-^ ?g_V?g X.?gVaX?gj?gp?g~-@?gD/b?gݦ?g,Y1ؔ?g[!?gQL?gWsU+?g@?gfxw?rr0r)r r%r9rr rEr )rhrrri0_0s r?rzTest_I0.test_simple{ si sGG H' ( ( * * * HIII J J8[[[\\BqEE8,,,BrFFH--- Hx,!:.!:.!:.!:. 0 1 1  qEE Hz:.!:.!:.!:.!:. 0 1 1 2 2 2ubT{{TZ&&&25";;"77777rAc&tjdGfdd}ttj||sJtj}tj|}t ||dS)NrNc$eZdZjZdZdS)*Test_I0.test_non_array..array_likec|Sr8r)rhrs r?__array_wrap__z9Test_I0.test_non_array..array_like.__array_wrap__ s rAN)rrr__array_interface__rrsr? array_liker s-"#"7      rAr)r9r:rrar%r )rhrrr>ris @r?rozTest_I0.test_non_array s IaLL          "&.. ;;;;;eAhheJJLL!!3$$$$$rActjddg}tjtd5t |}ddddS#1swxYwYdS)Nry?@z#i0 not supported for complex valuesr)r9rr5rrdr%)rhrir>s r?rzTest_I0.test_complex s Ha[ ! ! ]9,Q R R R  Q%%C                  sAAAN)rrrrrorrrAr?rry sA8888%%% rArceZdZdZdZdS) TestKaisercttjtddt tddtjgt tddtjddgt tddtjgdt tddtjgd dS) NrRrrrP oF?r[)rvT=?rrrgg?)?YJ?rrr)rr9isfiniter(r rrs r?rzTestKaiser.test_simple s F1cNN++,,,F1cNNHRLL * * *F1cNNHj*%=>> @ @ @F1cNNH&>&>&>?? @ @ @ F1g..H&>&>&>?? @ @ @ @ @rAc&tdddS)NrZrN)r(rs r? test_int_betazTestKaiser.test_int_beta sq! rAN)rrrrrrrAr?rr s5 @ @ @rArceZdZdZdS) TestMsortc tjgdgdgdg}tjtd5t t |tjgdgdgdgddddS#1swxYwYdS) N)\?bCQ?הs?)w? kl?*dta?)ӯ?6H?ܘC?zmsort is deprecatedr)rrr)rrr)rrr)r9rr5warnsrwr r*rs r?rzTestMsort.test_simple s H:::::::::< = =\,4I J J J B B a>>>>>>>>>@AA B B B B B B B B B B B B B B B B B B B B Bs9A??BBNrSrrAr?rr s( B B B B BrArcPeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d S) TestMeshgridc tgdgd\}}t|tjgdgdgdgdgt|tjgdgdgdgdgdS)NrrNr[ryrzr)r[r[r[)ryryry)rzrzrzr)r r9rrhXYs r?rzTestMeshgrid.test_simple s)))\\\22A1bh (1 (1 (1 (344 5 5 5 1bh (1 (1 (1 (344 5 5 5 5 5rActtgd\}t|tjgddS)Nrr)rhrs r?test_single_inputzTestMeshgrid.test_single_input s9|||$$1bh|||4455555rAclg}tgt|tgt|ddidS)NrF)r r)rs r? test_no_inputzTestMeshgrid.test_no_input s>2x///2x:E::;;;;;rAc gd}gd}t||d\}}t|tjgdgdgdgt|tjgdgdgdgdd g}t t||d jd kt t||dd jd kt t|||d jd kt t|||dd jdkt tt||ddS)Nrrijindexingr)rPrPrPrP)rZrZrZrZrxr;rrRrZrN)rNrZrP)rZrNrPnotvalid)r)r r9rrrEr re)rhrrrrrs r? test_indexingzTestMeshgrid.test_indexing sn II LL!Q...A1bh (4 (4 (677 8 8 8 1bh (4 (4 (677 8 8 8 FAq!'61222A---a06&@AAAAq!!!$*i7888Aq400039YFGGGj(Aq:FFFFFFrAc tgdgdd\}}t|tjgdgt|tjdgdgdgdggdS) NrrTsparserNr[ryrzrrs r? test_sparsezTestMeshgrid.test_sparse ss)))\\\$???A1bh {334441bhaS1#s';<<=====rAcJtttgdgdddS)Nrrr)r)r rdr)rs r?test_invalid_argumentsz#TestMeshgrid.test_invalid_arguments s8 iiit = = = = = =rAc~tjddtj}tjddtj}tj||\}}t |j|jkt |j|jktj||d\}}t |j|jkt |j|jktj||d\}}t |j|jkt |j|jkdS)Nrrr!rT)rr)r9r:rrr)rr)rhrrrrs r?rzTestMeshgrid.test_return_type s Ia2: . . . Ib"BJ / / /{1Q117"###17"###{1QT***117"###17"###{1Qt,,,117"###17"#####rActjddg}tjddg}tj||dd\}}d|dddf<t|dddfdt|d ddf|dS) Nrrgffffff @g@FT)rrrrR)r9rr)r )rhrrrrs r?test_writebackzTestMeshgrid.test_writeback s Hc3Z  Hc3Z {1aD9991!QQQ$Qq!!!tWa   Qq!!!tWa     rAc4tjdtddD\}}}}}d}t|j|t|j|t|j|t|j|t|j|dS)Nc3"K|] }dg|zV dS)rNrr,rs r?rz-TestMeshgrid.test_nd_shape..' s(%C%C!qcAg%C%C%C%C%C%CrArRry)rPrRrZrNr[)r9r)rgr rE)rhrirrr\riexpected_shapes r? test_nd_shapezTestMeshgrid.test_nd_shape& s %C%CuQ{{%C%C%CD 1aA(QWn---QWn---QWn---QWn---QWn-----rActjdgddggd\}}}t|gdggdggt|gdggdggt|gdggdggdS)NrrRrPrYrrrOr9r)r rs r?test_nd_valueszTestMeshgrid.test_nd_values/ s+qcAq6999551aQ)))yyyk2333Q)))yyyk2333Q)))yyyk233333rActjdgddggdd\}}}t|gdgdggt|gdgd ggt|gdgdggdS) NrrRrPrYrrrrrOrrs r?test_nd_indexingzTestMeshgrid.test_nd_indexing5 s+qcAq6999tDDD1aQ)))YYY/0111Q)))YYY/0111Q)))YYY/011111rAN)rrrrrrrrrrrrrrrrAr?rr s 5 5 5666<<< GGG(>>> ===$$$0!!!...444 22222rArcDeZdZdZdZdZdZdZdZdZ dZ d Z d S) TestPiecewisec tddgddgdg}t|ddgtddgddggdg}t|ddgtddgtjddgdg}t|ddgtddgtjddgdg}t|ddgtddgtjddggdg}t|ddgtddgddggdg}t|ddgt t dtddgddgggt t dtddgddgggddS) NrTFrRcdSrrrs r?rz+TestPiecewise.test_simple..S srAraz1 or 2 functions are expectedr)r+r r9rrrers r?rzTestPiecewise.test_simple> s q!ftUmaS 1 11q!f%%% q!fe}os 3 31q!f%%% q!fbhe}55s ; ;1q!f%%% q!fbh1v.. 4 41q!f%%% q!frxA//01# 6 61q!f%%% q!ft}o } = =1q"g&&&J(G 1v  4 4 4J(G 1v   ; ; ; ; ;rAc`tddgddgddggddg}t|ddgdS)NrRrPTFrZrNr+r rs r?test_two_conditionsz!TestPiecewise.test_two_conditions[ sB q!fe}udm.subclassNrrrAr?rr rrArrrrN)rnrr)rnrnrrr)r9rr:rr+r r)rhrrrs r?rzTestPiecewise.test_subclasses s     rz    IbMM  x ( ( a!B$1}}} 5 5T!WWh'''Q.../////rAN) rrrrr rrrrrrrrrAr?rr< s;;;:&&&&&& ! ! ! 9 9 9 88800000rArceZdZdZdZdZdZdZdZdZ dZ d Z d Z e je d d Ze jdddggdgdZdS) TestBincountctjtjd}t|tjddS)NrN)r9bincountr:r rfrs r?rzTestBincount.test_simple s7 K ! % %1bgajj)))))rActjtjgd}t|tjgddS)N)rRr[rPrNrR)rrPrRrrRrR)r9rrr rs r? test_simple2zTestBincount.test_simple2 sF K11 2 21bh'9'9'9::;;;;;rActjd}tjgd}tj||}t ||dS)NrN)皙?333333?rir)r9r:rrr rhrrrs r?test_simple_weightzTestBincount.test_simple_weight sL IaLL H))) * * K1  1a     rActjgd}tjgd}tj||}t|tjgddS)NrRrPrNr[rPr#r$rirr#)rr#rirrirr9rrr r%s r?test_simple_weight2z TestBincount.test_simple_weight2 sc H___ % % H... / / K1  1bh'A'A'ABBCCCCCrActjgd}tj|d}t|tjgdg}tj|d}t|tjgdS)N)rrRrrRrRrZ minlength)rPrZrrr*rs r?test_with_minlengthz TestBincount.test_with_minlength s~ H___ % % KQ ' ' '1bhyyy11222  KQ ' ' '1bhrll+++++rActjgd}tj|d}t|tjgdtj|d}t|tjgddS)N)rrRrRrPrPrZrZrPr-)rRrPrPrPrr*rs r?)test_with_minlength_smaller_than_maxvaluez6TestBincount.test_with_minlength_smaller_than_maxvalue s H*** + + KQ ' ' '1bh|||44555 KQ ' ' '1bh|||4455555rActjgd}tjgd}tj||d}t|tjgddS)Nr(r)rx)rr#rirrirrrr*r%s r?test_with_minlength_and_weightsz,TestBincount.test_with_minlength_and_weights se H___ % % H... / / K1a 1bh'G'G'GHHIIIIIrActjgt}tj|}t ||dSr)r9rrrr rs r?rzTestBincount.test_empty s9 HRs # # # KNN1a     rActjgt}tj|d}t |tjdtdS)Nr!r[r-)r9rrrr rCrs r?test_empty_with_minlengthz&TestBincount.test_empty_with_minlength sO HRs # # # KQ ' ' '1bhq44455555rAc.tjgtttdfdtt dfdtjdttdfdtt dfddS) Nr!z"'str' object cannot be interpretedc0tjdSNfoobarr-r9rrsr?rz. BKX$F$F$FrAzmust not be negativec0tjdSNrar-r;rsr?rz. BKR$@$@$@rAr[c0tjdSr9r;rsr?rz. r<rAc0tjdSr>r;rsr?rz. r?rA)r9rrrrdrer:rs @r?test_with_incorrect_minlengthz*TestBincount.test_with_incorrect_minlength s HRs # # #I@FFFF H H H J2@@@@ B B B IaLLI@FFFF H H H J2@@@@ B B B B BrArrc\tjtjtj}tjtjtj}t dD]}tjgdttjtjtj|ttjtjtj|t dD]}tjgdgdttjtjtj|ttjtjtj|dS)Nrrr) rrr9rintprrgrr )rh intp_refcountdouble_refcountjs r?test_dtype_reference_leaksz'TestBincount.test_dtype_reference_leaks s6(9(9:: /"(29*=*=>>r # #A K " " " "S_RXbg%6%677GGGS_RXbi%8%899?KKKr . .A K 999 - - - -S_RXbg%6%677GGGS_RXbi%8%899?KKKKKrAvalsrPc0tj|}tt5tj|dddn #1swxYwYtt5tj|ddddS#1swxYwYdSr8)r9asarrayr rer)rhrIvals_arrs r?test_error_not_1dzTestBincount.test_error_not_1d s:d## : & & " " K ! ! ! " " " " " " " " " " " " " " " : & &   K                     s#A  AA)B  BBN)rrrrr!r&r+r/r1r3rr6rBr5r6rrrHr7rMrrAr?rr s$***<<<!!! DDD ,,,666JJJ !!! 666 BBB" [L(1IJJ L LKJ L [V1vh]3343rArceZdZdZdZdZdZdZej dddd ggd  d Z d Z dZ dZ dZdZdZdZdZdZdS) TestInterpcHtttdggtttddgddgtttdddgddgdtttdggdtttddgddgddS)NrrRrPr h)r rer'rs r?test_exceptionszTestInterp.test_exceptions sj&!R444j&!aS1a&999j&!aVaVAFFFFj&!RC@@@@j&!aS1a&EEEEEErActjddd}tjddd}tjddd}ttj||||dS)NrrRr[rp)r9r`r r'rhrrx0s r?rnzTestInterp.test_basic s_ K1a  K1a  [Ar " "BIb!Q//44444rActddD]+}tj|tj}tj|tj}tjdd|dz |gtj}|ddd}t |||}t |||}tjgdt}|ddd} t||t|| t |||d}t |||d}tjgdt}|ddd} t||t|| t |||d }t |||d }tjgd t}|ddd} t||t|| t |||dd }t |||dd }tjgd t}|ddd} t||t|| -dS)NrRrr!rarrleft)rrRrRrRrPright)rRrRrRrP)rXrZ)rrRrRrP) rgr9r:rrfrr'rr ) rhsizexpypincptsdecptsincresdecresinctgtdectgts r?test_right_left_behaviorz#TestInterp.test_right_left_behavior sM!RLL ) )D4ry111BRY///BXr1dQh5RYGGGFDDbD\FFB++FFB++FXlll%888FDDbD\F  ( ( (  ( ( (FB333FFB333FXlll%888FDDbD\F  ( ( (  ( ( (FB!444FFB!444FXlll%888FDDbD\F  ( ( (  ( ( (FB!<<L s")A,,rAc"t|dSrbrJrs r?rzTestInterp.M -1%%rAc"td|Srbrnrs r?rzTestInterp.N rorAcHt|tj|dS)NrX)rJr9multiplyrs r?rzTestInterp.O s-2;q"#5#566rA)rGz complex-realz complex-imagz complex-both)paramsidsc|jS)z( scale function used by the below tests )param)rhrequests r?scz TestInterp.scK s }rAc vttjdtjdg|ddg|tjttjddtjg|ddg|tjttjdddg|tjdg|tjttjdddg|dtjg|tjdS)z test that nans are propagated rirRrrN)r r9r'rZrhrxs r?test_non_finite_any_nanz"TestInterp.test_non_finite_any_nanZ sRYsRV!$4bbqb9I6J6JKKRRPRPVZZXXXRYs!RV$4bbqb9I6J6JKKRRPRPVZZXXXRYs!!$4bb"&b9I6J6JKKRRPRPVZZXXXRYs!!$4bbq"&9I6J6JKKRRPRPVZZXXXXXrAc ttjdtj tj g|ddg|tjttjdddg|tj tj g|tjttjdddg|tj tj g|tjttjdtj tj g|ddg|ddS)z2 Test that interp between opposite infs gives nan rirrrRNr r9r'rWrZrzs r?test_non_finite_infzTestInterp.test_non_finite_infa s/RYsbfWrvg$6!";M8N8NOOQSQSTVTZQ[Q[\\\RYs1A$6RVGbfW;M8N8NOOQSQSTVTZQ[Q[\\\RYs1A$6RVGbfW;M8N8NOOQSQSTVTZQ[Q[\\\ RYsbfWrvg$6";M8N8NOOQSQSTVQWQWXXXXXrAc ttjdtj dg|tj dg|tjttjdtj dg|tj dg|tjttjdtj dg|dtj g|tjttjdtj dg|dtj g|tjttjddtj g|tj dg|tjttjddtj g|tj dg|tjttjddtj g|dtj g|tjttjddtj g|dtj g|tjdS)z@ Test that interp where both axes have a bound at inf gives nan rirRrrNr}rzs r?test_non_finite_half_inf_xfz&TestInterp.test_non_finite_half_inf_xfj sXRYsbfWA$6RVG";M8N8NOOQSQSTVTZQ[Q[\\\RYsbfWA$6RVG";M8N8NOOQSQSTVTZQ[Q[\\\RYsbfWA$6!bfW;M8N8NOOQSQSTVTZQ[Q[\\\RYsbfWA$6!bfW;M8N8NOOQSQSTVTZQ[Q[\\\RYs1rvg$6RVG";M8N8NOOQSQSTVTZQ[Q[\\\RYs1rvg$6RVG";M8N8NOOQSQSTVTZQ[Q[\\\RYs1rvg$6!bfW;M8N8NOOQSQSTVTZQ[Q[\\\RYs1rvg$6!bfW;M8N8NOOQSQSTVTZQ[Q[\\\\\rAc Zttjdtj tj g|ddg|dttjdtj dg|ddg|dttjddtj g|ddg|dttjdtj tj g|ddg|ddS)z1 Test interp where the x axis has a bound at inf rirrrRNr r9r'rWrzs r?test_non_finite_half_inf_xz%TestInterp.test_non_finite_half_inf_xu sRYsbfWrvg$6Ar7 DDbbffMMMRYsbfWa$6Ar7 DDbbffMMMRYs1rvg$6Ar7 DDbbeeLLLRYsbfWrvg$6Ar7 DDbbeeLLLLLrAc ttjdddg|dtj g|tj ttjdddg|dtj g|tj ttjdddg|tj dg|tj ttjdddg|tj dg|tj ttjdddg|tj tj g|tj ttjdddg|tj tj g|tj dS)z1 Test interp where the f axis has a bound at inf rirrRrNrrzs r?test_non_finite_half_inf_fz%TestInterp.test_non_finite_half_inf_f| sRYsQFBBa"&/A,B,BCCRR[[QQQRYsQFBBa"&/A,B,BCCRR[[QQQRYsQFBBb/A,B,BCCRR[[QQQRYsQFBBb/A,B,BCCRR[[QQQRYsQFBB"&/A,B,BCCRR[[QQQRYsQFBB"&/A,B,BCCRR[[QQQQQrActjddd}tjddddtjdddzdzz}d}|d|zdzz}ttj||||d}d}ttj|||||d }d}ttj|||| |gd }gd }dd tjdg}dd tjdztjdg}ttj||||gd}gd}gd}gd}ttj|||d|dS)NrrRr[rVr$ra@@rWrrYrhr@?rNy?iLiViGr8rrimiBi^i)@?y$@@@@y@@)y@?ry!@?y@?ry @ @y @ @y@@rQr )r9r`r r'rW) rhrrrUy0rXrZr\ris r?test_complex_interpzTestInterp.test_complex_interp s K1a  K1a A Aq!(<(<$. . . .Xa^^ . . .rArxrrhrzrPrprN)rgr9rrr rrr)rhrr,s r? test_fractionzTestPercentile.test_fraction sD . .U1XX . . . M!Xa[[ ) )Q $$$T!WWh''' M!Xc]] + +QA'''T!WWh''' M!Xb\\ * *QA'''T!WWh''' M!hrll^ , ,Q(1a..!122333T!WWbj)))))rActjd}tj|ddddtj|dddddtjd}tj|dd|dddS)Nr[Flinearr0)r9rfr)rhr\os r?test_apizTestPercentile.test_api sn GAJJ aD$... aD$x888 GDMM aD!UH55555rAc\tjgdd}tttj|dtjgdd}tttj|dtjgdd}tttj|ddSN)y?@y@?y?ffffff@Gr!riDr)r9rr rdrrharr_cs r?rzTestPercentile.test_complex s777sCCCis;;;777sCCCis;;;777sCCCis;;;;;rActjgdgdgdgdgdg}ttj|ddgddS)Nr)rNrNrZrprr)r9rr rrs r?test_2DzTestPercentile.test_2D sg Hiiiiiiiiiii ! " " 2=BQ777CCCCCrArr c&tjdtjdddg|}tj|dd}tj|tjtj|j|jdS)NrA@D@I@r!rr)r9rKNANrtestingr r)rhrrr>s r?test_linear_nan_1Dz!TestPercentile.test_linear_nan_1D sj$dD9GGGm   RV,,,  3955555rAc*g|]}|tjfSr)r9r)r,int_types r?r-zTestPercentile. s/" ,rArr input_dtypeexpected_dtyperr) ) inverted_cdfr)averaged_inverted_cdf;@)closest_observationr)interpolated_inverted_cdfr)hazenr)weibull)rr)median_unbiasedr)normal_unbiasedg ;@c"tj|}tjdkrtj|tj}tjgd|}tj|d|}tj|| |d|dvr|dkrCtj tj|jtjdStj tj|jtj|dStj tj|jtj|dS) Nlegacy)rg4@rrrr!rrr)rrr) r9r_get_promotion_state promote_typesrrKrrr rr )rhrrrrrr s r?test_linear_interpolationz(TestPercentile.test_linear_interpolation sc".11 " $ $ 0 0-nbjIINj777{KKKsD888 && N''112 7 7 7 < < <c!! '' 6(:(:(@"*MMMMM '' 6(:(:(@(*(=(=????? J # #BJv$6$6$<$&H^$<$< > > > > >rActtjtjd|dddttjtjd|ddddS) Nrr!rplowerrrNhigherr[r r9rr:rhrs r?test_lower_higherz TestPercentile.test_lower_higher1 sR]29Ru#=#=#=r*133345 7 7 7R]29Ru#=#=#=r*244456 8 8 8 8 8rActtjtjd|dddttjtjd|dzd dd ttjtjd |ddd ttjtjd |d dd dS) Nrr!3midpointrr&r;rRrpr[r@rrs r? test_midpointzTestPercentile.test_midpoint8 s R]29Ru#=#=#=r*46667: < < <R]29Qe#<#<# IaLL3 R]1lll33^^^DDDDDrAc|tjddd}ttj|dgdgdgdgdg}ttj|dd |gd gd gd g}ttj|dd tj|jtjddddd}ttj|djdttj|djdttj|dd jdttj|dd jdttj|dd jdttj|dd jdttj|dd jdttj|ddjdttj|ddjdttj|dd djdttj|dddjdttj|dddjdttj|dddjdttj|dddjddS)NrQrZrN)rJrprh)g@rrI)rPrZrNr[r)rxr;rrrr)?rKrZ)@rrz)rrrrRrQr[ryrJrpr)rJrprrZ)rPrNr[ry)rPrZr[ryrP)rPrZrNry)rZrZr[ryrrrr)r9r:r{r rrrrE)rhrr0rs r?rzTestPercentile.test_axisW sa IbMM ! !!Q ' 'R]1m446G6G6GHHHllLLL... 9R]1m!<<R]1m!<<>> BM!R0011222 X))) * * ]1bq 9 9 9RRXrx((( Xnnn % % ]1bq 9 9 9B###RXrx(((hr))) M!RS 9 9 9QS!hq((( M!RaS A A AQS"hq((( M!RaS A A AQS"rAc tttjddgddtttjdgdtttjdgdtttjdgt t ddgztttjdgt t dd gzdS) NrRrP8r:rerarpr)r rer9rrrgrs r?test_exceptionzTestPercentile.test_exception sj"-!Q% ' ' ' 'j"-!c:::j"-!b999j"-!d599oo6MNNNj"-!d599oo6NOOOOOrAcPttjgddddS)NrrrR)r r9rrs r?test_percentile_listz#TestPercentile.test_percentile_list s(R]999a00!44444rActjgd}tjd}d}tj|||t tj|||tjgdgdg}tjd}tj||d|t tj||d|tjd }tj||d |t tj||d |tjd d d }tjgdgdg}tjd}t tj|dd||t ||tjgdgdg}tjd}t tj|dd ||t ||tjgdgdg}tjd|j}tj|ddd|}t ||t ||tjgdgdg}tjd|j}tj|ddd |}t ||t ||dS)NrrrrrrrrrZrPrRrQrZrN)rrrrr)rPrNr)rrrrrrrRrPrZrr!rr)rrNrx)rRr[r;) r9rrCrr r:r{rr)rhrrr,rrrrs r?test_percentile_outz"TestPercentile.test_percentile_out s HYYY   HTNN  a""""R]1a((!,,, Hiiiii! " " HV   a****R]1aa000!444 HV   a****R]1aa000!444 IbMM ! !!Q ' ' X)))+;+;+;< = =hvR]1hQC@@@"EEES" X+++->->->? @ @hvR]1hQC@@@"EEES"X~~~|||4 5 5hvQW--- M!XgA3 G G GQS" X{{{KKK0 1 1hvQW--- M!XgA3 G G GQS"rAc tjddddd}ttj|ddjdttj|ddjd ttj|ddjd ttj|dd jd ttj|dd jd ttj|ddjd ttj|ddjd ttj|ddjdttj|dddjd ttj|dddjd ttjtj|ddgdjdttjtj|ddgdjdttjtj|ddgdjdttjtj|ddgd jddS)NrrrRrPrprr)rRrPrR)rrPrR)rrRrRrZ)rrRrPrarXr^r|rrr)rPrRrPrR)rPrrPrR)rPrrRrR)rPrrRrP)r9r:r{r rrErrhr\s r?test_percentile_empty_dimz(TestPercentile.test_percentile_empty_dim s If   % %b!Q 2 22=BQ777=yIII2=BQ777=zJJJ2=BQ777=zJJJ2=BQ777=zJJJ2=BR888> KKK2=BR888> KKK2=BR888> KKK2=BR888> JJJ2=BQ0:<<< > >R]1bv666t<]1Q!!!AAAX;#6#6#8#8"== ? ? ?R]1bw777=]1QqqqAX;#6#6#8#8"== ? ? ?R]1bv666t<]1Qqqq!AAAX;#6#6#8#8"== ? ? ? ? ?rAc@tjd}ttjtj|ddttjtj|ddttjtj|ddttjtj|ddtt tj|ddtt tj|d dtt tj|d ddS) NrrrJ)rr>rrrNrrNrW)rara)rZra)r9rfr rrrers r?test_extended_axis_invalidz)TestPercentile.test_extended_axis_invalid4 s GM " "blBM12DDDDblBM17bIIIIblBM11CCCCblBM16RHHHHj"-2FFFFj"-RHHHHj"-BGGGGGGrActjd}ttj|dddjdttj|dddjdttj|dddjd ttj|dd djd ttj|dd d jdttj|dddjdttj|ddgddjdttj|ddgddjddS)NrrzTrrrbrRrRrzrrrRr[rzrRr0rZrRrzrr)rrrRrRrzrRrR)rPrRrRrzrR)rPrRr[rzrR)r9rfr rrErs r? test_keepdimszTestPercentile.test_keepdims? s GM " "R]1adTBBBH! # # #R]1aftDDDJ" $ $ $R]1aftDDDJ! # # #R]1adTBBBH" $ $ $R]1aEEEK! # # #R]1ai$GGGM! # # # R]1q!f9,022227 J J JR]1q!f6,022227 J J J J JrAr>rzrRrNrRr0rb)r^raargnames argvaluesctjd| djz}nCt|jt fdt jD}tj||z}tj|}tj||d|}||usJt|j|dS)Nrr0c3>K|]}|vrdn j|VdSrRNrEr,r axis_normr\s r?rz3TestPercentile.test_keepdims_out..d NLL89Q)^^LLLLLLrATrrr) r9rfrr6rrgrErrr )rhr>r shape_outrrr3r\s @@r?test_keepdims_outz TestPercentile.test_keepdims_outS s GM " " <qv II,T16::ILLLLL=B16]]LLLLLIHQKK)+ hy!!q!$3GGG}}}}V\9-----rAc ptjd}tjd}ttj|dd||ttj|ddd||tjd}ttj|dd||ttj|ddd||tjd }ttj|d ||ttj|d d||dS) NrNrrrrrrrRrrP)r9rCrfr rrhrr\s r?test_outzTestPercentile.test_outm s HTNN GFOOR]1a222A666R]1a9!DDDaHHH HTNNR]1a222A666R]1a9!DDDaHHH HRLLR]1aQ///333R]1a qAAA1EEEEErAc  tjd5tjddtt jd}t jd}tj|d<tt j |dd| |tt j |ddd | |t jd }tt j |d d | |tt j |d d d | |t jd}tt j |d | |tt j |d d | |ddddS#1swxYwYdS)NTrprrrsr9rrrrrrrrRr) rurvrwrr9rCrfrZr rr:s r? test_out_nanzTestPercentile.test_out_nanz s  $D 1 1 1 A A  #Hb. A A AAAfAdG q!QA666 : : :  aAiQ??? D D DA q!QA666 : : :  aAiQ??? D D D A q!333Q 7 7 7  a9!<<s r? test_nan_qzTestPercentile.test_nan_q s ]:-E F F F 2 2 M..."& 1 1 1 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 ]:-E F F F 4 4 M...26( 3 3 3 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 KT2 & &v! ]:-E F F F - - M...! , , , - - - - - - - - - - - - - - - - - -s5"A  AA0#BB#&B#*DDDzm8[D]zM8[s]posrr rc tjdd|}d||<tj|d}|j|ksJtj|sJtj|ddg}|j|ksJtj|sJtjdd|dd }d||d f<tj|dd }ttj|gd dS Nrr{r!NaTrr3HrarZrRrr<)r9r:rrisnatrr{r rhrrCrir>s r?test_nat_basiczTestPercentile.test_nat_basic s Ia5 ) ) )#mAr""yE!!!!x}}mABx((yE!!!!x}}  """"" IaU + + + 3 3B : :#q& mAr***28C==*>*>*>?????rAN)-rrrrnrrrrr5r6r7r9 typecodesrrrrrrH_F_TYPE_CODESr TYPE_CODESrrrrrrrrrrrrrr rr#r)r7r;r=r@rBrKrrAr?rr sDDD***&666<<<DDD [Wbl7&;<<66=<6&(l<&@J 3J 3J 3M2=9$BHSMM2:6 88N [m-=>OO [h 3     ! !>> ! !PO>2l+bl7.CCcIJ [Wj1188218  [Wj11::21: [Wj1199219 333EEE"I"I"IH...`PPP555'''R***8000EEE)))???@ H H HJJJ( [S1q!f+.. [      . .  /. . F F FAAA$;9;9;9z - - - [Ww&899 [UKKK00@@10:9@@@rAr) rrrrrrrrrrrrrc eZdZdZdZdZdZdZdZdZ e j de jd d Ze j d ed Zejee jejd dejdddddZdZe j d ee j dgddZe j d ee j dgddZdS) TestQuantilec||k|z Sr8r)rhrralphas r?r@zTestQuantile.V sQ%rAc|gd}tj|d}tj|dddS)N)rr#rg?g ףp= ?rR)maxulp)r9quantilerassert_array_max_ulp)rhrris r? test_max_ulpzTestQuantile.test_max_ulp sA OO K4  ''4':::::rActjddz}ttj|ddttj|ddttj|dddS)NrxrirrrRrr)r9r:r rUrs r?rnzTestQuantile.test_basic sn IaLL3 R[A&&+++R[A&&,,,R[C(($/////rActtjdg}tjdd}t||dtt ||jtjgd}tj||}t ||t|j|jdS)NTFr)FTT)r9rrUr rrr )rhritf_quant quant_ress r?test_correct_quantile_valuez(TestQuantile.test_correct_quantile_values HdV  ;tU++Xqt$$$T(^^QW--- H((( ) )K1%% 9a(((Y_ag.....rAc dtdD}tj|d}t|dtt |t tj|d}t|t ddtt |t tj|t dd}t|t ddtt |t tj|t ddg}t|tjt ddgtt |tjtj|t ddgg}t|tjt ddggtt |tjtjd}ttj|t ddt dddS)Nc.g|]}t|dSrrrs r?r-z.TestQuantile.test_fraction..rrArxrrRrzrPrN) rgr9rUr rrrrr:)rhrr>s r?rzTestQuantile.test_fractions . .U1XX . . . K1  QT!WWh''' K1  QA'''T!WWh''' K8Aq>> * *QA'''T!WWh''' KHQNN+ , ,Q(1a..!122333T!WWbj))) KXa^^,- . .Q8Aq>>"2!344555T!WWbj))) IaLLR[HQNN33Xa^^DDDDDrAc\tjgdd}tttj|dtjgdd}tttj|dtjgdd}tttj|ddSr)r9rr rdrUrs r?rzTestQuantile.test_complex)s777sCCCieS999777sCCCieS999777sCCCieS99999rActjgd}|}tjtjd|dt |||}|}tjtjd|dt ||dS)N)rrrmrirrrr)r9rrrUr:r r)rhp0r,s r?rz TestQuantile.test_no_p_overwrite2s X/// 0 0 GGII BIdOOQz::::1b!!! YY[[ HHJJ BIdOOQz::::1b!!!!!rArrc|tjtjddg|dgd}|j|ksJdS)NrRrPr!rirr)r9rUrr)rhrr>s r?test_quantile_preserve_int_typez,TestQuantile.test_quantile_preserve_int_type>sMk"(Aq6777#!*,,,yE!!!!!!rArcHtjddd}tjtjgddz||}t tj||tjgd||}t tj||dS)NrrRr)rrRrRrPrPrZrZrNr[r[rRrRr;r;r;rxrxrzrr)rrrr)r9r`rUrr rd)rhrrarUs r?test_quantile_monotonicz$TestQuantile.test_quantile_monotonicDs [As # #;rx)2)2)2 3 358 9:rs r?'test_quantile_add_and_multiply_constantz4TestQuantile.test_quantile_add_and_multiply_constantsui##D))  JJqMM K5 0 0 0   AE5@@@!a%HHH AE5@@@!a%HHH [!QYv 6 6 6 6 ^ # #E SU^^++8AI&&#a%i..1*<<<2;q%#I#I#IJJJJJ2;q%#H#H#HIIIII , , ,5yCE NN**2;q%#I#I#IJJJJJ!e)$$AI(:::r{1eack(CCCEEEEE 2;q%#H#H#HIIIII 2 2 2 Ar{1eack&III J J J J J y 5yCE NN**2;q%!A#+f#M#M#MNNNNN2;q%#G#G#GHHHHH w   Ar{1eHEEE F F F F F x   Ar{1eGDDD E E E E E Ar{1eFCCC D D D D DrAN)rrrr@rWrnr\rrrr5r6r7r9rLrcquantile_methodsre hypothesisgivenrrstintegersfloatsrnrprwrzrrAr?rPrP s!   ;;;000 ///EEE6::: " " " [Wbl<&@AA""BA"  [X'788 2 298 2ZRZ(R[1EEE ) %4:e!M!M!MNNNOOO 22 OO 2 222 [X'788 [Wooo66::7698:8 [X'788 [Wooo666E6E76986E6E6ErArPc &eZdZejejddddejddddejddddejdddddZejejddddejddddejdddd d Zejejddddejddddejdddd d Z d Z d S)TestLerpFrrR)rkrjrfrgrhri)t0t1rirctj|||}tj|||}||ks||kr ||ksJdS||k||kkr ||ksJdS||ksJdSr8rj_lerp)rhrrrirl0l1s r?+test_linear_interpolation_formula_monotonicz4TestLerp.test_linear_interpolation_formula_monotonicsYq!R  Yq!R  88qAvv8888882g1q5 ! !888888888888rA)trirc||kr'|tj|||cxkr|ksnJdS|tj|||cxkr|ksnJdSr8r)rhrrirs r?)test_linear_interpolation_formula_boundedz2TestLerp.test_linear_interpolation_formula_boundeds 66 !Q**////a//////// !Q**////a////////rActj||dd|z z }tj||d|z }t||dSr?)rjrr)rhrrirrXrZs r?+test_linear_interpolation_formula_symmetricz4TestLerp.test_linear_interpolation_formula_symmetricsLyAqAE{++ !QA&&e$$$$$rActjd}tjd}tjd}tj|||dksJdS)NrPr[r#g@)r9rrjr)rhrirrs r?+test_linear_interpolation_formula_0d_inputsz4TestLerp.test_linear_interpolation_formula_0d_inputssN HQKK HQKK HSMMyAq!!S((((((rAN) rrrr|r}r~rrrrrrrAr?rrs ZU5-.!==="U5-.!===#")eE.4GGG#")eE.4GGG HHHHHZ  E%,-<<<! E%,2eEEE! E%,2eEEE FFF 00 FF 0 Z  E%,-<<<! E%,2eEEE! E%,2eEEE FFF %% FF % )))))rArceZdZdZdZdZdZdZej dgdde j d ge j d fd Z d Zd ZdZej eddZdZdZdZdZej dgddZej ddgej dgddZdS) TestMedianctjd}tjd}tjddd}t tj|dt tj|dt tj|dt tj|dgd t tj|ddd gt tj|ddtjgd }t|d|dzd z tj|tjgd }t |dtj|tjgd}t |dtj|t tj|jdtj |d<t tj|jddS)NrRrPryrZrirjrrrKrjrrN)6-¦? 0?}r?g4 ?r)rrr)rrrra) r9rr:r{r medianrr rrZ)rha0rrris r?rnzTestMedian.test_basics Xa[[ Yq\\ Yq\\ ! !!Q ' 'RYr]]A&&& " s+++ " s+++ "1---/@/@/@AAARYr***QF333 "4000#666 H@@@ A AQqTAaD[B. ! === H555 6 6QqT29Q<<((( H555 6 6QrUBIaLL)))RYq\\&***v!RYq\\&*****rActjddgddgddgddgg}|tjdd d fD]i}|}tj|d t |jD]}tj|| t||jttj|d ddgttj|j d ddgttj|d ttj|d d ttj|j d dS)NrPrZrrRryrzrNr[rhrrrr) r9rrbrandintrrrgrr rr)rha3riorigaxs r?test_axis_keywordzTestMedian.test_axis_keywordsc X1v1v1v1v  bi''3Y'??@ ( (A6688D Iad # # # #AFmm & & !"%%%%% q$ ' ' ' ' "1---Aw777 "$Q///!a999 " s+++ "4000#666 "$-----rActjddgddgddgddgg}tjd}tjd}tjddd}t tj|d dt tj|d d t tj|d d t tj|d d gdt tj|d d ddgt tj|d d d t tj|d d ddgt tj|jd d ddgtjdtjd}tj | t tj|dtj|dd t tj|dtj|dd t tj|dtj|dd t tj|dtj|dd dS)NrPrZrrRryrzrNr[Trrirj)rrrr3r!rYr)rr) r9rr:r{rrrrrrbrr)rhrrrra4s r?test_overwrite_keywordz!TestMedian.test_overwrite_keywordsF X1v1v1v1v  Xa[[ Yq\\ Yq\\ ! !!Q ' ' "''))TBBBAFFF "''))TBBBCHHH "''))TBBBCHHH "''))TJJJ))) + + + IbggiiA > > >A H H H IbggiiD A A A3 H H H IbggiiA > > >Q I I I "$))++t!LLLQ ! ! !Yy 3 3 3 ; ;I F F "((**%%% "4000 "''))$MMM O O O "1--- "''))!TJJJ L L L "1--- "''))!TJJJ L L L "1--- "''))!TJJJ L L L L LrAcgd}ttj|d|g}ttj|dttj|d|dS)NrrPrr)r r9rr)rhrr]s r?rKzTestMedian.test_array_like9sg IIBIaLL!,,,SBIbMM1--- "1---q11111rAcGddtj}|gd}ttj|ddS)NceZdZddZddZdS),TestMedian.test_subclass..MySubClassNcbtj||}||_|Sr8)r9rKrinfo)cls input_arrayrobjs r?__new__z4TestMedian.test_subclass..MySubClass.__new__Ds+j--22377 rAcdS)Nr)rhrrrs r?rz1TestMedian.test_subclass..MySubClass.meanIsrrAr8)NNN)rrrrrrrAr? MySubClassrBs<           rArrr)r9rr r)rhrris r?rzTestMedian.test_subclass@sa         Jyyy ! !RYq\\2&&&&&rArrrrrcGddtj}tjtj||}t ||sJdS)z6Check that we return subclasses, even if a NaN scalar.ceZdZdS)-TestMedian.test_subclass2..MySubclassNrrrAr? MySubclassrSrrArN)r9rrrrr)rhrrrBs r?test_subclass2zTestMedian.test_subclass2Osq         Ibhsmm((44 5 5!Z(((((((rActjd}tjd}ttj|d||tjd}ttj|d||tjd}ttj|||dS)Nr9rrrrrRr)r9rCrfr rr:s r?r;zTestMedian.test_outYs HTNN GFOORYq!+++Q/// HTNNRYq!+++Q/// HRLLRYqa(((!,,,,,rAc2tjd5tjddtt jd}t jd}tj|d<tt j |d| |t jd }tt j |d | |t jd }tt j || |ddddS#1swxYwYdS) NTrprrrsr9rrrrrrRr) rurvrwrr9rCrfrZr rr:s r?r=zTestMedian.test_out_nanbs/  $D 1 1 1 1 1  #Hb. A A AAAfAdG 1aQ/// 3 3 3A 1aQ/// 3 3 3 A 1!,,,a 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1sC)D  DDchtjdt}tj|d<t tj|tjt tj|dtjtjdtddd}tj|d<tj|d <t tj|tjt tj|jdtjtjdtdddd}tj|d <tj|d <t tj|d|tjtjdtdddd }tj|d <tj|d <t tj|d |tjtjdtdddd}tj|d <tj|d<t tj|d|dS)Nr{r!rPrrrZrNrr?rrsrRrrV)r9r:rrZr rr{rrs r?r@zTestMedian.test_nan_behaviorns Ib & & &v!RYq\\26***RYqq)))26222 Ib & & & . .q!Q 7 7V' V'  RYq\\26***RYq\\&*** Ibi%00088AqAA1 E E&$&$RYq!__a((( Ibi%00088AqAA1 E E&$&$RYq!__a((( Ibi%00088AqAA6 J Jv!v!RYq&))1-----rAzfp errors don't work correctlyrc2tjgt}tjd5}tjddt ttj|tj t|dj t utt|ddddn #1swxYwYtjgtd }tjd5}tjddt ttj|tj t|dj t udddn #1swxYwYtjgtd }ttj|d |ttj|d |tjtj td }tjd5}tjddt ttj|d |t|dj t uddddS#1swxYwYdS) Nr!TrprrrsrrPrZ)rrrrR) r9rrrurvrwrr rrZrryrC)rhrirrs r?rzTestMedian.test_emptys HRu % % %  $D 1 1 1 $Q  #Hb. A A A 1rv . . . AaDM^3 4 4 4 Q # # #  $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ HRuA . . .  $D 1 1 1 5Q  #Hb. A A A 1rv . . . AaDM^3 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 HRuA . . .RYqq)))1---RYqq)))1--- HRV5 2 2 2  $D 1 1 1 5Q  #Hb. A A A 11---q 1 1 1 AaDM^3 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5s8BCC C?A)E44E8;E8A!J  JJc |tjd}tttj|t ttj|d<tttj|t tdS)NrrP) r9r:rrrrr=rrZ)rhrs r? test_objectzTestMedian.test_objects| IbMMRYqxx//00115999v!RYqxx//0011599999rAc N tjd}tj|gdz}t tj|dtj|tj|dd}t tj|dtj||dd }t tj|d tj|t tj|d tj|dt tj|d tj|dt tj|d tj|dtj d d}tj | t tj|d dtj|dddddddf t tj|dd tj|ddddd ddf t tj|ddtj|dddddddf t tj|ddtj|dddddddf t tj|ddtj|dd ddddf t tj|ddtj|dddddd f t tj|ddtj|ddddddf dS)Nr rrrbrrarrtrRrrrQrj)rarrrrrPrrrrrrU)r9rbrrr rrrrr:r{rrrrs r?rzTestMedian.test_extended_axiss I  (  + + IqcBh  RYqv... ! === K2q ! !RYqx000")A,,??? JJq!   ! ! # #RYqw///1>>>RYqy11129QT3J3J3JKKKRYqu---ry/C/C/CDDDRYqv... !"0E0E0EFFF In % % - -m < < !''))$$$RYqy111!4Yq111QQQ{224455 7 7 7RYqy111!4Yq111a{224455 7 7 7RYqz22215Yq111a{224455 7 7 7RYqy111!4Yq111QQQqqqz113344 6 6 6RYqv...t4YqAaaa{224455 7 7 7RYqw///5Yq111QQQ{224455 7 7 7RYqv...t4Yq111a{224455 7 7 7 7 7rActjd}ttjtj|dttjtj|dttjtj|dttjtj|dtt tj|ddS)Nrrrr!rNr"rW)r9rfr rrrers r?r#z%TestMedian.test_extended_axis_invalids GM " "blBIqr::::blBIqw????blBIqq9999blBIqv>>>>j")QV<<<<<K|]}|vrdn j|VdSr0r1r2s r?rz/TestMedian.test_keepdims_out..r4rATr5) r9rfrr6rrgrrr rE)rhrr6rrr3r\s @@r?r7zTestMedian.test_keepdims_outs GM " " <qv II,T16::ILLLLL=B16]]LLLLLIhy!!14$C@@@}}}}V\9-----rArzm8[s]rCrDctjdd|}d||<tj|}|j|ksJtj|sJtj|ddg}|j|ksJtj|sJtjdd|dd }d||d f<tj|d }ttj|gd dSrF) r9r:rrrIrrr{r rJs r?test_nat_behaviorzTestMedian.test_nat_behaviors  Ia5 ) ) )#illyE!!!!x}}mABx((yE!!!!x}}  """"" IaU + + + 3 3B : :#q& i"""28C==*>*>*>?????rAN)rrrrnrrrKrr5r6r7r9rZrr;r=r@rrrrrr#r)r7rrrAr?rrs+++,...$LLLB222 ' ' ' [U*llR,>>>>rArceZdZejejjdkdej e ddZ ejejjdkddZ dS)TestAdd_newdocrPPython running -OOrPyPy does not modify tp_doccjd}ttjjjjdt ||tt tjjj jdktt tj j j jdkdS)Nz"Current flat index into the array.r~) r r9coreflatiterindexrGrCrufuncidentityrX index_tricksmgrid)rhrs r? test_add_doczTestAdd_newdoc.test_add_docs3RW%+3ISXXI>DDDBGM*233c9:::BF'-566<=====rActjjjj}tjddd|tjjjjksJdS)Nz numpy.corer)rz bad docstring)r9rrrrG add_newdoc)rhprev_docs r?test_errors_are_ignoredz&TestAdd_newdoc.test_errors_are_ignored'sK7#)1 lJ0JKKK27+19999999rAN) rrrr5r6rrrrrrrrrrAr?rrs [ *a/8LMM [w'DEE>>FENM> [ *a/8LMM::NM:::rArceZdZejejjdkdeje ddZ ejejjdkddZ dS)TestAddDocstringrPrrrctjtjjtjjjd}tj||jdS)NcdS docstringNrrrAr?funcz6TestAddDocstring.test_add_same_docstring..func9 FrA)r9 add_docstringrflatrGrhrs r?test_add_same_docstringz(TestAddDocstring.test_add_same_docstring3sN "*/*ABBB    t|,,,,,rAc0tt5tjtjjddddn #1swxYwYd}tt5tj|dddddS#1swxYwYdS)Nzdifferent docstringcdSrrrrAr?rz=TestAddDocstring.test_different_docstring_fails..funcErrA)r rr9rrrrs r?test_different_docstring_failsz/TestAddDocstring.test_different_docstring_fails?s'< ( ( E E  RZ_.C D D D E E E E E E E E E E E E E E E   < ( ( : :  T#8 9 9 9 : : : : : : : : : : : : : : : : : :s#%AA  A (B  BBN) rrrr5r6rrrrrrrrrAr?rr0s [ *a/8LMM [(EFF--GFNM- [ *a/8LMM : :NM : : :rArc`eZdZejdgddZdZdS)TestSortComplexztype_in, type_out))lr)hr)Hr)rr)rr)rrctjgd|}tj|}tj||}t ||t |j|jdS)N)r[rZryrPrRr!)r9r sort_complexrdrr r)rhtype_intype_outrir rs r?test_sort_realzTestSortComplex.test_sort_realOsp H___G 4 4 4##71::$$X..VX&&&V\8>22222rActjgdd}tjgdd}tj|}t||t|j|jdS)N)r?rrrr!)rrrr)r9rrr r)rhrirr s r?test_sort_complexz!TestSortComplex.test_sort_complex_sq H555S A A A8<<22222rAN)rrrr5r6r7rrrrAr?rrMsc [03 3 3   33  333333rAr)r)rr)rrurr* fractionsrrr5r|hypothesis.extra.numpyrhypothesis.strategies strategiesr~ functoolsrnumpyr9r numpy.testingrr r r r r rrrrrrrnumpy.lib.function_baserX function_baserj numpy.randomr numpy.librrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5numpy.core.numericr6r@rJrLrrrrrr9rlrrrrrrrrrCrrrrrrrrr r6r7rLr#r6rGrMrUr]rrrrrrrrOrr{rPrrrrrrrrAr?rs   ))))))"""""" &%%%%%%%% 433333 A@A@A@A@A@A@A@A@Hd1d1d1d1d1d1d1d1N::::::::"::::::::$ 3 3 3 3 3 3 3 3FK4K4K4K4K4K4K4K4Z;$;$;$;$;$;$;$;$|l0l0l0l0l0l0l0l0^ ; ; ; ; ; ; ; ; : : : : : : : : @ @ @ @ @ @ @ @;;;;;;;;.EEEEEEEE(JJJJJJJJ6I?I?I?I?I?I?I?I?Xr>r>r>r>r>r>r>r>j@%@%@%@%@%@%@%@%F''''''''4>%>%>%>%>%>%>%>%B$"$"$"$"$"$"$"$"Rd3d3d3d3d3d3d3d3N&&&&&&&&RR8R8R8R8R8R8R8R8j555555556 S2< - - W0E Ejjj))i;i;i;i;i;i;i;*)i;X;-;-;-;-;-;-;-;-|########" D D D D D D D D''''''''"X&X&X&X&X&X&X&X&vg&g&g&g&g&g&g&g&T11111111h& B B B B B B B Bk2k2k2k2k2k2k2k2\a0a0a0a0a0a0a0a0H]]]]]]]]@AAAAAAAADc@c@c@c@c@c@c@c@LJEJEJEJEJEJEJEJEZ/)/)/)/)/)/)/)/)da@a@a@a@a@a@a@a@H >>>>>>>>::::::::(:::::::::3333333333rA