U
|
¬ý°dî® ã @ sB d dl mZ d dlZd dlZd dlZd dlZd dlm m Z
|
d dl m Z d dl Zd dl mZmZ d dlmZ d dlmZ d dlmZ ejZejdd Zejd d
|
Z ejd d Z ejd d Z ejdd Z ejdd Z ejdd Z!ejdd Z"ejdd Z#ejdd Z$ejdd Z%ejd d Z&ejd d Z'ejd!d" Z(ejd#d$ Z)ejd%d& Z*ejd'd( Z+ejd)d* Z,ejd+d, Z-ejd-d. Z.ejd/d0 Z/ejd1d2 Z0ejd3d4 Z1ejd5d6 Z2ejd7d8 Z3ejd9d: Z4ejd;d< Z5ejd=d> Z6ejd?d@ Z7G dAdB dBZ8ej9 :dCej;ej<fej=ej>fej?ej@fejAejBfejCejDfejEejFfg¡dDdE ZGej9 :dFdGdHdIdJdKdGdLdMdNdOg
|
¡dPdQ ZHej9 :dFdRdSdTdUdVdWdXdYdZg ¡ej9 :d[dd\d]g¡d^d_ ZIej9 :d`d dddddg¡dadb ZJej9 :d`ddd.g¡dcdd ZKG dedf dfZLG dgdh dhZMG didj djZNG dkdl dlZOG dmdn dnZPdodp ZQej9 :dqejRdrfejSdrfejTdsfejUdsfejVdsfejWdsfejXdtfejYdufejZdtfej[dufg
|
¡dvdw Z\ej9 :dxej]ej^ejSejUejWej_ej`ejaejbejcej[ejZejdejeejfg¡dydz Zgej9 :d{d|d g¡d}d~ Zhej9 :dde idddg¡e idgddg ¡g¡ej9 :ddd
g¡dd Zje
|
jke
|
jlej9 :dddg¡dd Zmej9 :dddg¡dd Znej9 :ddddg¡dd Zoej9 :dejpejqejrejsejteueddg¡ej9 :ddddddddg¡d d¡ ZvdS )¢é )ÚpartialN)Úis_integer_dtype)ÚSeriesÚisna)Únanops)Ú DatetimeArrayc c s, | ¡ }| tdd¡ d V W 5 Q R X d S )NÚ_USE_BOTTLENECKF)ÚcontextÚsetattrr )Z monkeypatchÚm© r úOd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/test_nanops.pyÚdisable_bottleneck s
|
r c C s dS )N©é é r r r r r Ú arr_shape s r c C s t j d¡ t jj| S )Néã+ )ÚnpÚrandomÚseedÚrandn©r r r r Ú arr_float$ s r c C s | | d S )Nù ð?r ©r r r r Ú arr_complex* s r c C s t j d¡ t j dd| ¡S )Nr éöÿÿÿé
|
©r r r Úrandintr r r r Úarr_int/ s r! c C s t j d¡ t j dd| ¡dkS )Nr r é r r r r r Úarr_bool5 s r# c C s t | ¡ d¡S )NÚS©r ÚabsÚastyper r r r Úarr_str; s r( c C s t | ¡ d¡S )NÚUr% r r r r Úarr_utf@ s r* c C s" t j d¡ t j dd| ¡ d¡S )Nr r é N úM8[ns]©r r r r r' r r r r Úarr_dateE s r. c C s" t j d¡ t j dd| ¡ d¡S )Nr r r+ úm8[ns]r- r r r r Ú
|
arr_tdeltaK s r0 c C s t t j| ¡S ©N)r ÚtileÚnanr r r r Úarr_nanQ s r4 c C s t | |g¡S r1 ©r Úvstack)r r4 r r r Ú arr_float_nanV s r7 c C s t | |g¡S r1 r5 )r4 r r r r Úarr_nan_float1[ s r8 c C s t | | g¡S r1 r5 ©r4 r r r Ú arr_nan_nan` s r: c C s
|
| t j S r1 )r Úinfr r r r Úarr_infe s r< c C s t | |g¡S r1 r5 )r r< r r r Ú arr_float_infj s r= c C s t | |g¡S r1 r5 ©r4 r< r r r Ú arr_nan_info s r? c C s t | ||g¡S r1 r5 )r r4 r< r r r Úarr_float_nan_inft s r@ c C s t | | |g¡S r1 r5 r> r r r Úarr_nan_nan_infy s rA c C sJ t | d¡| d¡| d¡| d¡| d¡| d¡| d¡| d¡g¡S )NÚO)r r6 r' )r r! r# r r( r* r. r0 r r r Úarr_obj~ s øÿrC c
|
C s0 t jdd | | d W 5 Q R £ S Q R X d S ©NÚignore©Úinvalidr ©r Úerrstater9 r r r Ú arr_nan_nanj s rJ c
|
C s2 t jdd t | |g¡W 5 Q R £ S Q R X d S ©NrE rF ©r rI r6 )r rJ r r r Úarr_complex_nan s rM c
|
C s, t jdd | d W 5 Q R £ S Q R X d S rD rH )r< r r r Ú arr_nan_infj s rN c
|
C s2 t jdd t | |g¡W 5 Q R £ S Q R X d S rK rL )r rN r r r Úarr_complex_nan_infj¢ s rO c C s | d d
df S ©Nr r r r r r Ú arr_float_1d¨ s rQ c C s | d d
df S rP r r9 r r r Ú
|
arr_nan_1d s rR c C s | d d
df S rP r )r7 r r r Úarr_float_nan_1d² s rS c C s | d d
df S rP r )Úarr_float1_nanr r r Úarr_float1_nan_1d· s rU c C s | d d
df S rP r )r8 r r r Úarr_nan_float1_1d¼ s rV c @ s¨ e Zd Zdd Zdd ZdBddZdCd d
|
ZdDd d ZdEd dZdFddZ e
|
j de jejfe jejfg¡dd Zdd Zdd Zdd Ze
|
j ded¡d d Ze
|
j ded¡d d Zeje
|
j ded¡d d! Z e
|
j de j ej fe j ej fg¡d"d# Z!dGd$d%Z"d&d' Z#d(d) Z$dHd*d+Z%ejd,d- Z&ejd.d/ Z'd0d1 Z(d2d3 Z)d4d5 Z*d6d7 Z+d8d9 Z,ejd:d; Z-ejd<d= Z.ejd>d? Z/d@dA Z0dS )IÚTestnanopsDataFramec C sð t j d¡ dt_d}t jj| | _t jj| | _| j| jd | _t j dd|¡| _
|
t j dd|¡dk| _ t | j¡ d ¡| _t | j¡ d
|
¡| _t j dd |¡ d ¡| _t j dd |¡ d ¡| _t t j|¡| _t | j| jg¡| _t | j| jg¡| _t | j| jg¡| _t | j| jg¡| _| jt j | _t | j| jg¡| _ t | j| jg¡| _ t | j| j| jg¡| _ t | j| j| jg¡| _ t | j d¡| j
|
d¡| j d¡| j d¡| j d¡| j d¡| j d¡| j d¡g¡| _ t j!ddL | j| jd | _"t | j| j"g¡| _#| jd | _$t | j| j$g¡| _%W 5 Q R X | j| _&| j| _'| j| _(| j| _)| j| _*| j| _+| jd d
df | _,| jd d
df | _-| jd d
df | _.| jd d
df | _/| jd d
df | _0| jd d
df | _1d S )Nr Fr r r r r r" r$ r) r+ r, r/ rB rE rF )2r r r r r r r Ú
|
arr_float1r r r! r# r& r' r( r* r. r0 r2 r3 r4 r6 r7 rT r8 r: r; r< r= r? r@ rA rC rI rJ rM rN rO Ú arr_float_2dÚ arr_float1_2dÚ
|
arr_nan_2dÚarr_float_nan_2dÚarr_float1_nan_2dÚarr_nan_float1_2drQ Ú arr_float1_1drR rS rU rV )Úselfr r r r Ú setup_method sd
|
|
|
|
|
|
|
|
øÿ z TestnanopsDataFrame.setup_methodc C s
|
t t_d S r1 )Úuse_bnr r ©r` r r r Úteardown_method s z#TestnanopsDataFrame.teardown_methodTc C s> t |d|}|dkrLt|drL|jrL|j|jkrLtj||jd gddd }ztj|||d W nØ tk
|
r8 t|dr|j dkr t|dr|j j
|
dkr |j j
|
d krî|j j
|
d krÆ| |j ¡}qüttd
|
rÔd nd }| |¡}| |¡}n|j j
|
d krü tjt |¡t |¡|d tjt |¡t |¡|d Y nX d S ) NZasm8r Úshape©Úaxis©Ú check_dtypeÚdtyper/ )ÚcrB rB Z
|
complex128Úc16Úf8)ÚgetattrÚhasattrÚndimre r ÚsplitÚtmÚassert_almost_equalÚAssertionErrorrj Úkindr' ÚrealÚimag)r` ÚtargÚresrg ri Z
|
cast_dtyper r r Ú check_results s6 ÿþý
|
ü
|
z!TestnanopsDataFrame.check_resultsNc K s¤ t t|jd g D ]8} |r"|n|}
|
|rP|rPt|
|
¡ rP||
|
fd| i|} n||
|
fd| i|} |
|
jtkr¢|tjks|tjkr¢t | tj
|
r| t ¡} nt | } ||f| |d|} | j | | | |d |rô||fd| i|} | j | | | |d | d kr$||fd|i|} | j | | | |d |r| d kr||f|} | j | | | |d q|jdkr`d S tj|ddd} tj|ddd}| j||| |f|||d | d S )
|
Nrg ©rg Úskipnarh r| é r éÿÿÿÿrf )r| ri Úempty_targfunc)ÚlistÚrangerp r Úallrj Úobjectr ÚanyÚ
|
isinstanceÚndarrayr' Úboolrz ÚtakeÚcheck_fun_data)r` ÚtestfuncÚtargfuncÚ testarvalÚ targarvalr| ri r Úkwargsrg Z targartempvalrx ry Z
|
testarval2Z
|
targarval2r r r r , sP
|
ÿÿ
|
üùøz"TestnanopsDataFrame.check_fun_datac
|
K sb |}| d¡r,t| |d d
r,|d d
}t| |}t| |} | j|||| f||d| d S )NZ_nanéüÿÿÿ)r| r )Úendswithro rn r )
|
r` r r Ztestarr| r r Ztargarr r r r r Ú check_fund s
|
|
üúùz TestnanopsDataFrame.check_func K s¬ | j ||d|f| | j ||d|f| | j ||d|f| | j ||d|f| | j d¡| j d¡| j d¡g}
|
|r| j ||d|f| |rà| j ||d|f| | j ||d|f| |rÎ| j ||d |f| |
|
| j d¡g7 }
|
|r|| j | j ||d
|
|f| |
|
| j d¡g7 }
|
|rhz|| j W n tk
|
r@ Y n(X | j ||d |f| |
|
| j d¡g7 }
|
|r¨t
|
|
|
¡| _ |d krt | j ||d }| j ||d|f| d S )Nr r7 r! r# rB r4 r rM rJ r. r0 Úconvert)ÚfuncÚ allow_complexrC )r r r' r! r# r r. r0 Ú TypeErrorr r6 rC r Ú _badobj_wrap) r` r r r| r Ú allow_all_nanÚ
|
allow_dateÚ allow_tdeltaÚ allow_objr Úobjsr r r Ú
|
check_funsw sH
|
|
|
ý
|
|
ÿz TestnanopsDataFrame.check_funsc K s2 |j jdkr&|r | d¡}n
|
| d¡}||f|S )NrB rl rm )rj ru r' )r` Úvaluer r r r r r r ° s
|
|
z TestnanopsDataFrame._badobj_wrapz nan_op,np_opc C s | j |||ddd d S )NF)r r )r ©r` Znan_opZnp_opr| r r r Útest_nan_funcs¸ s z"TestnanopsDataFrame.test_nan_funcsc C s | j tjtj|ddtjd d S )NF)r ri r )r r Únansumr Úsum©r` r| r r r Ú test_nansum¾ s úz TestnanopsDataFrame.test_nansumc C s | j tjtj|ddd d S )NF)r r )r r Únanmeanr Úmeanr¢ r r r Ú test_nanmeanÈ s ÿz TestnanopsDataFrame.test_nanmeanc C sB t jdd, t dt¡ | jtjtj|dddd W 5 Q R X d S )NT©ÚrecordrE Fr )r r r ) ÚwarningsÚcatch_warningsÚ simplefilterÚRuntimeWarningr r Ú nanmedianr Úmedianr¢ r r r Útest_nanmedianÍ s úz"TestnanopsDataFrame.test_nanmedianÚddofé c C s | j tjtj|ddd|d d S ©NFr )r r r r° )r r Únanvarr Úvar©r` r° r| r r r Ú test_nanvarÙ s ùz TestnanopsDataFrame.test_nanvarc C s | j tjtj|ddd|d d S r² )r r Únanstdr Ústdrµ r r r Ú test_nanstdå s ùz TestnanopsDataFrame.test_nanstdc C sD ddl m} tjdd" | jtj||dddd|d W 5 Q R X d S )Nr )ÚsemrE rF Fr )r r r r r° )Ú scipy.statsrº r rI r r Únansem)r` r° r| rº r r r Ú test_nansemñ s øz TestnanopsDataFrame.test_nansemc C s: t jdd$ t dt¡ | j|||dd W 5 Q R X d S )NTr§ rE F©r )r© rª r« r¬ r r r r r Útest_nanops_with_warnings s z-TestnanopsDataFrame.test_nanops_with_warningsc C sV |||}t ||¡}t|}|jr.d||< n$t|dr@| ¡ sNt|dsR|rRd}|S )Nr~ r )r Úminr rp ro r )r` r rg r ry ZnansZnullnanr r r Ú_argminmax_wrap
|
s
|
|
ÿþýüz#TestnanopsDataFrame._argminmax_wrapc C sL t jdd6 t dt¡ t| jtjd}| jt j
|
||dd W 5 Q R X d S ©NTr§ rE ©r Fr¾ ) r© rª r« r¬ r rÁ r Zargmaxr r Ú nanargmax©r` r| r r r r Útest_nanargmax s z"TestnanopsDataFrame.test_nanargmaxc C sL t jdd6 t dt¡ t| jtjd}| jt j
|
||dd W 5 Q R X d S r ) r© rª r« r¬ r rÁ r Zargminr r Ú nanargminrÅ r r r Útest_nanargmin s z"TestnanopsDataFrame.test_nanargminc C st t |jjtjs| d¡}|||dd}t |tjrXd|tj||dtj||dk< |S t |¡t |¡krpdS |S )Nrm F)rg Zbiasr rf ç ) r
rj Útyper Zfloatingr' r ÚmaxrÀ )r` Úvaluesrg r Úresultr r r Ú_skew_kurt_wrap% s
|
z#TestnanopsDataFrame._skew_kurt_wrapc C sN ddl m} t| j|d}tjdd | jtj||dddd W 5 Q R X d S )Nr )Úskewrà rE rF F©r r r ) r» rÏ r rÎ r rI r r Únanskew)r` r| rÏ r r r r Ú test_nanskew1 s úz TestnanopsDataFrame.test_nanskewc C sZ ddl m} t|dd}t| j|d}tjdd | jtj||dddd W 5 Q R X d S )
|
Nr )ÚkurtosisT)Zfisherrà rE rF FrÐ ) r» rÓ r rÎ r rI r r Únankurt)r` r| rÓ Zfunc1r r r r Ú test_nankurt@ s úz TestnanopsDataFrame.test_nankurtc C s | j tjtj|ddtjd d S )NF)r r r )r r Únanprodr Úprodr¢ r r r Ú test_nanprodP s úz TestnanopsDataFrame.test_nanprodc K s~ || j | jf|}|| j | jfdt| j d i|}t ||¡ t ||¡ || j| jf|}|| j| jfdt| j d i|}t ||¡ t ||¡ tj} || j | jf|}
|
|| j | j f|} || j | j f|} || j| j
|
f|} || j| j
|
fdt| j d i|}|| j | jfdt| j d i|}t | |
|
¡ t | | ¡ t | | ¡ t | | ¡ t | |¡ t | |¡ d S ©NZ min_periodsr} ) rY rZ Úlenrr rs r\ r] r r3 r[ r^ ©r` ZcheckfunÚtarg0Útarg1r Zres00Zres01Zres10Zres11Útarg2Zres20Zres21Zres22Zres23Zres24Zres25r r r Úcheck_nancorr_nancov_2dZ sb þ ýü þ ýü þ ýüþ ýü z+TestnanopsDataFrame.check_nancorr_nancov_2dc K s~ || j | jf|}|| j | jfdt| j d i|}t ||¡ t ||¡ || j| jf|}|| j| jfdt| j d i|}t ||¡ t ||¡ tj} || j | jf|}
|
|| j | j f|} || j | j f|} || j| j
|
f|} || j| j
|
fdt| j d i|}|| j | jfdt| j d i|}t | |
|
¡ t | | ¡ t | | ¡ t | | ¡ t | |¡ t | |¡ d S rÙ ) rQ r_ rÚ rr rs rS rU r r3 rR rV rÛ r r r Úcheck_nancorr_nancov_1d sb þ ýü þ ýü þ ýüþ ýü z+TestnanopsDataFrame.check_nancorr_nancov_1dc C s t | j| j¡d }t | jj| jj¡d }| tj||¡ t | j| j ¡d }t | jj| j j¡d }| j
|
tj||dd d S ©N©r r} Zpearson©Úmethod© r ÚcorrcoefrY rZ Úflatrß r ÚnancorrrQ r_ rà ©r` rÜ rÝ r r r Ú test_nancorr´ s z TestnanopsDataFrame.test_nancorrc C s t | j| j¡d }t | jj| jj¡d }| jtj||dd t | j| j ¡d }t | jj| j j¡d }| j
|
tj||dd d S rá rå ré r r r Útest_nancorr_pearson¼ s z(TestnanopsDataFrame.test_nancorr_pearsonc C s ddl m} || j| jd }|| jj| jjd }| jtj||dd || j| j d }|| jj| j jd }| j
|
tj||dd d S )Nr )Ú
|
kendalltauZkendallrã ) r» rì rY rZ rç rß r rè rQ r_ rà )r` rì rÜ rÝ r r r Útest_nancorr_kendallÄ s z(TestnanopsDataFrame.test_nancorr_kendallc C s ddl m} || j| jd }|| jj| jjd }| jtj||dd || j| j d }|| jj| j jd }| j
|
tj||dd d S )Nr )Ú spearmanrZspearmanrã ) r» rî rY rZ rç rß r rè rQ r_ rà )r` rî rÜ rÝ r r r Útest_nancorr_spearmanÏ s z)TestnanopsDataFrame.test_nancorr_spearmanc C sb t | j| j¡d }t | jj| jj¡d }d}tjt|d | jt j
|
||dd W 5 Q R X d S )Nrâ z;Unknown method 'foo', expected one of 'kendall', 'spearman'©ÚmatchÚfoorã ) r ræ rY rZ rç ÚpytestÚraisesÚ
|
ValueErrorrà r rè )r` rÜ rÝ Úmsgr r r Útest_invalid_methodÚ s
|
z'TestnanopsDataFrame.test_invalid_methodc C s| t | j| j¡d }t | jj| jj¡d }| tj||¡ t | j| j ¡d }t | jj| j j¡d }|
|
tj||¡ d S )Nrâ ) r ZcovrY rZ rç rß r ZnancovrQ r_ rà ré r r r Ú test_nancovâ s z TestnanopsDataFrame.test_nancov)T)TN)N)TTTTT)T)NN)NN)1Ú__name__Ú
|
__module__Ú __qualname__ra rd rz r r r r ró ÚmarkÚ parametrizer Únananyr r Únanallr r r£ r¦ r¯ r r¶ r¹ ÚtdZskip_if_no_scipyr½ ÚnanminrÀ ÚnanmaxrË r¿ rÁ rÆ rÈ rÎ rÒ rÕ rØ rß rà rê rë rí rï r÷ rø r r r r rW Á sn @
|
. ø
|
9 ÿ
|
÷
|
9
|
ÿ
|
|
|
|
ÿ
|
|
|
|
|
|
--
|
|
|
|
|
rW zop,nanopc C sT | d¡}| d¡}|||}| d¡}| d¡}| d¡} | d¡}
|
| d¡} |jrP|||} t || ¡ |jdkrt ||g¡} nt ||g¡} || |
|
}tj| |dd |}|| | }tj||dd tj|d d
|
d }tj|d d
|
d }tj|d d
|
d }tj|d d
|
d }tj| d d
|
d } tj|
|
d d
|
d }
|
tj| d d
|
d } tj|d d
|
d }qPd S ) Nr r4 r: r7 r8 r} Frh r r~ rf ) Úgetfixturevaluerp rr rs r r6 ÚhstackÚassert_numpy_array_equalr )ÚrequestÚopZnanopr r rX rÜ r4 r: r7 rT r8 Úres0rÝ Zres1rÞ Zres2r r r Útest_nan_comparisonë s6
|
|
|
|
|
|
|
|
|
|
|
|
r z arr, correct)r F)r! F)r# F)r( F)r* F)rM F)rJ F)rN T)rO Tc C sX | |¡}t|ddrTt |¡}|r.|s6tn|r6tt|dsBqTtj|ddd}q
|
d S ©Nrp Tr r~ rf )r rn r Ú _has_infsrt ro r r )r ÚarrÚcorrectr Úvalr r r r Útest_has_infs_non_float s
|
|
|
|
r )r F)r4 F)r7 F)r: F)r= T)r< T)r? T)r@ T)rA Tr' Zf4Úf2c C sj | |¡}|d k r | |¡}t|ddrft |¡}|r@|sHtn|rHtt|dsTqftj|ddd}q d S r
|
) r r' rn r r rt ro r r )r r r r' r r r r r r Útest_has_infs_floats9 s
|
|
|
|
|
r Úfixturec C s | | ¡}t |jd¡s td S ©NÚtest©r r Ú _bn_ok_dtyperj rt ©r r r Úobjr r r Útest_bn_ok_dtypeZ s
|
r c C s | | ¡}t |jd¡r td S r r r r r r Útest_bn_not_ok_dtypeb s
|
r c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
ÚTestEnsureNumericc C s: t d¡dkstt d¡dks$tt d¡dks6td S )Nr} çñ?y ð? @)r Ú_ensure_numericrt rc r r r Útest_numeric_valuesp s z%TestEnsureNumeric.test_numeric_valuesc C s° t dddg¡}t t |¡|¡s&t| t¡}t t |¡|¡sFtt jdddgtd}t t |¡|¡sptt jdd d
|
gtd}d }tj t
|
|d t |¡ W 5 Q R X d S ) Nr} r" r± Ú1Ú2Ú3©rj rò ÚbarZbazz Could not convert .* to numericrð ) r ÚarrayÚallcloser r rt r' r ró rô r )r` rÌ Zo_valuesZs_valuesrö r r r Ú test_ndarrayz s
|
z TestEnsureNumeric.test_ndarrayc C sF t t d¡d¡stt t d¡d¡s,tt t d¡d¡sBtd S )Nr ç ð?z1.1r z1+1jy ð? ð?)r r% r r rt rc r r r Útest_convertable_values s z)TestEnsureNumeric.test_convertable_valuesc C sx d}t jt|d t d¡ W 5 Q R X d}t jt|d t i ¡ W 5 Q R X t jt|d t g ¡ W 5 Q R X d S )Nz Could not convert foo to numericrð rò z%argument must be a string or a number)ró rô r r r )r` rö r r r Útest_non_convertable_values s z-TestEnsureNumeric.test_non_convertable_valuesN)rù rú rû r r& r( r) r r r r r o s
|
r c @ s¢ e Zd Zejdd Zejdd Zdd Zdd Zd d
|
Z d d Z
|
d d Z ej ded¡ej ded¡dd Zej ded¡dd Zedd ZdS )ÚTestNanvarFixedValuesc C s dS )Ng @r rc r r r Úvariance¢ s z TestNanvarFixedValues.variancec C s | j j|d ddS )Nç à?i )ZscaleÚsize)ÚprngÚnormal)r` r+ r r r Úsamples¦ s z TestNanvarFixedValues.samplesc C s t |¡}tj||dd d S )Nç{®Gáz?©Zrtol)r r³ rr rs )r` r0 r+ Úactual_variancer r r Útest_nanvar_all_finiteª s
|
z,TestNanvarFixedValues.test_nanvar_all_finitec C sj t jt d|jd ¡ }||d d d
< tj|dd}tj||dd tj|dd}tj|t jdd d S )Nr" r T©r| r1 r2 F)r r3 Úonesre r r³ rr rs )r` r0 r+ Ú samples_testr3 r r r Útest_nanvar_nans® s z&TestNanvarFixedValues.test_nanvar_nansc C sn t jt d|jd ¡ }||d d d
< tj|dd}tj||d dd tj|dd}tj|t jdd d S ) Nr" r Tr5 r, r1 r2 F) r r3 r6 re r r· rr rs r³ )r` r0 r+ r7 Z
|
actual_stdr r r Útest_nanstd_nans¸ s z&TestNanvarFixedValues.test_nanstd_nansc C sN | j j|jd d}t ||g¡}tj|dd}tj|t |dg¡dd d S )Nr ©r- r} rf çUUUUUUµ?r1 r2 )
|
r. Úuniformre r r6 r r³ rr rs r$ )r` r0 r+ Z samples_unifr3 r r r Útest_nanvar_axis s ÿz&TestNanvarFixedValues.test_nanvar_axisc C s d}| j jd|d fd}tj|d d
df< tj|dddd ¡ }tj|dddd ¡ }tj|ddd d ¡ }d
|
}tj||d d tj||d | | d d tj||d |d | d d d S )Né i' r} r: r~ Tr )rg r| r° r" r; r1 r2 r' g @) r. r< r r3 r r³ r¥ rr rs )r` Únr0 Z
|
variance_0Z
|
variance_1Z
|
variance_2r´ r r r Útest_nanvar_ddofÌ s z&TestNanvarFixedValues.test_nanvar_ddofrg r" r° r± c C s t d¡}t dddgdddgdd d
|
gg¡|d d
d d
f< t j |d < |d d
d f< t d d dgdddgdddggdddgdddgdd d ggg¡}tj|d ||d }t |d d
|||f ¡ t |d ¡sÔt tj
|
|d ||d }t |d d
|||f d ¡ t |d ¡st d S )!N)é rA g*ßùf#ï?g§| 9þË?gôÖ«Çá?gävý¼Zç?g§Û×?g¤Åצʺï?gã«6bÚ·?gEó-9ã?g$mx´PÄ?r± gÓõÁ?g/-²:Ŭ?g¥¼?½½?g-u±ållÊ?g(I0ìµ?güÍïM6Æ?g-u±ållÚ?gFºìÅ?gmÉúM6Ö?gý
|
â^¸?g)õÜ Å?gìµZ&ª?gÍ@#à)GÂ?g/·æTSÏ?g6æØîܳ?gÍ@#à)GÒ?g/·æTSß?gSW_ÙÜÃ?T)r| rg r° r, ) r Úemptyr$ r3 r r³ rr rs Úisnanrt r· )r` rg r° r0 r+ r´ r¸ r r r Útest_ground_truthß s2
|
ýÿýýúÿ z'TestNanvarFixedValues.test_ground_truthc C s. t dt d¡ }|j|d}|dks*td S )NiÂìµ-r )r° rÉ )r r r6 r¸ rt )r` r° ÚdatarÍ r r r Útest_nanstd_roundoff s z*TestNanvarFixedValues.test_nanstd_roundoffc C s t j d¡S ©NiÒ ©r r Z RandomStaterc r r r r. s zTestNanvarFixedValues.prngN)rù rú rû ró r r+ r0 r4 r8 r9 r= r@ rü rý r rD rF Úpropertyr. r r r r r* s
|
|
|
|
|
&
|
r* c @ sx e Zd Zejdd Zejdd Zej ddddg¡d d
|
Z d d Z
|
d d Z dd Z dd Z dd Zedd ZdS )ÚTestNanskewFixedValuesc C s t t ddd¡¡S ©Nr r} éÈ ©r ÚsinZlinspacerc r r r r0 s z TestNanskewFixedValues.samplesc C s dS )NgÌ5ôîÈ¿r rc r r r Ú actual_skew s z"TestNanskewFixedValues.actual_skewr çfffff¨@ç¨@ç ¨@c C s( |t d¡ }t |¡}|dks$td S ©Ni, rÉ )r r6 r rÑ rt )r` r rE rÏ r r r Útest_constant_series s
|
z+TestNanskewFixedValues.test_constant_seriesc C s\ d\}}| j j||dd}t |¡dk s,td\}}| j j||dd}t |¡dksXtd S ©N)ç333333Ó?ç¹?éd r: r )rW rV )r. Úbetar rÑ rt ©r` ÚalpharY Z left_tailedZ right_tailedr r r Útest_all_finite& s z&TestNanskewFixedValues.test_all_finitec C s t |¡}t ||¡ d S r1 )r rÑ rr rs ©r` r0 rO rÏ r r r rD / s
|
z(TestNanskewFixedValues.test_ground_truthc C sH t |t jt t|¡ g¡}tj|dd}t |t |t jg¡¡ d S ©Nr} rf )
|
r r6 r3 r6 rÚ r rÑ rr rs r$ r] r r r Ú test_axis3 s z TestNanskewFixedValues.test_axisc C s0 t |t jg¡}tj|dd}t |¡s,td S ©NFr5 )r r r3 r rÑ rC rt )r` r0 rÏ r r r Ú test_nans8 s z TestNanskewFixedValues.test_nansc C s. t |t jg¡}tj|dd}t ||¡ d S ©NTr5 )r r r3 r rÑ rr rs r] r r r Útest_nans_skipna= s z'TestNanskewFixedValues.test_nans_skipnac C s t j d¡S rG rH rc r r r r. B s zTestNanskewFixedValues.prngN)rù rú rû ró r r0 rO rü rý rT r\ rD r_ ra rc rI r. r r r r rJ s
|
|
|
rJ c @ sx e Zd Zejdd Zejdd Zej ddddg¡d d
|
Z d d Z
|
d d Z dd Z dd Z dd Zedd ZdS )ÚTestNankurtFixedValuesc C s t t ddd¡¡S rK rM rc r r r r0 J s z TestNankurtFixedValues.samplesc C s dS )Ng|vÂKó¿r rc r r r Ú actual_kurtN s z"TestNankurtFixedValues.actual_kurtr rP rQ rR c C s( |t d¡ }t |¡}|dks$td S rS )r r6 r rÔ rt )r` r rE Úkurtr r r rT R s
|
z+TestNankurtFixedValues.test_constant_seriesc C s\ d\}}| j j||dd}t |¡dk s,td\}}| j j||dd}t |¡dksXtd S rU )r. rY r rÔ rt rZ r r r r\ Y s z&TestNankurtFixedValues.test_all_finitec C s t |¡}t ||¡ d S r1 )r rÔ rr rs ©r` r0 re rf r r r rD b s
|
z(TestNankurtFixedValues.test_ground_truthc C sH t |t jt t|¡ g¡}tj|dd}t |t |t jg¡¡ d S r^ )
|
r r6 r3 r6 rÚ r rÔ rr rs r$ rg r r r r_ f s z TestNankurtFixedValues.test_axisc C s0 t |t jg¡}tj|dd}t |¡s,td S r` )r r r3 r rÔ rC rt )r` r0 rf r r r ra k s z TestNankurtFixedValues.test_nansc C s. t |t jg¡}tj|dd}t ||¡ d S rb )r r r3 r rÔ rr rs rg r r r rc p s z'TestNankurtFixedValues.test_nans_skipnac C s t j d¡S rG rH rc r r r r. u s zTestNankurtFixedValues.prngN)rù rú rû ró r r0 re rü rý rT r\ rD r_ ra rc rI r. r r r r rd G s
|
|
|
rd c @ sJ e Zd Zejddddgddd Zdd Zej d
|
d d g¡d d Z dS )ÚTestDatetime64NaNOpsÚsÚmsÚusÚns)Úparamsc C s |j S r1 )Úparam)r` r r r r Úunit{ s zTestDatetime64NaNOps.unitc C s t jddd |¡}|d }|t|t|fD ]}t |¡}||ks.tq.| dt j ¡}|t|t|fD ]}t |¡}||ksjtqjd S )Nz
|
2016-01-01r± )Zperiodsr} )
|
ÚpdZ
|
date_rangeZas_unitr r r r¤ rt ÚinsertÚNaT)r` ro ZdtiÚexpectedr rÍ Zdti2r r r r¦ s
|
|
z!TestDatetime64NaNOps.test_nanmeanÚ constructorZM8Zm8c C sÖ | d| d}t d¡ t j¡ |¡ dd¡}d|d< tj|dd }t |¡sTt |j
|
|ksbt tj|d
|
dd }t j dd dg|j
|
d }t ||¡ tj|ddd }t |d |d |d |d g¡}t ||¡ d S )Nú[ú]é rA r± rr )r~ r~ Fr5 r r{ r> r" r} râ )r} r} )r" r} )r Zaranger' Úint64ÚviewZreshaper r¤ Zisnatrt rj r$ rr r )r` rt ro rj r rÍ rs r r r Útest_nanmean_skipna_false s "z.TestDatetime64NaNOps.test_nanmean_skipna_falseN)
|
rù rú rû ró r ro r¦ rü rý rz r r r r rh z s
|
|
rh c C sV t jrRt dd¡ t d¡s"tW 5 Q R X t dd¡ t d¡rHtW 5 Q R X d S )NZuse_bottleneckTF)r Z_BOTTLENECK_INSTALLEDrp Zoption_contextÚ
|
get_optionrt r r r r Útest_use_bottleneck£ s
|
r| znumpy_op, expectedr g @r} rA c C s$ | t ddddg}||ks td S )Nr} r" r± rA )r rt )Znumpy_oprs rÍ r r r Útest_numpy_ops¬ s r} Ú operationc C sD t ddtjdtjdg}| ¡ }| |}| ||d}||ks@td S )Nr} r" r± rA )Úmask)r r r3 r rt )r~ Úserr Zmedian_expectedZ median_resultr r r Ú%test_nanops_independent_of_mask_paramÁ s
|
r Ú min_countr~ c C s" t dd | ¡}d}||ks td S )N)é é% F©r Zcheck_below_min_countrt )r rÍ Úexpected_resultr r r Ú5test_check_below_min_count_negative_or_zero_min_countÞ s r r FTé zmin_count, expected_result)r} F)ée Tc C s" d}t || |¡}||ks td S )N)r r r
)r r r re rÍ r r r Ú-test_check_below_min_count_positive_min_countæ s r )l l$=O Tc C s$ d}t j|d | d}||ks td S )N)i?" iå )r r r
)r r re rÍ r r r Ú&test_check_below_min_count_large_shapeñ s r r r¤ r c C s t t | ¡j|¡rtd S r1 )r r r rj rÊ rt )Zany_real_numpy_dtyper r r r Ú test_check_bottleneck_disallowû s r r l lüÿÿÿ l Ô|H%[<c C sR t |tdtjd}| ¡ }|j ¡ }||ks2t||ks>t|jtjksNtd S )Niô )Úindexrj ) r r r rx r¥ rÌ rt rj Úfloat64)r r r rÍ Z np_resultr r r Útest_nanmean_overflow s
|
r rj Zfloat128rä r¥ r¸ r´ rÏ rf rÀ rË c C sb |d krt d¡ ttd|d}t|| }t|rP|dkrP|jtjks^t n|j|ks^t d S )Nznp.float128 not availabler r" )rÀ rË )
|
ró Úskipr r rn r rj r r rt )r rj rä r rÍ r r r Útest_returned_dtype s
|
r )wÚ functoolsr Úoperatorr© Únumpyr ró Z pandas.util._test_decoratorsÚutilZ_test_decoratorsr Zpandas.core.dtypes.commonr Zpandasrp r r Zpandas._testingZ_testingrr Z pandas.corer Zpandas.core.arraysr r rb r r r r r r! r# r( r* r. r0 r4 r7 r8 r: r< r= r? r@ rA rC rJ rM rN rO rQ rR rS rU rV rW rü rý ÚeqZnaneqÚneZnanneÚgtZnangtÚgeZnangeÚltZnanltÚleZnanler r r r r r r* rJ rd rh 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 Zskip_if_windowsZ skip_if_32bitr r r Úint16Úint32rx Úfloat32r rn r r r r r Ú<module> s
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
.
|
|
|
|
|
|
úþ
|
$öþ
|
÷þ ÿ
|
ýþ
|
0u33) öþ
|
ñþ
|
|
&ÿ
|
|
|
úþ
|
