.i' d dl Z d dlZd dlZd dlZd dlmZ d dlmZm Z d dl
m Z m Z m
Z
mZmZmZ ej dej ddej dgd d
ej d d d
gej ej ej dej dgdej ej ddej gg Z ej g d ej g d ej ddg ej g d gZ ej g dg dg dg dg Z ej g dg dg d g d g Z G d d Z G d d! Z G d" d# Z ej d$ ej g d% d&Ze D ]Z e d'( ej d)ej d* ej d+ z d,z ej d-e e! . G d/ d0 Z" G d1 d2 Z# G d3 d4e# Z$ G d5 d6e# Z% G d7 d8e# Z&d9Z' e(ej d+ Z)e)d: e'D z
Z) G d; d< Z* G d= d> Z+ G d? d@ Z,ej dA ej ej dBej ej- g ej g dC f ej g dDej. E dFf ej g dC dFf ej ej dBgej ej- ggej/ E ej d'dFgd'dFgg fg dG Z0dH Z1dS )I N)normalize_axis_tuple) _nan_mask _replace_nan)assert_ assert_equalassert_almost_equal
assert_raisesassert_array_equalsuppress_warningss?&S:??ʡE? ?绸zG?L
F%u?6٬?:M
S㥛?z):? h"?)r r
r r )r r r r r )r r r )r ?r
r r r )r r r r r r )r r r r r r )r r r r r r )r r
r r r )r r r r r r )r r r r r r )r r r r r r c e Zd Zej ej ej ej ej ej ej
ej ej ej
ej ej ej ej ej ej ej ej ej ej ej ej ej ej ej ej ej ej iZ d e D Z!e"d d Z#e$j% & de ' e! d Z(d Z)dS )
TestSignatureMatchc g | ] }|j
S __name__).0ks u/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/lib/tests/test_nanfunctions.py
z TestSignatureMatch.9 s
(
(
(!1:
(
(
( ...c < g }t j | }|j D ]Y}|j t j j u r| | 0| | | Zt j | S )z?Construct a signature and replace all default parameter-values.)default)
inspect signature
parametersvaluesr) Parameteremptyappendreplace Signature)funcr) prm_listr+ prms r$
get_signaturez TestSignatureMatch.get_signature; s %d++ '..00 > >C{g / 555$$$$ G < <==== ***r& z
nan_func,funcidsc | | }| | }t j || d S N)r6 nptestingr )selfnan_funcr3 r+
nan_signatures r$ test_signature_matchz'TestSignatureMatch.test_signature_matchG sH
&&t,, **844
=99999r& c t j t | j t t j j j dS )z4Validate that all nan functions are actually tested.N)r; r< r setIDSlib nanfunctions__all__r= s r$ test_exhaustivenessz&TestSignatureMatch.test_exhaustivenessP sC
MM3rv 2 :;;
r& N)r' )*r!
__module__ __qualname__r; nanminaminnanmaxamax nanargminargmin nanargmaxargmaxnansumsumnanprodprod nancumsumcumsum
nancumprodcumprodnanmeanmean nanmedianmedian
nanpercentile
percentile nanquantilequantilenanvarvarnanstdstdNANFUNCSrC staticmethodr6 pytestmark parametrizeitemsr@ rH r r& r$ r r ( s
27
27
bi
bi
26
BG
bi
rz
BG
bi
"-
26
26 H )
(x
(
(
(C + + + \ + [ _hnn.>.>C HH: : IH:
r& r c
x e Zd Zej ej gZej ej gZ d Z
d Z d Z d Z
d Zej dg d ej dej d ej d
ej ej ej d ej gd d
g d Zd Zd Zd Zd Zej dej d d Zej dej d d Z dS )TestNanFunctions_MinMaxc t }| j D ]"} || t |t #d S r: _ndatcopynanfuncsr r= ndatfs r$
test_mutationz%TestNanFunctions_MinMax.test_mutation\ J zz|| & &A
AdGGG u % % % % & &r& c t j d }t | j | j D ]C\ }}dD ];} |||d } |||d }t |j |j k TestNanFunctions_MinMax.test_result_values.. ! (((Q22a55(((r& r} r r rs r _rdatrq r r= r r r r s @r$ test_result_valuesz*TestNanFunctions_MinMax.test_result_values n $- 77 * *FB((((%(((C"U###C S ) ) ) ) * *r& r r| r AllFloatarrayr{ r{ 0d2dr7 c ~ | |j dk rt j d | | }d}| j D ]|}t j t | 5 ||| }d d d n # 1 swxY w Y t j | sJ |j
|j
k sJ }d S Nr *`axis != None` not supported for 0d arraysAll-NaN slice encounteredmatchr ) r ri skipastypers warnsRuntimeWarningr; isnanallr r= r r r r r3 r s r$ test_allnansz$TestNanFunctions_MinMax.test_allnans s
a KE F F F U##+M , ,D nE:::
-
-d5t,,,
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-8C==$$&& & &&9 +++++ , , A77A; >A; c t j t }|j }t j fD ]z} ||d } |t d }t || t |j | t t j |
{d S )Nr} r ) r; ma fix_invalidrq _maskrr rK r r isinfany)r= r mskrv r r s r$ test_maskedz#TestNanFunctions_MinMax.test_masked s e &&inn ) - -A!Ca...C!E"""C c " " " C ( ( (
))+++ , , , , - -r& c R | j D ] }t |d dk d S Nr rs r r= rv s r$ test_scalarz#TestNanFunctions_MinMax.test_scalar : ! !A AAbEERK ! !r& c G d dt j }t j d | }| j D ]} ||d }t t
|| t |j dk ||d }t t
|| t |j dk || }t |j dk t j |d< | j D ]y}t j
d
5 }t j d ||d }t t
|| t t j t j
| t t | dk d d d n # 1 swxY w Y t j
d
5 }t j d ||d }t t
|| t t j
|d o5t j
|d ot j
|d t t | dk d
t t |d j t" d d d n # 1 swxY w Y t j
d
5 }t j d || }t |j dk t |t j k t t | dk d d d n # 1 swxY w Y {d S )Nc e Zd ZdS )8TestNanFunctions_MinMax.test_subclass..MyNDArrayNr! rI rJ r r& r$ MyNDArrayr Dr& r r{ r r r{ r} r Trecordalways no warning raised)r; ndarrayr viewrs r
isinstanceshapenanwarningscatch_warnings simplefilterr r len
issubclasscategoryr )r= r minerv r ws r$
test_subclassz%TestNanFunctions_MinMax.test_subclass s
vayy~~i(( % %A!Dq///C JsI.. / / / CI% & & &!Dq///C JsI.. / / / CI% & & &!D''C CIO $ $ $ $ &Q % %A(555
%%h///a1ooo
3 22333BF28C== 1 11222A! $$$
%
%
%
%
%
%
%
%
%
%
%
%
%
%
% (555
C%h///a1ooo
3 22333Q(( 1#a&1A1A-A 1 "Q 0 0 02 2 2A! %8999
1Q4=.AABBB
C
C
C
C
C
C
C
C
C
C
C
C
C
C
C (555
%%h///agg R(((rv
&&&A! $$$
%
%
%
%
%
%
%
%
%
%
%
%
%
%
%# % %s9 BF>>G G C*KK K 4A5M55M9 <M9 c t j ddgt j dgt j t j ggt }t t j | d t t j |d ddg t
j d 5 }t
j d t t t j |d
ddt j g t t | d
k d t t |d j
t d d d d S # 1 swxY w Y d S ) Nr g @ @r r r Tr r r} r )r; r r objectr rK r r r listr r r r r )r= arrr s r$ test_object_arrayz)TestNanFunctions_MinMax.test_object_array sZ hc
RVSMBFBF3CDFSSSRYs^^S)))RYs+++c3Z888
$D
1
1
1 ?Q !( + + + bi! 4 4 455S"&7I J J J CFFaK!4 5 5 5 Jqt}n== > > > ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?s B!EEEc G d dt j }t j d | }t j |d d< | j D ]k}|t j u rdnd} ||| }|j |k sJ ||k sJ || | | }|j |k sJ ||k sJ ld S )Nc e Zd ZdS )7TestNanFunctions_MinMax.test_initial..MyNDArrayNr r r& r$ r r r r& r d r initial) r; r aranger r rs rM r r )r= r r arrv r ret1ret2s r$ test_initialz$TestNanFunctions_MinMax.test_initial s
Yq\\
'
'2A2 # #A ")^^ccG1R)))D:&&&&7????1RWWY''999D:&&&&7????? # #r& c G d dt j }t j d dd | }t j |dd d f< t j |t j }d|d d df< | j D ]m}|t j u rdnd } |||d
}|j
|k sJ ||k sJ || | |d
}|j
|k sJ ||k sJ nd S ) Nc e Zd ZdS )5TestNanFunctions_MinMax.test_where..MyNDArrayNr r r& r$ r r r r& r r r{ r r F r wherer ) r; r r reshaper r ones_likebool_rs rK r r ) r= r r r r rv referencer r s r$
test_wherez"TestNanFunctions_MinMax.test_where sF
Yq\\
!
!!Q
'
'
.
.u
5
561aaa4 Rrx000aaad % %A ")^^I1Rua000D:&&&&9$$$$1RWWY''ua@@@D:&&&&9$$$$$ % %r& N) r! rI rJ r; rK rM rs minmaxr rw r r r r ri rj rk typecodesr r fullr r r r r r r r r& r$ rn rn W s 29%H H& & &. . .* * * $ $ $* * * [ V\\\22 [ Wbl:&>?? [ W '
$<
,
, @? 32
,- - -! ! !)% )% )%V
?
?
? [ Wbl:&>??# # @?#$ [ Wbl:&>??% % @?% % %r& rn c
D e Zd Zej ej gZd Zd Ze j
dg d e j
dej d e j
d ej
ej ej dej gd d
g d Zd
Zd Zd Ze j
dej d d Ze j
dej d d ZdS ) TestNanFunctions_ArgminArgmaxc t }| j D ]"} || t |t #d S r: rp rt s r$ rw z+TestNanFunctions_ArgminArgmax.test_mutation rx r& c : t | j t j t j g D ]\ }}t
D ]}t
5 }| t d || }|| }t t j
| t ||| t t j ||d | d d d n # 1 swxY w Y d S )Nz invalid value encountered in)
r rs r; greaterlessrq r filterr r r r equal)r= rv fcmprowsupindvals r$ r z0TestNanFunctions_ArgminArgmax.test_result_values s\ 4=2:rw*?@@
@
@GAt
@
@&(( @CJJ~/MNNN!C&&C c(C
-...S# 2 2 4 4 4555c$3$i 8 8 < < > > >???@ @ @ @ @ @ @ @ @ @ @ @ @ @ @
@
@
@s B:DD D r r| r r r r r r r7 c | |j dk rt j d | | }| j D ]B}t j t d 5 ||| d d d n # 1 swxY w Y Cd S r )r ri r r rs raises
ValueError)r= r r r r3 s r$ r z*TestNanFunctions_ArgminArgmax.test_allnans s
a KE F F F U##M ' 'D z1LMMM
'
'U&&&&
'
'
'
'
'
'
'
'
'
'
'
'
'
'
' ' 's A55A9 <A9 c t j d }| j D ]S}dD ]}t t ||| dD ]1} ||| }t |t j d 2Td S )Nr r{ r Nr r} r )r; r rs r r r )r= r rv r r s r$
test_emptyz(TestNanFunctions_ArgminArgmax.test_empty. s hv / /A!
=
= j!St<<<<<
/
/a$''' S"(1++....
/ / /r& c R | j D ] }t |d dk d S r r r s r$ r z)TestNanFunctions_ArgminArgmax.test_scalar7 r r& c G d dt j }t j d | }| j D ]} ||d }t t
|| t |j dk ||d }t t
|| t |j dk || }t |j dk d S ) Nc e Zd ZdS )>TestNanFunctions_ArgminArgmax.test_subclass..MyNDArrayNr r r& r$ r r$ < r r& r r{ r r r r} r )r; r r r rs r r r )r= r r rv r s r$ r z+TestNanFunctions_ArgminArgmax.test_subclass; s
vayy~~i(( % %A!Dq///C JsI.. / / / CI% & & &!Dq///C JsI.. / / / CI% & & &!D''C CIO $ $ $ $ % %r& c t j d | }t j |d d< | j D ];}|t j u rdnd} ||d }|j |j k sJ ||k sJ ?? [ W '
$< ' ' @? 32 '/ / /! ! !% % %" [ Wbl:&>??$ $ @?$ [ Wbl:&>?? $ $ @? $ $ $r& r
r ) ' ] W . )r 1dF)writer
AllIntegerr Or r7 c e Zd Zej ej ej ej ej ej ej
ej ej ej
ej ej ej ej ej ej ej ej ej ej ej ej ej ej i Z d e D Z e j de ! e ej" d d Z#e j dej$ ej% fej& ej' fgddg d Z(e j dej ej fej ej fgd
d g d Z)d
S ) TestNanFunctions_NumberTypesc g | ] }|j
S r r )r" is r$ r% z'TestNanFunctions_NumberTypes. s 000!1:000r& z nanfunc,funcr7 ignore)overc | | } || } || }t || |dk r"t | t | u sJ d S |j |j k sJ d S )Nr3 r r r r r= r r nanfuncr3 r r s r$ test_nanfuncz)TestNanFunctions_NumberTypes.test_nanfunc s jj d3iigcllC%%% C<<99S ))))))9 ))))))r& ra r_ c | | | }|j j dk r2t t ||d t t ||d d S ||d } ||d }t || |dk r"t
| t
| u sJ d S |j |j k sJ d S )Nr r} qr3 )r r kindr TypeErrorr r r< s r$ test_nanfunc_qz+TestNanFunctions_NumberTypes.test_nanfunc_q s jj 9>S )T3! 4 4 4 4 )WcQ 7 7 7 7 7 7 $sa...C'####C S ) ) )||CyyDII------yCI------r& rc re c | | } ||d } ||d }t || |dk r"t | t | u sJ d S |j |j k sJ d S )N ?ddofr3 r; r< s r$ test_nanfunc_ddofz.TestNanFunctions_NumberTypes.test_nanfunc_ddof s jj d3S!!!gc$$$C%%% C<<99S ))))))9 ))))))r& N)*r! rI rJ r; rK r rM r rO rP rQ rR rS rT rU rV rW rX rY rZ r[ r\ r] r^ rc rd re rf rs nanfunc_idsri rj rk rl errstater> ra rb r_ r` rD rI r r& r$ r5 r5 m s 26
26
bi
bi
26
BG
bi
rz
BG
bi
26
26
H 10x000K [ ^X^^-=-=; OOR[h * * PO * [
."+ &)92=(IJ
O ,
. .
." [
)RV ry"& 12
x
* *
* * *r& r5 c D e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
S ) SharedNanFunctionsTestsMixinc t }| j D ]"} || t |t #d S r: rp rt s r$ rw z*SharedNanFunctionsTestsMixin.test_mutation rx r& c t j d }t | j | j D ]C\ }}dD ];} |||d } |||d }t |j |j k . r r& r} r r r s @r$ r z/SharedNanFunctionsTestsMixin.test_result_values r r& c R | j D ] }t |d dk d S r r r s r$ r z(SharedNanFunctionsTestsMixin.test_scalar r r& c G d dt j }t j d }| | }| j D ]} ||d j } ||d }t
t || t
|j |k ||d j } ||d }t
t || t
|j |k || j } || }t
t || t
|j |k d S )Nc e Zd ZdS )=SharedNanFunctionsTestsMixin.test_subclass..MyNDArrayNr r r& r$ r r`
r r& r r{ r r r} )r; r r r rs r r r )r= r r r rv expected_shaper s r$ r z*SharedNanFunctionsTestsMixin.test_subclass s_
q zz)$$ 1 1A Qu1 - - - 3N!Dq///C JsI.. / / / CI / 0 0 0 Qu1 - - - 3N!Dq///C JsI.. / / / CI / 0 0 0 QuXX^N!D''C JsI.. / / / CI / 0 0 0 0 1 1r& N) r! rI rJ rw r r rW rY r r r r r r& r$ rM rM s & & &. . .* * *( ( ("( ( (" $ $ $* * * ! ! !1 1 1 1 1r& rM c
H e Zd Zej ej gZej ej gZ e
j dg d e
j dej
d e
j d ej ej ej dej gddg d
Zd Ze
j dej
d d Ze
j dej
d d
ZdS )TestNanFunctions_SumProdr r| r r r r r r r7 c | |j dk rt j d | | }t | j ddg D ]>\ }} ||| }t
j ||k sJ |j |j k sJ ?d S )Nr r r} r r ri r r r rs r; r r r= r r r r3 identityr s r$ r z%TestNanFunctions_SumProd.test_allnans% s
a KE F F F U## !$-!Q 8 8 , ,ND($u4(((C6#/** * **9 +++++ , ,r& c L t t j t j gddg D ]z\ }}t j d }|gdz } ||d }t || g } ||d }t || |} ||d }t || {d S )Nr r} r r{ r )r r; rS rU r r r= rv tgt_valuer r r s r$ r z#TestNanFunctions_SumProd.test_empty5 s BJ 7!Q @ @
#
#LAy(6""C +a-C!Ca...C c " " "C!Ca...C c " " "C!Cd###C c " " " "
#
#r& c t j d | }t j |d d< | j D ]6}|t j u rdnd} ||d }|j |k sJ ||k sJ 7d S )Nr r i
r r )r; r r r rs rS r r& s r$ r z%TestNanFunctions_SumProd.test_initialB s
Yq\\
'
'2A2 $ $A 29nn$I!B"""C9%%%%)##### $ $r& c r t j d dd | }t j |dd d f< t j |t j }d|d d df< | j D ]7}|t j u rdnd} |||d }|j |k sJ ||k sJ 8d S )
Nr r{ r r F i r r )
r; r r r r r r rs rS r )r= r r r rv r r' s r$ r z#TestNanFunctions_SumProd.test_whereM s
Yq\\
!
!!Q
'
'
.
.u
5
561aaa4 Rrx000aaad $ $A 29nn$I!BeQ///C9%%%%)##### $ $r& N)r! rI rJ r; rS rU rs rT rV r ri rj rk r
r r r r r r r r r& r$ rc rc sX 2:&H H [ V\\\22 [ Wbl:&>?? [ W '
$< , , @? 32 , # # # [ Wbl:&>??$ $ @?$ [ Wbl:&>??
$
$ @?
$
$
$r& rc c
e Zd Zej ej gZej ej gZ e
j dg d e
j dej
d e
j d ej ej ej dej gddg d
Zd Zd Zd
Zd ZdS )TestNanFunctions_CumSumProdr r| r r r r r r r7 c | |j dk rt j d | | }t | j ddg D ]<\ }} || }t
j ||k sJ |j |j k sJ =d S )Nr r r} re rf s r$ r z(TestNanFunctions_CumSumProd.test_allnans` s
a KE F F F U## !$-!Q 8 8 , ,ND($u++C6#/** * **9 +++++ , ,r& c p t | j ddg D ]\ }}t j d }|t j d z } ||d }t || |} ||d }t || t j d } ||d }t || d S )Nr r} r r )r rs r; r onesr ri s r$ r z&TestNanFunctions_CumSumProd.test_emptyp s
1v 6 6
#
#LAy(6""CBGFOO+C!Ca...C c " " "C!Ca...C c " " "(A--C!Cd###C c " " " "
#
#r& c Z t | j | j D ]W\ }}t j d }dD ];} |||d } |||d }t |j |j k Zd S )Nr{ )r~ r r r} r r )r; r r rs r r )r= r r r r r r r s r$ r z$TestNanFunctions_CumSumProd.test_out s fQii$- 77 . .FBVAYYF&
.
.b4(((b4V444#C000#C----
. . .r& N)r! rI rJ r; rW rY rs rX rZ r ri rj rk r
r r r r r r r r r r& r$ rp rp [ s bm,H 2:&H [ V\\\22 [ Wbl:&>?? [ W '
$< , , @? 32 , # # #7 7 7&* * *. . . . .r& rp c
e Zd Zej ej ej gZej ej ej
gZ d Z d Z
d Zd Zej dg d ej dej d ej d ej ej ej d
ej gd d g
d Zd Zej dej d d ZdS )TestNanFunctions_MeanVarStdc | j D ]F}t j t j t j fD ] }t t |t d| !Gd S )Nr} )r r )rs r; r int_object_r rC rq )r= rv r s r$ test_dtype_errorz,TestNanFunctions_MeanVarStd.test_dtype_error sa H HA (BGRZ8
H
H iEGGGGG
H H Hr& c | j D ]l}t j t j t j fD ]F}t j t j d | }t t |t d| Gmd S )Nr r r} r )
rs r; r r r r/ rq r r rC )r= rv r r s r$ test_out_dtype_errorz0TestNanFunctions_MeanVarStd.test_out_dtype_error s{ D DA (BGRZ8
D
Dhu{1~U;;; iEsCCCCC
D D Dr& c t j t j g}t j t j g}t || D ]A\ }dD ]9fdt D } |t d }t || :Bd S )Nr r} c * g | ]} | S )rG r )r" r rH r s r$ r% z9TestNanFunctions_MeanVarStd.test_ddof.. s( 777Arr!$'''777r& r} r rH ) r; rc re rd rf r r rq r )r= rs r r r r rH r s @@r$ test_ddofz%TestNanFunctions_MeanVarStd.test_ddof s Iry)FBF#(H-- . .FB
.
.77777777bQT222#C----
. . .r& c t j t j g}t j t j g}d t
D }t
|| D ]\ }}t d D ] t 5 }| t | t j fd|D } |t d }t t j | | t! | r&t# t% |j dk n%t# t% |j dk d d d n # 1 swxY w Y d S )Nc , g | ]}t | S r )r r" r s r$ r% zATestNanFunctions_MeanVarStd.test_ddof_too_big.. s '''AQ'''r& r c g | ]}|k S r r )r" r rH s r$ r% zATestNanFunctions_MeanVarStd.test_ddof_too_big.. s 444419444r& r} r r )r; rc re rd rf r r ranger r r r rU rq r r r r r log)
r= rs r dsizer r r r r rH s
@r$ test_ddof_too_bigz-TestNanFunctions_MeanVarStd.test_ddof_too_big s Iry)FBF#'''''(H-- 3 3FB a
3
3&(( 3CJJ~...JJr 01114444e444C "U666C # 4443xx 3 CG 12222 CG 1222 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3
3 3 3s 8CEE "E r r| r r r r r r r7 c | |j dk rt j d | | }d}| j D ]}t j t | 5 ||| }d d d n # 1 swxY w Y t j | sJ |t j
u r|j |j k sJ |j t j | j k sJ d S )Nr r z:(Degrees of freedom <= 0 for slice.)|(Mean of empty slice)r r )
r ri r r rs r r r; r r r[ r absr s r$ r z(TestNanFunctions_MeanVarStd.test_allnans s<
a KE F F F U##LM
8
8D nE:::
-
-d5t,,,
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-8C==$$&& & && rz!!yEK/////yBF5MM$777777
8
8r c t j d }| j D ]g}dD ]}t j d 5 }t j d t
t j ||| t
t | dk t
t |d j t d d d n # 1 swxY w Y d D ]}t j d 5 }t j d t ||| t j g t
t | dk d d d n # 1 swxY w Y id S
Nr r Tr r r r} r r )r; r rs r r r r r r r r r r r )r= r rv r r s r$ r z&TestNanFunctions_MeanVarStd.test_empty s hv ) )A!
G
G ,D999 GQ )(333BHQQs%6%6%6 7 7 ; ; = =>>>CFFaK(((Jqt}n E EFFF G G G G G G G G G G G G G G G
)
) ,D999 )Q )(333 3T!2!2!2BHRLLAAACFFaK((() ) ) ) ) ) ) ) ) ) ) ) ) ) )
) ) )s% B C$$C( +C( A"E77E; >E; c t j d dd | }t j |dd d f< t j |t j }d|d d df< t | j | j D ]t\ }} ||| dd }|t j
u r|n |j j } ||| }|j |k sJ t j
|| ud S )Nr r{ r r Fr )r )r; r r r r r r r rs r r[ realr r< assert_allclose) r= r r r rv f_stdr dtype_referencer' s r$ r z&TestNanFunctions_MeanVarStd.test_where s
Yq\\
!
!!Q
'
'
.
.u
5
561aaa4 Rrx000aaad DM4=99 7 7HAu bim,,I'(BJeeBGMO!Be$$$C9//// J & &sI 6 6 6 6
7 7r& N)r! rI rJ r; r[ rc re rs r\ rd rf r r r r r ri rj rk r
r r r r r r r r& r$ r r sn
BIry1H(HH H H
D D D . . .3 3 3" [ V\\\22 [ Wbl:&>?? [ W '
$< 8 8 @? 32 8$
)
)
) [ Wbl:&>?? 7 7 @? 7 7 7r& r )
YMWDhmsmsusnspsfsasc g | ]}d | d S )zm8[]r )r" units r$ r% r% s 666$
d
666r& c ` e Zd Zd Zd Zej dg d ej d d Z d Z
d Z d
Z ej dg d ej d e
d
Zd Zd Zd Zd ZdS )TestNanFunctions_Medianc t }t j | t |t d S r: )rq rr r; r] r r= ru s r$ rw z%TestNanFunctions_Median.test_mutation s5 zz||
TT5!!!!!r& c t j d }dD ]O}t j ||d d }t j ||d d }t |j |j k Pt j d }t j d t j |j d d d f z }|
t j }t j |t | < t 5 }| t t j |d d }t# |j d t j |d
d }t# |j d t j |d d }t# |j d
t j |dd }t# |j d t j |dd }t# |j d t j |dd }t# |j d d d d d S # 1 swxY w Y d S )Nr{ r| Fr r overwrite_inputru r Tr~ r} r} r} r} r r} r} rv rw r r} r rv r} r r{ r} rv rw r r} r r{ r r} r{ r} r} rv r} )r; r r^ r] r r rs rx r r r r* r tupler r r r r= r r r r r r r s r$ r z%TestNanFunctions_Median.test_keepdims s= fQii * *D)CdeLLLC,s4OOOC CH( ) ) ) ) GM " " I X & &!'):):111d7)C C
HHRW f%((
2C JJ~ & & &,qtd;;;C L 1 1 1,qv===C M 2 2 2,qv===C L 1 1 1,qtd;;;C M 2 2 2,q|dCCCC L 1 1 1,qy4@@@C L 1 1 1
2
2
2
2
2
2
2
2
2
2
2
2
2
2
2
2
2
2s D#H77H;>H;r Nr} r r )r argnames argvalues#ignore:All-NaN slice:RuntimeWarningc \ t j d t j d t j j d d d f z }| t j }t j t | < | dj z }nCt |j t fdt j D }t j | }t j
|d| }||u sJ t |j | d S )Nru r r c 3 > K | ]}|v rd n j | V dS r} Nr r" r7 axis_normr s r$ z.F N L L89Q)^^ L L L L L Lr& Tr r r )r; rs rx r r r r* r r r r r r/ r] r )r= r r shape_outr resultr r s @@r$ test_keepdims_outz)TestNanFunctions_Median.test_keepdims_out1 s'
GM " " I X & &!'):):111d7)C C
HHRW f%(( < qv
II,T16::I L L L L L=B16]] L L L L LIhy!! adTsCCC}}}}V\9-----r& c t j dd }t j |ddgt j d }t j d }t j |d }t j |d| }t || t || t j d }t j |d }t j |d | }t || t || t j |d| }t || t || d S ) Nr{ r r r} r r r r ) r; rx rz insertr r r^ r] r r= r nan_matr r r s r$ r z TestNanFunctions_Median.test_outM s innQ"")C!Qa888!i!$$$l7777C(((C%%%" i$'''l76:::C(((C%%%l7V<<<C(((C%%%%%r& c dD ]}t j d| }t j d|j |j dz }t j | |< d|d d df< g }|D ]S}t j t j | | }| t j
|d Tt t j |d
| d S ) N)r 3 r i r r r )sizer T)r r r )
r; rx randnrandintr r ravelcompressr r0 r^ r
r] )r= r r r r xnonans r$ test_small_largez(TestNanFunctions_Median.test_small_large_ s ' > >A 1%%A !!!QV!&A+!>>A 6AGGIIaLAaaacFC
C
C RXa[[L!44
29UDAAABBBB r|AB 7 7 7 = = = = > >r& c ~ d t D }t j t d }t || d S )Nc 6 g | ]}t j | S r )r; r^ r s r$ r% z>TestNanFunctions_Median.test_result_values..p s ///A29Q<<///r& r} r )r r; r] rq r r= r r s r$ r z*TestNanFunctions_Median.test_result_valueso s? /////C,u1---C S ) ) ) ) )r& r| r c t j dt j | }t 5 }| t t j || }|j |j k sJ t j |
sJ |&t t |j
dk n%t t |j
dk t j t j | d }t j | }|j |j k sJ t j | sJ |&t t |j
dk n%t t |j
dk d d d d S # 1 swxY w Y d S )Nr r r} r{ r r r )r; r r r r r r r] r r r r r r r )r= r r r r outputscalar
output_scalars r$ r z$TestNanFunctions_Median.test_allnanst s gfbf%%,,U33
+C JJ~ & & &\#D111F<39,,,,8F##'')) ) ))|CG )****CG )*** Xbf%%,,U33B7F L 0 0M &&,66668M** * **|CG )****CG )***+ + + + + + + + + + + + + + + + + + +s E8GG
G
c
t j d }dD ]}t j d 5 }t j d t t j t j || t t | dk t t |d j t d d d n # 1 swxY w Y d D ]}t j d 5 }t j d t t j || t j g t t | dk d d d n # 1 swxY w Y d S r )r; r r r r r r r] r r r r r r r= r r r s r$ r z"TestNanFunctions_Median.test_empty s hv C CD(555
C%h/// c!=!=!=>>BBDDEEEA! $$$
1Q4=.AABBB
C
C
C
C
C
C
C
C
C
C
C
C
C
C
C
% %D(555
%%h/// R\#D 9 9 928B<<HHHA! $$$
%
%
%
%
%
%
%
%
%
%
%
%
%
%
% % %s% B)C##C' *C' A+E??F F c P t t j d dk d S r )r r; r] rG s r$ r z#TestNanFunctions_Median.test_scalar s% R B&'''''r& c t j d }t t j t j |d t t j t j |d t t j t j |d t t j t j |d t t
t j |d d S )Nru r r r r r r r} r} )r; rs r AxisErrorr] r r= r s r$ test_extended_axis_invalidz2TestNanFunctions_Median.test_extended_axis_invalid s GM " "blBL!"====blBL!'BBBBblBL!!<<<<blBL!&AAAAj",??????r& c
t 5 }| t t j t j fD ]b}t j |t j gt j t j gg }t t j |d |t j g t t j |d |t j g t t j | | t j t j t j |gt j t j |gg }t t j | | t t j |d t j t j |g t t j |d | t j ||g||gg }t t j |d | t j |d| dgdt j t j dgdt j t j |ggt j }|dk rLt t j |d d
d | d g t t j | d
nKt t j |d dd | dg t t j | d t t j |d dd|g t dd D ]t}t dd D ]_}t j t j g|z |g|z z gdz }t t j | | t t j |d | t t j |d t j g|z |g|z z t j t j g|z | g|z z gdz }t t j | | t t j |d | t t j |d t j g|z | g|z z avd d d d d S # 1 swxY w Y d S )Nr r r} rv iir r r r g @ @g @g $g "g r g
r ) r r r r; infr r r r] float32r )r= r r ar7 js r$ test_float_specialz*TestNanFunctions_Median.test_float_special s
+ BC JJ~ & & &( )
B )
BHsRVnrvrv.> ?@@ R\!! 4 4 4sRVnEEE R\!! 4 4 4sRVnEEE R\!__c222 Hrvrvs 3 fbfc 2 4 5 5 R\!__c222 R\!! 4 4 4rvrvs6KLLL R\!! 4 4 4c::: HsCj3* 566 R\!! 4 4 4c:::HsAtR 0 "BFBFA 6 "&"&# 6 8 %'J0 0 0 77 aa!8!8!82rC4:LMMM a #6666 aa!8!8!84cT3:OPPP a $777 R\!" 5 5 5T37GHHH q" B BA "1b\\ B B Hx!| &B%Ca%G H H$R\!__c:::$R\!!%<%<% > > * * *
[ V\\\22 [ Wk22+ + 32 32+2 % % % ( ( (@ @ @,B ,B ,B ,B ,Br& r c
\ e Zd Zd Zd Zej ddddgg ej dg d ej d d
Z d Z
d Z d
Z ej dg d ej de
j d ej d e
j e
j e
j de
j gddg d Zd Zd Zd Zd ZdS ) TestNanFunctions_Percentilec t }t j |d t |t d S )N )rq rr r; r_ r r s r$ rw z)TestNanFunctions_Percentile.test_mutation s8 zz||
r"""T5!!!!!r& c t j d }dD ]Q}t j |d|d d }t j |d|d d }t |j |j k Rt j d }t j d t j |j d d d f z }|
t j }t j |t | < t 5 }| t t j |dd d
}t# |j d t j |dd d
}t# |j d
t j |ddd
}t# |j d t j |ddd
}t# |j d t j |ddd
}t# |j d t j |ddd
}t# |j d d d d d S # 1 swxY w Y d S )Nr{ r| F Fr ru r Z Tr~ r r r r r r r r r r )r; r r` r_ r r rs rx r r r r* r r r r r r r s r$ r z)TestNanFunctions_Percentile.test_keepdims sb fQii * *D-Rd057 7 7C"3438: : :C CH( ) ) ) ) GM " " I X & &!'):):111d7)C C
HHRW f%((
2C JJ~ & & &"1btdCCCC L 1 1 1"1bvEEEC M 2 2 2"1bvEEEC L 1 1 1"1btdCCCC M 2 2 2"1b|dKKKC L 1 1 1"1by4HHHC L 1 1 1
2
2
2
2
2
2
2
2
2
2
2
2
2
2
2
2
2
2s D)H??IIrA rv r} r r r r c t j d t j d t j j d d d f z }| t j }t j t | < | dj z }nCt |j t fdt j D }t j | |z }t j | }t j
||d| }||u sJ t |j | d S )Nru r r c 3 > K | ]}|v rd n j | V dS r r r s r$ r z@TestNanFunctions_Percentile.test_keepdims_out.. r r& Tr )r; rs rx r r r r* r r r r r r/ r_ r ) r= rA r r r r r r r s @@r$ r z-TestNanFunctions_Percentile.test_keepdims_out s<
GM " " I X & &!'):):111d7)C C
HHRW f%(( < qv
II,T16::I L L L L L=B16]] L L L L LIHQKK)+ hy!!!!QTDcJJJ}}}}V\9-----r& c t j dd }t j |ddgt j d }t j d }t j |dd }t j |dd| }t || t || t j d }t j |dd }t j |dd | }t || t || t j |dd | }t || t || d S )
Nr{ r r r} r * r r r ) r; rx rz r r r r` r_ r r s r$ r z$TestNanFunctions_Percentile.test_out s, innQ"")C!Qa888!mC!,,, w???C(((C%%%" mC$/// w6BBBC(((C%%% wVDDDC(((C%%%%%r& c \ t j g dd }t t t j |d t j g dd }t t t j |d t j g dd }t t t j |d d S N)y ? @y @ ?y?ffffff@Gr rF r F)r; r r rC r_ r= arr_cs r$ test_complexz(TestNanFunctions_Percentile.test_complex- s 777sCCCi!15#>>>777sCCCi!15#>>>777sCCCi!15#>>>>>r& c d t D }t j t dd }t || t j d t D }t j t dd }t || d S )Nc 8 g | ]}t j |d S )rl r; r` r s r$ r% zBTestNanFunctions_Percentile.test_result_values..6 s$ 333r}Q##333r& rl r} r c 8 g | ]}t j |d S )rl b r r s r$ r% zBTestNanFunctions_Percentile.test_result_values..: s$ FFF1BM!X 6 6FFFr& r )r r; r_ rq r transposer s r$ r z.TestNanFunctions_Percentile.test_result_values5 s 33U333 ubq111C%%%lFFFFFGG uhQ777C%%%%%r& r| r Floatr r r r r7 c z | |j dk rt j d | | }t j t
d 5 t
j |d| }d d d n # 1 swxY w Y t
j | sJ |j
|j
k sJ d S )Nr r r r < r ) r ri r r r r r; r_ r r r r= r r r r s r$ r z(TestNanFunctions_Percentile.test_allnans> s
a KE F F F U##
\.0K
L
L
L 9 9"5"4888C 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9x}} """""yEK'''''' A66A:=A:c t j d }dD ]}t j d 5 }t j d t t j t j |d| t t | dk t t |d j t d d d n # 1 swxY w Y d
D ]}t j d 5 }t j d t t j |d| t j g t t | d k d d d n # 1 swxY w Y d S ) Nr r Tr r ( r r} r r )r; r r r r r r r_ r r r r r r r s r$ r z&TestNanFunctions_Percentile.test_emptyN s hv C CD(555
C%h///!1#r!E!E!EFFJJLLMMMA! $$$
1Q4=.AABBB
C
C
C
C
C
C
C
C
C
C
C
C
C
C
C
% %D(555
%%h/// R -c2D A A A28B<<PPPA! $$$
%
%
%
%
%
%
%
%
%
%
%
%
%
%
% % %s% B*C$$C( +C( A,FF F c t t j dd d t j d }t j |dd }t |d t t j | d S )Nr r 2 r r g @)r r; r_ r r isscalar)r= r rs r$ r z'TestNanFunctions_Percentile.test_scalar\ so R%b#..333 IaLL Q + + +Q A r& c t j d }t t j t j |dd t t j t j |dd t t j t j |dd t t j t j |dd t t
t j |dd d S ) Nru r r rA r r r r r )r; rs r r r_ r r s r$ r z6TestNanFunctions_Percentile.test_extended_axis_invalidc s GM " "blB$4a12FFFFblB$4a17KKKKblB$4a11EEEEblB$4a16JJJJj""2AHHHHHHr& c ddg}t j d }t j |z }t j d }d|d d dddd d f< |d d d d dd fxx dz cc< d D ]}d
D ]}t 5 }| t
d t j |||| }t j |||| } t | j |j t j |||| }t j |||| } t | | d d d n # 1 swxY w Y Όt j d
}
t t j |
|d j d d S )Nr" r )r r{ )r{ r r r r r r{ r| )FTr r~ )r{ r r r! )r} r r )r r{ r! )
r; rs r r r r r` r_ r r ) r= percr r large_matr keepdimr r nan_valmegamats r$ test_multiple_percentilesz5TestNanFunctions_Percentile.test_multiple_percentilesk s Cygfoo&3,GI&& !" !!!QqUAAA+ !!!QQQ(q
/
/D(
/
/&(( /CJJ~/JKKK -TwOOOC .w48? A A AG ::: - 4d18: : :C .y$T8? A A AG #... / / / / / / / / / / / / / / /
/ ',''R%gt&AAAGSSSSSs :B&D,,D0 3D0 N)r! rI rJ rw r ri rj rk r r r r r r; r
r r r r r r r r- r r& r$ r r s " " " 2 2 2: [ S1q!f+.. [
[ EFF. . GF /..&& & &$? ? ?& & & [ V\\\22 [ Wbl7&;<< [ W '
$< ( ( =< 32 ( % % % I I IT T T T Tr& r c
f e Zd Zd Zd Zd Zd Zej dg d ej de
j d ej d e
j e
j
e
j d
e
j
gd d g
d ZdS )TestNanFunctions_Quantilec t j d ddd t }t j |d d< t
t j |d t j |d t
t j |dd
t j |d d
t
t j |dd
t j |d d
t
t j |dgd
t j |d gd
t
t j |g d d
t j |g d d
d S )
N r r{ r r r} rF r@ r" r& ) ?rF ?) r" K ) r; r r r floatr r ra r_ )r= r s r$ test_regressionz)TestNanFunctions_Quantile.test_regression ss
Yr]]
"
"1a
+
+
2
25
9
961aR^B#...0@r0J0J0JKKKR^B#A666%bBQ777 9 9 9R^B#A666%bBQ777 9 9 9R^B3%a888%bRDq999 ; ; ;R^B*;*;*;!DDD%bLLLqAAA C C C C Cr& c t j d dz }t t j |d d t t j |d d t t j |d d d S )Nr rF r r r} g @g ?)r; r r ra )r= r s r$
test_basicz$TestNanFunctions_Quantile.test_basic sn IaLL3 R^Aq))2...R^Aq))3///R^As++T22222r& c \ t j g dd }t t t j |d t j g dd }t t t j |d t j g dd }t t t j |d d S r )r; r r rC ra r s r$ r z&TestNanFunctions_Quantile.test_complex s 777sCCCi <<<777sCCCi <<<777sCCCi <<<<> + +T>>>>" "r& c t j t j t j fD ]7} t j ddg| }t |d \ }}|J ||u sJ 8t j t j fD ]} t j ddg| }t |d \ }}|dk sJ ||usJ t || t j ddt j
g| }t |d \ }}t |t j g d ||usJ t |t j g d t j |d sJ dS )
z] Test that _replace_nan returns the original array if there are no
NaNs, not a copy.
r r} r Nr F)FFT)r r} r r ) r; r int32int64r r r float64r r r r )r r r maskarr_nan
result_nanmask_nans r$ test__replace_nanrO s (BHbh/ h1vU+++#C++ |||}}}}}*bj)
%
%h1vU+++#C++
""$$$$$S VS!!!(Aq"&>777 +GQ 7 7
HXrx(<(<(< = =>>>((((Z)))!4!4555x $$$$$$
%
%r& )2r ri r* numpyr; numpy.core.numericr numpy.lib.nanfunctionsr r
numpy.testingr r r r r
r r r rq r r r r rn r
_TEST_ARRAYSr- _vsetflagsrj rk r
keysr5 rM rc rp r _TIME_UNITSr r r r r/ r rI complex64rF rO r r& r$ rZ sW
3 3 3 3 3 3 : : : : : : : : 62666266B7BFFFFC626266267C26266626BD E E
222 3 3 ;;; < < 67# $ $ *** + + - RXAAAEEE<<<>>>@ A A
bhBBBFFF===???A B B ,
,
,
,
,
,
,
,
^m% m% m% m% m% m% m% m%`Z$ Z$ Z$ Z$ Z$ Z$ Z$ Z$~ "(1++
"((((
)
)
BKKeK L j!99C? 3 3 5 5<;L;L;N;NOOB* B* B* B* B* B* B* PO
B*Jf1 f1 f1 f1 f1 f1 f1 f1R8$ 8$ 8$ 8$ 8$ ; 8$ 8$ 8$vF. F. F. F. F."> F. F. F.R^7 ^7 ^7 ^7 ^7"> ^7 ^7 ^7B
d2<
+,, 66+666 6 FB FB FB FB FB FB FB FBRlT lT lT lT lT lT lT lT^9( 9( 9( 9( 9( 9( 9( 9(v
RXrvsBFBF+,,
RX((())+ RXlll"(+++ RX((()) RX} !(* 6 6 6
RXt}t} + " "! "% % % % %r&