U
|
ý°d< ã @ sV d dl Zd dlZd dlmZ d dlZd dlmZmZm Z d dl
|
m Z G dd dZ dS )é N)Úunion_categoricals)Ú CategoricalÚCategoricalIndexÚSeriesc @ sÖ e Zd Zej dedededfdddgddd gdddddd gfdd
|
dgdd d gdd
|
ddd d gfd d ejd gd ejdgd d ejd d ejdgfe
|
dd¡e
|
dd¡e
|
dd¡fe j
|
dddde j
|
dddde j
|
ddddfe dd¡e dd¡e dd¡fg¡ej de e eg¡dd Zdd Zdd Zd d Zd d Zd d! Zej d"g d#gg¡d$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zd.d/ Zd0d1 Zd2d3 Z d4d5 Z d6d7 Z d8d9 Z d:d; Z d<d= Z!d>d? Z"d@dA Z#dBS )CÚTestUnionCategoricalsza, b, combinedÚabcZabdZabcabdr é é é é ç333333ó?g333333 @ÚbÚaÚcz
|
2014-01-01z
|
2014-01-05z
|
2014-01-06z
|
2014-01-07z
|
US/Central)ÚtzÚboxc C s4 t |t||t|g}t|}t ||¡ d S ©N©r r ÚtmÚassert_categorical_equal)Úselfr r Zcombinedr ÚresultÚexpected© r úcd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/reshape/test_union_categoricals.pyÚtest_union_categorical s z,TestUnionCategoricals.test_union_categoricalc C s\ t dddg}t dddg}t||g}t ddddddgddddddgd}t ||¡ d S )NÚxÚyÚzr r r ©Ú
|
categories©r r r r ©r ÚsÚs2r r r r r Ú)test_union_categorical_ordered_appearance3 s ÿz?TestUnionCategoricals.test_union_categorical_ordered_appearancec C sX t dddgdd}t dddgdd}t||g}t ddddddgdd}t ||¡ d S )Nr r r T©Úorderedr! r" r r r Ú#test_union_categorical_ordered_true= s
|
z9TestUnionCategoricals.test_union_categorical_ordered_truec C sJ t dddg}t dddg}d}tjt|d t||g W 5 Q R X d S )Nr r r r
|
r z$dtype of categories must be the same©Úmatch)r ÚpytestÚraisesÚ TypeErrorr )r r# r$ Úmsgr r r Ú"test_union_categorical_match_typesD s
|
z8TestUnionCategoricals.test_union_categorical_match_typesc C s* d}t jt|d tg W 5 Q R X d S )NzNo Categoricals to unionr) )r+ r, Ú
|
ValueErrorr )r r. r r r Ú test_union_categorical_emptyL s z2TestUnionCategoricals.test_union_categorical_emptyc C s t tddtjgtddtjgg}tddtjddtjg}t ||¡ t tddgtddtjgg}tddddtjg}t ||¡ t d¡t d¡tjg}tjt d¡t d¡g}t t|t|g}t|| t d¡t d¡t d¡gd }t ||¡ t ttj tjtjgt
|
d
|
td gg}ttjtjd g}t ||¡ t ttjtjgttjtjgg}ttjtjtjtjg}t ||¡ d S ) Nr r r
|
ÚAÚBz
|
2011-01-01z
|
2011-03-01z
|
2011-02-01r ©ZdtypeÚX) r r ÚnpÚnanr r ÚpdÚ TimestampZNaTÚarrayÚobject)r ÚresÚexpZval1Zval2r r r Útest_union_categoricals_nanQ sB ÿ ÿ ýþ þÿ ÿz1TestUnionCategoricals.test_union_categoricals_nanÚvalÚ1c C s, t tg t|g}t|}t ||¡ d S r r )r r? r< r= r r r Ú test_union_categoricals_empty} s z3TestUnionCategoricals.test_union_categoricals_emptyc C s| t ddddgddddgd}t dddtjgddddgd}t||g}t dddddddtjgddddgd}t ||¡ d S )Nr r r
|
r r ©r r6 r7 r r r ©r Úc1Úc2r< r= r r r Ú$test_union_categorical_same_category s
|
&z:TestUnionCategoricals.test_union_categorical_same_categoryc C sj t dddgdddgd}t dddgdddgd}t||g}t ddddddgdddgd}t ||¡ d S )Nr r r r r! rC r r r Ú(test_union_categorical_same_category_str s
|
z>TestUnionCategoricals.test_union_categorical_same_category_strc C sj t dddgdddgd}t dddgdddgd}t||g}t ddddddgdddgd}t ||¡ d S )Nr r r r r! ©r rD rE r r r r r Ú6test_union_categorical_same_categories_different_order s ÿzLTestUnionCategoricals.test_union_categorical_same_categories_different_orderc C s8 t dddgdd}t dddgdd}d}tjt|d t||g W 5 Q R X t||g}t ddddddgdd}t ||¡ t dddtjgdd}t ddgdddgdd }t||g}t dddtjddgdd}t ||¡ t dddgdd}t dddgdddgdd }d
|
}tjt|d t||g W 5 Q R X d S ) Nr r r
|
Tr& Fú$Categorical.ordered must be the samer) ©r r' ú>to union ordered Categoricals, all categories must be the same) r r+ r, r- r r r r6 r7 )r rD rE r. r< r= r r r Ú test_union_categoricals_ordered s$ z5TestUnionCategoricals.test_union_categoricals_orderedc C s t dddgdd}t dddgdd}t||gdd}t ddddddg}t ||¡ d}tjt|d t||gdd W 5 Q R X t||gdd}t ddddddg}t ||¡ t||gdd}t ddddddgdddgdd
|
}t ||¡ t dddtjgdd}t ddgdddgdd
|
}t||gdd}t dddtjddg}t ||¡ t dddgdd}t dddgdddgdd
|
}t||gdd}t ddddddg}t ||¡ t||gddd }t ddddddgdddgd }t ||¡ t dddgdd}t d ddgdd}t||gdd}t dddd ddg}t ||¡ d}tjt|d t||gdd W 5 Q R X tjt|d t||g W 5 Q R X d S )Nr r r
|
Tr& F)Ú ignore_orderrJ r) rK )rN Úsort_categoriesr r é é rL ) r r r r r+ r, r- r6 r7 )r rD rE r< r= r. r r r r r Ú$test_union_categoricals_ignore_order· sJ z:TestUnionCategoricals.test_union_categoricals_ignore_orderc C s° t dddg}t dddg}t||gdd}t ddddddgddddddgd }t ||¡ t ddgdddgd }t ddgdddgd }t||gdd}t ddddgdddgd }t ||¡ t ddgdddgd }t ddgdddgd }t||gdd}t ddddgdddgd }t ||¡ t ddgdddgd }t ddgdddgd }t||gdd}t ddddgdddgd }t ||¡ t dtjg}t tjdg}t||gdd}t dtjtjdgddgd }t ||¡ t tjg}t tjg}t||gdd}t tjtjg}t ||¡ t g }t g }t||gdd}t g }t ||¡ t ddgdddgdd
|
}t ddgdddgdd
|
}d }tjt|d t||gdd W 5 Q R X d S ) Nr r r r r r T©rO r rK z9Cannot use sort_categories=True with ordered Categoricalsr) ) r r r r r6 r7 r+ r, r- ©r rD rE r r r. r r r Ú test_union_categoricals_sortë sV ÿ z2TestUnionCategoricals.test_union_categoricals_sortc C s` t dddg}t dddg}t||gdd}t ddddddgddddddgd }t ||¡ d S )
|
Nr r r r r r FrS r r! rH r r r Ú"test_union_categoricals_sort_false! s ÿz8TestUnionCategoricals.test_union_categoricals_sort_falsec C sf t ddgdddgd}t ddgdddgd}t||gdd}t ddddgdddgd}t ||¡ d S ©Nr r r r FrS r! rH r r r Ú+test_union_categoricals_sort_false_fastpath+ s
|
zATestUnionCategoricals.test_union_categoricals_sort_false_fastpathc C sf t ddgdddgd}t ddgdddgd}t||gdd}t ddddgdddgd}t ||¡ d S rW r! rH r r r Ú-test_union_categoricals_sort_false_skipresort3 s
|
zCTestUnionCategoricals.test_union_categoricals_sort_false_skipresortc C sX t dtjg}t tjdg}t||gdd}t dtjtjdgddgd}t ||¡ d S )Nr r FrS r rB rH r r r Ú*test_union_categoricals_sort_false_one_nan; s
|
z@TestUnionCategoricals.test_union_categoricals_sort_false_one_nanc C sH t tjg}t tjg}t||gdd}t tjtjg}t ||¡ d S ©NFrS rB rH r r r Ú+test_union_categoricals_sort_false_only_nanB s
|
zATestUnionCategoricals.test_union_categoricals_sort_false_only_nanc C s8 t g }t g }t||gdd}t g }t ||¡ d S r[ r! rH r r r Ú(test_union_categoricals_sort_false_emptyI s
|
z>TestUnionCategoricals.test_union_categoricals_sort_false_emptyc C sl t ddgdddgdd}t ddgdddgdd}t||gdd}t ddddgdddgdd}t ||¡ d S )Nr r r TrK FrS r! rH r r r Ú/test_union_categoricals_sort_false_ordered_trueP s
|
ÿzETestUnionCategoricals.test_union_categoricals_sort_false_ordered_truec C s¸ t ddg}tddgdd}t||g}t ddddg}t ||¡ t|}t||g}t ||¡ t|}t||g}t ||¡ d}tjt|d t|dddgg W 5 Q R X d S )Nr r r Úcategoryr4 z-all components to combine must be Categoricalr) ) r r r r r r r+ r, r- rT r r r Ú test_union_categorical_unwrapY s z3TestUnionCategoricals.test_union_categorical_unwrapN)$Ú__name__Ú
|
__module__Ú __qualname__r+ ÚmarkZ parametrizeÚlistr6 r7 r8 Z
|
date_rangeZ period_ranger r r r r% r( r/ r1 r> rA rF rG rI rM rR rU rV rX rY rZ r\ r] r^ r` r r r r r s\
|
ý
|
|
|
ýý
|
|
|
ýíþ
|
,
|
|
46
|
r )Únumpyr6 r+ Zpandas.core.dtypes.concatr Zpandasr8 r r r Zpandas._testingZ_testingr r r r r r Ú<module> s
|
