U
|
ý°dÑ# ã @ sN d dl Zd dlmZ d dlZd dlmZmZmZ d dl m
|
Z G dd dZ dS )é N)ÚCategoricalDtype)Ú CategoricalÚ DataFrameÚSeriesc @ sd e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dS )ÚTestCategoricalConcatc C s t tjddd dd¡dddgd }t tjd
|
dd d d ¡ddgd }d d dd ddd g}tt||d< tj||fdd|d}t dddddddd dddddg ddd dddtjtjtjtjtjtjtjg d d dd d
|
d ddd d dd dg d gd | d }|d |d j
|
¡|d< t ||¡ d S ) Né Úint64©Údtypeé é ÚaÚbÚc)Úcolumnsé é é ZoneZtwoÚhr T)ÚaxisÚ ignore_indexÚsorté é é é é é
|
é é é é é é )r r r r ) r ÚnpÚarangeZreshaper r ÚpdÚconcatÚnanÚastyper
|
ÚtmÚassert_frame_equal)Úselfr Údf1Údf2Z
|
cat_valuesÚresÚexp© r1 úcd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/reshape/concat/test_categorical.pyÚtest_categorical_concat s< ÿ" ó îÿz-TestCategoricalConcat.test_categorical_concatc C sÊ dddg}t dddg}tdddg}tddd g}tjt|||gd|d
|
}|jd k}td d d g|d}t ||¡ |jdk}td d d g|d}t ||¡ |jdk}td d d g|d}t ||¡ d S )NÚcatÚobjÚnumr r r r r r )r ÚkeysÚobjectFT©Úindexr Úcategory)r r r& r' Zdtypesr* Úassert_series_equal)r, r: r4 r5 r6 ÚdfÚresultÚexpectedr1 r1 r2 Ú test_categorical_concat_dtypes4 s
|
|
|
|
z4TestCategoricalConcat.test_categorical_concat_dtypesc C sÖ dddddg}t dtjddg|dd}t dtjddg|dd}t dtjddg|dd}tj|||gdd}tjddddg|d}tddtjtjgtjddtjgtjtjddgd dddg|d
|
}t ||¡ d S ) Nr r r r r ©Ú
|
categoriesr9 ©r )r r r )r r: ) r r& ÚCategoricalIndexr' r r$ r( r* r+ )r, rB r r r r> Zexp_idxr0 r1 r1 r2 Ú test_concat_categoricalindexH s ýù z2TestCategoricalConcat.test_concat_categoricalindexc C s8 t tddd}t tddd}t td}tj||gdd}t ||¡ t tddd}tj||gdd}t ||¡ t tdd d
|
d d d
|
d gdd }t ||g¡}t ||¡ t tjd dd}t td}t|| t
|
td¡d}t ||g¡}tt ||g¡t ||g¡ t
|
td¡d}t ||¡ d S )NÚabcr; r ZabdZabcabdT©r Zabcabcr r r )r: r
|
r r ÚaabbcaÚcab©ÚAÚB) r Úlistr& r' r* r< r$ r% r r) r r+ )r, ÚsÚs2r0 r/ r r r. r1 r1 r2 Ú test_categorical_concat_preserve^ s* þÿz6TestCategoricalConcat.test_categorical_concat_preservec C s t tjddd}t td}t|| ttd¡d d¡}t ||g¡}tt ||g¡t ||g¡ ttd¡d d¡}t
|
||¡ t|t |tdd d d¡}t ||g¡}t |j |j t¡d
|
d |j |j t¡d
|
d g¡}t
|
||¡ d S ) Nr r r rH rI rJ rL ZaberA r rC )r r$ r% rM r r) r Z set_indexr& r' r* r+ r Zset_axisr: r8 )r, r r r. r> r? Zdf3r1 r1 r2 Ú test_categorical_index_preserver} s8 ÿþ þÿû ÿþþÿz6TestCategoricalConcat.test_categorical_index_preserverc C sj t tjdddd}t ddgdd}tj||gd d
|
}t tjddd tjd dd ddg}t ||¡ d S ) Nz
|
2017-01-01r z
|
US/Pacific)ZperiodsÚtzr r r; r TrG )rR z
|
2017-01-02)r r& Z
|
date_ranger' Ú Timestampr* r< )r, r r r> r? r1 r1 r2 Útest_concat_categorical_tz s üÿz0TestCategoricalConcat.test_concat_categorical_tzc C s t tdddgddd}tddd gddd
|
gd d }tj||gdd }t tdddtjgddtddtjd gddd}t ||¡ d S )Nr r r r; rK )r
|
Únamer r r r rL )r: rU rC r ÚfloatrJ )r r r& r' r$ r( r* Ú assert_equal)r, r= Zserr> r? r1 r1 r2 Ú!test_concat_categorical_unchanged© s þÿz7TestCategoricalConcat.test_concat_categorical_unchangedc C sè t ddddddgtdd}t|d |d
|
< |d
|
j d d d g¡ |dd
}|dd
}t |d
|
jj|d
|
jj¡ t |d
|
jj|d
|
jj¡ t ||g¡}t |d
|
jj|d
|
jj¡ |
|
|¡}t |d
|
jj|d
|
jj¡ d S )Nr r r r r! r Zabbaae)ÚidÚ raw_graderZ ZgradeÚer r r ) r rM r r4 Zset_categoriesr* Zassert_index_equalrB r& r' Ú_append)r, r= r- r. ZdfxZdfar1 r1 r2 Ú test_categorical_concat_gh7864¸ s
|
z4TestCategoricalConcat.test_categorical_concat_gh7864c C sÜ t dddgitddgd}t dddgitddgd}t ||g¡}t dddddgiddddgd}t ||¡ tddgtddgd}tddgtddgd}t ||g¡}tddddgddddgd}t ||¡ d S ) NZfoor r Úbarr9 r r Zbaz)r r r& r' r* rW r )r, r r r/ r0 r1 r1 r2 Ú test_categorical_index_upcastË s z3TestCategoricalConcat.test_categorical_index_upcastc C s¢ t ddddgi}t dddgtdddg d¡d}tj||gdd }tdg}t ddddddgtjd
|
d
|
d
|
d d d g|d dd ddd ddgd }t ||¡ d S )NÚf1r r r r r; )r` Úf2T)r éÿÿÿÿr r r9 )
|
r r r) r& r' r r Z
|
from_codesr* r+ )r, r- r. r> r
|
r? r1 r1 r2 Ú'test_categorical_missing_from_one_frameà s $
|
þûz=TestCategoricalConcat.test_categorical_missing_from_one_framec C sª t jddgddgdd}t jddgddgdd}t jddddgddgdd}tdddgi|d}tdd d
|
gi|d}t ||f¡}tdddd d
|
gi|d}t ||¡ d S ) Nr r F)rB ZorderedrK r r r9 r r )r& rD r r' r* r+ )r, Zc1Úc2Úc3r- r. r> r? r1 r1 r2 Ú7test_concat_categorical_same_categories_different_orderï s
|
ÿzMTestCategoricalConcat.test_concat_categorical_same_categories_different_orderN)Ú__name__Ú
|
__module__Ú __qualname__r3 r@ rE rP rQ rT rX r] r_ rc rf r1 r1 r1 r2 r s % r ) Únumpyr$ Zpandas.core.dtypes.dtypesr Zpandasr& r r r Zpandas._testingZ_testingr* r r1 r1 r1 r2 Ú<module> s
|
|
