U
|
P ±dܺ ã @ s` d dl Z d dlZd dlZd dlZd dlmZ d dlmZm Z d dl
|
m Z m Z m Z mZmZmZ e dejddejdgd d
|
ejd d d gejejejdejdgdejejddejgg¡Ze ddddg¡e d d
|
d d d g¡e ddg¡e dddg¡gZe ddddddgd d
|
dd d d gddddddgddddddgg¡Ze ddddddgd d
|
dd d d gddddddgddddddgg¡ZG dd dZG dd dZG dd dZe d¡e d d d d d g¡d!Ze ¡ D ]Z e j d"d# qÈej d$ej d% ej d& d' ¡ej j d(e ¡ e !¡ d)G d*d+ d+Z"G d,d- d-Z#G d.d/ d/e#Z$G d0d1 d1e#Z%G d2d3 d3e#Z&d4Z'e(ej d& Z)e)d5d6 e'D 7 Z)G d7d8 d8Z*G d9d: d:Z+G d;d< d<Z,ej d=e ejd>ejej-g¡e d"d?d"d?g¡fejd@ddAdBgej.dCd?fe d"d?d"d?g¡d?fejejd>gejej-ggej/dCe d"d?gd"d?gg¡fg¡dDdE Z0dFdG Z1dS )Hé N)Únormalize_axis_tuple)Ú _nan_maskÚ _replace_nan)Úassert_Ú assert_equalÚassert_almost_equalÚ assert_raisesÚassert_array_equalÚsuppress_warningsg³êsµûã?g&S£:Ñ?gॽÁ?gÁÊ¡E¶ó½?gÞ á?g»¸ðî¿gáz®GáÊ?gL¦
|
F%uÞ?g6<½R!Ò?g[B>èÙ¬º?g:M ã¿gÏ÷Sã¥Ä?gzÇ):ËÇ?gÔ h"Ô?ç ð?ç c @ s e Zd Zejejejejejej ej
|
ej ej ej ejejejejejejejejejejejejejejej ej ej ej iZ dd e D Z!e"d ddZ#e$j%j&de '¡ e!ddd Z(d
|
d Z)d S )ÚTestSignatureMatchc C s g | ]
|
}|j qS © ©Ú__name__)Ú.0Úkr r úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/lib/tests/test_nanfunctions.pyÚ
|
<listcomp>9 s z TestSignatureMatch.<listcomp>ú...c C sT g }t | ¡}|j ¡ D ]0}|jt jjkr6| |¡ q| |j|d¡ qt |¡S )z?Construct a signature and replace all default parameter-values.)Údefault)
|
ÚinspectÚ signatureÚ
|
parametersÚvaluesr Ú ParameterÚemptyÚappendÚreplaceÚ Signature)Úfuncr Zprm_listr Zprmr r r Ú get_signature; s
|
z TestSignatureMatch.get_signaturez nan_func,func©Úidsc C s& | |¡}| |¡}tj ||¡ d S ©N)r! ÚnpÚtestingr )ÚselfZnan_funcr r Z nan_signaturer r r Útest_signature_matchG s
|
|
z'TestSignatureMatch.test_signature_matchc C s" t j t| jtt jjj¡ dS )z4Validate that all nan functions are actually tested.N)r% r& r ÚsetÚIDSÚlibZ nanfunctionsÚ__all__©r' r r r Útest_exhaustivenessP s ÿz&TestSignatureMatch.test_exhaustivenessN)r )*r Ú
|
__module__Ú __qualname__r% ÚnanminZaminÚnanmaxZamaxÚ nanargminÚargminÚ nanargmaxÚargmaxÚnansumÚsumÚnanprodÚprodÚ nancumsumÚcumsumÚ
|
nancumprodÚcumprodÚnanmeanÚmeanÚ nanmedianÚmedianÚ nanpercentileÚ
|
percentileÚ nanquantileÚquantileÚnanvarÚvarÚnanstdÚstdZNANFUNCSr* Ú staticmethodr! ÚpytestÚmarkÚ parametrizeÚitemsr( r. r r r r r ( sF ò
|
r c @ s e Zd ZejejgZejejgZ dd Z
|
dd Z dd Z dd Z d d
|
Zej d d d dg¡ej dejd ¡ejjde ej¡e dej¡gddgddd Zdd Zdd Zd d Zd d Zej dejd ¡d d! Zej dejd ¡d"d# Z d S )$ÚTestNanFunctions_MinMaxc C s* t ¡ }| jD ]}|| t|t qd S r$ ©Ú_ndatÚcopyÚnanfuncsr ©r' ÚndatÚfr r r Ú test_mutation\ s
|
z%TestNanFunctions_MinMax.test_mutationc C s\ t d¡}t| j| jD ]>\}}dD ]0}|||dd}|||dd}t|j|jk q$qd S ©Né ©Nr é T©ÚaxisÚkeepdims©r% ÚeyeÚziprT Ústdfuncsr Úndim©r' ÚmatÚnfÚrfr^ ÚtgtÚresr r r Ú test_keepdimsc s
|
z%TestNanFunctions_MinMax.test_keepdimsc C s^ t d¡}t| j| jD ]@\}}t d¡}||dd}||d|d}t|| t|| qd S ©NrZ r\ ©r^ ©r^ Úout©r% ra rb rT rc Úzerosr ©r' rf rg rh Úresoutri rj r r r Útest_outk s
|
|
|
z TestNanFunctions_MinMax.test_outc C s d}t | j| jD ]x\}}|D ]j}tjd|d}||ddjj}||ddjj}t||k ||d djj}||d djj}t||k q qd S )NÚefdgFDGrZ ©Údtyper\ rm ©rb rT rc r% ra rw Útyper ©r' Úcodesrg rh Úcrf ri rj r r r Útest_dtype_from_inputt s z-TestNanFunctions_MinMax.test_dtype_from_inputc sD t | j| jD ]0\} fddtD }|tdd}t|| qd S )Nc s g | ] } |qS r r ©r Úd©rh r r r s z>TestNanFunctions_MinMax.test_result_values.<locals>.<listcomp>r\ rm ©rb rT rc Ú_rdatrR r ©r' rg ri rj r r r Útest_result_values s z*TestNanFunctions_MinMax.test_result_valuesr^ Nr r\ rw ÚAllFloatÚarray©rZ rZ Ú0dÚ2dr" c
|
C s |d k r |j dkr t d¡ | |¡}d}| jD ]L}tjt|d |||d}W 5 Q R X t |¡ ¡ slt
|
|j |j ks0t
|
q0d S ©Nr ú*`axis != None` not supported for 0d arraysúAll-NaN slice encountered©Úmatchrm ) rd rL ÚskipÚastyperT ÚwarnsÚRuntimeWarningr% ÚisnanÚallÚAssertionErrorrw ©r' r^ rw r r r ro r r r Ú test_allnans s
|
|
|
z$TestNanFunctions_MinMax.test_allnansc C sj t j t¡}|j ¡ }t jfD ]F}||dd}|tdd}t|| t|j| tt |¡
|
¡ q d S )Nr\ rm ) r% ÚmaZ fix_invalidrR Z_maskrS r1 r r ÚisinfÚany)r' rf ZmskrW rj ri r r r Ú test_masked s
|
|
z#TestNanFunctions_MinMax.test_maskedc C s | j D ]}t|ddk qd S ©Nr ©rT r ©r' rW r r r Ú test_scalar£ s
|
z#TestNanFunctions_MinMax.test_scalarc
|
C s G dd dt j}t d¡ |¡}| jD ]j}||dd}tt|| t|jdk ||dd}tt|| t|jdk ||}t|jdk q(t j|d< | jD ]T}t j
|
d d
|
P}t d ¡ ||dd}tt|| tt t |¡¡ tt|dk W 5 Q R X t j
|
d d
|
}t d ¡ ||dd}tt|| tt |d ¡opt |d ¡ opt |d ¡ tt|dkd tt|d jt W 5 Q R X t j
|
d d
|
D}t d ¡ ||}t|jdk t|t jk tt|dk W 5 Q R X q¤d S )Nc @ s e Zd ZdS )z8TestNanFunctions_MinMax.test_subclass.<locals>.MyNDArrayN©r r/ r0 r r r r Ú MyNDArray¨ s r¡ rZ r rm ©rZ r\ r T©ÚrecordÚalwaysé úno warning raised)r% Úndarrayra ÚviewrT r Ú
|
isinstanceÚshapeÚnanÚwarningsÚcatch_warningsÚ simplefilterr r ÚlenÚ
|
issubclassÚcategoryr )r' r¡ ÚminerW rj Úwr r r Ú test_subclass§ sD
|
|
|
|
$ÿ
|
z%TestNanFunctions_MinMax.test_subclassc C sº t jddgt jdgt jt jggtd}tt |¡d tt j|ddddg tjddV}t d ¡ tt t j|d
|
dddt jg t
|
t |d
|
kd t
|
t |d j t W 5 Q R X d S ) Nr g @ç @rv r rm Tr£ r¥ r\ r§ )r% r r¬ Úobjectr r1 r r® r¯ Úlistr r° r± r² r )r' Úarrr´ r r r Útest_object_arrayÒ s &
|
z)TestNanFunctions_MinMax.test_object_arrayc C s¤ G dd dt j}t d¡ |¡}t j|d d
< | jD ]h}|t jkrHdnd}|||d}|j|ksft||ksrt|| |¡|d}|j|kst||ks6tq6d S )Nc @ s e Zd ZdS )z7TestNanFunctions_MinMax.test_initial.<locals>.MyNDArrayNr r r r r r¡ à s r¡ é é éd r ©Úinitial)
|
r% r¨ Úaranger r¬ rT r2 rw r r© )r' rw r¡ ÚarrW r¿ Úret1Úret2r r r Ú test_initialÞ s
|
z$TestNanFunctions_MinMax.test_initialc C sÔ G dd dt j}t d¡ dd¡ |¡}t j|dd d
f< t j|t jd}d|d d
df< | jD ]l}|t j krtdnd }|||d
|
d }|j
|
|kst ||ks t || |¡|d
|
d }|j
|
|ksÂt ||ksbt qbd S ) Nc @ s e Zd ZdS )z5TestNanFunctions_MinMax.test_where.<locals>.MyNDArrayNr r r r r r¡ ó s r¡ r» rZ r rv Fé é r¼ ©Úwherer¿ ) r% r¨ rÀ Úreshaper r¬ Ú ones_likeÚbool_rT r1 rw r r© ) r' rw r¡ rÁ rÈ rW Ú referencer rà r r r Ú
|
test_whereñ s
|
z"TestNanFunctions_MinMax.test_where) r r/ r0 r% r1 r2 rT ÚminÚmaxrc rX rk rt r} r rL rM rN Ú typecodesr r¬ Úfullr r r rµ rº rÄ rÍ r r r r rP W s0
|
þý
|
+
|
rP c @ sÔ e Zd ZejejgZdd Zdd Ze j
|
ddddg¡e j
|
d ej d
|
¡e j
|
j d e ej¡e d ej¡gd dgddd Zdd Zdd Zdd Ze j
|
d ej d
|
¡dd Ze j
|
d ej d
|
¡dd ZdS ) Ú TestNanFunctions_ArgminArgmaxc C s* t ¡ }| jD ]}|| t|t qd S r$ rQ rU r r r rX s
|
z+TestNanFunctions_ArgminArgmax.test_mutationc C s t | jtjtjgD ]\}}tD ]t}t d}| td¡ ||}|| }t t
|
|¡ t ||| ¡ t t ||d |
¡ ¡ W 5 Q R X q qd S )Nz invalid value encountered in) rb rT r% ZgreaterZlessrR r
|
Úfilterr r r r Úequal)r' rW ZfcmpÚrowÚsupÚindÚvalr r r r s z0TestNanFunctions_ArgminArgmax.test_result_valuesr^ Nr r\ rw r
r r r r r" c
|
C s\ |d k r |j dkr t d¡ | |¡}| jD ]*}tjtdd |||d W 5 Q R X q,d S r )rd rL r r rT ZraisesÚ
|
ValueError)r' r^ rw r r r r r r s
|
|
|
z*TestNanFunctions_ArgminArgmax.test_allnansc C sZ t d¡}| jD ]D}dD ]}tt|||d qdD ] }|||d}t|t d¡ q2qd S )N©r rZ ©r Nrm ©r\ r )r% rq rT r rÙ r )r' rf rW r^ rj r r r Ú
|
test_empty. s
|
|
z(TestNanFunctions_ArgminArgmax.test_emptyc C s | j D ]}t|ddk qd S r r r r r r r 7 s
|
z)TestNanFunctions_ArgminArgmax.test_scalarc C s G dd dt j}t d¡ |¡}| jD ]j}||dd}tt|| t|jdk ||dd}tt|| t|jdk ||}t|jdk q(d S ) Nc @ s e Zd ZdS )z>TestNanFunctions_ArgminArgmax.test_subclass.<locals>.MyNDArrayNr r r r r r¡ < s r¡ rZ r rm r¢ r\ r )r% r¨ ra r© rT r rª r« )r' r¡ r³ rW rj r r r rµ ; s
|
z+TestNanFunctions_ArgminArgmax.test_subclassc C sh t d¡ |¡}t j|d d
< | jD ]>}|t jkr6dnd}||dd}|j|jksVt||ks$tq$d S )Nr» r¼ rÆ T)r_ )r% rÀ r r¬ rT r3 rd r ©r' rw rÁ rW rÌ Úretr r r rk L s
|
z+TestNanFunctions_ArgminArgmax.test_keepdimsc C st t d¡ |¡}t j|d d
< | jD ]J}t jdt jd}|t jkrFdnd}|||d}||ksbt||ks$tq$d S )Nr» r¼ r rv rÆ ©ro ) r% rÀ r r¬ rT rq Úintpr3 r )r' rw rÁ rW ro rÌ rß r r r rt W s
|
z&TestNanFunctions_ArgminArgmax.test_out)r r/ r0 r% r3 r5 rT rX r rL rM rN rÐ r r¬ rÑ r rÝ r rµ rk rt r r r r rÒ s&
|
þý
|
|
rÒ r¼ é é' é] éW é. )r Z1dF)Úwriterw Z
|
AllIntegerr
ÚOrf r" c @ s e Zd Zejejejejejej ej
|
ej ej ej ejejejejejejejejejejejejejeji Z dd e D Z e j j de !¡ e dej"dddd Z#e j j dej$ej%fej&ej'fgd d
|
gdd d Z(e j j dejejfejejfgd dgddd Z)dS )Ú TestNanFunctions_NumberTypesc C s g | ]
|
}|j qS r r ©r Úir r r r s z'TestNanFunctions_NumberTypes.<listcomp>z nanfunc,funcr" Úignore)Zoverc C sV | |¡}||}||}t|| |dkrBt|t|ksRtn|j|jksRtd S )Nrè ©r r ry r rw ©r' rf rw Znanfuncr ri ro r r r Ú test_nanfunc s
|
|
z)TestNanFunctions_NumberTypes.test_nanfuncrE rC c C s^ | |¡}||dd}||dd}t|| |dkrJt|t|ksZtn|j|jksZtd S )Nr\ ©Úqrè rí rî r r r Útest_nanfunc_q s
|
|
z+TestNanFunctions_NumberTypes.test_nanfunc_qrG rI c C s^ | |¡}||dd}||dd}t|| |dkrJt|t|ksZtn|j|jksZtd S )Nç à?©Úddofrè rí rî r r r Útest_nanfunc_ddof s
|
|
z.TestNanFunctions_NumberTypes.test_nanfunc_ddofN)*r r/ r0 r% r1 rÎ r2 rÏ r3 r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? r@ rA rB rG rH rI rJ rT Z nanfunc_idsrL rM rN rO Zerrstaterï rE rF rC rD rò rö r r r r ré m sR ô
|
ý
|
ýré c @ sT e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dS )Ú SharedNanFunctionsTestsMixinc C s* t ¡ }| jD ]}|| t|t qd S r$ rQ rU r r r rX ² s
|
z*SharedNanFunctionsTestsMixin.test_mutationc C s\ t d¡}t| j| jD ]>\}}dD ]0}|||dd}|||dd}t|j|jk q$qd S rY r` re r r r rk ¹ s
|
z*SharedNanFunctionsTestsMixin.test_keepdimsc C s^ t d¡}t| j| jD ]@\}}t d¡}||dd}||d|d}t|| t|| qd S rl rp rr r r r rt Á s
|
|
|
z%SharedNanFunctionsTestsMixin.test_outc C sâ t d¡}d}t| j| jD ]À\}}|D ]²}t ¢}|t jt jhkrX|dkrX| t j ¡ ||t
|
|¡ddj
|
j }||t
|
|¡ddj
|
j }t ||k ||t
|
|¡d dj
|
j }||t
|
|¡d dj
|
j }t ||k W 5 Q R X q(q d S ©NrZ ru ZFDGr\ )rw r^ © r% ra rb rT rc r
|
rI rG rÓ ÚComplexWarningrw ry r © r' rf r{ rg rh r| rÖ ri rj r r r Útest_dtype_from_dtypeÊ s
|
z2SharedNanFunctionsTestsMixin.test_dtype_from_dtypec C sÊ t d¡}d}t| j| jD ]¨\}}|D ]}t }|t jt jhkrX|dkrX| t j ¡ |||ddj
|
j }|||ddj
|
j }t ||k |||d dj
|
j }|||d dj
|
j }t ||k W 5 Q R X q(q d S rø rù rû r r r Útest_dtype_from_charÛ s
|
z1SharedNanFunctionsTestsMixin.test_dtype_from_charc C s d}t | j| jD ]\}}|D ]t}tjd|d}||ddjj}||ddjj}t||kd||f ||d djj}||d djj}t||k q qd S )Nru rZ rv r\ rm zres %s, tgt %srx rz r r r r} ì s z2SharedNanFunctionsTestsMixin.test_dtype_from_inputc sD t | j| jD ]0\} fddtD }|tdd}t|| qd S )Nc s g | ] } |qS r r r~ r r r r û s zCSharedNanFunctionsTestsMixin.test_result_values.<locals>.<listcomp>r\ rm r r r r r r ù s z/SharedNanFunctionsTestsMixin.test_result_valuesc C s | j D ]}t|ddk qd S r r r r r r r ÿ s
|
z(SharedNanFunctionsTestsMixin.test_scalarc C sÐ G dd dt j}t d¡}| |¡}| jD ]}||ddj}||dd}tt|| t|j|k ||ddj}||dd}tt|| t|j|k ||j}||}tt|| t|j|k q,d S )Nc @ s e Zd ZdS )z=SharedNanFunctionsTestsMixin.test_subclass.<locals>.MyNDArrayNr r r r r r¡ s r¡ rZ r rm r\ )r% r¨ ra r© rT r« r rª )r' r¡ r r³ rW Zexpected_shaperj r r r rµ s
|
|
|
|
z*SharedNanFunctionsTestsMixin.test_subclassN) r r/ r0 rX rk rt rü rý r} r r rµ r r r r r÷ ± s r÷ c @ sÀ e Zd ZejejgZejejgZ e
|
j ddddg¡e
|
j dej d ¡e
|
j j de ej¡e dej¡gd d
|
gd d d Zdd Ze
|
j dej d ¡dd Ze
|
j dej d ¡dd ZdS )ÚTestNanFunctions_SumProdr^ Nr r\ rw r
r r r r r" c C sr |d k r |j dkr t d¡ | |¡}t| jddgD ]6\}}|||d}t ||k¡s\t|j |j ks6tq6d S )Nr r r\ rm ©
|
rd rL r r rb rT r% r r rw ©r' r^ rw r r Úidentityro r r r r s
|
|
z%TestNanFunctions_SumProd.test_allnansc C s t tjtjgddgD ]f\}}t d¡}|gd }||dd}t|| g }||dd}t|| |}||d d}t|| qd S )Nr r\ rÚ rZ rm )rb r% r7 r9 rq r ©r' rW Z tgt_valuerf ri rj r r r rÝ / s
|
|
|
|
z#TestNanFunctions_SumProd.test_emptyc C sf t d¡ |¡}t j|d d
< | jD ]<}|t jkr6dnd}||dd}|j|ksTt||ks$tq$d S )Nr» r¼ é i r¦ r¾ )r% rÀ r r¬ rT r7 rw r rÞ r r r rÄ < s
|
z%TestNanFunctions_SumProd.test_initialc C s t d¡ dd¡ |¡}t j|dd d
f< t j|t jd}d|d d
df< | jD ]>}|t jkrbdnd}|||dd }|j |kst
|
||ksPt
|
qPd S )
|
Nr» rZ r rv Fé iÀ r¦ rÇ ) r% rÀ rÉ r r¬ rÊ rË rT r7 rw r )r' rw rÁ rÈ rW rÌ rß r r r rÍ G s
|
z#TestNanFunctions_SumProd.test_where)r r/ r0 r% r7 r9 rT r8 r: rc rL rM rN rÐ r r¬ rÑ r rÝ rÄ rÍ r r r r rþ s
|
þý
|
|
|
rþ c @ s e Zd ZejejgZejejgZ e
|
j ddddg¡e
|
j dej d ¡e
|
j j de ej¡e dej¡gd d
|
gd d d Zdd Zdd Zdd Zdd ZdS )ÚTestNanFunctions_CumSumProdr^ Nr r\ rw r
r r r r r" c C sn |d k r |j dkr t d¡ | |¡}t| jddgD ]2\}}||}t ||k¡sXt|j |j ks6tq6d S )Nr r r\ rÿ r r r r r Z s
|
|
z(TestNanFunctions_CumSumProd.test_allnansc C s t | jddgD ]p\}}t d¡}|t d¡ }||dd}t|| |}||dd}t|| t d¡}||d d}t|| qd S )Nr r\ rÚ rm )rb rT r% rq Úonesr r r r r rÝ j s
|
|
|
|
z&TestNanFunctions_CumSumProd.test_emptyc C sÔ t | j| jD ]H\}}t d¡}dD ]0}|||d d}|||d d}t|j|jk q$q| jD ]p}t d¡}tj d¡}tj
|
||j |j dk < ||d d}t |j d t d ¡D ] }|||d}t |j d q°q^d S )
|
NrZ r[ rn ©rZ r¼ é é r ró rm )i rÅ )rb rT rc r% ra r rd r ÚrandomZ RandomStater¬ Úrandr« r rÀ ) r' rW Úgrf r^ ri rj r Úrsr r r rk w s
|
|
|
z)TestNanFunctions_CumSumProd.test_keepdimsc C sZ dD ]P}t jt|d}t jt|d}t|| t jt|d}t jt|d}t|| qd S )N)éþÿÿÿéÿÿÿÿr r\ Nrm ) r% r> Ú
|
_ndat_onesr= rR r r< Ú _ndat_zerosr; )r' r^ ri rj r r r r s
|
z.TestNanFunctions_CumSumProd.test_result_valuesc C sh t d¡}t| j| jD ]J\}}t d¡}dD ]2}|||d}||||d}t|| t|| q.qd S )NrZ )r r r r\ rm rn )r% ra rb rT rc r )r' rf rg rh rs r^ ri rj r r r rt s
|
|
|
z$TestNanFunctions_CumSumProd.test_out)r r/ r0 r% r; r= rT r< r> rc rL rM rN rÐ r r¬ rÑ r rÝ rk r rt r r r r r U s
|
þý
|
r c @ sÌ e Zd ZejejejgZejej ej
|
gZ dd Z dd Z dd Zdd Zej d d
|
d d g¡ej d ejd ¡ejjde ej¡e dej¡gddgddd Zdd Zej d ejd ¡dd Zd
|
S )ÚTestNanFunctions_MeanVarStdc C s8 | j D ],}tjtjtjfD ]}tt|td|d qqd S )Nr\ )r^ rw )rT r% rË Úint_Úobject_r Ú TypeErrorrR )r' rW rw r r r Útest_dtype_error£ s
|
z,TestNanFunctions_MeanVarStd.test_dtype_errorc C sL | j D ]@}tjtjtjfD ]*}tjtjd |d}tt |td|d qqd S )Nr rv r\ rn )
|
rT r% rË r r r rR r« r r )r' rW rw ro r r r Útest_out_dtype_error¨ s
|
z0TestNanFunctions_MeanVarStd.test_out_dtype_errorc sf t jt jg}t jt jg}t||D ]>\}dD ]0 fddtD }|td d}t|| q.q"d S )N©r r\ c s g | ]}| d qS )rô r r~ ©rõ rh r r r ³ s z9TestNanFunctions_MeanVarStd.test_ddof.<locals>.<listcomp>r\ ©r^ rõ ) r% rG rI rH rJ rb r rR r )r' rT rc rg ri rj r r r Ú test_ddof® s z%TestNanFunctions_MeanVarStd.test_ddofc sÒ t jt jg}t jt jg}dd tD }t||D ]\}}tdD ] t z}| t
|
¡ | t j ¡ fdd|D }|t d d}tt |¡| t|r®tt|jdk ntt|jdk W 5 Q R X q@q0d S )Nc S s g | ] }t |qS r )r° r~ r r r r º s zATestNanFunctions_MeanVarStd.test_ddof_too_big.<locals>.<listcomp>r¼ c s g | ] } |kqS r r r~ rô r r r À s r\ r r )r% rG rI rH rJ r rb Úranger
|
r¤ r rÓ rú rR r r r r r° Úlog) r' rT rc Zdsizerg rh rÖ ri rj r rô r Útest_ddof_too_big· s
|
z-TestNanFunctions_MeanVarStd.test_ddof_too_bigr^ Nr r\ rw r
r r r r r" c
|
C s¤ |d k r |j dkr t d¡ | |¡}d}| jD ]n}tjt|d |||d}W 5 Q R X t |¡ ¡ slt
|
|tj kr|j |j kst
|
q0|j t |¡j ks0t
|
q0d S )Nr r z:(Degrees of freedom <= 0 for slice.)|(Mean of empty slice)r rm )rd rL r r rT r r r% r r r r? rw Úabsr r r r r È s
|
|
|
|
z(TestNanFunctions_MeanVarStd.test_allnansc C sØ t d¡}| jD ]Â}dD ]d}tjddN}t d¡ tt |||d¡ ¡ tt |dk tt
|
|d j t W 5 Q R X qd D ]N}tjdd8}t d¡ t |||dt g ¡ tt |dk W 5 Q R X qqd S ©
|
NrÚ rÛ Tr£ r¥ rm r\ r rÜ )r% rq rT r r® r¯ r r r r° r± r² r r )r' rf rW r^ r´ r r r rÝ à s
|
|
|
|
z&TestNanFunctions_MeanVarStd.test_emptyc C s¸ t d¡ dd¡ |¡}t j|dd d
f< t j|t jd}d|d d
df< t| j| j D ]Z\}}||| dd
}|t j
|
kr|n|j j }|||d}|j |ks¤t t j ||¡ qXd S )Nr» rZ r rv Fr¦ )rÈ )r% rÀ rÉ r r¬ rÊ rË rb rT rc r? Úrealrw r r& Zassert_allclose) r' rw rÁ rÈ rW Zf_stdrÌ Zdtype_referencerß r r r rÍ ï s z&TestNanFunctions_MeanVarStd.test_where)r r/ r0 r% r? rG rI rT r@ rH rJ rc r r r r rL rM rN rÐ r r¬ rÑ r rÝ rÍ r r r r r s$
|
þýr ) ÚYÚMÚWÚDÚhÚmÚsÚmsÚusÚnsZpsÚfsÚasc C s g | ]}d | dqS )zm8[ú]r )r Úunitr r r r s r c @ s¬ e Zd Zdd Zdd Zejjddddd d
|
gd ej d ¡d d Z dd Z
|
dd Z dd Z ej ddddg¡ej de ¡dd Zdd Zdd Zd d Zd d ZdS )!ÚTestNanFunctions_Medianc C s t ¡ }t |¡ t|t d S r$ )rR rS r% rA r ©r' rV r r r rX
|
s
|
z%TestNanFunctions_Median.test_mutationc C sX t d¡}dD ]8}t j||d dd}t j||d dd}t|j|jk qt d¡}t j d¡t |j ¡d d
d f }|
|
t j ¡}t j |t |< t ¸}| t¡ t j|d dd}t|j d t j|d
|
dd}t|j d t j|d dd}t|j d t j|ddd}t|j d t j|ddd}t|j d t j|ddd}t|j d W 5 Q R X d S )NrZ r[ F©r^ ro Úoverwrite_inputr ©rÅ éÈ Tr] ©r\ r\ r\ r\ r ©r\ r\ r r rÚ ©r\ r¼ r r\ rÜ ©rZ r\ r r ©r r\ r¦ rZ ©r r\ rZ ©r\ r\ r r\ )r% ra rB rA r rd r r
|
r r« r rá r¬ Útupler
|
rÓ r r ©r' rf r^ ri rj r r´ rÖ r r r rk s.
|
|
$
|
z%TestNanFunctions_Median.test_keepdimsr^ Nr\ rÜ r ©éýÿÿÿr ©ÚargnamesZ argvaluesú#ignore:All-NaN slice:RuntimeWarningc s¾ t d¡t j d¡t j¡d d
d f }| t j¡}t jt|< |d kr\dj }n*t
|
|j t fddt j D }t |¡}t j |d|d}||ks®tt|j| d S )Nr r4 rÜ c 3 s$ | ] }| krd nj | V qdS ©r\ N©r« rê ©Z axis_normr r r Ú <genexpr>@ s z<TestNanFunctions_Median.test_keepdims_out.<locals>.<genexpr>T©r^ r_ ro )r% r r
|
r r« r rá r¬ r= rd r r r rA r r )r' r^ r´ Ú shape_outro Úresultr rF r Útest_keepdims_out+ s
|
$ ÿ
|
z)TestNanFunctions_Median.test_keepdims_outc C sÆ t j dd¡}t j|ddgt jdd}t d¡}t j|dd}t j|d|d}t|| t|| t d¡}t j|d d}t j|d |d}t|| t|| t j|d|d}t|| t|| d S ) NrZ r r¦ r\ rm rn r r ) r% r
|
r Úinsertr¬ rq rB rA r ©r' rf Únan_matrs ri rj r r r rt G s
|
|
|
|
|
|
|
z TestNanFunctions_Median.test_outc C s dD ]}t j d|¡}t jjd|j|jd d}t j| ¡ |< d|d d
df< g }|D ],}t t |¡ |¡}| t j
|
|dd¡ qVt t j |d d
|
| qd S ) N)r¼ é é3 r5 iè rÅ r r¼ )Úsizer T)r3 r rm ) r% r
|
ZrandnÚrandintrQ r¬ ZravelÚcompressr r rB r rA )r' r( r r´ ri ÚxZnonanr r r Útest_small_largeY s z(TestNanFunctions_Median.test_small_largec C s* dd t D }tjtdd}t|| d S )Nc S s g | ]}t |¡qS r )r% rB r~ r r r r j s z>TestNanFunctions_Median.test_result_values.<locals>.<listcomp>r\ rm )r r% rA rR r ©r' ri rj r r r r i s z*TestNanFunctions_Median.test_result_valuesr rw c C sþ t dt j¡ |¡}t Ú}| t¡ t j||d}|j|jksDt t
|
|¡ ¡ sVt |d krrt t |jdk nt t |jdk t t j¡ |¡d }t |¡}|j|jks´t t
|
|¡sÂt |d krÞt t |jdk nt t |jdk W 5 Q R X d S )Nr rm r\ rZ r r¦ rÅ )r% rÑ r¬ r r
|
r¤ r rA rw r r r r r° r r )r' rw r^ rf rÖ ÚoutputZscalarZ output_scalarr r r r n s
|
|
z$TestNanFunctions_Median.test_allnansc
|
C sÐ t d¡}dD ]f}tjddP}t d¡ tt t j||d¡ ¡ tt |dk tt
|
|d j t W 5 Q R X qd D ]P}tjdd:}t d¡ t t j||dt g ¡ tt |dk W 5 Q R X qzd S r )r% rq r r® r¯ r r rA r r° r± r² r r ©r' rf r^ r´ r r r rÝ s
|
|
|
z"TestNanFunctions_Median.test_emptyc C s t t d¡dk d S r )r r% rA r- r r r r s z#TestNanFunctions_Median.test_scalarc C sp t d¡}tt jt j|dd tt jt j|dd tt jt j|dd tt jt j|dd ttt j|dd d S )Nr éûÿÿÿrm ©r rY rÅ ©r rÅ ©r\ r\ )r% r r Ú AxisErrorrA rÙ ©r' r r r r Útest_extended_axis_invalid s
|
z2TestNanFunctions_Median.test_extended_axis_invalidc C sä t Ò}| t¡ tjtj fD ]°}t |tjgtjtjgg¡}ttj|dd|tjg ttj|dd|tjg tt |¡| t tjtj|gtjtj|gg¡}tt |¡| ttj|ddtjtj|g ttj|dd| t ||g||gg¡}ttj|dd| tj|d| dgdtjtjdgdtjtj|ggtj d }|dkrttj|ddd
|
d | d g tt |¡d n.ttj|dddd | dg tt |¡d ttj|dddd|g t
|
ddD ]ø}t
|
ddD ]æ}t tjg| |g| gd ¡}tt |¡| ttj|dd| ttj|ddtjg| |g| t tjg| | g| gd ¡}tt |¡| ttj|dd| ttj|ddtjg| | g| qèqÚq"W 5 Q R X d S )Nr rm r\ r i÷ÿÿÿiöÿÿÿr¼ rÅ rv r¶ g @ç @g @g $Àg "Àg Àr g ð¿é
|
r¦ ) r
|
rÓ r r% Úinfr r¬ r rA Úfloat32r )r' rÖ rb Úarë Újr r r Útest_float_special¢ sR
|
|
ÿ þý
|
"ÿ$ÿz*TestNanFunctions_Median.test_float_special)r r/ r0 rX rk rL rM rN ÚfilterwarningsrK rt rU r Ú _TYPE_CODESr rÝ r r_ rf r r r r r0 s. ûþ
|
|
r0 c @ sì e Zd Zdd Zdd Zej ddddgg¡ejjdd dd
|
d d gd ej d¡dd Z dd Z
|
dd Z ej dd ddg¡ej de j d ¡ejjde e j¡e de j¡gddgd d d Zd d Zd!d" Zd#d$ Zd%d& Zd S )'ÚTestNanFunctions_Percentilec C s" t ¡ }t |d¡ t|t d S )Né )rR rS r% rC r r1 r r r rX Ó s z)TestNanFunctions_Percentile.test_mutationc C sh t d¡}dD ]<}t j|d|d dd}t j|d|d dd}t|j|jk qt d¡}t j d¡t |j ¡d d
d f }|
|
t j ¡}t j |t |< t Ä}| t¡ t j|dd d d
|
}t|j d t j|dd d d
|
}t|j d t j|ddd d
|
}t|j d t j|ddd d
|
}t|j d t j|ddd d
|
}t|j d t j|ddd d
|
}t|j d W 5 Q R X d S )NrZ r[ éF Fr2 r r4 éZ Tr] r6 r r7 rÚ r8 rÜ r9 r: r; r< )r% ra rD rC r rd r r
|
r r« r rá r¬ r= r
|
rÓ r r r> r r r rk Ù s6
|
ÿ ÿ
|
$
|
z)TestNanFunctions_Percentile.test_keepdimsrñ r r\ r^ NrÜ r r? rA rC c sÎ t d¡t j d¡t j¡d d
d f }| t j¡}t jt|< |d kr\dj }n*t
|
|j t fddt j D }t |¡| }t |¡}t j ||d|d}||ks¾tt|j| d S )Nr r4 rÜ c 3 s$ | ] }| krd nj | V qdS rD rE rê rF r r rG s z@TestNanFunctions_Percentile.test_keepdims_out.<locals>.<genexpr>TrH )r% r r
|
r r« r rá r¬ r= rd r r r rC r r )r' rñ r^ r´ rI ro rJ r rF r rK ö s
|
$ ÿ
|
z-TestNanFunctions_Percentile.test_keepdims_outc C sÐ t j dd¡}t j|ddgt jdd}t d¡}t j|ddd}t j|dd|d}t|| t|| t d¡}t j|dd d}t j|dd |d}t|| t|| t j|dd |d}t|| t|| d S )
|
NrZ r r¦ r\ rm é* rn r r ) r% r
|
r rL r¬ rq rD rC r rM r r r rt s
|
|
|
|
|
|
|
z$TestNanFunctions_Percentile.test_outc C sZ dd t D }tjtddd}t|| t dd t D ¡}tjtddd}t|| d S )Nc S s g | ]}t |d ¡qS )r ©r% rD r~ r r r r ( s zBTestNanFunctions_Percentile.test_result_values.<locals>.<listcomp>r r\ rm c S s g | ]}t |d ¡qS )©r éb rn r~ r r r r , s ro )r r% rC rR r Z transposerV r r r r ' s
|
z.TestNanFunctions_Percentile.test_result_valuesr rw r
r r r r r" c C sv |d k r |j dkr t d¡ | |¡}tjtdd tj|d|d}W 5 Q R X t |¡ ¡ sbt
|
|j |j ksrt
|
d S )Nr r r r é< rm ) rd rL r r r r r% rC r r r rw ©r' r^ rw r ro r r r r 0 s
|
|
z(TestNanFunctions_Percentile.test_allnansc
|
C sÔ t d¡}dD ]h}tjddR}t d¡ tt t j|d|d¡ ¡ tt |dk tt
|
|d j t W 5 Q R X qd
|
D ]R}tjdd<}t d¡ t t j|d|dt g ¡ tt |d k W 5 Q R X q|d S ) NrÚ rÛ Tr£ r¥ é( rm r\ r rÜ )r% rq r r® r¯ r r rC r r° r± r² r r rX r r r rÝ @ s
|
|
|
z&TestNanFunctions_Percentile.test_emptyc C sH t t dd¡d t d¡}tj|ddd}t |d tt |¡ d S )Nr r½ é é2 r rm g @)r r% rC rÀ r Zisscalar)r' rd Úrr r r r N s
|
|
|
z'TestNanFunctions_Percentile.test_scalarc C sz t d¡}tt jt j|ddd tt jt j|ddd tt jt j|ddd tt jt j|ddd ttt j|ddd d S ) Nr r¼ rY ©rñ r^ rZ rÅ r[ r\ )r% r r r] rC rÙ r^ r r r r_ U s
|
z6TestNanFunctions_Percentile.test_extended_axis_invalidc C s ddg}t d¡}t j| }t d¡}d|d d
ddd
d d
f< |d d
d d
dd
f d9 < d D ]}d
|
D ]}t r}| td ¡ t j||||d }t j||||d } t| j |j t j||||d }t j||||d } t| | W 5 Q R X qpqht d ¡}
|
tt j|
|
|ddj d d S )Nru r½ )rÅ rZ )rZ rÅ r¼ r r¦ rÅ rZ r[ )FTr r] )rZ rÅ r¼ rt )r\ r¦ rm )r¦ rZ rt )
|
r% r r¬ r
|
rÓ r rD rC r r« ) r' Úpercrf rN Z large_matr^ ZkeepdimrÖ rØ Znan_valZmegamatr r r Útest_multiple_percentiles] s0
|
|
|
"
|
ÿ
|
ÿ
|
ÿ
|
z5TestNanFunctions_Percentile.test_multiple_percentiles)r r/ r0 rX rk rL rM rN rg rK rt r r% rÐ r r¬ rÑ r rÝ r r_ ry r r r r ri Ñ s: ûþ
|
|
|
þý
|
ri c @ s e Zd Zdd Zdd Zdd Zej ddd d
|
g¡ej d e j
|
d ¡ejjd e e j ¡e de j ¡gddgddd ZdS )ÚTestNanFunctions_Quantilec C sâ t d¡ ddd¡ t¡}t j|d d< tt j|ddt j|d d tt j|ddd
|
t j|d dd
|
tt j|ddd
|
t j|d dd
|
tt j|dgdd
|
t j|d gdd
|
tt j|d dd gdd
|
t j|d d dgdd
|
d S )Né r¦ rZ rÅ r r\ ró rð ru rw ç Ð?ç è?é éK ) r% rÀ rÉ r Úfloatr¬ r rE rC )r' rÁ r r r Útest_regression{ s ÿÿÿÿz)TestNanFunctions_Quantile.test_regressionc C sH t d¡d }tt |d¡d tt |d¡d tt |d¡d d S )NrÆ ró r r r\ g @g ü?)r% rÀ r rE )r' rT r r r Ú
|
test_basic s z$TestNanFunctions_Quantile.test_basicc C sp t dddddg¡}| ¡ }t jt d¡|dd t|| | ¡ }| ¡ }t jt d¡|dd t|| d S ) Nr r} r| ró r g Y@Zmidpoint)Úmethod)r% r rS rE rÀ r Útolist)r' Zp0Úpr r r Útest_no_p_overwrite s
|
z-TestNanFunctions_Quantile.test_no_p_overwriter^ Nr r\ rw r
r r r r r" c C sv |d k r |j dkr t d¡ | |¡}tjtdd tj|d|d}W 5 Q R X t |¡ ¡ sbt
|
|j |j ksrt
|
d S )Nr r r r r\ rm ) rd rL r r r r r% rE r r r rw rr r r r r s
|
|
z&TestNanFunctions_Quantile.test_allnans)r r/ r0 r r r rL rM rN r% rÐ r r¬ rÑ r r r r r rz x s
|
þýrz z arr, expectedr` Tr\ r r» rv c C sP d t j| jt jdfD ]4}t| |d}t|| t|t jk r|dkstqd S )Nrv rà T) r% r r« rË r r ry r¨ r )r¹ Úexpectedro Úactualr r r Útest__nan_mask« s
|
|
r c C s t jt jt jfD ]<} t jddg| d}t|d\}}|dks@t||kstqt jt jfD ]¸} t jddg| d}t|d\}}|dk ¡ st||k stt
|
|| t jddt j g| d}t|d\}}t
|
|t dddg¡ ||k sêtt
|
|t dddg¡ t |d ¡sZtqZdS ) z] Test that _replace_nan returns the original array if there are no
|
NaNs, not a copy.
|
r r\ rv Nr¦ FTr ) r% rË Úint32Úint64r r r rc Úfloat64r r r¬ r )rw r¹ rJ ÚmaskZarr_nanZ
|
result_nanZmask_nanr r r Útest__replace_nanÆ s"
|
r )2r rL r Únumpyr% Znumpy.core.numericr Znumpy.lib.nanfunctionsr r Z numpy.testingr r r r r r
|
r r¬ rR r r r r rP rÒ Z _TEST_ARRAYSr Z_vZsetflagsrM rN rÐ Úkeysré r÷ rþ r r Z _TIME_UNITSr¸ rh r0 ri rz rb r Z complex64r r r r r r Ú<module> s ý ýýý/ 1_þ þ?i;Ia J (3ÿÿÿ
|
ÿÿ
|
ÿþõ
|
|
