U
|
ý°dÐ? ã @ sZ d dl mZ d dlZd dlZd dlZd dlmZmZm Z m
|
Z
|
d dl m Z G dd dZdS )é )ÚtimezoneN)Ú DataFrameÚIndexÚSeriesÚ
|
date_rangec
|
@ s6 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Ze j
|
d dde j gddegddegdde j gg¡dd Zdd Zdd Zdd Zdd Zdd Zd d Zd d Zd d! Zd;d#d$Ze j
|
d%d&d'g¡e j
|
d(d)d*d"g¡e j
|
d+d)d*g¡e j
|
d,d-d.d/d0g¡d1d2 Zd3d4 Zd5d6 Zd7d8 Z d9d: Z d"S )<ÚTestDataFrameAlignc C s: t ddddd}t ddddd}ttj t|d¡|}ttj t|d¡|}| |¡\}}|jj|jjksrt |jj|jjkst |
|
d¡}| |¡\}}|jjt j ks°t |jjt j ksÂt |j|d d d
|
\}}|jjt j ksêt |jjt j ksüt |d j|d d
|
|jjt j ks"t |jjt j ks6t d S ) NZ2001é ÚHz
|
US/Eastern)ZperiodsÚfreqÚtzZ2Hé z
|
US/Centralr ©Úaxis) r r ÚnpÚrandomZrandnÚlenÚalignÚindexr ÚAssertionErrorZ
|
tz_convertr Úutc)ÚselfZidx1Zidx2Údf1Údf2Znew1Znew2Z df1_central© r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_align.pyÚtest_frame_align_aware s"
|
z)TestDataFrameAlign.test_frame_align_awarec s | |¡\}}|j|jk s t|j |dd\}}|sF|j|jksVtn|j|jk sVt|jd d
d d
f }|j |ddd\}}t |j|j¡ |j |j¡}|j |¡}|
|
|¡j }|dk ¡ sÆt|j |ddd \}}t |j|j¡ t |j|j¡ t |j|j¡ |jd d
d d
f ¡ }|j |d
|
d \}}t |j|j¡ t |j|j¡ |j |j¡}|j |¡}|
|
|¡j }|dk ¡ st|j |d d
|
d \}}t |j|j¡ |j |d d
|
d d\}}t |j|j¡ |j |jd d
df d d
|
d d d\}}t |jtg ¡ |j |jd d
df d d
|
d dd\}}t |jtg ¡ d} tjt| d$ |j |jdd d
f d dd W 5 Q R X |j}
|
ttt|
|
|
|
d |j dd \} } t | j|j¡ t | j|j¡ t| tsàt|j d
|
d\} } t | j|j¡ fdd|jD } t| |j|jd} t | | ¡ td
|
ddgdddgd}||d dk }tddggd
|
gd d gd} t || ¡ | |d dkd¡}tdddgdddgd} t || ¡ d S ) NF)Úcopyéûÿÿÿr r éÿÿÿÿ)r Ú
|
fill_valueÚright)Újoinr é r ÚinnerÚpad©r! r Úmethod©r! r r& r z)No axis named 2 for object type DataFrame©Úmatché ©r )Zbroadcast_axisc s i | ]
|
}| qS r r )Ú.0Úc©Úsr r Ú
|
<dictcomp>s s z7TestDataFrameAlign.test_align_float.<locals>.<dictcomp>©r Úcolumnsé r é ©ÚaÚbr6 r7 )r Z_mgrr ÚilocÚtmÚassert_index_equalr2 r r! Ú
|
differenceÚreindexÚvaluesÚallr r ÚpytestÚraisesÚ
|
ValueErrorr Úranger Ú
|
isinstancer Úassert_frame_equalÚwhere)r Ú float_frameZusing_copy_on_writeÚafÚbfÚotherZjoin_idxZdiff_aZ diff_a_valsÚmsgÚidxÚleftr ÚexpectedÚdfÚresultr r. r Útest_align_float+ s ÿ
|
ÿ
|
( ÿ z#TestDataFrameAlign.test_align_floatc C s@ t tddddgd}|j|dddd \}}t |j|j¡ d S )
|
Nr ÚAÚBÚCr1 r# r" r$ r% )r rB r r9 r: r2 )r Z int_framerI rG rH r r r Útest_align_int s z!TestDataFrameAlign.test_align_intc C s* |j |dddd\}}t |j|j¡ d S )Nr# r" r$ r% )r r9 r: r2 )r Zfloat_string_framerG rH r r r Útest_align_mixed_type s ÿ
|
z(TestDataFrameAlign.test_align_mixed_typec C sR t tddddgd}|j|jd d
df ddd dd \}}t |jtg ¡ d S ©
|
Nr rQ rR rS r1 r r# r" r' ©r rB r r8 r9 r: r r )r Zmixed_float_framerI rG rH r r r Útest_align_mixed_float s ÿ
|
z)TestDataFrameAlign.test_align_mixed_floatc C sR t tddddgd}|j|jd d
df ddd dd \}}t |jtg ¡ d S rV rW )r Zmixed_int_framerI rG rH r r r Útest_align_mixed_int s ÿ
|
z'TestDataFrameAlign.test_align_mixed_intz l_ordered,r_ordered,expectedTFc C s¶ t tjdddttd tjtd|d¡d d¡}t tjd ddttd
|
tjtd|d¡d d¡}| |¡\}}t
|
|j |st t
|
|j |s¢t t |j |j ¡ d S ) Nr4 Úint64©ZdtypeZaabbcaZcab)Zordered)rQ rR rR r Zbabca)r r Úaranger ÚlistZastypeÚpdZCategoricalDtypeZ set_indexr rC r r r9 r: )r Z l_orderedZ r_orderedrM Zdf_1Zdf_2Z aligned_1Z aligned_2r r r Útest_align_categorical¡ s, ÿþÿù ÿþÿù z)TestDataFrameAlign.test_align_categoricalc C s t jjtdtdtdgdd}ttddd}ttjddd |d
|
}ttjddd |d
|
}|j|d d \}}|j|d d \}}|} t
|
| |¡ t
|
| |¡ tddddtj tj gd |d
|
}
|
t
|
|
|
|¡ t
|
|
|
|¡ |j|d d \}}|j|d d \}}t jjtdtdtdgdd} tddddddddg| d
|
} t
|
| |¡ t
|
| |¡ tddddgd | d
|
}
|
t
|
|
|
|¡ t
|
|
|
|¡ d S )Nr* r )r6 r7 r- )Únamesr7 ©Únameé rZ r[ r+ rL )r! r r r" r4 é é é ) r^ Ú
|
MultiIndexÚ from_productrB r r r r\ r r9 rD Únan) r ZmidxrK r r Zres1lZres1rZres2lZres2rZexplÚexprZexp_idxr r r Útest_align_multiindex s6 ÿ ÿ z(TestDataFrameAlign.test_align_multiindexc C sà t dddgdddgdtdd}tdddgtd d
|
d }|j|d d \}}t dtjdtjdgdtjdtjdgdtdd}tddtjdtjgtdd
|
d }t ||¡ t ||¡ | |¡\}}t ||¡ t ||¡ d S )Nr" r r r5 ZACEr+ r* r3 ZABDÚx©r rb r r ZABCDE) r r] r r r ri r9 rD Úassert_series_equal)r rN r/ Zres1Zres2Zexp1Zexp2r r r Ú test_align_series_combinationså s "$þ z1TestDataFrameAlign.test_align_series_combinationsc C s t dddgdd}t ddgdd}tddg|dd}td t d
|
¡itj ||g¡d }tddgd |jdd}|j |d d \}}t
|
||¡ t
|
||¡ d S )Nr" r* r Úfoora ÚbarÚ
|
foo_seriesrm Úcolr4 r+ r r © r r r r r\ r^ rg rh r r r9 rD rn ©r Ú foo_indexÚ bar_indexÚseriesrN Z
|
expected_rZresult_lZresult_rr r r Ú7test_multiindex_align_to_series_with_common_index_levelù s þ zJTestDataFrameAlign.test_multiindex_align_to_series_with_common_index_levelc C s° t dddgdd}t ddgdd}tddddgt ddddgdddd }td
|
t d ¡itj ||g¡d }tddgd |jdd }|j |d d\}}t
|
||¡ t
|
||¡ d S )Nr" r* r rp ra rq r3 rr rm rs r4 r+ r r rt ru r r r ÚGtest_multiindex_align_to_series_with_common_index_level_missing_in_left
|
s
|
ÿ þ zZTestDataFrameAlign.test_multiindex_align_to_series_with_common_index_level_missing_in_leftc C s´ t dddgdd}t ddddgdd}tddgt ddgdddd }td
|
t d ¡itj ||g¡d }tddtjtjgd |j dd }|j
|
|d d\}}t ||¡ t ||¡ d S )Nr" r* r rp ra r3 rq rr rm rs rc r+ r r ©r r r r r\ r^ rg rh ri r r r9 rD rn ru r r r ÚHtest_multiindex_align_to_series_with_common_index_level_missing_in_right s þ ÿ z[TestDataFrameAlign.test_multiindex_align_to_series_with_common_index_level_missing_in_rightc C s² t dddgdd}t dddgdd}tdddgt dddgdddd }td
|
t d ¡itj ||g¡d }tdtjdgd |j dd }|j
|
|d d\}}t ||¡ t ||¡ d S )Nr" r* r rp ra r3 rq rr rm rs rf r+ r r r{ ru r r r ÚGtest_multiindex_align_to_series_with_common_index_level_missing_in_both0 s ÿ þ zZTestDataFrameAlign.test_multiindex_align_to_series_with_common_index_level_missing_in_bothc C s° t dddgdd}t ddgdd}tddg|dd}tt d ¡ d
|
d¡tj ||g¡d }d d d g|_ tddgd |j
|
dd}|j |dd\}}t ||¡ t |j |j ¡ d S )Nr" r* r rp ra rq rr rm é r4 r+ ZcfooZcbarr r )r r r r r\ Úreshaper^ rg rh r2 r r r9 rn r: )r rv rw rx rN rM Z result_leftZ result_rightr r r ÚGtest_multiindex_align_to_series_with_common_index_level_non_unique_colsC s þ zZTestDataFrameAlign.test_multiindex_align_to_series_with_common_index_level_non_unique_colsc C sJ t t d¡ d¡}tt d¡}tjtdd | |¡ W 5 Q R X d S )Né2 )é
|
r r z axis=0 or 1r( ) r r r\ r r r? r@ rA r )r rN rx r r r Ú)test_missing_axis_specification_exceptionU s z<TestDataFrameAlign.test_missing_axis_specification_exceptionNc C sÜ |j ||||||d\}} d\}
|
} || } } |d ks<|dkrf|jj|j|d}
|
| j|
|
d} | j|
|
d} |d ksv|dkr |jj|j|d} | j| d} | j| d} | j|||d} | j|||d} t || ¡ t | | ¡ d S ) N)r r! r& ÚlimitÚ fill_axis)NNr )Úhowr+ r" )r2 )r r& r )r r r! r< r2 Zfillnar9 rD )r r6 r7 r r
r r& r ZaaÚabZ
|
join_indexZ join_columnsZeaZebr r r Ú _check_align\ s, ÿ
|
|
z TestDataFrameAlign._check_alignÚmethr$ ZbfillÚaxr r" Úfaxr r# ÚouterrL r c C s |}| |||||¡ d S )N)Ú_check_align_fill)r r r r r rF rN r r r Útest_align_fill_methodt s z)TestDataFrameAlign.test_align_fill_methodc C sþ |j dd
d d
f }|j dd
dd
f }|j d d
d d
f }| j||||||d | j||||||dd | j||||||d | j||||||dd | j||||||d | j||||||dd | j||||||d | j||||||dd d S ) Nr r3 r r* r4 )r r
r r& r" )r r
r r& r )r8 r ) r ÚframeÚkindr r r rL r Úemptyr r r r | sV ÿ ÿ ÿ ÿz$TestDataFrameAlign._check_align_fillc C sR t dddgi}tdgdd}| ¡ }|j|dd\}}d|jd< t ||¡ d S )Nr r" r* ra r éd )r r r r r8 r9 rn )r rN ÚserrM rO rI r r r Ú test_align_series_check_copy s
|
z/TestDataFrameAlign.test_align_series_check_copyc C sb t dddgi}tddg}|j|dd\}}t ||¡ t ||¡ ||k sRt||k s^td S )Nr6 r" r* r r3 r r ©r r r r9 rD rn r ©r rN r rO Zresult2r r r Ú%test_align_identical_different_object¡ s z8TestDataFrameAlign.test_align_identical_different_objectc C sf t dddgi}tdgdgd}|j|dd\}}t ||¡ t ||¡ ||k sVt||k sbtd S )Nr6 r" r* r+ r r r r r r Ú-test_align_identical_different_object_columns« s z@TestDataFrameAlign.test_align_identical_different_object_columns)N) Ú__name__Ú
|
__module__Ú __qualname__r rP rT rU rX rY r? ÚmarkZ parametrizer^ ZCategoricalIndexr r_ rk ro ry rz r| r} r r r r r r r r r r r r r sB X
|
|
üþ
|
#
|
|
r )Údatetimer Únumpyr r? Zpandasr^ r r r r Zpandas._testingZ_testingr9 r r r r r Ú<module> s
|
