U
|
ý°d# ã @ sN d dl Zd dlZd dlZd dlmZmZmZmZ d dl m
|
Z G dd dZ dS )é N)Ú DataFrameÚIndexÚSeriesÚconcatc @ sª e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z e j dddg¡e j dddg¡e j dddg¡d d Zd d Zd d! Zd"d# Zd$S )%ÚTestDataFrameConcatc C s t t d¡ddgtjd}t t d¡tjd}t||fddj}tt d¡gd t d
|
¡gd ddd dgd }t
|
||¡ d S ) N)é
|
é ÚfooÚbar)ÚdataÚcolumnsÚdtype)r r é ©ÚaxisÚfloat64r Úfloat32r ©Úindex) r ÚnpÚonesr r r Zdtypesr r ÚtmÚassert_series_equal©ÚselfÚdf1Údf2ÚresultsÚexpected© r úad:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/reshape/concat/test_dataframe.pyÚ"test_concat_multiple_frames_dtypes s
|
þz6TestDataFrameConcat.test_concat_multiple_frames_dtypesc C sz t t d¡tdd}t t d¡d tdd}t||fddgd}t d d d
|
d
|
d
|
d d d d
|
d
|
d
|
d d }t ||¡ d S ) N)r r ZAB©r )é r r )ÚbeeÚbah)r$ Úboo©Úkeysç ð?g @))r$ r% r )r$ r% r )r$ r& r )r$ r& r )r$ r& r )ÚAÚB)r r r Úlistr r Úassert_frame_equalr r r r Útest_concat_tuple_keys s& ûûøÿz*TestDataFrameConcat.test_concat_tuple_keysc C s t ddgddgd}tddgdd }t||g|d
|
}t ddddgddddgdtjjddgd dgfdd gd d }t ||¡ tddgd d }t||g|dgd}t ||¡ t||g|d
|
}t ddddgddddgdtjjddgd dgfd d gd d }t ||¡ d S )Nr r ç¹?çÉ?)r r
|
ÚaÚbZbaz©Únamer' r ©Únamesr ©r( r6 )r r r ÚpdÚ
|
MultiIndexÚ from_productr r- ) r Údfr Zconcatted_named_from_keysZexpected_namedZ index_no_nameZconcatted_named_from_namesZconcatted_unnamedZexpected_unnamedr r r Útest_concat_named_keys3 s" þ þz*TestDataFrameConcat.test_concat_named_keysc C s t dddgitdd}t dddgitdd}t dddddgidd dd gd}t||gd
|
d }t ||¡ t||gd d }t ||¡ t||gdd }t ||¡ t ddgddggdd gddgd }t||gdd }t ||¡ t||gd d } t | |¡ tddg}
|
tddg} tddddgdd dd gd} t|
|
| gd
|
d } t | | ¡ t|
|
| gd d }t || ¡ t|
|
| gdd }t || ¡ t ddgddggdd gdd gd }t|
|
| gdd }t ||¡ t|
|
| gd d }t ||¡ tjt dd t|
|
| gdd W 5 Q R X d S )Nr* r/ r0 r r g333333Ó?gÙ?r r r r Úrows)r r r z No axis named)ÚmatchZ something)
|
r Úranger r r- r r ÚpytestZraisesÚ
|
ValueError)r r r Zexpected_indexZconcatted_indexZ concatted_rowZ concatted_0Zexpected_columnsZconcatted_columnsZ concatted_1Zseries1Zseries2Zexpected_index_seriesZconcatted_index_seriesZconcatted_row_seriesZconcatted_0_seriesZexpected_columns_seriesZconcatted_columns_seriesZconcatted_1_seriesr r r Útest_concat_axis_parameterI sL ÿ ÿ z.TestDataFrameConcat.test_concat_axis_parameterc C s¨ t dtdidtjjdddgddd ggd
|
d gd d }t|jd d
d d
f |jdd
d d
f f}t ddd
|
ddgidtjjddddgd
|
d gd d }t ||¡ d S )NÚcolé Úint32ÚA0ÚA1ÚA2ÚB0ÚB1ÚB2r r r5 )r r éþÿÿÿr é é )rF rI )rF rJ )rH rJ )rH rK )
|
r r? r8 r9 r: r ZilocÚ from_tuplesr r- ©r r; Úresultr r r r Útest_concat_numerical_names s$
|
ÿü 0
|
ÿýz/TestDataFrameConcat.test_concat_numerical_namesc C s` t ddig}t||gdd}| d¡}t t ddg¡ dd¡ddgd d¡}t ||¡ d S )Nr1 r2 r r Úcategoryr r" )r r Úastyper ÚarrayZreshaper r- rP r r r Útest_concat_astype_dup_col s
|
ÿþz.TestDataFrameConcat.test_concat_astype_dup_colc C s~ t dtdddgtdddgdd d
|
i}t dtd d gtddgdd d
|
i}t||gdd dg|d}t|jddgksztd S )NÚvaluer r r# r1 r2 ÚcÚidr3 r rM rN Út1Út2)r r( Úsort)rZ rW )r[ rW )r r r r r, r ÚAssertionError)r r\ rZ r[ rQ r r r Ú test_concat_dataframe_keys_bug£ s "ÿ$z2TestDataFrameConcat.test_concat_dataframe_keys_bugc C s\ t tddddgdd}t tdddgdd}t||g}t| d¡|g}t ||¡ d S )NTFÚbool)r r r Úint64)r r r rT r r- ©r r r rQ r r r r Útest_concat_bool_with_int s
|
z-TestDataFrameConcat.test_concat_bool_with_intc C s dddg}dddg}t ||d}t|gdgddgd}tjjdg|gddgd }t ||d}t ||¡ t |jj d t
|
ddgdd
|
¡ d S ) Nr r# r )r r r* ZIDÚdater7 r5 r3 ) r r r8 r9 r: r r- Zassert_index_equalr Úlevelsr )r r r r; rQ Úmir r r r Ú)test_concat_duplicates_in_index_with_keys· s
|
|
z=TestDataFrameConcat.test_concat_duplicates_in_index_with_keysÚ ignore_indexTFÚorderÚCÚFr r r c C s` t tjdtj|d}t|gd ||dd}|s\| ¡ D ]"}| ¡ D ]}t ||¡rDtqDq8d S )N)i' éÈ )r rh é T)r rg Úcopy)r r Úzerosr r Z_iter_column_arraysZ shares_memoryr] ) r r rh rg Zusing_copy_on_writer; ÚresZarrZarr2r r r Útest_concat_copiesà s z&TestDataFrameConcat.test_concat_copiesc C sf t dgdgdd}t ddgi}t||gdddd}t d tjgddgd tjgd
|
}t ||¡ d S ) Nr r ©r* r+ r r* éd TÚouter©rg Újoinr\ r) )r r* r+ )r r r Únanr r- ra r r r Útest_outer_sort_columnsÑ s
|
z+TestDataFrameConcat.test_outer_sort_columnsc C s^ t dgdgdd}t dgdd}t||gdddd }t ddgddgd
|
}t ||¡ d S ) Nr r rq rr r ©r* r TÚinnerrt ©r r* ©r r r r- ra r r r Útest_inner_sort_columnsÙ s
|
z+TestDataFrameConcat.test_inner_sort_columnsc C sD t dgdd}t|gdddd}t dgdgd}t ||¡ d S )Nrr r rx Try rt rz r{ )r r rQ r r r r Útest_sort_columns_one_dfá s z,TestDataFrameConcat.test_sort_columns_one_dfN)Ú__name__Ú
|
__module__Ú __qualname__r! r. r< rB rR rV r^ rb rf r@ ÚmarkZ parametrizerp rw r| r} r r r r r s :
|
|
r ) Únumpyr r@ Zpandasr8 r r r r Zpandas._testingZ_testingr r r r r r Ú<module> s
|
|
