U
|
ý°d ! ã @ sà d dl Z d dlZd dlZd dlZd dlmZmZmZm Z m
|
Z
|
m Z m Z m Z mZmZmZmZmZ d dlmZ d dlmZ d dlmZmZ dd Z dd Z d d
|
Z d d Z d d Z dd Z!dd Z"dd Z#dd Z$ej% &ddddg¡dd Z'ej% &dd d d ge(e)d g¡d!d" Z*ej% &d#d d d ged d d gdd$fe(e)d ed dd%gdd$fg¡d&d' Z+ej% &d(d)d*idfi d+fd)d,id+fd)did-fg¡d.d/ Z,ej% &d0d1d2d3g¡ej% &d4dd%g¡ej% &ddd5g¡d6d7 Z-ej%j&d8e ed9e
|
d:ge e d;e
|
d<ggd=d> d?d@dA Z.ej% &dBd e /d ddC¡g¡dDdE Z0ej% &dFedGd dHe dId<dJgged9d dHedKdLdMggg¡dNdO Z1ej% &dBdPdQg¡ej% &dRe ej2fej3ej4fe(ej5fg¡dSdT Z6ej% &dUd%dVdWg¡dXdY Z7dS )Zé N) Ú CategoricalÚ DatetimeIndexÚIntervalÚ IntervalIndexÚNaTÚSeriesÚTimedeltaIndexÚ TimestampÚcutÚ
|
date_rangeÚisnaÚqcutÚtimedelta_range)ÚCategoricalDtype)ÚDayÚNanoc C s t j d¡} t| ddd\}}t | ddddd g¡}|jjj}t j||d d
|
d d sZt |jj
|
j}t j||d d
d d st t | |dd}t ||¡ d S )Néè é T©Zretbinsr ç Ð?ç à?ç è?ç ð?éÿÿÿÿg{®Gáz?)Zatolé )Zinclude_lowest)ÚnpÚrandomÚrandnr ZquantileÚ
|
categoriesÚleftÚvaluesZallcloseÚAssertionErrorÚrightr
|
ÚtmÚassert_categorical_equal)ÚarrÚlabelsÚ_Zex_binsÚresultÚ ex_levels© r* úUd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/reshape/test_qcut.pyÚ test_qcut s
|
|
r, c C s4 t j d¡} t| ddd}tt |¡dks0td S )Nr é
|
F©r& )r r r r ÚlenÚuniquer! )r% Úfactorr* r* r+ Útest_qcut_bounds1 s r2 c C s: t j d¡} t| dddddg}t| d}t ||¡ d S )Néd r r r r r r )r r r r r# r$ )r% r1 Úexpectedr* r* r+ Útest_qcut_specify_quantiles8 s
|
r5 c C s< t jtdd$ tddddddddddg
|
d W 5 Q R X d S )Nz edges.*unique©Úmatchr é )ÚpytestÚraisesÚ
|
ValueErrorr r* r* r* r+ Útest_qcut_all_bins_same@ s r< c C sN t d¡} t| d}ttddtddtddtddg}t |j|¡ d S )Nr- r çü©ñÒMbP¿g @g @g @é )r Úaranger r r r# Úassert_index_equalr )r Úiir) r* r* r+ Útest_qcut_include_lowestE s
|
|
üÿrB c C s@ t j d¡} t j| d d
< t| d}t|d d
¡ s<td S )Nr3 é r )r r r Únanr r Úallr! )r% r( r* r* r+ Ú test_qcut_nasT s
|
rF c C s> t ddgd} tddtddg}t|dd}t | |¡ d S )Nr é r= r T©Zordered)r r r r# r$ )r( Ú intervalsr4 r* r* r+ Útest_qcut_index\ s rJ c C sä | t j ddd¡}t |¡}t|d}g }g }t |¡D ]8}|j}|j}||ksVt |
|
t |¡ |
|
t |¡ q:t t |d d
|dd
t |d d
|dd
D ]4\\} }
|
\} } | |
|
k sÆt | | k sÒt | |
|
ksªt qªd S )NZreshapeÚdataz cut_data.csvrC r r ) ÚosÚpathÚjoinr Zloadtxtr r0 r r" r! ÚappendÚfloatÚzip) ÚdatapathZcut_filer% r( ZstartsZendsZlevÚsÚeÚspZsnÚepÚenr* r* r+ Útest_qcut_binning_issuesd s$
|
|
ÿ rX c C s t ddddddddd g } t| dd
|
d dg}t td d td dtdd g¡}t | dddddddddg ¡ tdd¡}t ||¡ d S )Nr r rG r8 r é é é é g Zd;ßOÕ?g Zd;ßOå?r= g Zd;ßO@g Zd;ßO@TrH )
|
r r r Úarrayr ZtakeÚastypeÚCDTr# Úassert_series_equal)ÚserÚresZ
|
exp_levelsÚexpr* r* r+ Útest_qcut_return_intervals} s ÿ,rd r& Zfoor Tc C s8 t d}d}tjt|d t|d| d W 5 Q R X d S )NrY zJBin labels must either be False, None or passed in as a list-like argumentr6 r r. ©Úranger9 r: r; r ©r& r Úmsgr* r* r+ Útest_qcut_incorrect_labels s ri ÚaÚbÚcr8 c C s8 t d}d}tjt|d t|d| d W 5 Q R X d S )Nr- z9Bin labels must be one fewer than the number of bin edgesr6 r r. re rg r* r* r+ Ú test_qcut_wrong_length_labels s rm zlabels, expectedrH rG c C s& t d}t|d| d}t ||¡ d S )Nr8 r. )rf r r# r$ )r& r4 r r( r* r* r+ Útest_qcut_list_like_labels s rn z
|
kwargs,msgÚ
|
duplicatesÚdropzBin edges must be uniqueÚraisez(invalid value for 'duplicates' parameterc C s| dddddddg}|d k rDt jt|d t|df| W 5 Q R X n4t|df| }ttddtddg}t |j|¡ d S )Nr r rG r8 r6 r= ) r9 r: r; r r r r# r@ r )Úkwargsrh r r( r4 r* r* r+ Útest_qcut_duplicates_bin¨ s rs zdata,start,end)ç "@gshí|ÿ!@rt )ç r= ru )ç "Àgn "Àrv ÚlengthFc C sv t | g| }t|d|d}|d krRtt||g| dd}t | tdd¡}nt dg| tjd}t ||¡ d S ) Nr r. r" )ÚclosedTrH r ©Údtype)
|
r r r r r^ r_ r Zintpr# r` ) rK ÚstartÚendrw r& ra r( rI r4 r* r* r+ Útest_single_quantile¾ s r} ra Z20180101Z20180103z0 daysz2 daysc C s
|
t | jS )N)Ústrrz )Úxr* r* r+ Ú<lambda>× ó r )Úidsc C sf t | d t | d t ftj| d t | d fg¡}tt|dd}t| d}t
|
||¡ d S )Nr rG TrH ) r Ú from_tuplesr r r rD r r r r# r` )ra rI r4 r( r* r* r+ Ú test_qcut_natÑ s
|
4ÿ
|
r Úbinsr c
|
C s d}t tdd|d}t|| }t tttd|dtd|dttd|dtd|dttd|dtd |dg td
|
d ¡}t ||¡ d S ) Nz
|
US/EasternZ20130101r8 )ÚperiodsÚtzz 2012-12-31 23:59:59.999999999)r z2013-01-01 16:00:00z2013-01-02 08:00:00z2013-01-03 00:00:00TrH )
|
r r r r r r r^ r_ r# r` )r
r ra r( r4 r* r* r+ Útest_datetime_tz_qcutä s.
|
|
|
þ
|
|
þ
|
|
þ÷ÿÿïr zarg,expected_binsZ1day)r z1 daysz3 daysz
|
2018-01-01z
|
2018-01-02z
|
2018-01-03c C s* t | }t|ddd\}}t ||¡ d S )NrG Tr )r r r# r@ )ÚargZ expected_binsra r( Z result_binsr* r* r+ Útest_date_like_qcut_bins s r rZ r[ z box, comparec C sV |dddddgd }|dddddgd }t || dd}t || dd}||| d S )Nr r r- FTrp )ro )r )r
ZboxÚcompareZ data_expectedZ data_resultr4 r( r* r* r+ Ú test_qcut_bool_coercion_to_int s
|
r ÚqrY r- c C sN t jt d¡|d}t j|d d d
< t|| }t| t¡| }t ||¡ d S )Nr3 ry rG )
|
Úpdr] r r? ZNAr r^ rP r# r$ )r Zany_numeric_ea_dtyper% r( r4 r* r* r+ Útest_qcut_nullable_integer& s
|
|
r )8rL Únumpyr r9 Zpandasr r r r r r r r r r
|
r r r r Zpandas._testingZ_testingr# Zpandas.api.typesr r_ Zpandas.tseries.offsetsr r r, r2 r5 r< rB rF rJ rX rd ÚmarkZ parametrizeri Úlistrf rm rn rs r} r Zlinspacer r r` r] r$ Z assert_equalr r r* r* r* r+ Ú<module> s <
|
|
þþ
|
|
|
|
üþ
|
ÿþú
|
|
|
þ
|
þûþ
|
|
ýþ
|
