U
|
®ý°d0c ã @ s@ d dl m Z mZ d dlZd dlZd dlmZmZmZm Z d dl
|
m Z d dl mZ ej dd de ddgg¡d d
|
Zd d Zd d Zdd Zdd Zej dddgddgg¡dd Zdd Zej dddddddd ddd d g
|
fd dddd d d dd d d g
|
fd d d d dd d d dd d g
|
fd d d d dd d d dd d g
|
fd d d d d d d d d d dg
|
fd!d dd d d d d d d d g
|
fd"dd d d dd dd d d g
|
fd#dd dd dd d d d d g
|
fg¡d$d% Zej dd d ddd ddd gfd&d dd d d dd gfg¡d'd( Zej dd d ejdd ejdd gfd&d ejd d ejdd gfg¡d)d* Zd+d, Z d-d. Z d/d0 Z d1d2 Z ej d3d4d5dddd6d7d gfd8d5dddd9d:dgfd4d5d6ddd6d:dgfd8d5d6ddd9d:dgfd4d;dd<dd9d:dgfd8d;d d9dd6d7dgfg¡d=d> Z d?d@ Z!ej dAd4d8g¡dBdC Z"ej dDd4d7d7d ggd8d6d7dEggg¡dFdG Z#dHdI Z$ej dJdEd9dddejdKgfd d6dLdMdMejdMgfdddLdNdOejdPgfd6dQdEdRdSejdTgfd6d dLdUdVejdVgfg¡dWdX Z%ej dJdEd9ddejdejejejejejgfdd7dLdYejdZejejejejejgfg¡d[d\ Z&ej d]dEd6dd^d_d`dMejgfdEd6dadbdcdddaejgfdEdEdadedfdgdaejgfdEd7dadedfdgdaejgfdLddadhdidjdaejgfddkdadldmdndaejgfdod<dadpdqdrdaejgfdsd6dadldcdtdaejgfg¡dudv Z'ej dDdwdxdyejdzggd{d|d}ejd~ggdddejdzggg¡dd Z(ej dDdwdxejdyggd{dejd
ggddejdggg¡dd Z)ej dDdwdddggd{dddggddddggg¡dd Z*ej dddddgfddMddgfg¡dd Z+ej dddddgfddddMgfg¡dd Z,dd Z-d d¡ Z.d¢d£ Z/d¤d¥ Z0d¦d§ Z1d¨d© Z2ej dªd«ddd¬ejdgfd®dd¯d°ejd±gfg¡d²d³ Z3d´dµ Z4d¶d· Z5ej d¸ddgd¹fddd7gdºfddd»gd¼fddd»d7gdºfdxe d½d7d7gd¼fg¡d¾d¿ Z6ej d¸d7ejgdÀfe d½d7d7gdÁfed7gdÂfg¡dÃdÄ Z7dÅdÆ Z8dÇdÈ Z9dÉdÊ Z:dËdÌ Z;dÍdÎ Z<dÏdÐ Z=dÑdÒ Z>dÓdÔ Z?dS )Õé )ÚdatetimeÚ timedeltaN)Ú DataFrameÚIndexÚ
|
MultiIndexÚSeries)Ú StringMethodsÚpatternTÚfooÚbarc C sl t ddg}dt| j }tjt|d |j | ¡ W 5 Q R X tjt|d |j | ¡ W 5 Q R X d S )Nr
|
r z expected a string or tuple, not ©Úmatch) r ÚtypeÚ__name__ÚpytestÚraisesÚ TypeErrorÚstrÚ
|
startswithÚendswith)r ÚserÚmsg© r úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/strings/test_strings.pyÚ)test_startswith_endswith_non_str_patterns s r c C sZ t ddtjdg| d}|j d¡}| dkr0tjnd}t dd tjd
|
g|d}t ||¡ d S ) Nr
|
ZfoofooZfoooofooofommmfoo©Údtypezf[o]+ÚobjectÚInt64é é é )r ÚnpÚnanr ÚcountÚfloat64ÚtmÚassert_series_equal©Úany_string_dtyper ÚresultÚexpected_dtypeÚexpectedr r r Ú
|
test_count! s
|
r- c
|
C sf t dtjddt ¡ dd ddg td} | j d¡}t dtjdtjtjdtjtjtjg }t ||¡ d S ) NÚaÚbTr
|
r ç @r r )
|
r r" r# r Útodayr r r$ r& r' ©r r* r, r r r Útest_count_mixed_object) s þ &r3 c C s t ddtjdtjdg| d}|j d¡}t ddtjd tjd
|
g| d}t ||¡ |j d d dd ddg¡}t ddtjdtjdg| d}t ||¡ d S )Nr. r/ ÚcÚdr é ÚaaaÚbbbÚcccZdddr r r! é é ÚbbZccccZdddddd)r r" r# r Úrepeatr& r' ©r) r r* r, r r r Ú test_repeat3 s ÿ ÿr? c
|
C sb t dtjddt ¡ dd ddg } | j d¡}t dtjd tjtjd
|
tjtjtjg }t ||¡ d S ) Nr. r/ Tr
|
r r0 r6 r7 r8 Z foofoofoo) r r" r# r r1 r r= r& r' r2 r r r Útest_repeat_mixed_objectC s ÿr@ z arg, repeatr! r/ c C sB t d|g| d}|j d|g¡}t dtjg| d}t ||¡ d S )Nr. r r6 r7 )r r r= r" r# r& r' )r) Úargr= r r* r, r r r Útest_repeat_with_nullL s rB c C s t | d }}| dkr,t dd}t td}nt dd}t dd}t td}t td}t }t ||j |¡¡ d|j ¡ kstt ||j ¡ ¡ t ||j
|
d¡¡ t ||j d¡¡ t ||j d¡¡ t ||j d¡¡ t ||j ¡ ¡ t ||j ¡ ¡ t ||j dd¡¡ t ||j d ¡¡ t ||j d
|
¡¡ t td g| d |jjd dd¡ t td dg| d |jjddd¡ t ||jjd dd¡ t td dg| d |jjddd¡ t |jg dd|j ¡ ¡ t ||j d¡¡ t ||j ¡ ¡ t ||j d¡¡ t ||j d¡¡ t ||j d¡¡ t ||j d¡¡ t ||j d¡¡ t ||j d¡¡ t ||j d¡¡ t ||jj ddd¡ t ||j d¡¡ t ||jj!ddd¡ t ||j !d¡¡ t ||jj"dd¡ t ||jj"dd¡ t ||j #¡ ¡ t ||j $¡ ¡ t ||j %¡ ¡ t ||j &d¡¡ t ||j 'd ¡¡ t ||j (d¡¡ t ||j )d¡¡ t ||j *¡ ¡ t ||j +¡ ¡ t ||j ,¡ ¡ t ||j -¡ ¡ t ||j .¡ ¡ t ||j /¡ ¡ t ||j 0¡ ¡ t ||j 1¡ ¡ t ||j 2¡ ¡ t ||j 3¡ ¡ t ||j 4¡ ¡ t ||j 5d¡¡ t 6dd¡}t ||j 7|¡¡ d S )Nr r Úint64r ÚbooleanÚ r. r/ r6 z^ar )Úcolumnsr z()T)Úexpandr z()()F©Zaxisé* ©Ústop)ÚstepÚasciiÚNFC)8r Úboolr r r& r' r ÚcatÚAssertionErrorÚtitler$ Úcontainsr r ÚlowerÚupperÚreplacer= r Zassert_frame_equalÚextractZset_axisZ get_dummiesÚjoinÚlenÚfindallÚfindÚrfindÚpadÚcenterÚsplitÚrsplitÚ partitionÚ
|
rpartitionÚsliceÚstripÚlstripÚrstripÚwrapÚgetÚdecodeÚencodeÚisalnumÚisalphaÚisdigitÚisspaceÚislowerÚisupperÚistitleÚ isnumericÚ isdecimalÚ
|
capitalizeÚswapcaseÚ normalizeÚ maketransÚ translate) r) Z empty_strÚemptyZ empty_intZ
|
empty_boolZ empty_objectZ empty_bytesZempty_dfÚtabler r r Útest_empty_str_methodsU s
|
|
|
|
|
þþþ r{ zmethod, expectedrk Frl rm rr rn ro rp rq c s| t ddddddddd d
|
g
|
|d }|d kr,d nd}t ||d }t|j }t ||¡ fdd|D }t||ksxtd S )NÚAr/ ZXyÚ4Z3ArE ZTTZ55ú-z r r rO rD c s g | ]}t | qS r ©Úgetattr©Ú.0Úitem©Úmethodr r Ú
|
<listcomp>Ï s z"test_ismethods.<locals>.<listcomp>©r r r r& r' ÚlistrQ ©r
r, r) r r+ r* r r r Útest_ismethods¦ s ÿ r rs c sv t dddddddg|d}|d kr&d
|
nd }t ||d}t|j }t ||¡ fd d |D }t||ksrtd S )Nr| Ú3õ ¼õ â
u á¸õ ï¼Úfourr r rO rD c s g | ]}t | qS r r r r r r r æ s z*test_isnumeric_unicode.<locals>.<listcomp>r r r r r Útest_isnumeric_unicodeÓ s r c C s\ dt jddt jddg}t||d}|dkr.dnd}t||d}t|j| }t ||¡ d S ) Nr| r r r r r r rD )r" r# r r r r& r' )r
r, r) Úvaluesr r+ r* r r r Ú test_isnumeric_unicode_missingê s r c C sD t ddtjdg| d}|j d¡j d¡}| t¡}t ||¡ d S )NZa_b_cZc_d_eZf_g_hr Ú_)
|
r r" r# r r_ rX Úastyper r& r' r> r r r Útest_spilt_join_roundtripú s
|
r c
|
C sj t dtjddt ¡ dd ddg } | j d¡j d¡}t dtjdtjtjdtjtjtjg }t ||¡ d S )NÚa_bÚ asdf_cas_asdfTr
|
r r0 r )
|
r r" r# r r1 r r_ rX r& r' r2 r r r Ú&test_spilt_join_roundtrip_mixed_object s ÿ ÿr c C sb t dddtjdddg| d}|j ¡ }| dkr2d nd
|
}t d d d tjdd dg|d}t ||¡ d S )Nr
|
ZfoooZfoooooZfooooooozfoo
|
u ãr r r% r r6 r! r; é r )r r" r# r rY r& r' r( r r r Útest_len s þ
|
r c
|
C s` t dtjddt ¡ dd ddg } | j ¡ }t dtjdtjtjdtjtjtjg }t ||¡ d S ) Nr r Tr
|
r r0 r6 é ) r r" r# r r1 r rY r& r' r2 r r r Útest_len_mixed s ÿ
|
&r z method,sub,start,end,expectedÚindexZEFr6 r Úrindexr: é ÚEr c
|
s |ddddg|d}|dkr"t jnd}|||d}t|j } |tkr\t | |¡ n t | |¡ fdd |D }t| |kst d S )
|
NÚABCDEFGÚBCDEFEFÚ DEFGHIJEFÚEFGHEFr r r c s g | ]}t | qS r r r ©Úendr
ÚstartÚsubr r r : s z test_index.<locals>.<listcomp>)
|
r" rC r r r r& r' Úassert_index_equalr rQ )
|
r
r¨ r§ r¦ Úindex_or_seriesr) r, Úobjr+ r* r r¥ r Ú
|
test_index s
|
ÿ r¬ c C s> | ddddg|d}t jtdd |j d¡ W 5 Q R X d S ) Nr¡ r¢ r£ r¤ r zsubstring not foundr ZDE)r r Ú
|
ValueErrorr r )rª r) r« r r r Útest_index_not_found_raises> s
|
ÿr® r
c C s> | g |d}d}t jt|d t|j|d W 5 Q R X d S )Nr z!expected a string object, not intr r )r r r r r )rª r) r
r« r r r r Ú test_index_wrong_type_raisesF s r¯ z method, expr c C s\ t dddtjg| d}| dkr$tjnd}t|j|d}t |tjg |d}t ||¡ d S )NZabcbÚabZbcber r r r/ )r r" r# r% r r r& r' )r) r
Úexpr r+ r* r, r r r Útest_index_missingO s
|
r² c C sj t dg| d}|j d¡}t dddggtd}t ||¡ |jjdddd }t d
|
g| d}t ||¡ d S ) NzA|B|Cr ú|r| ÚBÚCú F)ÚregexzA B C)r r r_ r r& r' rV r> r r r Útest_pipe_failures_ s r¸ zstart, stop, step, expectedZbazéÿÿÿÿrE ZowtoofaaZowtrabaaZxuqzabaaé
|
ZotoZatoZaqxZofaZabac C sB t ddtjdg|d}|j | ||¡}t ||d}t ||¡ d S )NÚaafootwoÚaabartwoZaabazquxr )r r" r# r rc r& r' )r§ rK rL r, r) r r* r r r Ú
|
test_slicel s r½ ZoofZrabc C sF t dtjddt ¡ d ddg}|j | ||¡}t |}t ||¡ d S )Nr» r¼ Tr r0 ) r r" r# r r1 r rc r& r' )r§ rK rL r, r r* r r r Útest_slice_mixed_object} s r¾ zstart,stop,repl,expectedZshrtz a it longerZevnlongerthanthatÚzZshzrtz a zit longerZevznlongerthanthatZshzortz a zbit longerZevzenlongerthanthatZshorzz a bit longezZevenlongerthanthazéþÿÿÿZzrtZzerZzatr; Zshortzz a bit zngerZevenlozerthanthatiöÿÿÿZevenlongzerthanthatc C sD t ddddtjg|d}t ||d}|j | ||¡}t ||¡ d S )NÚshortz a bit longerZevenlongerthanthatrE r )r r" r# r Z slice_replacer& r' )r§ rK Úreplr, r) r r* r r r Útest_slice_replace s þ rà rd Zaar< Úccre zaa zbb
|
úcc rf z aaz bbc C s@ t ddtjdg| d}t|j| }t || d}t ||¡ d S )Nz aa z bb
|
rÅ r )r r" r# r r r& r' ©r) r
r± r r* r, r r r Útest_strip_lstrip_rstrip¢ s rÇ zaa zbb
|
c C s\ t dtjddt ¡ d ddg}t|j| }t |tjtjtjtjtjg }t ||¡ d S )Nz aa z bb
|
Tr r0 ) r r" r# r r1 r r r& r' )r
r± r r* r, r r r Ú%test_strip_lstrip_rstrip_mixed_object² s rÈ ÚABCz BNSDzLDFJH ZABCxxúLDFJH xxZxxABCúxx BNSDc C s> t dddg| d}t|j|d}t || d}t ||¡ d S )NZxxABCxxrË rÊ r Úx)r r r r& r' rÆ r r r Ú test_strip_lstrip_rstrip_args s rÍ zprefix, expectedr. z b cÚbcr° úa b cc C s: t dddg| d}|j |¡}t || d}t ||¡ d S ©Nr° rÏ rÎ r )r r Z removeprefixr& r' )r) Úprefixr, r r* Ú ser_expectedr r r Útest_removeprefixÒ s rÓ zsuffix, expectedr4 za b c C s: t dddg| d}|j |¡}t || d}t ||¡ d S rÐ )r r Z removesuffixr& r' )r) Úsuffixr, r r* rÒ r r r Útest_removesuffixÜ s rÕ c
|
C s t dddddtjdddg | d }|jd
|
}|j d
|
¡}t ||¡ |jd d
}|jjd d }t ||¡ |jd d d
}|jjd dd}t ||¡ d S )NZYYYr´ rµ Z
|
YYYYYYbYYYZBYYYcYYYZCYYYBYYYZdogZcYYYtr r r6 rJ r r¹ )r§ rL )r r" r# r rh r& r' rc r> r r r Ú test_string_slice_get_syntaxæ s þ
|
rÖ c C s8 t dddg} | jd }t dtjdg}t ||¡ d S )N)r r )r )r6 r! r: r r r! ©r r r" r# r& r' r2 r r r Ú&test_string_slice_out_of_bounds_nestedù s
|
rØ c C s@ t dddg| d}|jd }t dtjdg| d}t ||¡ d S )Nr
|
r/ Úbar r Úor. r× r> r r r Ú test_string_slice_out_of_bounds s
|
rÛ c C sD t dddg| dj d¡}|j d¡}| dd ¡}t ||¡ d S )Nr. r/ u aär úutf-8c S s
|
| d¡S )NrÜ ©ri ©rÌ r r r Ú<lambda>
|
ó z$test_encode_decode.<locals>.<lambda>)r r rj ri Úmapr& r' r> r r r Útest_encode_decode s râ c C sh t dddg| d}d}tjt|d |j d¡ W 5 Q R X |j dd¡}| d d
|
¡}t ||¡ d S ) Nr. r/ u aÂr z['charmap' codec can't encode character '\\x9d' in position 1: character maps to <undefined>r Úcp1252Úignorec S s | dd¡S ©Nrã rä )rj rÞ r r r rß rà z*test_encode_errors_kwarg.<locals>.<lambda>) r r r ÚUnicodeEncodeErrorr rj rá r& r' )r) r r r* r, r r r Útest_encode_errors_kwarg s ÿrç c C sd t dddg} d}tjt|d | j d¡ W 5 Q R X | j dd¡}| dd ¡}t ||¡ d S )
|
Nó aó bs azS'charmap' codec can't decode byte 0x9d in position 1: character maps to <undefined>r rã rä c S s | dd¡S rå rÝ rÞ r r r rß ( rà z*test_decode_errors_kwarg.<locals>.<lambda>) r r r ÚUnicodeDecodeErrorr ri rá r& r' )r r r* r, r r r Útest_decode_errors_kwarg s ÿrë zform, expectedÚNFKCÚ123õ ã¢ã¤ã¨rN õ ＡＢＣõ ï¼ï¼ï¼õ ｱｲｴc C sX t dddtjdgddddd g|d
|
}t |ddddd g|d
|
}|j | ¡}t ||¡ d S ) NrÉ rï rð rñ r. r/ r4 r5 Úe©r r )r r" r# r rv r& r' )Úformr, r) r r* r r r Útest_normalize, s ý rõ c C sN t dddtjdgddddd g| d
|
}tjtd d |j d ¡ W 5 Q R X d S )NrÉ rï rð rñ r. r/ r4 r5 rò ró zinvalid normalization formr Zxxx)r r" r# r r r r rv ©r) r r r r Ú test_normalize_bad_arg_raises> s ýr÷ c C s8 t dddg} t dddg}| j d¡}t ||¡ d S )Nrï rð rñ rÉ rí rî rì )r r rv r& r© )Úidxr, r* r r r Útest_normalize_indexH s rù zvalues,inferred_typeÚstringz mixed-integergÍÌÌÌÌÌô?ÚmixediÛ c C s2 || }|t kr |j|ks tt|jts.td S )N)r Ú inferred_typerQ Ú
|
isinstancer r )r rü rª r« r r r Ú"test_index_str_accessor_visibilityO s rþ ZfloatingZ
|
datetime64Z timedelta64c C sF || }|t kr |j|ks td}tjt|d |j W 5 Q R X d S )Nz-Can only use .str accessor with string valuesr )r rü rQ r r ÚAttributeErrorr )r rü rª r« r r r r Ú0test_index_str_accessor_non_string_values_raisesa s r c C sD t ddg¡} | jdks td}tjt|d | j W 5 Q R X d S )N)r. r/ rû z5Can only use .str accessor with Index, not MultiIndexr )r Ú from_tuplesrü rQ r r rÿ r )rø r r r r Ú)test_index_str_accessor_multiindex_raisesu s
|
r c C s6 t td| d}tjtdd d|j_W 5 Q R X d S )NZaabbcder z You cannot add any new attributer r. )r r r r rÿ r Zxlabelrö r r r Ú#test_str_accessor_no_new_attributes s r c C sb t t tdd¡ t¡} t t tdd¡ t¡}d}tjt|d | j
|
|¡ W 5 Q R X d S )NÚabcZS1Údefz9Cannot use .str.cat with values of inferred dtype 'bytes'r ) r r" Úarrayr r r r r r r rP )ÚlhsÚrhsr r r r Útest_cat_on_bytes_raises s
|
r c C s> t tdd} tdddg}| jdd dd }t ||¡ d S )
|
Nr r zA/DzB/EzC/Fc S s d | j ¡ ¡S )Nú/)rX r rU )Úfr r r rß rà z1test_str_accessor_in_apply_func.<locals>.<lambda>r rH )r Úzipr Úapplyr& r' )Zdfr, r* r r r Ú test_str_accessor_in_apply_func s r c C sn t ddddtjg} t dddtjtjg}t | j d¡|¡ t dd g} t d
|
d g}t | j d ¡|¡ d S ) Nú-1Ú1Z1000rº z-01Z001r6 ú-2ú+5z-0002z+0005r: )r r" r# r& r' r Úzfill©Úvaluer, r r r Ú
|
test_zfill s r c C sJ t ddg} d}dt|j }tjt|d | j |¡ W 5 Q R X d S )Nr r r. z#width must be of integer type, not r )r r r r r r r r )r Zwidr r r r Ú$test_zfill_with_non_integer_argument¡ s
|
r c C s4 t dddg} t dddg}t | j d¡|¡ d S )Nz-catr z+dogz-0catz-0001z+0dogr: )r r& r' r r r r r r Ú test_zfill_with_leading_sign© s r c C sn t ddddddddig} | j d¡}t ddd g}t ||¡ | j d¡}t dddg}t ||¡ d S ) NZHelloZWorld)Únamer ZGoodbyeZPlanetr ZSear )r r rh r& r' )Úsr* r, r r r Útest_get_with_dict_label¯ s ýÿ r )@r r Únumpyr" r Zpandasr r r r Zpandas._testingZ_testingr& Z pandas.core.strings.accessorr ÚmarkZ parametrizer r- r3 r? r@ rB r{ r r r# r r r r r r¬ r® r¯ r² r¸ r½ r¾ rà rÇ rÈ rÍ rÓ rÕ rÖ rØ rÛ râ rç rë rõ r÷ rù rþ r r r r r r r r r r r r r Ú<module> sp
|
|
|
Qþþþþþþéþ
|
þþ
|
þþ
|
úþ
|
|
þþ
|
ûþ
|
|
&&þþ
|
øþ
|
|
ýþ
|
ýþ
|
ýþ
|
ÿ
|
ÿ
|
þþ
|
|
|
ûþ
|
|
ýþ
|
|
|
