U
|
®ý°dÄ ã
|
@ sH d dl m Z d dlZd dlZd dlZd dlmZ d dlZd dlm Z m
|
Z dd Z dd Z d d
|
Zej d dejfd d ddejejfg¡ej dddg¡dd Zdd Zdd Zej dddg¡ej d dd g¡ej d dejejg¡ej d ddg¡d d! Zej d dddg¡d"d# Zej dddg¡ej d dd g¡ej d dejejg¡ej d ddg¡d$d% Zej d dddg¡d&d' Zd(d) Zd*d+ Zd,d- Z d.d/ Z ej d0dd1d2d3ig¡ej d4d2d3dgd2d3d5d6gg¡d7d8 Z d9d: Z ej d0d;d< d=d< d{d>d<g¡d?d@ Z dAdB Z!dCdD Z"dEdF Z#dGdH Z$dIdJ Z%dKdL Z&ej dMddNdNejgfddNdejgfg¡dOdP Z'dQdR Z(dSdT Z)dUdV Z*dWdX Z+dYdZ Z,ej dddg¡d[d\ Z-d]d^ Z.d_d` Z/dadb Z0dcdd Z1dedf Z2dgdh Z3didj Z4dkdl Z5dmdn Z6dodp Z7dqdr Z8dsdt Z9dudv Z:dwdx Z;dydz Z<dS )|é )ÚdatetimeN)ÚPerformanceWarning)ÚSeriesÚ_testingc C sÔ t jdt jdddgt jd}t|| d}d}|j |¡}| dkrDdnd}tt jd t jd
|
d
|
d gt jd|d}t ||¡ |jj|d d }tt jd t jd d d
|
gt jd|d}t ||¡ tt jdd ddgt d| d}|j |¡}| dkrðt j
|
nd}tt d d d
|
d
|
g¡|d}t ||¡ tt jd dddgt d| d}t t | dk¡ |jjdd d}W 5 Q R X tt d
|
d d
|
d
|
g¡|d}t ||¡ |jjdd d d}tt d
|
d d
|
d g¡|d}t ||¡ tt jdt jddgt jd| d}d}|j |¡}| dkrdnd}tt jd t jd
|
d
|
gt jd|d}t ||¡ |jj|d d}| dkrTt j
|
nd}tt d d d
|
d
|
g¡|d}t ||¡ tt jdd ddgt jd| d}|j |¡}tt d d d
|
d
|
g¡|d}t ||¡ d S )NÚfooZ fooommm__fooZmmm_z foommm[_]+bar©Údtypezmmm[_]+ÚobjectÚbooleanFT©ÚregexZxyzZFooZxYzZ fOOomMm__fOoZMMM_ústring[pyarrow]zFOO|mmm©Úcase)r r ©Úna) ÚnpÚarrayÚnanÚobject_r ÚstrÚcontainsÚtmÚassert_series_equalr Úbool_Úmaybe_produces_warningr )Úany_string_dtypeÚvaluesÚpatÚresultÚexpected_dtypeÚexpected© r" ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/strings/test_find_replace.pyÚ test_contains s ÿ þ þ þ þ ÿ þ ÿ þ r$ c C sx t tjdtjddt ¡ dd ddg td} | j d¡}t tjd tjd tjtjdtjtjtjg tj d}t
|
||¡ d S )
|
NÚaÚbTr é ç @r ÚoF) r r r r r Útodayr r r r r r ©Úmixedr r! r" r" r# Útest_contains_object_mixed` s þÿ þÿr- c C sè t ddddtjgdd} | jjddd}t dddddg}t ||¡ | jjddd}t dddddg}t ||¡ t ddddtjg} | jjddd}t dddddg}t ||¡ | jjddd}t dddddg}t ||¡ d S ) Nr% r& ÚcÚcategoryr Tr F©r r r r r r r )r r r! r" r" r# Ú*test_contains_na_kwarg_for_object_categoryq s r1 z na, expected)TT)FF)r F)é Tr TFc C sP t ddddtjg| d}|jjd||d}t dddd|gdd}t ||¡ d S ) Nr% r& r. r )r r TFr
|
r0 )Únullable_string_dtyper r! r r r r" r" r# Ú0test_contains_na_kwarg_for_nullable_string_dtype s r4 c C s t ddddddtjddd g
|
| d
|
}|j d ¡}| d kr:d nd }t ddddddtjdddg
|
|d
|
}t ||¡ t t| dk¡ |jjd dd}W 5 Q R X t ddddddtjdddg
|
|d
|
}t ||¡ |j d¡}t ddddddtjdddg
|
|d
|
}t ||¡ |j d¡}t ddddddtjdddg
|
|d
|
}t ||¡ t t| dk¡ |jjddd}W 5 Q R X t ddddddtjdddg
|
|d
|
}t ||¡ d S )NÚAÚBÚCÚAabaÚBacaÚ ÚCABAÚdogÚcatr r% r r
|
FTr r ZAaÚba) r r r r r r r r r ©r Úsr r r! r" r" r# Útest_contains_moar sV þ þ ÿþ þ þ ÿþrA c C s t tjtjtjg| d}|jjddd}| dkr6tjnd}t dddg|d}t ||¡ |jjddd}t dddg|d}t ||¡ |jjddd}| dkr´t dddgtjd}nt dddgdd}t ||¡ |j d¡}| dkrêdnd}t tjtjtjg|d}t ||¡ d S )Nr r Fr r r
|
T) r r r r r r r r r r? r" r" r# Útest_contains_nanÒ s" rB r r )r Zbazr r/ Ú
|
null_valuer c
|
|
C sì t d|ddd|dg|d}|j | ¡}t dtjdddtjdg}t ||¡ |jj| |d }t d|ddd|dg}t ||¡ tjd
|
tjd dt ¡ dd d d g tj
|
d}t |j d¡}t dtjdtjtjdtjtjtjg } t || ¡ d S ©NÚomÚfoo_nomÚnomÚbar_foor r FTr r% r& r' r( Úf) r r Ú
|
startswithr r r r r r r* r ©
|
r r rC r r r Úexpr, ÚrsZxpr" r" r# Útest_startswithñ s" þ þ&rN c
|
C s t dd dddd dddg | d}|jjd|d }t d
|
|d d
|
d
|
|d d
|
d
|
g d d}t ||¡ |jjd|d }t d
|
|d
|
d
|
d
|
|d
|
d
|
d g d d}t ||¡ d S © NrE rF rG rH r r zrege.r r FTr
|
)r r rJ r r ©r3 r r r rL r" r" r# Ú%test_startswith_nullable_string_dtype s þ ÿ ÿrQ c
|
|
C sê t d|ddd|dg|d}|j | ¡}t dtjdddtjdg}t ||¡ |jj| |d }t d|ddd|dg}t ||¡ tjd
|
tjd dt ¡ dd d d g t
|
d}t |j d¡}t dtjdtjtjdtjtjtjg } t || ¡ d S rD ) r r Úendswithr r r r r r r* r rK r" r" r# Ú test_endswith& s" þ þ&rS c
|
C s t dd dddd dddg | d}|jjd|d }t d
|
|d
|
d
|
d |d d
|
d
|
g d d}t ||¡ |jjd|d }t d
|
|d
|
d
|
d
|
|d
|
d
|
d g d d}t ||¡ d S rO )r r rR r r rP r" r" r# Ú#test_endswith_nullable_string_dtypeC s þ ÿ ÿrT c C sF t dtjg| d}|jjdddd}t dtjg| d}t ||¡ d S )NÚfooBAD__barBADr úBAD[_]*r: Tr Úfoobar©r r r r Úreplacer r ©r Úserr r! r" r" r# Ú test_replace[ s r\ c C sz t dtjg| d}t dtjg| d}|jjddddd}t ||¡ t d tjg| d}|jjd
|
ddd d}t ||¡ d S ) NrU r Ú foobarBADrV r: r' T©Únr Z foo__barBADÚBADFrX ©r r[ r! r r" r" r# Ú test_replace_max_replacementsc s rb c
|
C sl t dtjddt ¡ dd ddg } t | jjdddd }t d
|
tjd tjtjd tjtjtjg }t ||¡ d S ) NÚaBADÚbBADTÚfooBADr' r( rV r: r r% r& r ) r r r r r* r rY r r ©r[ r r! r" r" r# Útest_replace_mixed_objecto s ÿ&rg c C sj t d d¡g| d}t d d¡g| d}t t| dk¡ |jjddtjdd }W 5 Q R X t ||¡ d S )
|
Nó abcd,Ã úutf-8r ó abcd, Ã r ú(?<=\w),(?=\w)ú, T©Úflagsr )
|
r Údecoder r r r rY ÚreÚUNICODEr ra r" r" r# Útest_replace_unicodex s ÿ rr Úreplr2 r% r& Údatar. Úadc C s< d}||| d}t jt|d |j d|¡ W 5 Q R X d S )Nz!repl must be a string or callabler ©Úmatchr% )ÚpytestÚraisesÚ TypeErrorr rY )r Úindex_or_seriesrs rt ÚmsgÚobjr" r" r# Ú#test_replace_wrong_repl_type_raises s r~ c C sl t dtjg| d}dd }t t| dk¡ |jjd|ddd }W 5 Q R X t d
|
tjg| d}t ||¡ d S ) NrU r c S s | d¡ ¡ S ©Nr ©ÚgroupÚswapcase©Úmr" r" r# Ú<lambda> ó z'test_replace_callable.<locals>.<lambda>r ú [a-z][A-Z]{2}é Tr^ ÚfoObaD__baRbaD© r r r r r r r rY r )r r[ rs r r! r" r" r# Útest_replace_callable s ÿ r c C s d S ©Nr" r" r" r" r# r
r r
c C s d S r r" )r Úxr" r" r# r
r c C s d S r r" )r r Úyr" r" r# r
r c
|
C sb t dtjg| d}d}tjt|d4 t t| dk¡ |j j
|
d|dd W 5 Q R X W 5 Q R X d S ) NrU r zO((takes)|(missing)) (?(2)from \d+ to )?\d+ (?(3)required )positional arguments?rv r r% Tr ) r r r rx ry rz r r r r rY )r rs r r| r" r" r# Ú test_replace_callable_raises s ÿ ÿr c C sn t dtjg| d}d}dd }t t| dk¡ |jj||dd}W 5 Q R X t d tjg| d}t ||¡ d S )
|
Nz Foo Bar Bazr z,(?P<first>\w+) (?P<middle>\w+) (?P<last>\w+)c S s | d¡ ¡ S )NZmiddler r r" r" r# r
± r z4test_replace_callable_named_groups.<locals>.<lambda>r Tr ZbARr )r r[ r rs r r! r" r" r# Ú"test_replace_callable_named_groups s ÿ r c C sº t dtjg| d}t d¡}t t| dk¡ |jj |ddd}W 5 Q R X t dtjg| d}t
|
||¡ t t| dk¡ |jj |dd dd
|
}W 5 Q R X t d tjg| d}t
|
||¡ d S ) NrU r úBAD_*r r: Tr rW r' r^ r] © r r r rp Úcompiler r r r rY r )r r[ r r r! r" r" r# Útest_replace_compiled_regexº s
|
ÿ ÿ r c
|
C sv t d¡} tdtjddt ¡ dd ddg }t|jj| ddd }td
|
tjd tjtjd tjtjtjg }t
|
||¡ d S ) Nr rc rd Tre r' r( r: r r% r& r ) rp r r r r r r* r rY r r )r r[ r r! r" r" r# Ú(test_replace_compiled_regex_mixed_objectÏ s
|
ÿ&r c C sv t d d¡g| d}t d d¡g| d}tjdtjd}t t| dk¡ |jj |dd d
|
}W 5 Q R X t
|
||¡ d S ) Nrh ri r rj rk ©rn r rl Tr ) r ro rp r rq r r r r rY r )r r[ r! r r r" r" r# Ú#test_replace_compiled_regex_unicodeÙ s ÿ r c C s° t dtjg| d}t d¡}d}tjt|d |jj |dtj
|
dd W 5 Q R X tjt|d |jj |dd dd
|
W 5 Q R X tjt|d |jj |dddd
|
W 5 Q R X d S ) NZfooBAD__barBAD__badr r z9case and flags cannot be set when pat is a compiled regexrv r: Trm F©r r ) r r r rp r rx ry Ú
|
ValueErrorr rY Ú
|
IGNORECASE©r r[ r r| r" r" r# Ú"test_replace_compiled_regex_raisesä s
|
r c C sv t dtjg| d}dd }t d¡}t t| dk¡ |jj ||ddd }W 5 Q R X t d
|
tjg| d}t
|
||¡ d S ) NrU r c S s | d¡ ¡ S r r r r" r" r# r
ù r z6test_replace_compiled_regex_callable.<locals>.<lambda>r r r Tr^ r r )r r[ rs r r r! r" r" r# Ú$test_replace_compiled_regex_callableö s
|
ÿ r zregex,expectedZbaoc C sB t ddtjg|d}t ||d}|jjdd| d}t ||¡ d S )Nzf.or r zf.r> r rX )r r! r r[ r r" r" r# Útest_replace_literal s r c C sH t g | d}dd }d}tjt|d |jjd|dd W 5 Q R X d S ) Nr c S s | d¡ ¡ S r r r r" r" r# r
r z6test_replace_literal_callable_raises.<locals>.<lambda>z2Cannot use a callable replacement when regex=Falserv ÚabcFr )r rx ry r r rY )r r[ rs r| r" r" r# Ú$test_replace_literal_callable_raises s
|
r c C sJ t g | d}t d¡}d}tjt|d |jj|ddd W 5 Q R X d S )Nr r zCCannot use a compiled regex as replacement pattern with regex=Falserv r: Fr )r rp r rx ry r r rY r r" r" r# Ú$test_replace_literal_compiled_raises s
|
|
r¡ c C s t ddddddtjddd g
|
| d
|
}|j dd ¡}t d ddd ddtjd dd g
|
| d
|
}t ||¡ t t| dk¡ |jjdd dd}W 5 Q R X t d dddddtjd ddg
|
| d
|
}t ||¡ t t| dk¡ |jjddddd}W 5 Q R X t ddddddtjdddg
|
| d
|
}t ||¡ d S ) Nr5 r6 r7 r8 r9 r: r; r< r= r ZYYYZYYYabaZCYYYBYYYr Fr Z
|
YYYYYYbYYYZBYYYcYYYZcYYYtz^.a|dogzXX-XX Tr zXX-XX bazXX-XX cazXX-XX BAzXX-XX t) r r r r rY r r r r rZ r" r" r# Útest_replace_moar sd þþ ÿ ö ó ÿ ö ór¢ c C sÄ t ddddtjg| d}t t| dk¡ |jjddd d d
|
}W 5 Q R X t d d d d tjg| d}t ||¡ t t| dk¡ |jjdd d d d
|
}W 5 Q R X t d d ddtjg| d}t ||¡ d S ) NzA.za.ZAbÚabr r r% r. Fr zc.Úcbr rZ r" r" r# Ú)test_replace_not_case_sensitive_not_regexY s ÿ ÿ r¥ c C sR t dddtjdg| d}|jjdddd}t dddtjdg| d}t ||¡ d S ) Nr% r& Úacr: r z^.$Tr rX )r r@ r r! r" r" r# Útest_replace_regexl s r§ c C sp t dddtjdg|d}|jjdd| d}| rHt dddtjdg|d}nt d ddtjdg|d}t ||¡ d S )
|
Nza.bÚ.r& r: r r% r ZaaaZaabrX )r r r@ r r! r" r" r# Ú#test_replace_regex_single_charactert s r© c C s8 | dkr dnd}t dtjdg| d}|j d¡}t dtjdg|d}t ||¡ t dd tjdg| d}|j d
|
¡}t ddtjdg|d}t ||¡ |j d ¡}t ddtjdg|d}t ||¡ t dd tjdg| d}|j d ¡}t ddtjdg|d}t ||¡ |j d¡}t ddtjdg|d}t ||¡ d S )Nr r
|
rU r r ú.*(BAD[_]+).*(BAD)TFÚBAD_BADleroybrownú.*BAD[_]+.*BADz BAD[_]+.*BADz^BAD_BADleroybrownz ^BAD[_]+.*BADz\^BAD[_]+.*BAD)r r r r rw r r )r r r r r! r" r" r# Ú
|
test_match s2 ÿ ÿ r c
|
C st t dtjddt ¡ dd ddg } t | j d¡}t dtjdtjtjdtjtjtjg }t|t sdtt
|
||¡ d S ) NZaBAD_BADZ BAD_b_BADTr r' r( rª F) r r r r r* r rw Ú
|
isinstanceÚAssertionErrorr r r+ r" r" r# Útest_match_mixed_object§ s$ ÷ÿ ÿr° c C s t ddtjg| d}|jjddd}| dkr2tjnd}t dddg|d}t ||¡ |j d¡}| dkrldnd}t ddtjg|d}t ||¡ d S ) Nr% r& r Fr r r
|
T)r r r r rw r r r r? r" r" r# Útest_match_na_kwarg½ s r± c C sv t ddddg| d}t t| dk¡ |jjddd}W 5 Q R X | d krNtjnd
|
}t d d d d g|d}t ||¡ d S ) Nr£ ÚABr ÚABCr r Fr r r
|
T) r r r r r rw r r r )r r r r r! r" r" r# Útest_match_case_kwargÌ s ÿr´ c C sX t ddtjdg| d}|j d¡}| dkr.dnd}t dd tjd g|d}t ||¡ d S )
|
NrU r« r r r¬ r r
|
TF)r r r r Ú fullmatchr r ©r r[ r r r! r" r" r# Útest_fullmatchÜ s ÿ r· c C s\ t ddtjdg| d}|jjddd}| dkr4tjnd }t d
|
dddg|d}t ||¡ d S ) NrU r« r r r¬ Fr r r
|
T)r r r r rµ r r r r¶ r" r" r# Útest_fullmatch_na_kwargç s ÿr¸ c C sà t ddddg| d}| dkr"tjnd}t dd d d g|d}|jjddd
|
}t ||¡ t ddd d g|d}t t| d k¡ |jjdd d
|
}W 5 Q R X t ||¡ t t| d k¡ |jjdt j
|
d }W 5 Q R X t ||¡ d S ) Nr£ r² r r³ r r r
|
TFr r r ) r r r r rµ r r r r rp r )r r[ r r! r r" r" r# Útest_fullmatch_case_kwargñ s$ ÿ ÿ r¹ c C sJ t dtjddg| d}|j d¡}t ddgtjg dgg}t ||¡ d S )NrU r r` r rV ÚBAD__)r r r r Úfindallr r rZ r" r" r# Ú test_findall s r¼ c
|
C sh t dtjddt ¡ dd ddg } | j d¡}t ddgtjg tjtjdgtjtjtjg }t ||¡ d S ) NrU r Tr` r' r( rV rº ) r r r r r* r r» r r rf r" r" r# Útest_findall_mixed_object s4 ÷ÿ ÷ÿr½ c C s t dddddg| d}| dkr$tjnd}|j d ¡}t d
|
d d d dg|d}t ||¡ tjdd t |¡D tjd}t tj|tjd|¡ |j d ¡}t d
|
ddd
|
dg|d}t ||¡ tjdd t |¡D tjd}t tj|tjd|¡ |j d d ¡}t d
|
d dd
|
dg|d}t ||¡ tjdd t |¡D tjd}t tj|tjd|¡ |j d d ¡}t d
|
ddd
|
dg|d}t ||¡ tjdd t |¡D tjd}t tj|tjd|¡ |j d d d¡}t d
|
d dd
|
dg|d}t ||¡ tjdd t |¡D tjd}t tj|tjd|¡ |j d d d¡}t d
|
d dd
|
dg|d}t ||¡ tjdd t |¡D tjd}t tj|tjd|¡ d S )NÚABCDEFGZBCDEFEFÚ DEFGHIJEFZEFGHEFÚXXXXr r ÚInt64ÚEFé r2 r' r éÿÿÿÿc S s g | ]}| d ¡qS ©r ©Úfind©Ú.0Úvr" r" r# Ú
|
<listcomp>D s z test_find.<locals>.<listcomp>é é c S s g | ]}| d ¡qS rÅ ©ÚrfindrÈ r" r" r# rË J s c S s g | ]}| d d¡qS ©r r2 rÆ rÈ r" r" r# rË P s c S s g | ]}| d d¡qS rÐ rÎ rÈ r" r" r# rË V s é c S s g | ]}| d dd¡qS ©r r2 rÑ rÆ rÈ r" r" r# rË \ s c S s g | ]}| d dd¡qS rÒ rÎ rÈ r" r" r# rË b s )
|
r r Úint64r rÇ r r r Zassert_numpy_array_equalrÏ ©r r[ r r r! r" r" r# Ú test_find; sF ÿ rÕ c C s\ t g | d}tjtdd |j d¡ W 5 Q R X tjtdd |j d¡ W 5 Q R X d S )Nr z!expected a string object, not intrv r )r rx ry rz r rÇ rÏ )r r[ r" r" r# Útest_find_bad_arg_raisesf s
|
rÖ c C sh t dtjdtjdg| d}| dkr(tjnd}|j d¡}t dtjd tjd
|
g|d}t ||¡ |j d¡}t dtjd tjd
|
g|d}t ||¡ |j dd ¡}t dtjd tjd
|
g|d}t ||¡ |j dd ¡}t dtjd tjd
|
g|d}t ||¡ |j dd d ¡}t dtjd
|
tjd
|
g|d}t ||¡ |j dd d ¡}t dtjd
|
tjd
|
g|d}t ||¡ d S )Nr¾ r¿ rÀ r r rÁ r rà r' rÄ rÍ r2 rÑ ) r r r Úfloat64r rÇ r r rÏ rÔ r" r" r# Ú test_find_nano s. ÿ rØ c C sP | ddddg|d}t dd¡}|j |¡}| dd d
|
d g|d}t ||¡ d S ) NZabcdefgZabccZcdddfgZcdefgggr r ÚcdeZcdedefgZcdeeZedddfgZedefggg)r Ú maketransÚ translater Z assert_equal)r{ r r} Útabler r! r" r" r# Útest_translate s
|
ÿ
|
ÿrÝ c C sJ t ddddg} t dd¡}t dddtjg}| j |¡}t ||¡ d S ) Nr% r& r. g333333ó?r rÙ ÚdÚe)r r rÚ r r rÛ r r )r@ rÜ r! r r" r" r# Útest_translate_mixed_object s
|
rà c C sT dddt jd}t|| d}d}| dk}|jj|tjdd }|jd
|
¡ d d d gksXt t
|
t |¡ |jj |tjd}W 5 Q R X |d
|
st t
|
t |¡ |jj|tjd}W 5 Q R X |d
|
sÄt |jj|tjd}|d
|
d
|
dksêt |jj|tjd}|d
|
dkst d}t
|
jt|| d |jj|tjd}W 5 Q R X |d
|
sPt d S )Nzdave@google.comzsteve@gmail.comz rob@gmail.com)ZDaveZSteveZRobZWesr z,([A-Z0-9._%+-]+)@([A-Z0-9.-]+)\.([A-Z]{2,4})r T)rn Úexpandr ÚdaveÚgoogleÚcomr )râ rã rä r' zhas match groups)rw Zraise_on_extra_warnings)r r r r Úextractrp r ZilocÚtolistr¯ r r r rw rµ r» ÚcountZassert_produces_warningÚ UserWarningr )r rt r Z using_pyarrowr r| r" r" r# Útest_flags_kwarg« s8 ü ÿ ré )N)=r rp Únumpyr rx Z pandas.errorsr ZpandasÚpdr r r r$ r- r1 ÚmarkZ parametrizeZNAr r4 rA rB rN rQ rS rT r\ rb rg rr 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> s¦
|
L
|
úþ 2
|
|
|
ÿ
|
|
ÿ
|
9
|
|
&+ $
|
