U
|
®ý°d*G ã @ s d dl mZ d dlZd dlZd dlZd dlmZmZ d dl m
|
m Z d dl mZmZmZmZmZ d dlmZ d dlmZ ejdd Zejde d d
|
d d ejd d d ejd
|
g
|
¡gde ddddejdddejdg
|
¡gde ddddejdddejdg
|
¡gde ddddejdddejdg
|
¡gde ddddejdddejdg
|
¡ggddd Z ejd d d d d!ej d"e d#¡d$ej d%e d#¡d$gdd&d' Z G d(d) d)Z ej! "d*dgd+gfddgd,d+gfddgd+d+gfdddgd-d.d+gfdddgd,d+d+gfdddgd+d.d-gfdddddgd-d-d+d+d.gfddddddgd-d-d.d.d+d+gfd/d0d1d2d3gd4d5d6d7d+gfg ¡d8d9 Z#ej! "d*dgd+gfddgd,d+gfddgd,d,gfdddgd-d.d+gfdddgd-d.d.gfdddgd+d.d-gfdddddgd4d4d7d7d6gfddddddgd:d:d,d,d;d;gfd/d0d1d2d3gd4d5d6d7d+gfg ¡d<d= Z$ej! "d*dgd+gfddgd,d+gfddgd+d+gfdddgd-d.d+gfdddgd-d+d+gfdddgd+d.d-gfdddddgd5d5d+d+d6gfddddddgd-d-d.d.d+d+gfd/d0d1d2d3gd4d5d6d7d+gfg ¡d>d? Z%ej! "d*dgd+gfddgd,d+gfddgd@d@gfdddgd-d.d+gfdddgd-d;d;gfdddgd+d.d-gfdddddgdAdAdBdBd6gfddddddgdCdCdDdDdEdEgfd/d0d1d2d3gd4d5d6d7d+gfg ¡dFdG Z&ej! "d*dgd+gfddgd,d+gfddgd,d+gfdddgd-d.d+gfdddgd-d.d+gfdddgd+d.d-gfdddddgd4d5d7d+d6gfddddddgd:d-d,d.d;d+gfd/d0d1d2d3gd4d5d6d7d+gfg ¡dHdI Z'ej!j(dJdK Z)dS )Lé )ÚchainN)ÚInfinityÚ NegInfinity)ÚNAÚNaTÚSeriesÚ TimestampÚ
|
date_range)ÚCategoricalDtypec C s t ddddtjdddtjdg
|
S )Né é é é é )r ÚnpÚnan© r r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/methods/test_rank.pyÚser s r Úaverageg ø?g @ç @g @g @Úminr r é r é Úmaxr é r ÚfirstÚdense)Úparamsc C s | j S ©N©Úparam©Úrequestr r r Úresults s
|
r$ ÚobjectÚfloat64Úint64ÚFloat64ZInt64úfloat64[pyarrow]Úpyarrow©Zmarkszint64[pyarrow]c C s | j S r r r" r r r Údtype* s r, c @ sú e Zd Zejdd Zdd Zdd Zej
|
dde g¡d d
|
Z ejej
|
d d d g¡ej
|
ddddddg¡ej
|
ddddg¡ej
|
ddde e fdejejej fdeejej fejd eejej e d ¡d g¡d d Zd!d" Zejej
|
ddddddg¡ej
|
d#ejd$gejd%gejd&gg¡d'd( Zej
|
d)d*gd*gfd+gd*gfd$gd*gfd+d+gd*d*gfd*d+d,gd*d+d,gfd-d+d*gd,d+d*gfd*d*d.d.d,gd*d*d,d,d+gfd/d0d1d2d3gd*d+d,d-d.gfg¡d4d5 Zd6d7 Z d8d9 Z d:d; Z d<d= Z dS )>ÚTestSeriesRankc C sN ddl m} tj|d d d
< d|d dd
< | ¡ }| d¡ ¡ }t ||¡ t |¡}| tj
|
¡}t |||j dd }tj||< t ||¡ t t d
|
¡ d¡}| ¡ } | t¡ ¡ }t | |¡ t t d
|
¡d }|d }|jd d} t | |¡ t t dd¡}t t dd¡}|jd d} t | |¡ | d¡}tj|d< t t dd¡}tj|d< |jd d} t | |¡ t t d
|
¡d }tj|d< |d }|jd d} t | |¡ t t tjd¡}| ¡ }|jd d} t | |¡ t t d
|
¡ddd }tj|d< |d }|jd d} t | |¡ tdd
|
d}
|
t t d
|
¡|
|
ddd }tj|jd< |d }|jd d} t | |¡ t ddddd d g}t dddd
|
dd g}| ¡ } t | |¡ t d d d!tgd"d}t dddtjg}| ¡ } t | |¡ tjd#d$d%d&d'dd(dd)dd*g d+d} tj t| ¡} t | | }t | d d+d}| ¡ } t | |¡ d S ),Nr ©Úrankdatar ç @é
|
r ÚOÚts)ÚindexÚnamer ç ð?ç @T©Úpctr éd g)\Âõ(à?Úfloatg¥@_)à?r ©r, z1/1/1990)Zperiodsg ¸ÔJzî5g0ä.ÿ++g#B ¡Ç;g{®Gáz?gIf ¡Ç;g¹?ç @z3 dayz 1 day 10mz-2 dayzm8[ns]iÎÿÿÿéÿÿÿÿg#B ¡Ç»gÙ}ÚõÐò¾ºg ¸ÔJzîµgWw'&l¡7g»½×Ùß|Û=é( r& )Ú scipy.statsr/ r r ÚrankÚastypeÚtmÚassert_series_equalÚisnanZfillnaÚinfr r4 ÚarangeÚrepeatr; Úcopyr Zilocr ÚarrayÚrandomZ permutationÚlen) ÚselfZdatetime_seriesr/ ÚranksZoranksÚmaskZfilledÚexpÚiseriesZiranksÚrngÚvaluesZ random_orderr r r Ú test_rank: s
|
|
|
|
|
|
|
þ zTestSeriesRank.test_rankc
|
C sV t ddddddg}t ddddddg}t ddd d
|
d d g tddd d
|
d d gd d¡}t | ¡ |¡ t |jdd|¡ t ddd d
|
d d g tddd d
|
d d gdd¡}t ddddddg}| ¡ }t ||¡ t ddddddg tddddddgd¡}t ddddddg}| ¡ } t | |¡ t ddd d
|
d d tjg tddd d
|
d d dgd ¡}
|
t dddddddg} t dddddddg} t ddddddtjg} t |
|
jdd| ¡ t |
|
jdd| ¡ t |
|
jd d| ¡ t dddddddg} t dddddddg} t ddddddtjg} t |
|
jddd | ¡ t |
|
jddd | ¡ t |
|
jd dd | ¡ d }tj t
|
|d |
|
jd dd W 5 Q R X tj t
|
|d |
|
jd dd W 5 Q R X t ddd d
|
tjg tddd d
|
gd ¡}
|
t d!d"d#dd$g} t d$d!d"d#dg} t d%d&d'dtjg} t |
|
jdd d(| ¡ t |
|
jdd d(| ¡ t |
|
jd d d(| ¡ d S ))Nr6 g @g @r0 r7 r= r ÚsecondÚthirdZfourthZfifthZsixthT)Ú
|
categoriesÚorderedF©Ú ascendingr r r r r r Zseventhr Útop)Ú na_optionÚbottomÚkeep)r\ rZ z3na_option must be one of 'keep', 'top', or 'bottom'©ÚmatchÚbadçÙ?ç333333ã?çé?çÉ?ç Ð?ç à?ç è?)r\ r9 ) r rB r
|
rC rD rA r ÚNaNÚpytestÚraisesÚ
|
ValueError)rM rP Zexp_descrX Ú unorderedZ exp_unorderedÚresZ
|
unordered1Zexp_unordered1Zres1Zna_serZexp_topZexp_botZexp_keepÚmsgr r r Útest_rank_categorical s ÿþýÿþý ÿ ÿþý ÿÿz$TestSeriesRank.test_rank_categoricalc C sD t ddg}|jdd d}tjt|d | d¡ W 5 Q R X d S )Nr r r ©Úmethodz,No axis named average for object type Seriesr_ )r rA rj rk rl )rM Úsro r r r Útest_rank_signatureî s
|
z"TestSeriesRank.test_rank_signaturer, Nc C s> |\}}|d kr|n| |¡}|j|d}t |t|¡ d S )Nrq )rB rA rC rD r )rM r r$ r, rr rP Úresultr r r Útest_rank_tie_methodsõ s z$TestSeriesRank.test_rank_tie_methodsrZ TFrr r r r r r r\ r[ r] r^ z!dtype, na_value, pos_inf, neg_infr% r& r( r) r* r+ c C s |dkr |dkrd}q d}nd}d} |g| |g| |g| }
|
t |
|
|d} dddgdddgd d d gfd
|
d
|
d
|
gd d d gd d d gfdddgd d d gdddgfd
|
ddgd dd gd d dgfd
|
d
|
d
|
gdddgdddgfd} | | } |dkrþ| d
|
| d | d g}n<|dkr | d | d | d
|
g}n| d tjg| | d
|
g}|rD|n |d d d
}tt |¡}| j|||d}t |t ||d¡ d S )Nr) r zuint64[pyarrow]r& r r< r r r r r r r é )r r r r r r[ r r] r> )rr r\ rZ ) r r r Úlistr Ú from_iterablerA rC rD )rM rr r\ rZ r, Zna_valueZpos_infZneg_infZ exp_dtypeÚchunkZin_arrrQ Z exp_ranksrN ÚorderÚexpectedru r r r Ú"test_rank_tie_methods_on_infs_nansü s0 û
|
z1TestSeriesRank.test_rank_tie_methods_on_infs_nansc C sN t dtjtjtj dg}|jdd}t dtjdddgdd }t ||¡ d S )
|
Nr é FrY r r r r& r< )r r r rF rA rC rD )rM rQ ru rP r r r Ú test_rank_desc_mix_nans_infs0 s z+TestSeriesRank.test_rank_desc_mix_nans_infsz op, valuer g .Agíµ ÷Æ°>c s´ ddl m} tj d¡ t fddtdddD ¡ tj ¡ dd tt D }| |}t ||d}|j
|
|d }|||d
|
kr|nd } t | |d d ¡}
|
t ||
|
¡ d S ) Nr r. rw c s g | ]} |d
qS r r ©Ú.0Úi©Úxsr r Ú
|
<listcomp>F s z;TestSeriesRank.test_rank_methods_series.<locals>.<listcomp>r c S s g | ]}t td | qS )Úa)ÚchrÚordr r r r r
I s )r4 rq r Zordinalr& )r@ r/ r rK ZrandnZ concatenateÚrangeÚshufflerL r rA rB rC rD ) rM rr ÚopÚvaluer/ r4 Úvalsr3 ru Zsprankr| r r r Útest_rank_methods_series8 s
|
z'TestSeriesRank.test_rank_methods_seriesúser, expr r r r r éûÿÿÿéüÿÿÿéýÿÿÿéþÿÿÿr> c C s: t | |¡}|jdd}t | |j¡}t ||¡ d S )Nr rq ©r rB rA r, rC rD )rM r, r rP rs ru r| r r r Útest_rank_dense_methodQ s z%TestSeriesRank.test_rank_dense_methodc
|
C s~ |\}}d|kr| ¡ }n
|
| |¡}|jdd}| ¡ | ¡ }t ||¡ | ¡ | j|d}|j|dd} t | |¡ d S )Nr FrY rq )rr rZ )ÚdropnarB rA r rC rD )
|
rM r r$ r, rr Ú_rs rn r| Zres2r r r Útest_rank_descendingd s
|
|
z#TestSeriesRank.test_rank_descendingc C sF |\}}| ¡ d¡}|j|d}t| ¡ }|j|_t ||¡ d S )NÚi8rq )r rB rA r r4 rC rD )rM r r$ rr rP rs ru r| r r r Ú test_rank_ints s z TestSeriesRank.test_rank_intc C s@ t tjgd t¡jdd t tjgd t¡jdd d S )Né TrY F)r r r rB r% rA )rM r r r Útest_rank_object_bug| s z#TestSeriesRank.test_rank_object_bugc C s4 t tdtg}| ¡ }| ¡ |}t ||¡ d S )Nz2017-01-05 10:20:27.569000)r r r rI rA rC rD )rM rs r| ru r r r Útest_rank_modify_inplace s
|
z'TestSeriesRank.test_rank_modify_inplace) Ú__name__Ú
|
__module__Ú __qualname__ÚtdZskip_if_no_scipyrT rp rt rj ÚmarkÚ parametrizer% rv r r r r rF r r! Ú
|
skip_if_nor} r ÚoperatorÚaddÚmulr r r r r r r r r r r- 9 sj
|
[X
|
ûüþ!ýþ
|
|
|
øþ
|
r- r r6 rg gUUUUUUÕ?gUUUUUUå?r r r r r> re rb rc rd c C s< t | | ¡}|jddd}t | |j¡}t ||¡ d S )Nr T©rr r9 r ©r, r rP rs ru r| r r r Útest_rank_dense_pct s rª gUUUUUUÅ?g«ªªªªªê?c C s< t | | ¡}|jddd}t | |j¡}t ||¡ d S )Nr Tr¨ r r© r r r Útest_rank_min_pct¦ s r« c C s< t | | ¡}|jddd}t | |j¡}t ||¡ d S )Nr Tr¨ r r© r r r Útest_rank_max_pct» s r¬ rh g333333Ó?gÍÌÌÌÌÌì?rf g«ªªªªªâ?gUUUUUUí?c C s< t | | ¡}|jddd}t | |j¡}t ||¡ d S )Nr Tr¨ r r© r r r Útest_rank_average_pctÐ s r c C s< t | | ¡}|jddd}t | |j¡}t ||¡ d S )Nr Tr¨ r r© r r r Útest_rank_first_pctå s r® c C s. t t d¡} | jdd ¡ }|dks*td S )Ni Tr8 r )r r rG rA r ÚAssertionError)rs ru r r r Útest_pct_max_many_rowsú s r° )*Ú itertoolsr r¥ Únumpyr rj Zpandas._libs.algosr r Z pandas.util._test_decoratorsÚutilZ_test_decoratorsr¡ Zpandasr r r r r Zpandas._testingZ_testingrC Zpandas.api.typesr
|
Zfixturer rJ r r$ r! r¤ r, r- r¢ r£ rª r« r¬ r r® Z
|
single_cpur° r r r r Ú<module> sÒ
|
$$$$$ûÿ
|
ùÿ
|
Z
|
÷þ
|
|
÷þ
|
|
÷þ
|
|
÷þ
|
|
÷þ
|
|
