uj<HddlZddlZddlmZddlZddlmZ GddZ y)N)union_categoricalsceZdZejj dedededfgdgdgdfgdgd gd fd d ejd gd ejd gd d ejd d ejd gfe jdde jdde jddfe jddde jddde jdddfe jdde jdde jddfgejj de je je jgdZdZdZdZdZdZejj dgdggdZdZdZd Zd!Zd"Zd#Zd$Zd%Zd&Zd'Zd(Z d)Z!d*Z"d+Z#y,)-TestUnionCategoricalsza, b, combinedabcabdabcabd)rr )rr r r r r r333333?r )r 333333 @r )rrr r rr bacz 2014-01-01z 2014-01-05z 2014-01-06z 2014-01-07z US/Central)tzboxct|tj||tj|g}tj|}tj||y)N)rpd Categoricaltmassert_categorical_equal)selfrrcombinedrresultexpecteds w/var/www/html/python-recuperauto-other/env/lib/python3.12/site-packages/pandas/tests/reshape/test_union_categoricals.pytest_union_categoricalz,TestUnionCategoricals.test_union_categorical sN>$S):%;SPQAR=S$TU>>(+ ##FH5ctjgd}tjgd}t||g}tjgdgd}tj||y)Nxyzrrrr$r%r&rrr categoriesrrrrrrss2rrs r)test_union_categorical_ordered_appearancez?TestUnionCategoricals.test_union_categorical_ordered_appearance.sR NN? + ^^O ,#QG,>> *7U  ##FH5r!ctjgdd}tjgdd}t||g}tjgdd}tj||y)NrTordered)rrr rrr r+r,s r#test_union_categorical_ordered_truez9TestUnionCategoricals.test_union_categorical_ordered_true8sQ NN; 5 ^^K 6#QG,>>"8$G ##FH5r!ctjgd}tjgd}d}tjt|5t ||gdddy#1swYyxYw)Nrr z$dtype of categories must be the samematch)rrpytestraises TypeErrorr)rr-r.msgs r"test_union_categorical_match_typesz8TestUnionCategoricals.test_union_categorical_match_types?sR NN; ' ^^I &4 ]]9C 0 ( 2w ' ( ( (s A##A,c~d}tjt|5tgdddy#1swYyxYw)NzNo Categoricals to unionr5)r7r8 ValueErrorr)rr:s rtest_union_categorical_emptyz2TestUnionCategoricals.test_union_categorical_emptyGs2( ]]:S 1 # r " # # #s 3<c ttjddtjgtjddtjgg}tjddtjddtjg}t j ||ttjddgtjddtjgg}tjddddtjg}t j ||tjdtjdtjg}tjtjdtjdg}ttj|tj|g}tj||ztjdtjdtjdg }t j ||ttjtjtjtjgt tjd gtjd gt  g}tjtjtjd g}t j ||ttjtjtjgtjtjtjgg}tjtjtjtjtjg}t j ||y) Nr r r ABz 2011-01-01z 2011-03-01z 2011-02-01r)dtypeX) rrrnpnanrr TimestampNaTarrayobjectIndex)rresexpval1val2s rtest_union_categoricals_nanz1TestUnionCategoricals.test_union_categoricals_nanLs~ ^^Q266N +R^^Q266N-K L nnaBFFAq"&&9: ##C- ^^S#J 'c2668J)K L nnc3S"&&9: ##C- \*BLL,FO \2BLL4NO ".."6t8L!MNnn 4K \* \* \*  ##C-!rxx(8GHu3%v1NO   nnbffbffc23 ##C- ^^RVVRVV, -r~~rvvrvv>N/O P nnbffbffbffbff=> ##C-r!val1c.|r5|dgk(r/|jtjjdt t j gt j |g}t j |}tj||y)NrRz0TDOD(infer_string) object and strings dont match)reason) applymarkerr7markxfailrrrrr)rrQrequestusing_infer_stringrLrMs rtest_union_categoricals_emptyz3TestUnionCategoricals.test_union_categoricals_emptyxsx ##,    !!M"  !".."4bnnS6I!JKnnS! ##C-r!c 4tjgdgd}tjdddtjggd}t ||g}tjdddddddtjggd}t j ||y)N)r r r r r)r r r r rrrErFrrrrc1c2rLrMs r$test_union_categorical_same_categoryz:TestUnionCategoricals.test_union_categorical_same_categorysr ^^L\ B ^^Q1bff-, G "b*nnaAq!Q266:|T ##C-r!ctjgdgd}tjgdgd}t||g}tjgdgd}tj||y)N)r&r&r&r#r))r$r$r$)r&r&r&r$r$r$r+r]s r(test_union_categorical_same_category_strz>TestUnionCategoricals.test_union_categorical_same_category_strsQ ^^O H ^^O H "b*nn;X ##C-r!ctjgdgd}tjgdgd}t||g}tjgdgd}tj||y)Nr'r)rrr)rrrrrrr+rr^r_rrs r6test_union_categorical_same_categories_different_orderzLTestUnionCategoricals.test_union_categorical_same_categories_different_ordersU ^^O H ^^O H#RH->> *  ##FH5r!cntjgdd}tjgdd}d}tjt|5t ||gdddt ||g}tjgdd}t j||tjdd d tjgd}tjd d ggdd }t ||g}tjdd d tjd d gd}t j||tjgdd}tjgdgd d }d }tjt|5t ||gdddy#1swYBxYw#1swYyxYw)Nr r r Tr1F$Categorical.ordered must be the samer5r r r r r r r r r r*r2r r r >to union ordered Categoricals, all categories must be the same) rrr7r8r9rrrrErF)rr^r_r:rLrMs rtest_union_categoricals_orderedz5TestUnionCategoricals.test_union_categoricals_orderedsT ^^It 4 ^^Iu 54 ]]9C 0 ) Bx ( )!"b*nn/> ##C- ^^Q1bff-t < ^^QFy$ G "b*nnaArvvq!4dC ##C- ^^It 4 ^^I)T JN ]]9C 0 ) Bx ( ) )% ) )$ ) )sFF+F(+F4ctjgdd}tjgdd}t||gd}tjgd}tj||d}t j t|5t||gddddt||gd}tjgd}tj||t||gd}tjgdgdd }tj||tjd d d tjgd}tjd d ggdd }t||gd}tjd d d tjd d g}tj||tjgdd}tjgdgd d }t||gd}tjgd}tj||t||gdd}tjgdgd}tj||tjgdd}tjgdd}t||gd}tjgd}tj||d}t j t|5t||gddddt j t|5t||gdddy#1swYxYw#1swYHxYw#1swYyxYw)NrhTr1F) ignore_orderrjrir5rkr r r rl)rpsort_categoriesr))r )r r r r rrrsrm) rrrrrr7r8r9rErF)rr^r_rLrMr:rrs r$test_union_categoricals_ignore_orderz:TestUnionCategoricals.test_union_categoricals_ignore_orders ^^It 4 ^^Iu 5 "b=nn/0 ##C-4 ]]9C 0 = Bxe < =!"b=nn/0 ##C- "b>nn/ItT ##C- ^^Q1bff-t < ^^QFy$ G "b=nnaArvvq!45 ##C- ^^It 4 ^^I)T J "b=nn/0 ##C- "bdSnn/IF ##C- ^^It 4 ^^It 4#RH4@>>"45 ##FH5N ]]9C 0 = Bxe < =]]9C 0 ) Bx ( ) )O = =H = = ) )s$ L;1M$M;MMMctjgd}tjgd}t||gd}tjgdgd}tj||tjdd ggd }tjd d ggd }t||gd}tjgd gd}tj||tjdd ggd }tjd d ggd }t||gd}tjgd gd}tj||tjdd ggd}tjd d ggd}t||gd}tjgd gd}tj||tjdt j g}tjt j d g}t||gd}tjdt j t j d gd dg}tj||tjt j g}tjt j g}t||gd}tjt j t j g}tj||tjg}tjg}t||gd}tjg}tj||tjd dggd d}tjdd ggd d}d}tjt|5t||gddddy#1swYyxYw)Nr#r'Trqr()rrrr$r%r&r)rrrdrrrrr)rrrr$rkz9Cannot use sort_categories=True with ordered Categoricalsr5) rrrrrrErFr7r8r9rr^r_rrr:s rtest_union_categoricals_sortz2TestUnionCategoricals.test_union_categoricals_sorts ^^O , ^^O ,#RHdC>> *7U  ##FH5^^S#J? C ^^S#J? C#RHdC>>"6?S ##FH5 ^^S#J? C ^^S#J? C#RHdC>>"6?S ##FH5^^S#J? C ^^S#J? C#RHdC>>"6?S ##FH5 ^^S"&&M * ^^RVVSM *#RHdC>>3"<#sT ##FH5 ^^RVVH % ^^RVVH %#RHdC>>266266"23 ##FH5 ^^B  ^^B #RHdC>>"% ##FH5 ^^S#J?D Q ^^S#J?D QI ]]9C 0 ? Bx > ? ? ?s O++O4ctjgd}tjgd}t||gd}tjgdgd}tj||y)Nr#r'Frvr(r)r+res r"test_union_categoricals_sort_falsez8TestUnionCategoricals.test_union_categoricals_sort_false"sT ^^O , ^^O ,#RHeD>> *7U  ##FH5r!ctjddggd}tjddggd}t||gd}tjgdgd}tj||y) Nrrrdr)rFrvrwr+res r+test_union_categoricals_sort_false_fastpathzATestUnionCategoricals.test_union_categoricals_sort_false_fastpath,[ ^^S#J? C ^^S#J? C#RHeD>>"6?S ##FH5r!ctjddggd}tjddggd}t||gd}tjgdgd}tj||y) Nrrr'r)rFrvrwr+res r-test_union_categoricals_sort_false_skipresortzCTestUnionCategoricals.test_union_categoricals_sort_false_skipresort4r~r!cTtjdtjg}tjtjdg}t ||gd}tjdtjtjdgddg}t j ||y)Nr$rFrvr)r\res r*test_union_categoricals_sort_false_one_nanz@TestUnionCategoricals.test_union_categoricals_sort_false_one_nan<ss ^^S"&&M * ^^RVVSM *#RHeD>>3"<#sT ##FH5r!cDtjtjg}tjtjg}t ||gd}tjtjtjg}t j ||yNFrvr\res r+test_union_categoricals_sort_false_only_nanzATestUnionCategoricals.test_union_categoricals_sort_false_only_nanCse ^^RVVH % ^^RVVH %#RHeD>>266266"23 ##FH5r!ctjg}tjg}t||gd}tjg}tj||yrr+res r(test_union_categoricals_sort_false_emptyz>TestUnionCategoricals.test_union_categoricals_sort_false_emptyJsL ^^B  ^^B #RHeD>>"% ##FH5r!ctjddggdd}tjddggdd}t||gd}tjgd gdd}tj||y) NrrrdTrkrFrv)rrrrr+res r/test_union_categoricals_sort_false_ordered_truezETestUnionCategoricals.test_union_categoricals_sort_false_ordered_trueQse ^^S#J?D Q ^^S#J?D Q#RHeD>> _d  ##FH5r!c:tjddg}tjddgd}t||g}tjgd}t j ||tj |}t||g}t j ||tj|}t||g}t j ||d}tjt|5t|gd gdddy#1swYyxYw) NrrrcategoryrBrwz-all components to combine must be Categoricalr5r') rrSeriesrrrCategoricalIndexr7r8r9rxs rtest_union_categorical_unwrapz3TestUnionCategoricals.test_union_categorical_unwrapZs ^^S#J ' YYSz 4#RH->>"67 ##FH5   $#RH- ##FH5 YYr]#RH- ##FH5= ]]9C 0 6 O4 5 6 6 6s 8DDN)$__name__ __module__ __qualname__r7rV parametrizelistrErFr date_range period_rangerrrr r/r3r;r>rPrZr`rbrfrnrtryr{r}rrrrrrr!rrr s [[ %[$u+tH~ 6  #5 6 +'= >c2663'bffc"c2663RVVS9   lL9 lL9 lL9   lL\J lL\J lL\J    l; l; l; ' 8 [[UR^^R5H5H"))$TU6V9:6 66(# *.X [[UR#K0 .1 ...6)42)h4?l66666666r!r) numpyrEr7pandas.core.dtypes.concatrpandasrpandas._testing_testingrrrr!rrs  8b6b6r!