U
|
®ý°d%3 ã @ s d dl Z d dlZd dlZd dlmZmZmZmZm Z
|
m Z dd Z ej ddeeg¡dd Zdd Zd
|
d Zej d d dg¡ej dddg¡ej dddg¡dd Zej jddddgdddgdddggd d d gd ej jd eeed!d" gd#d$d%d&gd d'd( Zd)d* Zej d+d,d-d.d/g¡d0d1 Zej d+d,d-d.d/g¡d2d3 Zd4d5 Zd6d7 Zd8d9 Zej d:eeej eeg¡d;d< Z d=d> Z dS )?é N)Ú DataFrameÚIndexÚ
|
MultiIndexÚSeriesÚ_testingÚconcatc C s( t | trt | |¡ n t | |¡ d S )N)Ú
|
isinstancer ÚtmÚassert_series_equalZassert_index_equal)ÚleftÚright© r úTd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/strings/test_cat.pyÚ assert_series_or_index_equal s
|
r Úotherc C sH | }ddg}|r||}n|}||ddj j|dd}|jdksDtd S )NÚaÚbÚname)r ú,©Úsep)ÚstrÚcatr ÚAssertionError)Úindex_or_seriesr ÚboxÚvaluesÚresultr r r Útest_str_cat_name s
|
r c C sD | }|dddddt jg}|j ¡ }d}||ks4t|jjdd}d}||ksRt|jjdd d
|
}d }||ksrtt jdt jdd d t jgtd}|ddddddg}|jj|dd}t|| |jjt|dd}t|| d}t dddg}t
|
j t |d |j |j ¡ W 5 Q R X t
|
j t |d |j t|¡ W 5 Q R X d S )Nr r ÚcZaabbcú-©Úna_repzaabbc-Ú_ZNA)r r" Z a_a_b_b_c_NAÚdZfoo©ÚdtypeÚaaza-ÚbbZbdZcfooz--ú<If `others` contains arrays or lists \(or other list-likes.*Ú1Ú2Ú3©Úmatch)ÚnpÚnanr r r ÚarrayÚobjectr Úlistr ÚpytestÚraisesÚ
|
ValueErrorr )r r Úsr ÚexpectedÚtÚrgxÚzr r r Ú test_str_cat$ s.
|
|
|
r< c C sh | }|ddddg}d}t jt|d |j d¡ W 5 Q R X t jt|d |j d¡ W 5 Q R X d S ) Nr r r r$ z'Did you mean to supply a `sep` keyword?r- ú|z )r4 r5 r6 r r )r r r7 Úmessager r r Ú#test_str_cat_raises_intuitive_errorM s r? r Ú Ú dtype_targetr2 ÚcategoryÚ dtype_callerc C s. | }t ddddg|d}|t kr$|n
|
t||d}t ddddg|d}t dddd g}|t kr`|n
|
t||d}|jj|j|d
|
}t|| t|j|d}|jj||d
|
}t|| |jj|j|d
|
}t|| t|j|jd}t dddddg}|t krú|nt||jd d
d}|jj||d
|
}t|| d S ) Nr r r% ©Úindexr Úabr' r( Úacr é )r r r r r r ) r rC rA r r r7 r9 r8 r r r r Útest_str_cat_categoricalX s$
|
|
|
"rI ÚdatarH é é g¹?gÉ?g333333Ó?r ZintegersZfloatsÚmixed)Úidsr c C s t j| tdS )Nr% )r/ r1 r2 )Úxr r r Ú<lambda> ó rP r r r3 znp.arrayc C sJ t dddg}| |}d}tjt|d |jj|ddd W 5 Q R X d S ) Nr r r z;Concatenation requires list-likes containing only strings.*r- Úouterr ©Újoinr" )r r4 r5 Ú TypeErrorr r )r rJ r7 r9 Úmsgr r r Ú test_str_cat_wrong_dtype_raises{ s
|
rW c C sÌ | }t ddddg}|t kr |n
|
t||d}tdddd g|jd}t|t||dgd
|
d }t d d ddg}|t krv|nt|j|jd}|j |¡}t|| |j |j¡}t|| |j ||g¡}t|| |j ||jg¡}t|| ddddg|_|ddddg}|t kr|nt|j|jd}|j ||g¡}t|| |j ||jg¡}t|| ddddg|_|ddddg}|t kr|nt|j|jd}|j |¡}t|| d}tdddg}t||gd
|
d } tj t
|
|d |j | j¡ W 5 Q R X tj t
|
|d |j |j|jg¡ W 5 Q R X tj t
|
|d |j |j|g¡ W 5 Q R X d }tdt j dd g}
|
tj t |d |j |
|
d g¡ W 5 Q R X tj t |d |j |
|
|g¡ W 5 Q R X tj t |d |j |
|
|jg¡ W 5 Q R X tj t |d |j |
|
|
|
|gg¡ W 5 Q R X tj t |d |j t|
|
¡ W 5 Q R X tj t |d |j |
|
t|
|
g¡ W 5 Q R X tj t |d |j d
|
¡ W 5 Q R X tj t |d |j t|jt|g¡ W 5 Q R X d S ) Nr r r r$ rD ÚAÚBÚCÚDrH ©ZaxisZaAaZbBbZcCcZdDdZaDaZbAbZcBcZdCdZaDdZbAaZcBbZdCcr) r* r+ r, r- z*others must be Series, Index, DataFrame,.*Úu)r r r r r r r rE r4 r5 r6 r/ r0 rU ÚsetÚiterr3 ) r r r7 r9 r$ r8 r r: r; Úer] r r r Útest_str_cat_mixed_inputs sl
|
|
|
|
|
|
|
ra rT r rR Úinnerr c C s | }t ddddgddddgd}t dddd gddd
|
dgd}|j||d \}}|jj|d d }|tkr~t|}t|}t|}|jj||d d}t|| d S )Nr r r r$ rD r[ rX ÚErY r` ©rT r r! rS )r Zalignr r r r ) r rT r r7 r9 ÚsaÚtar8 r r r r Útest_str_cat_align_indexed÷ s rg c C sæ t ddddg}t ddddgdddd gd
|
}t||gd d }t d d dddg}|j|jj|j| d }|jj||g| dd}t ||¡ |jj|| dd}t ||¡ t
|
ddddg¡}t ddddd g}| d krà|j |j¡n*| d krö|j |j¡n|j |j |j¡¡}|j|jj|| d }|jj||g| dd}t ||¡ tjtd d |jj|t|g| d! W 5 Q R X d"} t d#d$d%gj}
|
tjt| d |jj|
|
| d! W 5 Q R X tjt| d |jj||
|
g| d! W 5 Q R X d S )&Nr r r r$ r` rL r é rH rD r\ ÚaaaÚbbbzc--Údddz-ee)Úhowr rS rX rY rZ r[ ZaaAZbbBzc-CZddDú-e-rb rR zothers must be Series,.*r- rd r) r* r+ r, )r r ÚlocrE rT r r r r
|
r/ r1 Ú intersectionÚunionÚappendÚ
|
differencer4 r5 rU r3 r r6 ) rT r7 r9 r$ Zexpected_outerr8 r r] Zrhs_idxr: r; r r r Ú test_str_cat_align_mixed_inputs s8 ÿÿû "rs c C sì | }|}t ddddg}|t kr$|n
|
t||d}|tjgd td}|t krP|n
|
t||d}|tkr~ttjgd |jtd}nt tjgd td}|jj|d d
|
}t|| |tkrèttjgd t|jd }|jj|d d
|
}t
|
||¡ d S ) Nr r r r$ rD rh r% )rE r& r rd )r& rE ) r r r/ r0 r2 rE r r r r r
|
)r Zindex_or_series2r r r7 r9 r8 r r r r Útest_str_cat_all_na= s
|
rt c C s° t ddddg} t ddddgdddd gd
|
}t d d d ddg}| jjt|| jgddd}t ||¡ t ddgddgd
|
}| jj|jdg |jdg gddd}t ||¡ d S )Nr r r r$ r` rL r rh rH rD ri rj zc-crk rm rR r rS zaa-zd-dr )r r r r_ r r r
|
rn )r7 r9 r8 r r r r Útest_str_cat_special_casesX s &ru c C s t tjddgdddggddgdd } | ¡ } | | jdk } | j d
|
¡}| j d
|
¡}|jj|d d }|j d d ksvt
|
|jj||gd d }|j d dkst
|
d S )NiÛ iÜ rH rK rL ÚyearÚmonth)ÚnamesrD r ú r z2011 2z2011 2 2) r r Z from_productZ reset_indexrw rv Zastyper r rn r )ZdfZstr_yearZ str_monthZstr_bothZ str_multipler r r Útest_cat_on_filtered_indexg s ÿÿ rz Úklassc C sB t dddg}|j | dddg¡}t ddd g}t ||¡ d S )
|
Nr r r rO Úyr; ZaxZbyÚcz)r r r r r
|
)r{ r7 r r8 r r r Útest_cat_different_classes| s r~ c C sJ t dddddg} t d¡}tjt|d | jj| jd W 5 Q R X d S ) NZAbCÚdeZFGHIÚjZkLLLmz£others must be Series, Index, DataFrame, np.ndarray or list-like (either containing only strings or containing only objects of type Series/Index/np.ndarray[1-dim])r- )Zothers)r ÚreÚescaper4 r5 rU r r )Zpsr> r r r Útest_cat_on_series_dot_str
s ÿr ) r Únumpyr/ r4 Zpandasr r r r r r r r ÚmarkZ parametrizer r< r? rI r3 rW ra rg rs rt ru rz Útupler1 r~ r r r r r Ú<module> sD
|
|
) ý
|
ý f
|
|
1
|
|
