U
|
ý°d6' ã @ sb d dl Zd dlZd dlmZmZmZ d dlmZm Z G dd dZ
|
G dd dZ G dd d Z dS )
|
é N)ÚNaTÚ TimestampÚisna)Ú SparseArrayÚ SparseDtypec @ s e Zd Zej ddddgddfdddgddfdddgdd fg¡d
|
d Zej ddddgddfdddgddfdddgdd fg¡d d Zej ddddgddfdddgddfd dd gdd fg¡dd Zej ddddgddfdddgddfd dd gdd fg¡dd Z dd Z
|
ej de dde j dg¡e dddg¡g¡ej ddde j g¡ej ddde j fg¡dd Zdd Zd d Zd d Zd!d" Zd#S )$ÚTestReductionsz data,pos,negTFé é r g ð?g @g c C sd t | ¡ }|stt ||d ¡ }|s,t||d< t | ¡ }|rHtt ||d ¡ }|r`td S ©N©Ú
|
fill_valuer )r ÚallÚAssertionError©ÚselfÚdataÚposÚnegÚout© r úad:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/sparse/test_reductions.pyÚtest_all s
|
zTestReductions.test_allc C s¢ t t|¡}|stt t||d¡}|s0t||d< t t|¡}|rNtt t||d¡}|rhtd}tjt|d t jt|t g ¡d W 5 Q R X d S ©Nr r ú$the 'out' parameter is not supported©Úmatch©r )Únpr r r ÚpytestÚraisesÚ
|
ValueErrorÚarray©r r r r r Úmsgr r r Útest_numpy_all' s
|
z TestReductions.test_numpy_allc C sd t | ¡ }|stt ||d ¡ }|s,t||d< t | ¡ }|rHtt ||d ¡ }|r`td S r
|
)r Úanyr r r r r Útest_anyC s
|
zTestReductions.test_anyc C s t t|¡}|stt t||d¡}|s0t||d< t t|¡}|rNtt t||d¡}|rhtd}tjt|d t jt||d W 5 Q R X d S r )r r% r r r r r r" r r r Útest_numpy_anyZ s
|
z TestReductions.test_numpy_anyc C sp t d¡ t¡}t| ¡ }|dks(tt j|d< t|dd ¡ }|dksNtt|t jd ¡ }|dksltd S )Né
|
ç F@é r r ç D@)r ÚarangeÚastypeÚfloatr Úsumr Únan©r r r r r r Útest_sumu s
|
zTestReductions.test_sumÚarrr zmin_count, expected)é r é c C sB t ||d}|j|d}t |¡r2t |¡s>tn ||ks>td S )Nr ©Ú min_count)r r/ r Úisnanr )r r3 r r7 ÚexpectedZsparrayÚresultr r r Útest_sum_min_count s
|
|
z!TestReductions.test_sum_min_countc C sL t ddgd tjdd}|jdd}|dks0t|jdd}t|sHtd S )NFTr* )Údtyper r r6 é )r r Zbool_r/ r r )r Z spar_boolÚresr r r Útest_bool_sum_min_count s
|
z&TestReductions.test_bool_sum_min_countc C sØ t d¡ t¡}t t|¡}|dks*tt j|d< t t|dd¡}|dksRtt t|t jd¡}|dksrtd}tj t
|
|d t jt|t j d W 5 Q R X d
|
}tj t
|
|d t jt||d W 5 Q R X d S ) Nr( r) r* r r r+ ú&the 'dtype' parameter is not supportedr ©r< r r ) r r, r- r. r/ r r r0 r r r Úint64©r r r r# r r r Útest_numpy_sum s
|
z TestReductions.test_numpy_sumc C sN t d¡ t¡}t| ¡ }|dks(tt j|d< t| ¡ }|dksJtd S )Nr( ç @r* çr Çq Ç@)r r, r- r. r Úmeanr r0 r1 r r r Ú test_mean« s
|
zTestReductions.test_meanc C s´ t d¡ t¡}t t|¡}|dks*tt j|d< t t|¡}|dksNtd}tj t
|
|d t jt|t j d W 5 Q R X d}tj t
|
|d t jt||d W 5 Q R X d S )
|
Nr( rE r* rF r@ r rA r r ) r r, r- r. rG r r r0 r r r rB rC r r r Útest_numpy_mean´ s
|
z TestReductions.test_numpy_meanN)Ú__name__Ú
|
__module__Ú __qualname__r ÚmarkÚ parametrizer r$ r& r' r2 r r! r0 r; r? rD rH rI r r r r r sX ýþ
|
ýþ
|
ýþ
|
ýþ
|
"þ r c @ sH e Zd Zej de d¡dgdgfe d¡ dgdgfe dddej dg¡dgdgfe ej gd ¡ej gej gfe g ¡ej gej gfg¡d d
|
Z
|
ej d d d g¡dd Z dd Z ej dddg¡ej de g ¡e ej ej g¡g¡ej de ejej ej fe ejdej fe deefe dedefg¡dd ZdS ) Ú
|
TestMinMaxz"raw_data,max_expected,min_expectedg @r5 r éüÿÿÿr r r* c C s t |}| ¡ }| ¡ }||ks$t||ks0t|jdd}|jdd}t |¡ ¡ rtt |¡sdtt |¡stn||kst||kstd S )NF©Zskipna)r ÚmaxÚminr r r8 r% )r Úraw_dataÚ max_expectedÚ min_expectedr3 Ú
|
max_resultÚ
|
min_resultr r r Útest_nan_fill_valueÇ s z TestMinMax.test_nan_fill_valuez$fill_value,max_expected,min_expected)éd rZ r )éÿÿÿr r[ c C sJ t t |ddg¡td|d}| ¡ }||ks2t| ¡ }||ksFtd S )Nr r ÚintrA )r r r! r rR r rS )r r rU rV r3 rW rX r r r Útest_fill_valueá s ÿ zTestMinMax.test_fill_valuec C s d}t t |||g¡td|d}t|jdks4t| ¡ |ksDt| ¡ |ksTt|jdd|ksht|jdd|ks|td S )NrZ r\ rA r FrQ ) r r r! r ÚlenZ_valid_sp_valuesr rR rS )r Zfvr3 r r r Útest_only_fill_valueò s z TestMinMax.test_only_fill_valueÚfuncrS rR r zdtype,expectedzdatetime64[ns]z
|
2018-05-05c C sJ t ||d}t|| }|tkr8|tksFt |¡sFtnt |¡sFtd S )NrA )r Úgetattrr r Zisnatr r8 )r r` r r< r9 r3 r: r r r Ú test_na_value_if_no_valid_valuesü s
|
z+TestMinMax.test_na_value_if_no_valid_valuesN)rJ rK rL r rM rN r r, r! r0 rY r] r_ r Úfloat64r r rb r r r r rO Æ s: ûþ
|
|
þþ
|
|
|
$ üþ rO c @ s e Zd Zej dedddddgddfedddddgddfeejdddejdgddfeejdddejdgddfeejdddejdgddddfeejdddejdgddddfeejdddejdgddddfeejdddejdgddddfeejdddejdgd dddfedgd
|
dg dddd
|
fedgd
|
dg dddd
|
fedgd
|
dg dddd
|
fedgdgd
|
ddddfedgdgd
|
ddddfedgdgd
|
ddddfedgdgd
|
ddddfg¡d d Z ej d eg dfeg dfg¡dd Z
|
dS )ÚTestArgmaxArgminz#arr,argmax_expected,argmin_expectedr r r éÿÿÿÿéþÿÿÿr* r r4 r( c C s, | ¡ }| ¡ }||ks t||ks(td S )N)ÚargmaxÚargminr )r r3 Zargmax_expectedZargmin_expectedZ argmax_resultZ argmin_resultr r r Útest_argmax_argmin s z#TestArgmaxArgmin.test_argmax_argminz
|
arr,methodrg rh c C sB d| d}t jt|d |dkr,| ¡ n| ¡ W 5 Q R X d S )Nzattempt to get z of an empty sequencer rg )r r r rg rh )r r3 Úmethodr# r r r Útest_empty_array- s z!TestArgmaxArgmin.test_empty_arrayN) rJ rK rL r rM rN r r r0 ri rk r r r r rd s4 ðþ
|
þrd ) Únumpyr r Zpandasr r r Zpandas.core.arrays.sparser r r rO rd r r r r Ú<module> s 8K
|
