U
|
ý°d». ã @ sT d dl Zd dlZd dlZd dlmZmZmZ d dlm Z
|
dddZ G dd dZ dS )é N)Ú DataFrameÚSeriesÚ
|
date_rangec C s& |d k rt | |¡ | j t| d S )N)ÚtmÚassert_frame_equalZdtypesÚstr)ÚresultÚexpected© r
|
ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/test_nonunique_indexes.pyÚcheck s r c @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zd d Zd d Zd!S )"Ú TestDataFrameNonuniqueIndexesc C sJ t j dd¡}ttd}t|ddgd}||_t||d}t|| d S )Né é ÚA©Úcolumns)ÚnpÚrandomÚrandnÚlistÚranger r r )ÚselfZarrÚidxÚdfr r
|
r
|
r Ú.test_setattr_columns_vs_construct_with_columns s zLTestDataFrameNonuniqueIndexes.test_setattr_columns_vs_construct_with_columnsc C s~ t dddd}tddddgddddgddddggd d d d gd
|
}||_tddddgddddgddddgg|d
|
}t|| d S ) NZ20130101é zQ-NOV)ZperiodsÚfreqé é r r Úar )r r r r )r r r r r
|
r
|
r Ú;test_setattr_columns_vs_construct_with_columns_datetimeindx s
|
ÿ*zYTestDataFrameNonuniqueIndexes.test_setattr_columns_vs_construct_with_columns_datetimeindxc C sT t ddddgddddgddddggddddgd}d |d
|
< t ddddd gddddd gddddd ggddddd
|
gd}t|| tjtd d " | d dtt|jd ¡ W 5 Q R X d|d< t ddddd dgddddd dgddddd dggddddd
|
dgd}t|| d|d< t ddddd dgddddd dgddddd dggddddd
|
dgd}t|| d|d< |d= t dddd dgdddd dgdddd dggdddd
|
dgd}t|| |d= t ddd dgddd dgddd dggddd
|
dgd}t|| | ¡ }t ddd dgddd dgddd dggddd
|
dgd}t|| | ddd¡ t dddd dgdddd dgdddd dggdddd
|
dgd}t|| tjtdd | ddd¡ W 5 Q R X |jddddd t ddddd dgddddd dgddddd dggddddd
|
dgd}t|| |d= t ddd dgddd dgddd dggddd
|
dgd}t
|
||¡ d S )Nr r r r ÚfooÚbarÚhellor ZbahÚstringzLength of value©Úmatchr Z AnotherColumnÚfoo2r Znew_colg @z cannot insertg @T)Zallow_duplicates) r r ÚpytestÚraisesÚ
|
ValueErrorÚinsertr ÚlenÚindexZ _consolidater r ©r r r r
|
r
|
r Ú"test_insert_with_duplicate_columns( s
|
þ& þ
|
&,þ
|
,þ
|
& þ
|
|
þ
|
|
þ
|
& þ
|
ýú
|
|
þz@TestDataFrameNonuniqueIndexes.test_insert_with_duplicate_columnsc C sD t ddddgddddgddddggdddd gd
|
}t| d |d < t ddddd gddddd gddddd ggdddd d gd
|
}t|| |d }t ddgddgddggddgd
|
}t|| d |d< t d dd dd gd dd dd gd dd dd ggdddd d gd
|
}t|| |d= t ddd gddd gddd ggdd d gd
|
}t|| d S )Nr ç ð?r ç @r ç @r" r# r$ r g @r( r% )r r )r r r r r
|
r
|
r Útest_dup_across_dtypes s:
|
þ& þ
|
"
|
ý ú
|
ÿz4TestDataFrameNonuniqueIndexes.test_dup_across_dtypesc C s t tj dd¡dddddgdd dgd
|
}|jt td ¡fD ]J}| ¡ }t |j
|
|jd }t ||d d dd dgd}||d< t || q<d S )Nr r r ÚbÚcÚdÚer ÚB©r. r Zedcba©r. )r r9 r ) r r r r r. ÚpdZIndexr Úcopyr Úvaluesr )r r r. Zthis_dfZ expected_serZ expected_dfr
|
r
|
r Útest_column_dups_indexes¤ s ý þz6TestDataFrameNonuniqueIndexes.test_column_dups_indexesc C s t tj dd¡ddgd}t dtdddgd}d|d< t|| t tj dd¡ddgd}t dtdddgd}d|d< t|| d S )Nr r Úthatr r1 r: r )r r r r r r Zrandr/ r
|
r
|
r Ú+test_changing_dtypes_with_duplicate_columnsµ s
|
zITestDataFrameNonuniqueIndexes.test_changing_dtypes_with_duplicate_columnsc C sÄ t ddgdtjgddgddggddgd}t ddgddgdtjgdd ggddgd}d
|
}tjt|d ||k W 5 Q R X | |¡}||k}t d d gd d gd d gd d ggddgd}t ||¡ d S )Nr r r r r r9 r r r zQCan only compare identically-labeled \(both index and columns\) DataFrame objectsr& FT) r r Únanr) r* r+ Z reindex_liker r )r Zdf1Zdf2ÚmsgZdf1rr r r
|
r
|
r Ú test_dup_columns_comparisonsÅ s **ÿ
|
þz:TestDataFrameNonuniqueIndexes.test_dup_columns_comparisonsc C s t tdddgdddgdtddddgddddgdtddddgddddgdd}tj|d |d
|
|d gd d }|d d
|
d g }t|| d S ) NTFr r5 r6 r; r7 )ÚoneZtwoÚthreerE rF r ©Zaxis)r r r< Úconcatr )r Zdfboolr r r
|
r
|
r Útest_mixed_column_selectionÚ s ýÿ z9TestDataFrameNonuniqueIndexes.test_mixed_column_selectionc C s® t t d¡ dd¡dddddgdd d
|
d d gd }|dd
|
dg ¡ }|jdddg }t t d¡ dd¡dddddgdd d
|
d d gd }|dd
|
dg }|jdddg }t|| d S )Ng 9@r r r5 r6 r7 r8 r r9 ÚCÚDÚEr: )r r ÚarangeÚreshaper= Úlocr )r r Úzr r r
|
r
|
r Útest_multi_axis_dupsè s ý ýz2TestDataFrameNonuniqueIndexes.test_multi_axis_dupsc C s¢ t ddggddgd}ddg|_t| t ddggddgd}t ||¡ t dddggdddgd}dddg|_t| t dddggdddgd}t ||¡ d S )Nr r r r úa.1r r5 ©r r r r r r/ r
|
r
|
r Útest_columns_with_dupsü s
|
z4TestDataFrameNonuniqueIndexes.test_columns_with_dupsc C sN t ddggddgd}ddg|_t| t ddggddgd}t ||¡ d S )Nr r r r r5 rS r/ r
|
r
|
r Útest_columns_with_dup_index s
|
|
z9TestDataFrameNonuniqueIndexes.test_columns_with_dup_indexc C sl t dddddddggddd d d
|
d d gd }td |_t| t dddddddggtd d }t ||¡ d S )Nr r r1 r2 r3 r" r# r r5 r7 r6 r ZABCDEFG)r r r r r r r/ r
|
r
|
r Útest_multi_dtype s þ
|
ÿz.TestDataFrameNonuniqueIndexes.test_multi_dtypec C sb t ddddggddddgd}dddd g|_t| t ddddggdddd gd}t ||¡ d S )
|
Nr r r" r# r r rR za.2za.3rS r/ r
|
r
|
r Útest_multi_dtype2! s
|
z/TestDataFrameNonuniqueIndexes.test_multi_dtype2c C sä t tj dd¡dd}t tj dd¡ d¡}t d|j|jd}t d|j|jd}t t d ¡|j|jd}tj |||||gd
|
d }|sºt
|
|j j t
|
|jks t t
|
|j jt
|
|jksºt tt
|
|jD ]}|jd d
|f qÈd S ) Né
|
r Úfloat64)ZdtypeÚint64Tr: r" Z20010101r rG )r r r r Zastyper. r r< Ú TimestamprH r- Z_mgrZblknosÚAssertionErrorZblklocsr Úiloc) r Zusing_array_managerZdf_floatZdf_intZdf_boolZ df_objectZdf_dtr Úir
|
r
|
r Útest_dups_across_blocks( s ÿz5TestDataFrameNonuniqueIndexes.test_dups_across_blocksc C sJ dddgdddgg}t |dddgd }t |}dddg|_t ||¡ d S )
|
Nr éÿÿÿÿr2 r éþÿÿÿr3 r r9 r )r r r r )r ÚvalsÚrsZxpr
|
r
|
r Ú test_dup_columns_across_dtype; s
|
z;TestDataFrameNonuniqueIndexes.test_dup_columns_across_dtypec C s
|
d }d}t t d¡ dd¡j}td|_|jd d
df }tj ||d d|jd d
df< W 5 Q R X t
|
|jd d
df |¡ t t d¡ dd¡j}dt dt dg|_|jd d
df }tj ||d d|jd d
df< W 5 Q R X t
|
|jd d
df |¡ d S ) Nz&will attempt to set the values inplaceé r ZAAAr r& r r ) r r rM rN ÚTr r r] r Zassert_produces_warningZassert_series_equalÚfloatr )r ÚwarnrC r r r
|
r
|
r Útest_set_value_by_indexC s
|
z5TestDataFrameNonuniqueIndexes.test_set_value_by_indexN)Ú__name__Ú
|
__module__Ú __qualname__r r! r0 r4 r? rA rD rI rQ rT rU rV rW r_ rd ri r
|
r
|
r
|
r r s
|
W% r )N) Únumpyr r) Zpandasr< r r r Zpandas._testingZ_testingr r r r
|
r
|
r
|
r Ú<module> s
|
|
