U
|
ý°dC ã @ sp d dl m Z mZ d dlZd dlZd dlmZmZ d dlm m
|
Z d dl m Z mZ d dlmZ G dd dZdS )é )ÚdatetimeÚ timedeltaN)ÚInfinityÚ NegInfinity)Ú DataFrameÚSeriesc @ sD e Zd Zeddddejdddejdg
|
ZeeedZe ddd d
|
ejd
|
dd ejdg
|
¡e ddd dejddd ejdg
|
¡e ddd dejddd ejdg
|
¡e ddd dejddd ejdg
|
¡e ddddejdddejdg
|
¡dZ
|
e j dddddgddd Z ejdd Zdd Zd d Zd d Zejd d! Zd"d# Zejd$d% Ze j d&d'd(d)g¡d*d+ Ze j d,d-dg¡e j d&d.eg¡d/d0 Ze j d1dd2d2d2gd2d3d4gd2d3d5ggfdd5d2d2gd5d5d4gd5d5d5ggfdd2d2d2gd2d4d4gd2d4d5ggfdd4d2d2gd4d3d4gd4d3d5ggfdd5d2d2gd4d5d4gd2d4d5ggfg¡d6d7 Z e jj d8d9 Z e j d:ej d;d<d=d>d?d-d@dAdBddCej g dDfej d;d<d=d>dEd-d@dAdBddCej g dFfe ej!¡j"dddGe ej!¡j#gdHfe ej$¡j"dId-ddJdKdLe ej$¡j#gdMfe% dNdOdPdQdRe& gdSfe'dTdde'dTdde'dTddgdUfg¡dVdW Z(dXdY Z)dZd[ Z*e j d\d]d^d_d2d`gfd]dad`d2d_gfdbd^d`d_d2gfdbdad2d_d`gfg¡dcdd Z+e j d\dbd^d2d`ded_gfdbdad2d`ded_gfd]d^d`d_d2degfd]dad`d_d2degfg¡dfdg Z,e j dhdddigdddgdjedkd2d`d_gifdidddigiee-dg dlfg¡dmdn Z.d.S )oÚTestRanké é é é é ©ÚAÚBç ø?g @ç @ç @g @é é é ©ÚaverageÚminÚmaxÚfirstÚdenser r r r r )Úparamsc C s |j S )z5
|
Fixture for trying all rank methods
|
)Úparam)ÚselfÚrequest© r! ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_rank.pyÚmethod" s zTestRank.methodc C sL dd l }ddl m} tj|jd d d
df< tj|jd d d
df< tj|jd d d
df< tj|jd d d
d
|
f< | ¡ }| d ¡}t |j¡}| tj ¡j}t
|
|d|¡}tj||< t
|
|d |¡} tj| |< t |j|¡ t |j| ¡ t tjjdd d d d¡}
|
|
|
¡ } |
|
t¡ ¡ } t | | ¡ |
|
d ¡} |
|
t¡ d ¡} t | | ¡ d S )Nr ©Úrankdatar r r
|
r r ÚCr ÚDr é( )Úsize)é
|
r )Ú scipy.statsr% ÚnpÚnanÚlocÚrankÚisnanÚvaluesÚfillnaÚinfÚapply_along_axisÚtmÚassert_almost_equalr ÚrandomÚrandintZreshapeÚastypeÚfloatÚassert_frame_equal) r Ú float_frameÚscipyr% Úranks0Úranks1ÚmaskÚfvalsÚexp0Úexp1ÚdfÚresultÚexpr! r! r" Ú test_rank) s.
|
|
|
|
zTestRank.test_rankc C s t dddgdddgg}t dddgdddggd }|jddd}t ||¡ t dddgdddgg}| d ¡d }|jd dd}t ||¡ t d
|
d d gd d d
|
gg}t dddgdddgg}|jdd d}t ||¡ t dddgdddgg}|jd d d}t ||¡ t d
|
tjd gd d d
|
gg}t dtjdgdddgg}|jdd d}t ||¡ t dtjdgdddgg}|jd d d}t ||¡ tdddtjtdddgtdddtdddtdddgg}t |}t dtjdgdddgg}|jdd dd}t ||¡ t dtjdgdddgg}|jdd d d}t ||¡ t d dddddddgi}t d ddddd d dgi}t | ¡ |¡ d S ) Nr r
|
r ç ð?ç @ç @T©Úpctr ÚbÚcÚaF©Ú numeric_onlyr éÑ r iÐ )rQ Ú ascendingç#B ¡Ç;éûÿÿÿr* gDóÂääécLg_ûðQïüPg ÂëþKH´9r g @g @r )r r/ r5 r; r, r- r )r rD ÚexpectedrE ÚdatarF r! r! r" Ú
|
test_rank2M sF þ zTestRank.test_rank2c C s: t tj dd¡dd}| ¡ }| ¡ |}t ||¡ d S )Nr* r
|
Úfloat64©Údtype)r r, r7 ZrandnÚcopyr/ r5 r; ©r rD rV rE r! r! r" Útest_rank_does_not_mutate~ s
|
z"TestRank.test_rank_does_not_mutatec C sR t ¡ |d< tddd|d< |jdd tjtdd |jdd W 5 Q R X d S )
|
Nr r )ÚdaysÚsecondsr FrP z"not supported between instances of©Úmatch©Úaxis)r Únowr r/ ÚpytestÚraisesÚ TypeError)r Zfloat_string_framer! r! r" Útest_rank_mixed_frame s
|
z TestRank.test_rank_mixed_framec C sÀ dd l }ddl m} tj|jd d d
df< tj|jd d d
df< tj|jd d d
df< tj|jd d d
d
|
f< |jd d }|jd d d }| tj¡j}t |d|¡}t |d |¡}t
|
|j|¡ t
|
|j|¡ |jdd }|jd dd }| | ¡ d ¡ ¡j} |j}
|
|
|
|
|
¡ d ¡ ¡j}
|
|
|
tj¡j}
|
t |d| ¡}t |d |
|
¡}t
|
|j|¡ t
|
|j|¡ |jddd}|jd ddd}| tj¡j}t |d| ¡}t |d | ¡}t
|
|j|¡ t
|
|j|¡ |jd dd}|jd d dd}| | ¡ d ¡ ¡j} |j}
|
|
|
|
|
¡ d ¡ ¡j}
|
|
|
tj¡j}
|
t |d| ¡}t |d |
|
¡}t
|
|j|¡ t
|
|j|¡ d} tjt| d |jddd W 5 Q R X tjt| d |jddd W 5 Q R X d S )Nr r$ r r r
|
r r r& r r' Úbottom)Ú na_optionr ÚtopF)rk rS z3na_option must be one of 'keep', 'top', or 'bottom'ra ÚbadT)r+ r% r, r- r. r/ r2 r3 r1 r4 r5 r6 r Úto_dictÚTZassert_numpy_array_equalrf rg Ú
|
ValueError) r r< r= r% r> r? rA rB rC Zfval0Zfval1Úmsgr! r! r" Útest_rank_na_option sZ z TestRank.test_rank_na_optionc C sP t ddgddgg}t |jdd|jdd¡ t |jdd|jdd¡ d S ) Nr r r r
|
r rc ÚindexÚcolumns)r r5 r; r/ )r rD r! r! r" Útest_rank_axisÝ s zTestRank.test_rank_axisc
|
C sØ dd l }ddl m} tj ddd¡}|d d }dd t|jd D }||d |d
|
fD ]x}t||d }d D ]b}d D ]X}|j||d} t ||||dkr|nd¡}
|
|
|
|
tj ¡}
|
t|
|
|d
|
d¡} t | | ¡ qvqnqZd S )Nr r$ é )éd é g $@c S s g | ]}t td | qS )Úz)ÚchrÚord)Ú.0Úir! r! r" Ú
|
<listcomp>ê s z4TestRank.test_rank_methods_frame.<locals>.<listcomp>r g .Agíµ ÷Æ°>)rt )r r r )rd r# r ZordinalrY )r+ r% r, r7 r8 ÚrangeÚshaper r/ r4 r9 rY r5 r; ) r r= r% ÚxsÚcolsÚvalsrD ZaxÚmrE ZsprankrV r! r! r" Útest_rank_methods_frameã s& ÿ z TestRank.test_rank_methods_framer[ ÚOÚf8Úi8c C s¦ d|kr| j ¡ |¡}n | j |¡}|jdd}| ¡ | ¡ }t ||¡ | ¡ | j|d}|dkr|j|ddd}t ||¡ |j|ddd}t ||¡ d S )Nr} F)rS )r# r T)r# rS rQ )rD Zdropnar9 r/ r r5 r; )r r# r[ rD ÚresrV Zres2Zres3r! r! r" Útest_rank_descendingù s z TestRank.test_rank_descendingrd r Nc C s@ | j }ddd}|d kr |n| |¡}||| j| ||d d S )Nr r c S s@ t ||d}|dkr"| j} |j}| j||d}t ||¡ d S )Nr r ©r# rd )r ro r/ r5 r; )rD rV r# rd Zexp_dfrE r! r! r" Ú_check2d s z3TestRank.test_rank_2d_tie_methods.<locals>._check2dr )r r )rD r9 Úresults)r r# rd r[ rD r Úframer! r! r" Útest_rank_2d_tie_methods s
|
|
z!TestRank.test_rank_2d_tie_methodsz
|
method,exprH g à?gUUUUUUå?gUUUUUUÕ?c C sF t dddgdddgdddgg}|j|dd}t |}t ||¡ d S ) NiÜ éB r
|
éA r r T)r# rL )r r/ r5 r; )r r# rF rD rE rV r! r! r" Útest_rank_pct_true s ! zTestRank.test_rank_pct_truec C sB t t d¡t ddd¡d}|jdd ¡ }|dk ¡ s>td S )Ni r éÿÿÿÿr TrK r )r r, Zaranger/ r ÚallÚAssertionError)r rD rE r! r! r" Útest_pct_max_many_rowsF s
|
ÿz TestRank.test_pct_max_many_rowszcontents,dtypeiÎÿÿÿr g#B ¡Ç»gÙ}ÚõÐò¾ºg ¸ÔJzîµgWw'&l¡7rT g»½×Ùß|Û=r( rY g¹j7Ö¶Úfloat32rw Úuint8iÿÿÿi' i g _ BÚint64Ú1r ZBAZBar& ÚobjectrR Ú
|
datetime64c C sÈ t jt jd t d¡d}t j||d}t jtt|ddd }||kr|| }t j tt|d¡}t |||¡}t ||t j¡}t j t|¡} ||| }
|
||| dd} |
|
|
¡ } t | | ¡ d S )NZnat)rY r r r rZ rY rH r ) r, r- r Úarrayr Úlenr7 ÚchoiceÚinsertZ permutationr/ r5 Ú assert_equal) r Úcontentsr[ Úframe_or_seriesZ dtype_na_mapr1 Z exp_orderZna_valueZ nan_indicesZ random_orderÚobjrV rE r! r! r" Útest_rank_inf_and_nanO s" >ü
|
z TestRank.test_rank_inf_and_nanc
|
C s¦ ddddddddd d
|
g
|
}ddddddddddg
|
}ddt jdddt jt jdt j g
|
}t||d |d d }| ¡ }| ¡ }|d ¡ |d< |d ¡ |d< t ||¡ d S )Nr r r
|
r r r r r é r* )Úcol1Úcol2r )rW rs r[ r§ r¨ )r, r- r3 r r/ r\ r5 r; )r rs r§ r¨ rD Z df_resultZ series_resultr! r! r" Ú"test_df_series_inf_nan_consistency¥ s "þúz+TestRank.test_df_series_inf_nan_consistencyc C sB t dtj dtjgi}t ddddgi}| ¡ }t ||¡ d S )NrO r rH rJ rI )r r, r3 r/ r5 r; r] r! r! r" Útest_rank_both_infº s zTestRank.test_rank_both_infz na_option,ascending,expectedrl TrI rJ Frj c C s> |t jt jt j g}|j|||d}||}t ||¡ d S )N©r# rk rS )r, r3 r- r/ r5 r¡ )r r£ r# rk rS rV r¤ rE r! r! r" Ú test_rank_inf_nans_na_optionÁ s z%TestRank.test_rank_inf_nans_na_optiong @c C s8 |ddd dg}|j d||d}||}t ||¡ d S )NZfoor r« )r/ r5 r¡ )r r£ rk rS rV r¤ rE r! r! r" Útest_rank_object_firstÒ s
|
z TestRank.test_rank_object_firstz data,expectedrO )rO rM rM )rs rt c C sF t |}tjtdd | ¡ W 5 Q R X |jdd}t ||¡ d S )Nz&'<' not supported between instances ofra TrP )r rf rg rh r/ r5 r; )r rW rV rD rE r! r! r" Útest_rank_mixed_axis_zeroá s
|
z"TestRank.test_rank_mixed_axis_zero)/Ú__name__Ú
|
__module__Ú __qualname__r r, r- Úsr rD r r rf Zfixturer# ÚtdZskip_if_no_scipyrG rX r^ ri rr ru r
ÚmarkZ parametrizer r r r Z
|
single_cpur r3 Ziinfor r r r r r r r¥ r© rª r¬ r r r® r! r! r! r" r s û
|
|
#1
|
M
|
|
ýþ þþýþîþ
|
|
óðóð$ ø
|
õ þÌþ<
|
üþ
|
üþ
|
& þþr )r r Únumpyr, rf Zpandas._libs.algosr r Z pandas.util._test_decoratorsÚutilZ_test_decoratorsr³ Zpandasr r Zpandas._testingZ_testingr5 r r! r! r! r" Ú<module> s
|
