U
|
ý°d¾\ ã @ sj d dl Zd dlZd dlZd dlmZmZmZmZm Z m
|
Z
|
m Z m Z m Z mZmZmZmZmZmZmZ d dlmZ d dlmZ d dlm m m Z dd Z ej! "de#ej$g¡dd Z%d d
|
Z&d d Z'd d Z(dd Z)dd Z*dd Z+ej! "deddde j,e de j-ge
|
.e j,e dfe de j-fg¡fdd dgej$e /ej0¡j,d e /ej0¡j-gd d e
|
.e /ej0¡j,d fd e /ej0¡j-fg¡fe 1dd ¡e 1d d ¡e 1dd ¡ge $e 1e /ej0¡j- d ¡e 1d d ¡e 1e /ej0¡j-d ¡g¡e
|
.e 1e /ej0¡j- d ¡e 1d d ¡fe 1d d ¡e 1e /ej0¡j-d ¡fg¡fg¡d d Z2d!d" Z3ej! "d#g d$d%fdd$dgd&d'fg¡d(d) Z4ej! "d*d$e 5d$¡ee 5d$¡g¡ej! "d+eeg¡d,d- Z6ej! "d.d dd$dd/ej7gej7 d dd$dd/gej7 d dd$dd/ej7gg¡d0d1 Z8d2d3 Z9ej! "d4d5d6d7d&d8dgd9fd:d d7d&d8d;gd<fg¡d=d> Z:d?d@ Z;dAdB Z<ej! "dCdd:g¡dDdE Z=dFdG Z>dHdI Z?ej! "dJdKdL dMdL fdNdL dOdL fg¡dPdQ Z@dRdS ZAej! "dTe BdU¡e BdU¡dV g¡dWdX ZCej! "dYdZd[d\d]g¡d^d_ ZDd`da ZEdbdc ZFej! "dddedfidfi dgfdedhidgfdediidjfg¡dkdl ZGej! "d.dmdndog¡ej! "dpdd$g¡dqdr ZHej! "dsdtdudvg¡dwdx ZIej! "dydzdL d{dL d|dL d}dL g¡d~d ZJej! "d.ee dddge Kd¡e Kd¡e Kd¡ge $e Kd¡e Kd¡e Kd¡g¡edddgg¡dd ZLej! "d
de de de de dgg¡ej! "de#ej$ee g¡dd ZMdd ZNdd ZOej! "ddddg¡dd ZPdd ZQej! "d
ddg¡ej! "de ejRfej$ejSfe#ejTfg¡dd ZUej! "dCdidd5g¡dd ZVej! "d
dd dd gg¡ej! "d9d5d:g¡ej! "d¡d5d:g¡d¢d£ ZWej! "d¤d d¥d¦gd§d¨d©dªgd«d¬d«gd dd gd«d¬gfdddgd d$d/ddgd$d dd$gd$d dgd dd$gfg¡d®d¯ ZXej! "d¤d d¥d¦gd§d¨d©dªgd°d¬d«gd dd$gd°d¬d«gfdddgd d$d/ddgdd dd$gd dd$gdd dd$gfg¡d±d² ZYd³d´ ZZdµd¶ Z[d·d¸ Z\d¹dº Z]d»d¼ Z^d½d¾ Z_d¿dÀ Z`dÁd ZadS )Ãé N)Ú CategoricalÚ DataFrameÚ DatetimeIndexÚIndexÚIntervalÚ IntervalIndexÚSeriesÚTimedeltaIndexÚ TimestampÚcutÚ
|
date_rangeÚinterval_rangeÚisnaÚqcutÚtimedelta_rangeÚ to_datetime)ÚCategoricalDtypec C sD t jddd} t| ddd}t dddddg¡}tj||dd d S ) Né Úint64©Údtypeé F©Úlabelsé )Z check_dtype)ÚnpZonesr ÚarrayÚtmÚassert_numpy_array_equal)ÚdataÚresultÚexpected© r" úTd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/reshape/test_cut.pyÚ test_simple s r$ Úfuncc C s | ddddddg}t |ddd \}}t | d¡¡}| d
|
d
|
d
|
d d d
|
g¡}t|dd }t ||¡ t |t
|
ddddg¡¡ d S )NçÉ?çffffffö?ç @çÍÌÌÌÌÌ@çffffff#@çÍÌÌÌÌÌ @é T©Úretbinsr r é ©Úorderedçü©ñÒMbÈ?g>waïîî
|
@gúÝè!""@) r r Ú from_breaksÚroundÚtaker r Úassert_categorical_equalÚassert_almost_equalr r )r% r r ÚbinsÚ intervalsr! r" r" r# Ú test_bins$ s r: c
|
C s t dddddddg¡} t| dd d d
|
\}}t | d ¡¡}t|d d }| d d d dd d d g¡}t ||¡ t
|
|t dddddg¡¡ d S )Nr& r' r( r) r* r+ ç@r T©Úrightr. r, r0 r r/ r2 çÍÌÌÌÌÌ@çÍÌÌÌÌL @) r r r r r3 r4 r r5 r r6 r7 ©r r r8 r9 r! r" r" r# Ú
|
test_right1 s rA c
|
C s t dddddddg¡} t| dd d
|
d \}}tj| d ¡d d}| ddddd ddg¡}t|d
|
d}t ||¡ t
|
|t dddddg¡¡ d S )Nr& r' r( r) r* r+ r; r FTr< r, Úleft©Úclosedr r/ r r0 r> r? g%Ck#@) r r r r r3 r4 r5 r r r6 r7 r@ r" r" r# Ú test_no_right= s rE c C sn t tdd} | }t td|jd}t ||¡ tjt | j d¡| jdd}t td|jd}t ||¡ d S )Nr r, ©r8 éÿÿÿÿT©Ú
|
categoriesr1 é )
|
r ÚrangerI r r6 r Ú
|
from_codesr ÚappendÚcodes)Úcr! r r" r" r# Ú test_bins_from_interval_indexI s ÿrP c C s t ddddddddd g ¡} t| d
|
d d d gd}t dddg¡}t |j|¡ tdddg|jd}t |j|¡ t |j t jdddgdd¡ d S )Né
|
é é é é é é é; é< r é é# éF rF )r rZ )rZ r[ )r[ r\ é é2 r r/ Zint8r )
|
r r r r Ú from_tuplesr Úassert_index_equalrI r rN )ZagesrO r! r r" r" r# Ú)test_bins_from_interval_index_doc_exampleV s ra c C sB d} t dddg¡}tjt| d tddg|d W 5 Q R X d S ) Nz)Overlapping IntervalIndex is not accepted)r rQ )r/ rT )r é ©Úmatchr rJ rF )r r_ ÚpytestÚraisesÚ
|
ValueErrorr )ÚmsgÚiir" r" r# Ú-test_bins_not_overlapping_from_interval_indexb s rj c C sD d} ddddddg}t jt| d t|d d
|
d d g W 5 Q R X d S ) Nz bins must increase monotonicallyr& r' r( r) r* r+ rc ç¹?ç ø?r rQ ©re rf rg r ©rh r r" r" r# Útest_bins_not_monotonick s ro zx, bins, expectedz
|
2017-12-31r, ©Úperiodsz
|
2018-01-01rG r r r Únsc C s t | |}t |j|¡ d S ©N)r r r` rI )Úxr8 r! r r" r" r# Ú#test_bins_monotonic_not_overflowings s 4
|
ru c C sL d} ddddddg}t jt| d t|d d
|
d gd d dgd W 5 Q R X d S )Nz9Bin labels must be one fewer than the number of bin edgesr& r' r( r) r* r+ rc r r rQ ÚfooÚbarZbazr rm rn r" r" r# Útest_wrong_num_labels« s rx z
|
x,bins,msgr/ zCannot cut empty arrayç à?z#`bins` should be a positive integerc C s( t jt|d t| | W 5 Q R X d S )Nrc rm )rt r8 rh r" r" r# Útest_cut_corner³ s rz ÚargÚcut_funcc C s, d}t jt|d || d W 5 Q R X d S )Nz!Input array must be 1 dimensionalrc r/ )re rf rg )r{ r| rh r" r" r# Útest_cut_not_1d_arg¿ s r} r r c C s. d}t jt|d t| dd W 5 Q R X d S )Nz?cannot specify integer `bins` when input data contains infinityrc r, rF rm )r rh r" r" r# Útest_int_bins_with_infÇ s
|
r~ c C sZ d} t dddddg| d}t|ddgdd}t tjtjtjdtjg| d}t ||¡ d S ) Nrt r rG r éýÿÿÿ©ÚnameFr )r r r Únanr Úassert_series_equal)r ÚserÚindÚexpr" r" r# Útest_cut_out_of_range_moreÖ s
|
r zright,breaks,closedTçü©ñÒMbP¿g Ð?ç è?r= Fgj¼tð?rB c C sJ t t ddd¡d¡}t|dd| d\}}tj||d}t |j|¡ d S )Nr g)\Âõ(ð?rk r T)r. r= rC ) r ÚtileÚaranger r r3 r r` rI )r= ZbreaksrD Úarrr r8 Ú ex_levelsr" r" r# Ú test_labelsá s r c C s4 d} t tj d¡| d}t|d}|j| ks0td S )Nrv éd r r )r r ÚrandomÚrandnr r ÚAssertionError)r r Zfactorr" r" r# Ú#test_cut_pass_series_name_to_factorð s
|
r c C sB t ddd¡} t| ddd}t ddd d
|
d g¡}t |j|¡ d S ) Nr g\Âõ(\ç?ç{®Gáz?r r/ ©Ú precisiongÖÿ9ÌG¿g
|
×£p=
|
Ç?g
|
×£p=
|
×?gHáz®Gá?g
|
×£p=
|
ç?)r r r r r3 r r` rI )r r r r" r" r# Útest_label_precisionø s r r c C sZ t ddd¡}t j|d d d
< t|d| d}t |¡}t t|t j|¡}t ||¡ d S )Nr r r r, r r ) r r r r ZasarrayÚwherer r r7 )r r r r! r" r" r# Útest_na_handling s
|
r c C s¼ t d¡} t| dd}t j ddt jg}t| |}t||}t |¡}t |j |¡ |d t
|
dt jkslt |d t
|
t j dkst |d t
|
dt jkst |d t
|
t j dks¸t d S )NrJ r r r/ r r r ) r r r Úinfr r r3 r r` rI r r )r Zdata_serr8 r Z
|
result_serZ
|
ex_uniquesr" r" r# Útest_inf_handling s
|
|
|
|
r c C sD t j d¡} t| dddg}t|}| dk | dkB }t ||¡ d S )Nr rG r r )r r r r r r r )r r ÚmaskZex_maskr" r" r# Útest_cut_out_of_bounds s
|
r zget_labels,get_expectedc C s | S rs r" r r" r" r# Ú<lambda>* ó r c C s" t dgddg ddg | ddS )NÚMediumr ÚSmallÚLargeTrH )r r r" r" r# r + s ýc C s t dddg| ¡S )Nr r r/ ©r rL r r" r" r# r 2 r c C s t dgddg ddg | ¡S )Nr r r r/ r£ r r" r" r# r 3 r c C sN ddddg}dddddd d
|
g}d d d g}t ||| |d}t |||¡ d S )Nr rV r^ r r rQ rR r] é r\ r¡ r r¢ r )r r r6 )Z
|
get_labelsZ get_expectedr8 r r r r" r" r# Útest_cut_pass_labels& s
|
|
r¥ c C sR dddddddg} dd d
|
g}t | d |d }t | d t||d dd }t ||¡ d S )Nr^ r rQ rR r] r¤ r\ ZGoodr ZBadr, r TrH )r r r r6 )r r r r r" r" r# Útest_cut_pass_labels_compat@ s
|
|
r¦ rt g &@g _ Bc C s t | d d S )Nr/ ©r )rt r" r" r# Útest_round_frac_just_worksJ s r¨ zval,precision,expected)g¢#¹ü]Àr, iÿÿÿ)ç¢#¹ü]@r, év )r© r/ rª )gÝ ÷~. ?r/ gi UMu ?c C s t j| |d}||kstd S )Nr )ÚtmodZ _round_fracr )Úvalr r! r r" r" r# Útest_round_fracP s r c C s t ddddddddd g } t| d}tjdd dd
|
d¡}|d d 8 < t tj|d d dddddddddg ¡ t dd¡}t
|
||¡ d S )Nr r r/ r, r r rJ é é )Únumgü©ñÒMb?r= rC Tr0 ) r r r Zlinspacer4 r r3 r5 ÚastypeÚCDTr r )r r Zexp_binsr! r" r" r# Útest_cut_return_intervals_ s
|
ÿÿür³ c C sZ t t d¡} t| ddd\}}t tjdddgdd d¡ tdd
|
¡}t
|
||¡ d S ) Nr r/ Tr- gú~j¼th¿rl r, r= rC r0 ) r r r r r r3 Úrepeatr± r² r r )r r r8 r! r" r" r# Útest_series_ret_binsn s ÿþrµ z
|
kwargs,msgÚ
|
duplicatesÚdropzBin edges must be uniqueÚraiserv z(invalid value for 'duplicates' parameterc C s ddddddg}t t dddd d
|
g¡d d d ddgd}|d k rhtjt|d t||f| W 5 Q R X n*t||f| }t|t |¡}t
|
||¡ d S )Nr r/ r rJ rQ r r, r r® é ÚaÚbrO ÚdÚe©Úindexrc ) r r r re rf rg r ÚpdÚuniquer r )Úkwargsrh r8 Úvaluesr r! r" r" r# Útest_cut_duplicates_biny s &rÄ g "@g "Àg Úlengthc C s@ t | g| }t|ddd}t dg| tjd}t ||¡ d S )Nr Fr r r )r r r Zintpr r )r rÅ r r r! r" r" r# Útest_single_bin s rÆ z#array_1_writeable,array_2_writeable)TT)TF)FFc C sR t ddd¡}| |j_t ddd¡}||j_t d¡}t t||t||¡ d S )Nr r rQ )r r ÚflagsZ writeabler r6 r )Zarray_1_writeableZarray_2_writeableZarray_1Zarray_2Zhundred_elementsr" r" r# Útest_cut_read_only s
|
ÿrÈ Úconvc C s t | S rs )r
|
©Úvr" r" r# r ¯ r c C s t | S rs )r rÊ r" r" r# r ° r c C s
|
t | ¡S rs )r Ú
|
datetime64rÊ r" r" r# r ± r c C s t | ¡ S rs )r
|
Z to_pydatetimerÊ r" r" r# r ² r c s t d¡t d¡g}dddg}tttt|d t|d tt|d t|d g td d
|
¡} fd d |D }tt||d }t
|
||¡ d S )Nz
|
2012-12-13z
|
2012-12-15z
|
2012-12-12z
|
2012-12-14z
|
2012-12-16r r r/ Tr0 c s g | ] } |qS r" r" )Ú.0rË ©rÉ r" r# Ú
|
<listcomp>Â s z%test_datetime_bin.<locals>.<listcomp>rF ) r rÌ r r r r
|
r± r² r r r )rÉ r Zbin_datar! r8 r r" rÎ r# Útest_datetime_bin¬ s
|
|
þÿÿù rÐ z
|
2013-01-01z
|
2013-01-02z
|
2013-01-03c C sn t | ddd\}}ttttdtdttdtdttdtdg tdd¡}t t||¡ d S ) Nr, Tr- ú2012-12-31 23:57:07.200000ú2013-01-01 16:00:00ú2013-01-02 08:00:00ú2013-01-03 00:00:00r0 ) r r r r r
|
r± r² r r )r r Ú_r! r" r" r# Útest_datetime_cutÇ s* þ ÿ ÿøÿÿñrÖ r8 z2013-01-01 04:57:07.200000z2013-01-01 21:00:00z2013-01-02 13:00:00z2013-01-03 05:00:00Úboxc
|
C s¢ d}t tdd|d}t| ts(|| } t|| }t tttd|dtd|dttd|dtd|dttd|dtd |dg t d
|
d ¡}t
|
||¡ d S ) Nz
|
US/EasternÚ20130101r, ©rq ÚtzrÑ )rÚ rÒ rÓ rÔ Tr0 ) r r Ú
|
isinstanceÚintr r r r
|
r± r² r r )r8 r× rÚ Úsr r! r" r" r# Útest_datetime_tz_cutò s2
|
|
|
|
þ
|
|
þ
|
|
þ÷ÿÿïrÞ c C s< d} t jt| d ttddddddgd W 5 Q R X d S )
|
Nz bins must be of datetime64 dtyperc rØ r, rp r r/ r rF )re rf rg r r ©rh r" r" r# Útest_datetime_nan_error s rà c
|
C sb t tdddtdddd} | j ¡ }t |t dg¡¡ | ¡ }t |t dddddg¡¡ d S ) NZ20130102r rp rØ r/ rF FT)r r rI r r r r r )r r r" r" r# Útest_datetime_nan_mask$ s
|
|
ÿ
|
rá rÚ ÚUTCz
|
US/Pacificc C sb t tdd| d}t|ddd\}}t||}t ||¡ tddd g}| | ¡}t ||¡ d S )
|
NZ20180101r, rÙ r/ Tr- z2017-12-31 23:57:07.200000z2018-01-02 00:00:00z2018-01-03 00:00:00)r r r r r r Z tz_localizer` )rÚ r r Ú result_binsr! Ú expected_binsr" r" r# Útest_datetime_cut_roundtrip0 s
|
ÿ
|
rå c C sV t tddd} t| ddd\}}t| |}t ||¡ tddd g}t ||¡ d S )
|
NZ1dayr, rp r/ Tr- z0 days 23:57:07.200000z2 days 00:00:00z3 days 00:00:00)r r r r r r r` )r r rã r! rä r" r" r# Ú test_timedelta_cut_roundtrip@ s
|
ÿræ rJ r® z box, comparec C sV |dddddgd }|dddddgd }t || dd}t || dd}||| d S )Nr r rQ FTr· )r¶ r§ )r8 r× ÚcompareZ data_expectedZ data_resultr! r r" r" r# Ú test_cut_bool_coercion_to_intN s
|
rè c C s8 t d}d}tjt|d t|d| d W 5 Q R X d S )Nr zJBin labels must either be False, None or passed in as a list-like argumentrc r r )rK re rf rg r )r rà rh r" r" r# Útest_cut_incorrect_labels` s ré r rR Úinclude_lowestc C sb t jjdddd t¡}t j|d d d
< ttj|dd| ||d}t|| ||d}t
|
||¡ d S ) Nr rQ r^ )Úsizer/ ÚInt64r )r= rê ) r r Úrandintr± Úfloatr r rÀ r r r6 )r8 r= rê rº r r! r" r" r# Útest_cut_nullable_integeri s ÿrï z3data, bins, labels, expected_codes, expected_labelsé é rb é rZ r] ÚAÚBr¯ c C s0 t | ||dd}tj||dd}t ||¡ d S ©NF©r8 r r1 rH ©r r rL r r6 ©r r8 r Zexpected_codesZexpected_labelsr r! r" r" r# Útest_cut_non_unique_labelsv s ÿrù ÚCc C s0 t | ||dd}tj||dd}t ||¡ d S rõ r÷ rø r" r" r# Útest_cut_unordered_labels s ÿrû c C s: d} t jt| d tddgdddgdd W 5 Q R X d S )
|
Nz.'labels' must be provided if 'ordered = False'rc ry r, r r r/ F)r8 r1 rm rß r" r" r# Ú3test_cut_unordered_with_missing_labels_raises_error s rü c C sf t dddddg} t ddddg}t dd d
|
g}t| ||d d }t ddd d d
|
gd d}t ||¡ d S )Nr r/ r, r r r rJ rº r» rO Frö Úcategoryr )r r r r )rÝ r8 r r r! r" r" r# Ú%test_cut_unordered_with_series_labels s rþ c C sf t dtj ddd¡i} dd tdddD }t d¡$ t| jtdd dd|d
|
| d < W 5 Q R X d S ) NÚvaluer r r] c S s g | ]}| d |d qS )z - r¹ r" )rÍ Úir" r" r# rÏ © s z(test_cut_no_warnings.<locals>.<listcomp>rQ Féi )r= r Úgroup) r r r rí rK r Zassert_produces_warningr rÿ )Zdfr r" r" r# Útest_cut_no_warnings§ s r c C s t tddddgd tddddtddddg dddddgd d
|
j ¡ } t dddddgdddddgd }t|dddgd d }t || ¡ d S )Nr r/ r= rC r, r r r rý )r¿ r r¾ T)r8 rê )r r ÚcatZ
|
as_orderedr r r )r! rÝ r r" r" r# Ú.test_cut_with_duplicated_index_lowest_included® s ÿ ü r c C s t tdd} t| d ¡ d¡}t| d ¡ d¡}t||d}tjt dddd t dd
|
dd t d
|
d dd t d d dd t d d dd t d ddd t dddd t dddd t dddd t dddd g
|
dd}tdgd t
|
j gd t
|
j gd dgd d|d}t ||¡ d S )Nr r rQ r )Ú1Ú2gòÒMbX¹¿gÍÌÌÌÌÌ#@r= rC gÍÌÌÌÌÌ3@g33333³=@gÍÌÌÌÌÌC@g ÀH@g33333³M@g33333SQ@gÍÌÌÌÌÌS@gfffffFV@éc Tr0 r¾ )r rK r Z value_countsÚheadÚtailr rÀ ZCategoricalIndexr r r r Zassert_frame_equal)r Zser1Zser2r r¿ r! r" r" r# Ú*test_cut_with_nonexact_categorical_indices¼ s. ö ó, ÿr c C s^ t dft dft dfg} tdddgddd d
|
g| d }tjd ddg| d d}t ||¡ d S )NrQ r] r¤ r/ r rJ r r, r r® )r8 r r Tr0 )r
|
r r rL r r6 )r r r! r" r" r# Ú$test_cut_with_timestamp_tuple_labelsÛ s r c C sP t tdtddd} tttdg| d}tjdg| dd }t |j|¡ d S )
|
Nz
|
2022-02-25z
|
2022-02-27Z1D)Úfreqz
|
2022-02-26rF r Tr0 ) r r
|
r r r rL r r6 r )r8 r r! r" r" r# Ú$test_cut_bins_datetime_intervalindexä s r c
|
C sv t dddddtjddgdd } ddddd
|
g}t |¡}t tjd ddddd ddg|d d }t| |d}t ||¡ d S )Nr r r/ r, r rJ r® rì r r¯ rG Tr0 rF )
|
r rÀ ZNAr r3 r rL r r r )Zseriesr8 r9 r! r r" r" r# Ú test_cut_with_nullable_int64í s
|
ÿ r )bÚnumpyr re ZpandasrÀ r r r r r r r r r
|
r r r r r r r Zpandas._testingZ_testingr Zpandas.api.typesr r² Zpandas.core.reshape.tileÚcoreZreshaper r« r$ ÚmarkZ parametrizeÚlistr r: rA rE rP ra rj ro ÚminÚmaxr_ Ziinfor Z timedelta64ru rx rz Zeyer} 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å ræ r r6 Z assert_equalrè ré rï rù rû rü rþ r r r r r r r" r" r" r# Ú<module> s¨ H
|
|
þÿý ÿ"ÿû
|
|
|
ý
|
ýÿ
|
þ
|
þûÿóìþ2
|
þþ
|
" ýþ
|
þþ
|
|
þ þ÷þ
|
|
|
üþ
|
|
|
|
üþ
|
ÿ
|
üþ
|
ýýÿ òþ
|
üþþ
|
|
ýþ
|
|
*0þþ
|
,2þþ
|
|
|
