K iR ddlZddlZddlmZddlZddlZddlmZddl m Z m Z ddl m Z mZddlmZmZmZmZmZmZmZej.dej0dd ej0d gd d ej0d ddgej0ej0ej0dej0dgdej0ej0ddej0ggZej.gdej.gdej.ddgej.gdgZej.gdgdgdgdgZej.gdgdgdgdgZGd d!ZGd"d#ZGd$d%Zej.d&ej.gd'd(Z e jCD]Z"e"jGd)*ejHjKd+ejLd,ejLd-zd.zejHjKd/e jCe jO0Gd1d2Z(Gd3d4Z)Gd5d6e)Z*Gd7d8e)Z+Gd9d:e)Z,d;Z-e.ejLd-Z/e/e-Dcgc]}d<|d= c}z Z/Gd>d?Z0Gd@dAZ1GdBdCZ2ejHjKdDej.ej0dEej0ejfgej.gdFfej.gdGejhHdIfej.gdFdIfej.ej0dEgej0ejfggejjHej.d)dIgd)dIggfgdJZ6dKZ7dLZ8ycc}w)MN)partial)normalize_axis_tuple) AxisErrorComplexWarning) _nan_mask _replace_nan)assert_assert_almost_equalassert_array_equal assert_equal assert_raisesassert_raises_regexsuppress_warningss?&S:??ʡE? ?绸zG?L F%u?6٬?:M S㥛?z):? h"?)rrrr)rrrrr)rrr)r?rrrr)rrrrrr)rrrrrr)rrrrrr)rrrrr)rrrrrr)rrrrrr)rrrrrrc8eZdZejej ej ejejejejejejejejejej ej"ej$ej&ej(ej*ej,ej.ej0ej2ej4ej6ej8ej:ej<ej>iZ e Dcgc]}|jc}}Z!e"ddZ#e$jJjMde jOe!dZ(dZ)ycc}}w)TestSignatureMatchcJg}tj|}|jjD][}|jtj j ur|j|;|j|j|]tj|S)z?Construct a signature and replace all default parameter-values.)default) inspect signature parametersvaluesr# Parameteremptyappendreplace Signature)funcr#prm_listr%prms g/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/numpy/lib/tests/test_nanfunctions.py get_signaturez TestSignatureMatch.get_signatureBs%%d+ ''..0 >C{{g//555$ G <=  >   **z nan_func,funcidsc|j|}|j|}tjj||yN)r1nptestingr )selfnan_funcr-r% nan_signatures r0test_signature_matchz'TestSignatureMatch.test_signature_matchNs8 &&t, **84   =9r2ctjjt|jttj j jy)z4Validate that all nan functions are actually tested.N)r7r8r setIDSlib_nanfunctions_impl__all__r9s r0test_exhaustivenessz&TestSignatureMatch.test_exhaustivenessWs6  M3rvv88@@A r2N)z...)*__name__ __module__ __qualname__r7nanminaminnanmaxamax nanargminargmin nanargmaxargmaxnansumsumnanprodprod nancumsumcumsum nancumprodcumprodnanmeanmean nanmedianmedian nanpercentile percentile nanquantilequantilenanvarvarnanstdstdNANFUNCSr? staticmethodr1pytestmark parametrizeitemsr<rD).0ks00r0r!r!/s% 277 277 bii bii 266 BGG bii rzz BGG bii "--   266 266H ( (!1:: (C + + [[_hnn.>CH:I: / )sFr!c eZdZejej gZejejgZ dZ dZ dZ dZ dZej j#dgdej j#dej$d ej j#d ej&ej(ej*d ej(gd d gdZdZdZdZdZej j#dej$d dZej j#dej$d dZy)TestNanFunctions_MinMaxc~tj}|jD]}||t|tyr6_ndatcopynanfuncsr r9ndatfs r0 test_mutationz%TestNanFunctions_MinMax.test_mutationc1zz| &A dG u % &r2ctjd}t|j|jD]D\}}dD]:}|||d}|||d}t |j |j k(<FyNNrTaxiskeepdimsr7eyeziprrstdfuncsr ndimr9matnfrfr~tgtress r0 test_keepdimsz%TestNanFunctions_MinMax.test_keepdimsjqffQi$--7 .FB$ .4$74$7CHH,- . .r2ctjd}t|j|jD]G\}}tj d}||d}||d|}t ||t ||IyNrzr|r~r~outr7rrrrrzerosr r9rrrresoutrrs r0test_outz TestNanFunctions_MinMax.test_outrkffQi$--7 *FBXXa[FSq/CSqf-C V , S )  *r2cd}t|j|jD]\}}|D]}tjd|}||dj j }||dj j }t||u||dj j }||dj j }t||uy)NefdgFDGrzdtyper|rrrrrr7rrtyper r9codesrrcrrrs r0test_dtype_from_inputz-TestNanFunctions_MinMax.test_dtype_from_input{s$--7 $FB $ffQa(1o++001o++00s #4(..334(..33s # $ $r2ct|j|jD]8\}}tDcgc] }|| }}|td}t ||:ycc}wNr|rrrrr_rdatrpr r9rrdrrs r0test_result_valuesz*TestNanFunctions_MinMax.test_result_valuesT$--7 *FB"'(Q2a5(C(U#C S ) *(Ar~r{rAllFloatarrayrzrz0d2dr3c|$|jdk(rtjd|j|}d}|jD]p}tj t |5|||}dddtjjsJ|j|jk(rpJy#1swYLxYwNr*`axis != None` not supported for 0d arraysAll-NaN slice encounteredmatchr) rrfskipastyperrwarnsRuntimeWarningr7isnanallrr9r~rrrr-rs r0 test_allnansz$TestNanFunctions_MinMax.test_allnanss   a KKD E U#+MM ,DnE: -5t, -88C=$$& &&99 + ++  , - -s $ B::C cxtjjt}|jj }tj fD]i}||d}|td}t||t|j|ttj|j kyr) r7ma fix_invalidrp_maskrqrHr r isinfany)r9rmskrurrs r0 test_maskedz#TestNanFunctions_MinMax.test_maskedsee&iinn)) -ACa.CE"C c " C (  ))++ ,  -r2cN|jD]}t|ddk(yNrrrr r9rus r0 test_scalarz#TestNanFunctions_MinMax.test_scalar% !A AbERK  !r2c jGddtj}tjdj|}|jD]}||d}t t ||t |jdk(||d}t t ||t |jdk(||}t |jdk(tj|d<|jD]}tjd 5}tjd ||d}t t ||t tjtj| t t|dk(dddtjd 5}tjd ||d}t t ||t tj|dxr4tj|d xrtj|d  t t|dk(d t t|dj t"dddtjd 5}tjd ||}t |jdk(t |tjk7t t|dk(dddy#1swY}xYw#1swYxYw#1swY6xYw)Nc eZdZy)8TestNanFunctions_MinMax.test_subclass..MyNDArrayNrErFrGr2r0 MyNDArrayr r2rrzrrrzr|rTrecordalwaysno warning raised)r7ndarrayrviewrrr isinstanceshapenanwarningscatch_warnings simplefilterrrlen issubclasscategoryr)r9rminerurws r0 test_subclassz%TestNanFunctions_MinMax.test_subclasss}    vvay~~i( %ADq/C JsI. / CII% &Dq/C JsI. / CII% &D'C CIIO $ %&&Q %A((5 %%%h/1o 3 23BFF288C=112A! $  %((5 C%%h/1o 3 23Q(1#a&1A-A1 "Q 002A! %89 1Q4==.AB C((5 %%%h/g R(rvv &A! $  % %# % % % C C % %s'A>L2C LA)L(L L% (L2 c tjddgtjdgtjtjggt}t tj |dt tj |dddgt jd5}t jd t ttj |d ddtjgtt|d k(d tt|djtdddy#1swYyxYw) Nr@@rrrTrrr|r)r7rrobjectr rHrrrlistr rrrr)r9arrrs r0test_object_arrayz)TestNanFunctions_MinMax.test_object_arrayshhc RVVSMBFFBFF3CDFSRYYs^S)RYYs+c3Z8  $ $D 1 ?Q  ! !( + bii!45S"&&7I J CFaK!4 5 Jqt}}n= >  ? ? ?s 'BD??EcGddtj}tjdj|}tj|dd|j D]k}|tj urdnd}|||}|j|k(sJ||k(sJ||j||}|j|k(sJ||k(rkJy)Nc eZdZy)7TestNanFunctions_MinMax.test_initial..MyNDArrayNrrr2r0rrrr2r drinitial) r7rarangerrrrrJrr)r9rrarrurret1ret2s r0 test_initialz$TestNanFunctions_MinMax.test_initials   YYq\  '2A #A"))^cGR)D::& &&7? "?RWWY'9D::& &&7? "? #r2c Gddtj}tjdjddj |}tj |dddf<tj |tj}d|dddf<|jD]m}|tjurdnd }|||d }|j|k(sJ||k(sJ||j||d }|j|k(sJ||k(rmJy) Nc eZdZy)5TestNanFunctions_MinMax.test_where..MyNDArrayNrrr2r0rrrr2rrrzrrFrwherer) r7rrreshaperr ones_likeboolrrrHrr) r9rrrrru referencerrs r0 test_wherez"TestNanFunctions_MinMax.test_wheres   YYq\ ! !!Q ' . .u 5661a4 Rrww/ad  %A"))^IRua0D::& &&9$ $$RWWY'ua@D::& &&9$ $$ %r2N)rErFrGr7rHrJrrminmaxrrvrrrrrfrgrh typecodesrrfullrrrrrrrrr2r0rmrm^sL 299%HH&.* $*  [[V\2 [[Wbll:&>? [[W' $< , @3 ,-!)%V ? [[Wbll:&>?#@#$ [[Wbll:&>?%@%r2rmc jeZdZejej gZdZdZe jjdgde jjdejde jjdejejejdejgd d g d Zd ZdZdZe jjdejddZe jjdejddZy)TestNanFunctions_ArgminArgmaxc~tj}|jD]}||t|tyr6rorss r0rvz+TestNanFunctions_ArgminArgmax.test_mutationrwr2c t|jtjtjgD]\}}t D]}t 5}|jtd||}||}ttj| t|||j ttj||d|j dddy#1swYxYw)Nzinvalid value encountered in) rrrr7greaterlessrprfilterrr rrequal)r9rufcmprowsupindvals r0rz0TestNanFunctions_ArgminArgmax.test_result_valuess4==2::rww*?@ @GAt @&(@CJJ~/MNC&Cc(C -.S# 2 2 445c$3i 8 < < >>?@@ @ @@@s BC00C9 r~r{rrrrrrr3c|$|jdk(rtjd|j|}|jD]0}tj t d5|||ddd2y#1swY=xYwr)rrfrrrrraises ValueError)r9r~rrr-s r0rz*TestNanFunctions_ArgminArgmax.test_allnans&su   a KKD E U#MM 'Dz1LM 'U& ' ' ' ' 's " A88B c tjd}|jD]M}dD]}ttd|||dD]+}|||}t |tjd-Oy)NrrzrNz*attempt to get argm.. of an empty sequencerr|r)r7rrrrrr )r9rrur~rs r0 test_emptyz(TestNanFunctions_ArgminArgmax.test_empty5svhhv /A! +#"D3T+ +  /$'S"((1+. /  /r2cN|jD]}t|ddk(yrrrs r0rz)TestNanFunctions_ArgminArgmax.test_scalarArr2cGddtj}tjdj|}|jD]}||d}t t ||t |jdk(||d}t t ||t |jdk(||}t |jdk(y) Nc eZdZy)>TestNanFunctions_ArgminArgmax.test_subclass..MyNDArrayNrrr2r0rr#Frr2rrzrrrr|r)r7rrrrrr rr)r9rrrurs r0rz+TestNanFunctions_ArgminArgmax.test_subclassEs    vvay~~i( %ADq/C JsI. / CII% &Dq/C JsI. / CII% &D'C CIIO $ %r2ctjdj|}tj|dd|jD]D}|tj urdnd}||d}|j |j k(sJ||k(rDJy)NrrrT)r)r7rrrrrrLrr9rrrurrets r0rz+TestNanFunctions_ArgminArgmax.test_keepdimsVs~ YYq\  '2A $A",,.AIB&C88rww& &&)# ##  $r2c8tjdj|}tj|dd|jD]T}tj dtj }|tjurdnd}|||}||usJ||k(rTJy)Nrrrrrr)r7rrrrrrintprL)r9rrrurrr&s r0rz&TestNanFunctions_ArgminArgmax.test_outas YYq\  '2A $A((2RWW-C",,.AIBC.C#: :)# ##  $r2N)rErFrGr7rLrNrrrvrrfrgrhr rrr rrrrrrrr2r0r r s& bll+H& @ [[V\2 [[Wbll:&>? [[W' $<' @3 ' /!%" [[Wbll:&>?$@$ [[Wbll:&>? $@ $r2r r)']W.)r1dF)writer AllIntegerrOrr3ceZdZejej ej ejejejejejejejejejej ej"ej$ej&ej(ej*ej,ej.ej0ej2ej4ej6i ZeDcgc]}|jc}}Zej>jAdejCeejDddZ#ej>jAdejHejJfejLejNfgddgdZ(ej>jAdej0ej2fej4ej6fgd d gd Z)ej>jAd ej0ej4gd Z*ycc}}w)TestNanFunctions_NumberTypesz nanfunc,funcr3ignore)overc|j|}||}||}t|||dk(rt|t|usJy|j|jk(sJy)Nr2rr rrr9rrnanfuncr-rrs r0 test_nanfuncz)TestNanFunctions_NumberTypes.test_nanfuncsajj3iclC% C<9S ) ))99 ) ))r2r^r\cX|j|}|jjdk(r'tt||dtt||dy||d}||d}t |||dk(rt |t |usJy|j|jk(sJy)Nrr|)qr2)rrkindr TypeErrorr rr9s r0test_nanfunc_qz+TestNanFunctions_NumberTypes.test_nanfunc_qs jj 99>>S )T3! 4 )WcQ 7sa.C##C S )|CyDI---yyCII---r2r`rbc|j|}||d}||d}t|||dk(rt|t|usJy|j|jk(sJy)N?ddofr2r8r9s r0test_nanfunc_ddofz.TestNanFunctions_NumberTypes.test_nanfunc_ddofsg jj3S!c$C% C<9S ) ))99 ) ))r2r:c"|j|}t||d||dd}tt|5||dddddtt|5||dddddy#1swY.xYw#1swYyxYw)NrB) correctionrCz5ddof and correction can't be provided simultaneously.)rDrGr|r)rr rr)r9rrr:err_msgs r0test_nanfunc_correctionz4TestNanFunctions_NumberTypes.test_nanfunc_correctionsjj CC ('#C*@ J W 5 3 Ccc 2 3!W 5 / CaA . / / 3 3 / /s A9$ B9BBN)+rErFrGr7rHrrJrrLrMrNrOrPrQrRrSrTrUrVrWrXrYrZr[r`rarbrcrr nanfunc_idsrfrgrhrierrstater;r^r_r\r]r@rErI)rjis00r0r4r4vs 266 266 bii bii 266 BGG bii rzz BGG bii 266 266 H(00!1::0K [[^X^^-=;OR[[h * P * [[ .."++ &)9)92==(IJ O , .  ." [[ ))RVV ryy"&&12 x  *  * [[BIIryy) / /q1sIr4cBeZdZdZdZdZdZdZdZdZ dZ d Z y ) SharedNanFunctionsTestsMixinc~tj}|jD]}||t|tyr6rorss r0rvz*SharedNanFunctionsTestsMixin.test_mutationrwr2ctjd}t|j|jD]D\}}dD]:}|||d}|||d}t |j |j k(<Fyryrrs r0rz*SharedNanFunctionsTestsMixin.test_keepdimsrr2ctjd}t|j|jD]G\}}tj d}||d}||d|}t ||t ||Iyrrrs r0rz%SharedNanFunctionsTestsMixin.test_outrr2c tjd}d}t|j|jD]?\}}|D]3}t 5}|tj tjhvr|dvr|jt||tj|djj}||tj|djj}t||u||tj|djj}||tj|djj}t||uddd6By#1swYExYwNrzrFDGr|)rr~ r7rrrrrrrbr`rrrrr r9rrrrrrrrs r0test_dtype_from_dtypez2SharedNanFunctionsTestsMixin.test_dtype_from_dtypes.ffQi$--7 (FB (&( (Cbii33U  >2S !<BBGGCS !<BBGGCC3J'S $?EEJJCS $?EEJJCC3J' ( ( ( ( ( (s DE<<F cptjd}d}t|j|jD]\}}|D]}t 5}|tj tjhvr|dvr|jt|||djj}|||djj}t||u|||djj}|||djj}t||udddy#1swYxYwrSrUrVs r0test_dtype_from_charz1SharedNanFunctionsTestsMixin.test_dtype_from_charsffQi$--7 (FB (&( (Cbii33U  >2S288==CS288==CC3J'S5;;@@CS5;;@@CC3J' ( ( ( ( ( (s CD,,D5 c d}t|j|jD]\}}|D]}tjd|}||dj j }||dj j }t||ud|d|||dj j }||dj j }t||uy)Nrrzrr|rzres z, tgt rrs r0rz2SharedNanFunctionsTestsMixin.test_dtype_from_input s$--7 $FB $ffQa(1o++001o++00s d3%vcU$;<4(..334(..33s # $ $r2ct|j|jD]8\}}tDcgc] }|| }}|td}t ||:ycc}wrrrs r0rz/SharedNanFunctionsTestsMixin.test_result_valuesrrcN|jD]}t|ddk(yrrrs r0rz(SharedNanFunctionsTestsMixin.test_scalarrr2c^Gddtj}tjd}|j|}|jD]}||dj }||d}t t||t |j |k(||dj }||d}t t||t |j |k(||j }||}t t||t |j |k(y)Nc eZdZy)=SharedNanFunctionsTestsMixin.test_subclass..MyNDArrayNrrr2r0rr_#rr2rrzrrr|)r7rrrrrrr r)r9rrrruexpected_shapers r0rz*SharedNanFunctionsTestsMixin.test_subclass"s    q zz)$ 1Au1-33NDq/C JsI. / CII/ 0u1-33NDq/C JsI. / CII/ 0uX^^ND'C JsI. / CII/ 0 1r2N) rErFrGrvrrrWrYrrrrrr2r0rNrNs/&.*("(" $* !1r2rNc eZdZejej gZejejgZ e jjdgde jjdejde jjdejejej dejgddg d Zd Ze jjdejdd Ze jjdejdd Zy)TestNanFunctions_SumProdr~r{rrrrrrr3c0|$|jdk(rtjd|j|}t |j ddgD]D\}}|||}t j||k(sJ|j|jk(rDJy)Nrrr|r rrfrrrrrr7rrr9r~rrr-identityrs r0rz%TestNanFunctions_SumProd.test_allnans>s   a KKD E U#!$--!Q8 ,ND(u4(C66#/* **99 + ++ ,r2c.ttjtjgddgD]f\}}tjd}|gdz}||d}t ||g}||d}t |||}||d}t ||hy)Nrr|rrzr)rr7rPrRrr r9ru tgt_valuerrrs r0rz#TestNanFunctions_SumProd.test_emptyNsBJJ 7!Q@ #LAy((6"C+/CCa.C c "CCa.C c "CCd#C c " #r2ctjdj|}tj|dd|jD]:}|tj urdnd}||d}|j |k(sJ||k(r:Jy)Nrri rr)r7rrrrrrPrr%s r0rz%TestNanFunctions_SumProd.test_initial[sy YYq\  '2A $A299n$IB"C99% %%)# ##  $r2ctjdjddj|}tj|dddf<tj |tj }d|dddf<|jD];}|tjurdnd}|||d }|j|k(sJ||k(r;Jy) NrrzrrFirr) r7rrrrrrrrrPr)r9rrrrurr&s r0rz#TestNanFunctions_SumProd.test_wherefs YYq\ ! !!Q ' . .u 5661a4 Rrww/ad  $A299n$IBeQ/C99% %%)# ##  $r2N)rErFrGr7rPrRrrrQrSrrfrgrhr rrr rrrrrr2r0rbrb9s! 2::&H H [[V\2 [[Wbll:&>? [[W' $<, @3 , # [[Wbll:&>?$@$ [[Wbll:&>? $@ $r2rbc eZdZejej gZejejgZ e jjdgde jjdejde jjdejejej dejgddg d Zd Zd Zd ZdZy)TestNanFunctions_CumSumProdr~r{rrrrrrr3c,|$|jdk(rtjd|j|}t |j ddgD]B\}}||}t j||k(sJ|j|jk(rBJy)Nrrr|rdres r0rz(TestNanFunctions_CumSumProd.test_allnansys   a KKD E U#!$--!Q8 ,ND(u+C66#/* **99 + ++ ,r2cPt|jddgD]\}}tjd}|tjdz}||d}t |||}||d}t ||tjd}||d}t ||y)Nrr|rr)rrrr7ronesr rhs r0rz&TestNanFunctions_CumSumProd.test_emptys 1v6 #LAy((6"Cbggfo-CCa.C c "CCa.C c "((1+CCd#C c " #r2ct|j|jD]Y\}}tjd}dD]:}|||d}|||d}t |j |j k(<[|jD]}tjd}tjjd}tj||j|jdk<||d}t|jdtjd D]"}|||}t|jd$y) Nrzr{rrzr rrBr)ir)rrrrr7rr rrrrandom RandomStaterrandrr r) r9rugrr~rrrrss r0rz)TestNanFunctions_CumSumProd.test_keepdimss t}}5 .DAq&&)C$ .$D1$D1CHH,- . . 7A &A&&q)B)+Agbggqww#% &AD/C G , !  7oSYY 6 7 7r2cdD]}tjt|}tjt|}t ||tj t|}tjt|}t ||y)N)rr|Nr) r7rW _ndat_onesrVrpr rU _ndat_zerosrT)r9r~rrs r0rz.TestNanFunctions_CumSumProd.test_result_valuesse( *D**Zd3C--D1C S )))Kd3C,,u40C S )  *r2ctjd}t|j|jD]N\}}tjd}dD]/}|||}||||}t ||t ||1Py)Nrz)r}r~rr|rr)r7rrrrrr )r9rrrrr~rrs r0rz$TestNanFunctions_CumSumProd.test_outs{ffQi$--7 .FBVVAYF& .4(4V4#C0#C-  . .r2N)rErFrGr7rTrVrrrUrWrrfrgrhr rrr rrrrrrr2r0rorots bmm,H 2::&H [[V\2 [[Wbll:&>? [[W' $<, @3 , #7&*.r2roc neZdZejej ej gZejejejgZ dZ dZ dZdZej"j%dgdej"j%dej&dej"j%d ej(ej*ej,d ej*gd d g dZdZej"j%dej&ddZdZy)TestNanFunctions_MeanVarStdc |jD]N}tjtjtjfD]}t t |td|Py)Nr|)r~r)rrr7rint_object_r r?rp)r9rurs r0test_dtype_errorz,TestNanFunctions_MeanVarStd.test_dtype_errorsK HA''277BJJ7 HiEG H Hr2c |jD]v}tjtjtjfD]B}tj t jd|}tt|t d|Dxy)Nrrr|r) rrr7rrrr)rprr r?)r9rurrs r0test_out_dtype_errorz0TestNanFunctions_MeanVarStd.test_out_dtype_errorsc DA''277BJJ7 Dhhu{{1~U;iEsC D Dr2c 0tjtjg}tjtjg}t ||D]B\}}dD]8}t Dcgc] }|||}}|td|}t||:Dycc}w)Nrr|rCr|r~rD) r7r`rbrarcrrrpr ) r9rrrrrrDrrrs r0 test_ddofz%TestNanFunctions_MeanVarStd.test_ddofsIIryy)FFBFF#(H- .FB .167Ar!$'77QT2#C- . .7s Bctjtjg}tjtjg}t Dcgc] }t |}}t||D]\}}tdD]}t5}|jt|jt|Dcgc]}||k\ } }|td|} ttj | | t#| r"t%t |j&dk(n!t%t |j&dk(dddycc}wcc}w#1swYxYw)Nrr|rr)r7r`rbrarcrrrrangerrrrrrpr rrr log) r9rrrrdsizerrrDrrrs r0test_ddof_too_bigz-TestNanFunctions_MeanVarStd.test_ddof_too_bigsIIryy)FFBFF#!&'AQ''(H- 3FBa 3&( 3CJJ~.JJ~..344194C4U6C #43xCGG  12CGG  12 3 3 3 3( 5 3 3s% E /E6 E A>EEE r~r{rrrrrrr3c|$|jdk(rtjd|j|}d}|jD]}tj t |5|||}dddtjjsJ|tjur|j|jk(rJ|jtj|jk(rJy#1swYxYw)Nrrz:(Degrees of freedom <= 0 for slice.)|(Mean of empty slice)rr) rrfrrrrrrr7rrrXrabsrs r0rz(TestNanFunctions_MeanVarStd.test_allnanss   a KKD E U#LMM 8DnE: -5t, -88C=$$& &&rzz!yyEKK///yyBFF5M$7$7777 8 - -s $ C;;D c tjd}|jD](}dD]}tjd5}tj dt tj|||jt t|dk(t t|djtdddd D]t}tjd5}tj dt|||tjgt t|dk(dddv+y#1swY1xYw#1swYxYw NrrTrrrr|rr)r7rrrrrrr rrrrrrr )r9rrur~rs r0rz&TestNanFunctions_MeanVarStd.test_emptys+hhv )A! G,,D9GQ))(3BHHQs%67;;=>CFaK(Jqt}}nEF GG G  ),,D9)Q))(3 3T!2BHHRLACFaK()) ) )GG ))sBE-AEE E% c,tjdjddj|}tj|dddf<tj |tj }d|dddf<t|j|jD]x\}}|||dd}|tjur|n|jj}|||}|j|k(sJtjj||zy)NrrzrrFr)r)r7rrrrrrrrrrrXrealrr8assert_allclose) r9rrrruf_stdrdtype_referencer&s r0rz&TestNanFunctions_MeanVarStd.test_wheres YYq\ ! !!Q ' . .u 5661a4 Rrww/ad DMM4==9 7HAubim,I'(BJJeBGGMMOBe$C99/ // JJ & &sI 6  7r2cJtjjd}|jddddz}tj|dddddf<tj d}tj d}tj ||dd }||usJtj||dd| }||usJ|j|jk(sJ|jdk(sJtj|dd }|j|jk(sJt||y) Ni rrB)rr|rr|T)rr~r)rr~rrYr}) r7rwrxrandnrrrXrbrr )r9rngAmean_outstd_outrYrcstd_olds r0test_nanstd_with_mean_keywordz9TestNanFunctions_MeanVarStd.test_nanstd_with_mean_keywords ii##D) IIb"a 3 &VV!Q' 88J'((:&zz!# $& 4ii "  #~~yyDJJ&&&yyJ&&&))AA5}} ***C)r2N)rErFrGr7rXr`rbrrrYrarcrrrrrrfrgrhr rrr rrrrrr2r0rrs BIIryy1H(HH D .3" [[V\2 [[Wbll:&>? [[W' $<8 @3 8$ ) [[Wbll:&>? 7@ 7"*r2r) YMWDhmsmsusnspsfsaszm8[]cNeZdZdZdZej jdgdej jddZ dZ d Z d Z ej jdgd ej jd e d ZdZdZdZdZy)TestNanFunctions_Mediancvtj}tj|t |tyr6)rprqr7rZr r9rts r0rvz%TestNanFunctions_Median.test_mutationMs#zz| TT5!r2ctjd}dD]V}tj||dd}tj||dd}t |j |j k(Xtj d}tjjdtj|jdddfz}|jtj}tj|t|<t5}|jt tj|dd}t#|jd tj|d d}t#|jd tj|d d}t#|jd tj|dd}t#|jdtj|dd}t#|jd tj|dd}t#|jddddy#1swYyxYw)Nrzr{Fr~roverwrite_inputrtrTr}r|r|r|r|rr|r|rurvrr|rrur|rrzr|rurvrr|rrzrr|rzr|r|rur|)r7rr[rZr rrrrwrrrr)rtuplerrrr r9rr~rrrrrs r0rz%TestNanFunctions_Median.test_keepdimsSsffQi  *D))CdeLC,,s4OC CHH( ) * GGM " II  X &!''):1d7)C C HHRWW ff%(  2C JJ~ &,,qtd;C L 1,,qv=C M 2,,qv=C L 1,,qtd;C M 2,,q|dCC L 1,,qy4@C L 1 2 2 2s D*II r~Nr|rr)r~argnames argvalues#ignore:All-NaN slice:RuntimeWarningchtjdtjjdtjjdddfz}|j tj }tjt|<|djz}n>t|jtfdtjD}tj|}tj|d|}||usJt|j|y)Nrtrrc3HK|]}|vrdnj|ywr|NrrjrL axis_normrs r0 z.+L89Q)^3L"Tr~rr)r7rrrwrrrr)rrrrrr)rZr )r9r~r shape_outrresultrrs @@r0test_keepdims_outz)TestNanFunctions_Median.test_keepdims_outns GGM " II  X &!''):1d7)C C HHRWW ff%( <qvv I,T166:IL=B166]LLIhhy!adTsC}}V\\9-r2cftjjdd}tj|ddgtjd}tj d}tj |d}tj|d|}t||t||tj d}tj |d}tj|d|}t||t||tj|d|}t||t||y) Nrzrrr|rrrr) r7rwryinsertrrr[rZr )r9rnan_matrrrs r0rz TestNanFunctions_Median.test_outsiinnQ"))C!Qa8!ii!$ll77C(C%"ii$'ll76:C(C%ll7V<C(C%r2cdD]}tjjd|}tjjd|j|jdz}tj |j |<d|dddf<g}|D]R}tjtj||}|jtj|dTttj|d |y) N)rr3rirrrsizerT)rr~r) r7rwrrandintrrravelcompressrr*r[r rZ)r9rrrrxnonans r0test_small_largez(TestNanFunctions_Median.test_small_larges' >A 1%A !!!QVV!&&A+!>A66AGGIaLAadGC C RXXa[L!4 299UDAB C r||AB7 = >r2ctDcgc]}tj|}}tjtd}t ||ycc}wr)rr7r[rZrpr )r9rrrs r0rz*TestNanFunctions_Median.test_result_valuess<%*+ryy|++ll5q)C%,sAr{rctjdtjj|}t 5}|j t tj||}|j|jk(sJtj|jsJ|"tt|jdk(n!tt|jdk(tjtjj|d}tj|}|j|jk(sJtj|sJ|"tt|jdk(n!tt|jdk(dddy#1swYyxYw)Nrrr|rzrrr)r7r rrrrrrZrrrr rrr)r9rr~rroutputscalar output_scalars r0rz$TestNanFunctions_Median.test_allnanssGggfbff%,,U3   +C JJ~ &\\#D1F<<399, ,,88F#'') ))|CGG )*CGG )*XXbff%,,U3B7FLL0M &&&,,6 6688M* **|CGG )*CGG )*+ + + +s E3F::Gc tjd}dD]}tjd5}tjdt tj tj||jt t|dk(t t|djtdddd D]}tjd5}tjdttj||tjgt t|dk(dddy#1swYHxYw#1swYxYwr)r7rrrrr rrZrrrrrr r9rr~rs r0rz"TestNanFunctions_Median.test_emptys hhv CD((5 C%%h/c!=>BBDEA! $ 1Q4==.AB  C C C  %D((5 %%%h/R\\#D9288B<HA! $ % % %  C C  % %sBE*A!E#E #E, cFttjddk(yr)r r7rZrCs r0rz#TestNanFunctions_Median.test_scalars R B&'r2cxtjd}tttj|dtttj|dtttj|dtttj|dtt tj|dy)Nrtrrrrrrr|r|)r7rrr rrZrr9rs r0test_extended_axis_invalidz2TestNanFunctions_Median.test_extended_axis_invalidsj GGM "iqr:iqw?iqq9iqv>j",,?r2c  t5}|jttjtj fD]|}tj |tj gtj tj gg}ttj|d|tj gttj|d|tj gttj||tj tj tj |gtj tj |gg}ttj||ttj|dtj tj |gttj|d|tj ||g||gg}ttj|d|tj |d| dgdtj tj dgdtj tj |ggtj }|dkDrFttj|dd d | d gttj|d nEttj|ddd | dgttj|dttj|ddd|gtdD]r}tddD]_}tj tj g|z|g|zzgdz}ttj||ttj|d|ttj|dtj g|z|g|zztj tj g|z| g|zzgdz}ttj|| ttj|d| ttj|dtj g|z| g|zzbu dddy#1swYyxYw)Nrrr|ruiirrrrg@@g@g$g"gr~grr) rrrr7infrrr rZfloat32r)r9rrarLjs r0test_float_specialz*TestNanFunctions_Median.test_float_specials  + BC JJ~ &() BHHsRVVnrvvrvv.>?@R\\!!4sRVVnER\\!!4sRVVnER\\!_c2HHrvvrvvs3 ffbffc245R\\!_c2R\\!!4rvvrvvs6KLR\\!!4c:HHsCj3*56R\\!!4c:HHsAtR0"BFFBFFA6 "&&"&&#68%'JJ07 aa!82rC4:LM a#6 aa!84cT3:OP a$7R\\!"5T37GHr BA"1b\ BHHx!| &B%Ca%GH$R\\!_c:$R\\!!% &  [[V\2 [[Wk2+33+2 %(@,Br2rc eZdZdZdZej jddddggej jdgdej jd d Z ej jd d d gdZ dZ ej jd d d gej jdd d gdZ ej jdgdej jde jdej jde je j e j"de j gddgdZdZdZdZdZej jdgd d!Zej jdgd"d#Zy$)%TestNanFunctions_Percentilecxtj}tj|dt |ty)N)rprqr7r\r rs r0rvz)TestNanFunctions_Percentile.test_mutations'zz| r"T5!r2ctjd}dD]X}tj|d|dd}tj|d|dd}t |j |j k(Ztj d}tjjdtj|jdddfz}|jtj}tj|t|<t5}|jt tj|ddd }t#|jd tj|dd d }t#|jd tj|ddd }t#|jdtj|ddd }t#|jdtj|ddd }t#|jd tj|ddd }t#|jddddy#1swYyxYw)Nrzr{FFrrtrZTr}rrrrrrrrrr)r7rr]r\r rrrrwrrrr)rrrrrr rs r0rz)TestNanFunctions_Percentile.test_keepdimssffQi  *D--Rd057C""3438:C CHH( )  * GGM " II  X &!''):1d7)C C HHRWW ff%(  2C JJ~ &""1btdCC L 1""1bvEC M 2""1bvEC L 1""1btdCC M 2""1b|dKC L 1""1by4HC L 1 2 2 2s D0I  Ir=rur|r~rrrctjdtjjdtjjdddfz}|j tj }tjt|<|djz}n>t|jtfdtjD}tj||z}tj|}tj||d|}||usJt|j|y)Nrtrrc3HK|]}|vrdnj|ywrrrs r0rz@TestNanFunctions_Percentile.test_keepdims_out..OrrTr)r7rrrwrrrr)rrrrrr)r\r ) r9r=r~rrrrrrs @@r0rz-TestNanFunctions_Percentile.test_keepdims_out9s GGM " II  X &!''):1d7)C C HHRWW ff%( <qvv I,T166:IL=B166]LLIHHQK)+ hhy!!!!QTDcJ}}V\\9-r2weightedFTctjjdd}tj|ddgtjd}tj d}|r1tj |dd}tj |dd}ni}i}tj|dfd di|}tj|dfd|d |}t||t||tj d }tj|dfd di|}tj|dfd|d |}t||t||tj|dfd |d |}t||t||y) Nrzrrr|r inverted_cdfweightsmethod*r~rrr) r7rwryrrrrr]r\r ) r9rrrrw_args nan_w_argsrrs r0rz$TestNanFunctions_Percentile.test_outXsciinnQ"))C!Qa8! !#c!2nMF<<0NJFJmmC6!6v6wMM*MC(C%"mmC9$9&9 R " 2<  C(C% R $& 4>  C(C%r2cZtjgdd}tttj|dtjgdd}tttj|dtjgdd}tttj|dyN)y?@y@?y?ffffff@GrrBrF)r7rr r?r\r9arr_cs r0 test_complexz(TestNanFunctions_Percentile.test_complexwsq=SIi!1!15#>=SIi!1!15#>=SIi!1!15#>r2use_outc |r:ttjd}ttjd}d}n#tj}tj}d}tDcgc]}||d||}}|rtj |nd}|t dd|t |} t| |tjtDcgc]}||d ||c}}|rtj |nd}|t d d|t |} t| |ycc}wcc}w) Nr r c,tj|Sr6)r7rrs r0 gen_weightszCTestNanFunctions_Percentile.test_result_values..gen_weightss||A&r2cyr6rrs r0rzCTestNanFunctions_Percentile.test_result_values..gen_weightssr2rk)r r|)r~r r)rkb) rr7r]r\r empty_likerpr transpose) r9rrr]r\rrrrrs r0rz.TestNanFunctions_Percentile.test_result_valuess  ~FJ#B$4$4^LM 'J,,M CHHQz!RQ8HH$+bmmC E2A$/$6CAC%ll%*, !'q(KNK,-$+bmmC E8!$/$6CAC%I ,s (D9D>r{rFloatrrrrr3c|$|jdk(rtjd|j|}tjt d5t j|d|}dddt jjsJ|j|jk(sJy#1swYJxYw)Nrrrr<r) rrfrrrrr7r\rrrr9r~rrrs r0rz(TestNanFunctions_Percentile.test_allnanss   a KKD E U# \\.0K L 9""5"48C 9xx}  """yyEKK''' 9 9 B55B>c tjd}dD]}tjd5}tjdt tj tj|d|jt t|dk(t t|d jtdddd D]}tjd5}tjdttj|d|tjgt t|d k(dddy#1swYJxYw#1swYxYw) NrrTrr(rr|rr)r7rrrrr rr\rrrrrr rs r0rz&TestNanFunctions_Percentile.test_emptys(hhv CD((5 C%%h/!1!1#r!EFJJLMA! $ 1Q4==.AB  C C C  %D((5 %%%h/R--c2DA288B<PA! $ % % %  C C  % %sBE+A"E%E" %E. cttjdddtjd}tj|dd}t|dt tj |y)Nrr2rrg@)r r7r\rr isscalar)r9rrs r0rz'TestNanFunctions_Percentile.test_scalarsTR%%b#.3 IIaL   Q +Q Ar2ctjd}tttj|ddtttj|ddtttj|ddtttj|ddtt tj|ddy) Nrtrrr=r~rrrr)r7rrr rr\rrs r0rz6TestNanFunctions_Percentile.test_extended_axis_invalids~ GGM "i!1!11Ci!1!11Hi!1!11Bi!1!11Gj""2"2AHr2c ddg}tjd}tj|z}tjd}d|dddddddf<|ddddddfxxdzcc<d D]}d D]}t5}|j t d tj |||| }tj|||| } t| j|jtj |||| }tj|||| } t| |dddtjd } ttj| |djdy#1swY xYw)Nr*r)rrz)rzrrrrrrzr{)FTrr})rzrrr))r|rr)rrzr)) r7rrrrrrr]r\r r) r9percrr large_matr~keepdimrrnan_valmegamats r0test_multiple_percentilesz5TestNanFunctions_Percentile.test_multiple_percentiless]Cyggfo&&3,GGI& !" !QqUA+!Q(q  /D( /&( /CJJ~/JK--TwOC ..w48?AG  :-- 4d18:C ..y$T8?AG #. / / / /'','   Wd 8 > >   / /s ;B'E00E: nan_weight)rr|rrzZbticdtjddg}tjd}tj|dd}t ||d|ddg}tj|d|d }t ||y) Nr|rrzrr*r rrr )r7rfloat64r\r )r9r6rr q_unweightedr q_weighteds r0test_nan_value_with_weightz6TestNanFunctions_Percentile.test_nan_value_with_weightso 1 C''2nE \6**c3 '%%aQ~N Z(r2rr|rc(tjjdtjjd}|j}tj|tj dtj ddf<tj gdgdg}tj|}d|tj|<tj||||d }tj|||d }t||d |tj|<tj||||d }t||tj|} tj||||| d } | | usJt| |y) Nr|)rcrrr?r)g4@gI@r)r!Pr )r~r r )r~r r7)r~r rr ) r7rwseedrqrrrrrr]r\r r) r9r~x_no_nanrpr  p_expected p_unweighted p_weightedrrs r0test_nan_value_with_weight_ndimz;TestNanFunctions_Percentile.test_nan_value_with_weight_ndimsP q99###6 MMO-/VV"))B-2 )* HHnl3 4,,q/ ! ]] adGNL '' qtN4  \:.$ %% qtW^E Z,mmJ' qtW#nNczzS*%r2N)rErFrGrvrrfrgrhrrrrrr7r rrr rrrrr5r<rHrr2r0rrs" 2: [[S1q!f+. [[   [[ EF.G /.& [[Z%7&8&<? [[Z%7 [[Y 6&78&8 [[V\2 [[Wbll7&;< [[W' $<( =3 ( % I 8 [[\+>? )@ ) [[VY/%&0%&r2rc eZdZejj dddgdZdZdZdZ ejj dgd ejj d e jd ejj d e je je jd e jgddgdZy)TestNanFunctions_QuantilerFTc (tjdjdddjt}tj |dd<|rtj |dd}ni}ttj|fd d i|tj|fd d i|ttj|fd dd |tj|fd dd |ttj|fd dd |tj|fd dd |ttj|fd gdd |tj|fd gdd |ttj|fgd dd |tj|fgddd |y)Nrrzrrr|r r r=rBr*r.)?rB?)r*K) r7rrrfloatrrr r^r\)r9rrrs r0test_regressionz)TestNanFunctions_Quantile.test_regressionsw YYr] " "1a + 2 25 9661a !#b!1^LFFR^^B8#88%%b9B9&9 ;R^^B@#A@@%%bABQA&A CR^^B@#A@@%%bABQA&A CR^^BB3%aB6B%%bCRDqCFC ER^^BN*;!NvN%%bKLqKFK Mr2ctjddz}ttj|ddttj|ddttj|ddy)NrrBrrr|g @g?)r7rr r^)r9rs r0 test_basicz$TestNanFunctions_Quantile.test_basic2sS IIaL3 R^^Aq)2.R^^Aq)3/R^^As+T2r2cZtjgdd}tttj|dtjgdd}tttj|dtjgdd}tttj|dyr)r7rr r?r^rs r0rz&TestNanFunctions_Quantile.test_complex8sk=SIi<=SIi<=SIi +T> !>"r2ctjtjtjfD]4}tjddg|}t |d\}}|J||ur4Jtj tjfD]}tjddg|}t |d\}}|dk(jsJ||usJt||tjddtjg|}t |d\}}t|tjgd||usJt|tjgdtj|d rJy) z] Test that _replace_nan returns the original array if there are no NaNs, not a copy. rr|rNrF)FFTr=r~) r7rint32int64rrrr9rr rr)rrrmaskarr_nan result_nanmask_nans r0test__replace_nanrgwsF''288RXX.hh1vU+#C+ ||}} **bjj) %hh1vU+#C+  ""$$$S   VS!((Aq"&&>7+GQ7 HXrxx(<=>(((Z)!45xx $$$ %r2ctjdt}t|j dd5}|j |j tj||j|j}tjtd5tj|tjd dddtjtd 5tj|tjd ddddddy#1swY`xYw#1swYxYw#1swYyxYw) Ng$@rzdata.binzw+b)rrzreduction operation fmaxrrr(zreduction operation fmin)r7rrQopenjoinr0tobytesmemmaprrrfrrrJrrH)tmpdirrfhmms r0test_memmap_takes_fast_routerps #U#A fkk*%u -+  YYr 8 ]]:-G H + IIbbhhqk * +]]:-G H + IIbbhhqk * + ++ + + + + ++s<A'E+D( #E,+D4E(D1 -E4D= 9EE )9r$r functoolsrrfnumpyr7numpy._core.numericrnumpy.exceptionsrrnumpy.lib._nanfunctions_implrr numpy.testingr r r r r rrrrrprrrr!rmr  _TEST_ARRAYSr'_vsetflagsrgrhr keysr4rNrbror _TIME_UNITSrrrrrJrrb complex64r_rgrp)units0r0r~s 46@ 6266662666B7BFFFFFC6626626662667C26626666266BD E 2 3 ; < 67# $ * + - RXXAE<>@A bhhBF=?AB , , ^m%m%`]$]$B "((1+ "((( )     BKKeK LLj!99C? 3 3 5<;L;L;NOR/R/P  R/jf1f1R8$;8$vF.">F.RB*">B*J 2<< +, +6$#dV1 66 FBFBRD&D&N?(?(B RXXrvvsBFFBFF+, RXX()+RXXl"((+  RXX()  RXX}!(* 6 RXXt}t} + "! "%6 +U7s O