Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/strings/__pycache__/test_find_replace.cpython-37.pyc
Назад
B �5�g�~ � @ s d dl m Z d dlZd dlZd dlZd dlZd dlmZmZ dd� Z dd� Z dd � Z ej �d dejfd d d dejejfg�ej �dddg�dd� ��Zdd� Zdd� Zej �dddg�ej �ddejejg�ej �dddg�d d � ���Zej �ddddg�d d � �Zej �dddg�ej �ddejejg�ej �dddg�d d!� ���Zej �ddddg�d"d#� �Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zej �d,dd-d.d/ig�ej �d0d.d/dgd.d/d1d2gg�d3d4� ��Z d5d6� Z ej �d,d7d8� d9d8� dxd:d8�g�d;d<� �Z d=d>� Z d?d@� Z dAdB� Z!dCdD� Z"dEdF� Z#dGdH� Z$ej �dIddJdJejgfddJdKejgfg�dLdM� �Z%dNdO� Z&dPdQ� Z'dRdS� Z(dTdU� Z)dVdW� Z*ej �ddddg�dXdY� �Z+dZd[� Z,d\d]� Z-d^d_� Z.d`da� Z/dbdc� Z0ddde� Z1dfdg� Z2dhdi� Z3djdk� Z4dldm� Z5dndo� Z6dpdq� Z7drds� Z8dtdu� Z9dvdw� Z:dS )y� )�datetimeN)�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�}|jjdd d�}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�|�}| dk�r�dnd}tt jd t jd d gt jd�|d�}t�||� |jj|d d�}| dk�r8t 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_zFOO|mmm)�case)r r )�na) �np�array�nan�object_r �str�contains�tm�assert_series_equalr �bool_)�any_string_dtype�values�pat�result�expected_dtype�expected� r �\/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/strings/test_find_replace.py� test_contains sd r c C sx t tjdtjddt�� dd ddg td��} | j�d�}t tjd tjd tjtjdtjtjtjg tj d��}t � ||� d S ) N�a�bTr � g @)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�category)r T)r F)r r r r r r r )r r r r r r �*test_contains_na_kwarg_for_object_categoryl s r' 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 )r r r r r r r )�nullable_string_dtyper r r r r r r r �0test_contains_na_kwarg_for_nullable_string_dtype� s r* c C s` t ddddddtjddd g | d �}|j�d �}| d kr:d nd }t ddddddtjdddg |d �}t�||� |jjd dd�}t ddddddtjdddg |d �}t�||� |j�d�}t ddddddtjdddg |d �}t�||� |j�d�}t ddddddtjdddg |d �}t�||� |jjddd�}t ddddddtjdddg |d �}t�||� d S )N�A�B�C�Aaba�Baca� �CABA�dog�cat)r r r r FT)r ZAa�ba)r r r r r r r )r �sr r r r r r �test_contains_moar� s: r6 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 )N)r r F)r r r T) r r r r r r r r r )r r5 r r r r r r �test_contains_nan� s" r7 r r& � null_valuer c C s� t d|ddd|dg| d�}|j�d�}t dtjdddtjdg�}t�||� |jjd|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 FT)r r r r g @�f) r r � startswithr r r r r r r" r ) r r8 r r r �expr# �rs�xpr r r �test_startswith� s &rB 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 ) Nr9 r: r; r< r r zrege.)r )r FTr )r r r> r r )r) r r r r? r r r �%test_startswith_nullable_string_dtype s rC c C s� t d|ddd|dg| d�}|j�d�}t dtjdddtjdg�}t�||� |jjd|d �}t d|ddd|dg�}t�||� tjd tjd dt� � dd d d g t d�}t |�j�d�}t dtjdtjtjdtjtjtjg �}t�||� d S )Nr9 r: r; r< r )r FT)r r r r g @r= ) r r �endswithr r r r r r r" r ) r r8 r r r r? r# r@ rA r r r � test_endswith s &rE 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 ) Nr9 r: r; r< r r zrege.)r )r FTr )r r rD r r )r) r r r r? r r r �#test_endswith_nullable_string_dtype6 s rF c C sF t dtjg| d�}|jjdddd�}t dtjg| d�}t�||� d S )N�fooBAD__barBAD)r zBAD[_]*r0 T)r �foobar)r r r r �replacer r )r �serr r r r r � test_replaceN s rK c C sz t dtjg| d�}t dtjg| d�}|jjddddd�}t�||� t d tjg| d�}|jjd ddd d�}t�||� d S ) NrG )r � foobarBADzBAD[_]*r0 r T)�nr Z foo__barBAD�BADF)r r r r rI r r )r rJ r r r r r � test_replace_max_replacementsV s rO c C sl t dtjddt�� dd ddg �} t | �jjdddd �}t d tjd tjtjd tjtjtjg �}t�||� d S ) N�aBAD�bBADT�fooBADr g @zBAD[_]*r0 )r r r r ) r r r r r" r rI r r )rJ r r r r r �test_replace_mixed_objectb s &rS c C sN t d�d�g| d�}t d�d�g| d�}|jjddtjdd�}t�||� d S ) Ns abcd,àzutf-8)r s abcd, àz(?<=\w),(?=\w)z, T)�flagsr )r �decoder rI �re�UNICODEr r )r rJ r r r r r �test_replace_unicodek s rX �replr( r r �datar% �adc C s< d}||| d�}t jt|d�� |j�d|� W d Q R X d S )Nz!repl must be a string or callable)r )�matchr )�pytest�raises� TypeErrorr rI )r �index_or_seriesrY rZ �msg�objr r r �#test_replace_wrong_repl_type_raisesr s rc c C sP t dtjg| d�}dd� }|jjd|ddd�}t d tjg| d�}t�||� d S ) NrG )r c S s | � d��� S )Nr )�group�swapcase)�mr r r �<lambda>� � z'test_replace_callable.<locals>.<lambda>z [a-z][A-Z]{2}� T)rM r �foObaD__baRbaD)r r r r rI r r )r rJ rY r r r r r �test_replace_callable| s rk c C s d S )Nr r r r r rg � rh rg c C s d S )Nr )rf �xr r r rg � rh c C s d S )Nr )rf rl �yr r r rg � rh c C sB t dtjg| d�}d}tjt|d�� |j�d|� W d Q R X d S )NrG )r zO((takes)|(missing)) (?(2)from \d+ to )?\d+ (?(3)required )positional arguments?)r\ r )r r r r] r^ r_ r rI )r rY r ra r r r � test_replace_callable_raises� s rn c C sR t dtjg| d�}d}dd� }|jj||dd�}t dtjg| d�}t�||� d S ) Nz Foo Bar Baz)r z,(?P<first>\w+) (?P<middle>\w+) (?P<last>\w+)c S s | � d��� S )N�middle)rd re )rf r r r rg � rh z4test_replace_callable_named_groups.<locals>.<lambda>T)r ZbAR)r r r r rI r r )r rJ r rY r r r r r �"test_replace_callable_named_groups� s rp c C s� t dtjg| d�}t�d�}|jj|ddd�}t dtjg| d�}t�||� |jj|dddd �}t d tjg| d�}t�||� d S ) NrG )r zBAD_*r0 T)r rH r )rM r rL ) r r r rV �compiler rI r r )r rJ r r r r r r �test_replace_compiled_regex� s rr c C sv t �d�} tdtjddt�� dd ddg �}t|�jj| ddd �}td tjd tjtjd tjtjtjg �}t � ||� d S ) NzBAD_*rP rQ TrR r g @r0 )r r r r ) rV rq r r r r r" r rI r r )r rJ r r r r r �(test_replace_compiled_regex_mixed_object� s &rs c C sV t d�d�g| d�}t d�d�g| d�}tjdtjd�}|j�|d�}t�||� d S )Ns abcd,àzutf-8)r s abcd, àz(?<=\w),(?=\w))rT z, ) r rU rV rq rW r rI r r )r rJ r r r r r r �#test_replace_compiled_regex_unicode� s rt c C s� t dtjg| d�}t�d�}d}tjt|d�� |jj |dtj d� W d Q R X tjt|d�� |jj |ddd � W d Q R X tjt|d�� |jj |dd d � W d Q R X d S ) NZfooBAD__barBAD__bad)r zBAD_*z9case and flags cannot be set when pat is a compiled regex)r\ r0 )rT F)r T) r r r rV rq r] r^ � ValueErrorr rI � IGNORECASE)r rJ r ra r r r �"test_replace_compiled_regex_raises� s rw c C sX t dtjg| d�}dd� }t�d�}|jj||dd�}t dtjg| d�}t�||� d S ) NrG )r c S s | � d��� S )Nr )rd re )rf r r r rg � rh z6test_replace_compiled_regex_callable.<locals>.<lambda>z [a-z][A-Z]{2}ri )rM rj ) r r r rV rq r rI r r )r rJ rY r r r r r r �$test_replace_compiled_regex_callable� s rx zregex,expectedZbaor c C sB t ddtjg|d�}t ||d�}|jjdd| d�}t�||� d S )Nzf.or )r zf.r4 )r )r r r r rI r r )r r r rJ r r r r �test_replace_literal� s ry c C sH t g | d�}dd� }d}tjt|d�� |jjd|dd� W d Q R X d S ) N)r c S s | � d��� S )Nr )rd re )rf r r r rg � rh z6test_replace_literal_callable_raises.<locals>.<lambda>z2Cannot use a callable replacement when regex=False)r\ �abcF)r )r r] r^ ru r rI )r rJ rY ra r r r �$test_replace_literal_callable_raises� s r{ c C sJ t g | d�}t�d�}d}tjt|d�� |jj|ddd� W d Q R X d S )N)r z [a-z][A-Z]{2}zCCannot use a compiled regex as replacement pattern with regex=False)r\ r0 F)r )r rV rq r] r^ ru r rI )r rJ r ra 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�||� |jjdd dd�}t d dddddtjd ddg | d �}t�||� |jjddddd�}t ddddddtjdddg | d �}t�||� d S )Nr+ r, r- r. r/ r0 r1 r2 r3 )r ZYYYZYYYabaZCYYYBYYYF)r Z YYYYYYbYYYZBYYYcYYYZcYYYtz^.a|dogzXX-XX T)r r zXX-XX bazXX-XX cazXX-XX BAzXX-XX t)r r r r rI r r )r rJ r r r r r �test_replace_moar� sH r} c C s� t ddddtjg| d�}|jjddddd �}t d d d d tjg| d�}t�||� |jjdd ddd �}t d d ddtjg| d�}t�||� d S ) NzA.za.ZAb�ab)r r r% F)r r zc.�cb)r r r r rI r r )r rJ r r r r r �)test_replace_not_case_sensitive_not_regex. s r� c C sl t dddtjdg| d�}d}tjt|d�� |j�dd�}W d Q R X t dddtjdg| d�}t�||� d S ) Nr r �acr0 )r zPThe default value of regex will change from True to False in a future version\.$)r\ z^.$) r r r r �assert_produces_warning� FutureWarningr rI r )r r5 ra r r r r r �"test_replace_regex_default_warning; s r� c C s� t dddtjdg|d�}| d krXt�d�}tjt|d�� |jj dd| d �}W d Q R X n|jj dd| d �}t d ddtjdg|d�}t� ||� d S ) Nza.b�.r r0 )r z�The default value of regex will change from True to False in a future version. In addition, single character regular expressions will *not* be treated as literal strings when regex=True.)r\ r )r Zaab) r r r rV �escaper r� r� r rI r )r r r5 ra r r r r r �#test_replace_regex_single_characterH 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 rG r )r z.*(BAD[_]+).*(BAD)TF�BAD_BADleroybrownz.*BAD[_]+.*BADz BAD[_]+.*BADz^BAD_BADleroybrownz ^BAD[_]+.*BADz\^BAD[_]+.*BAD)r r r r r\ r r )r r r r r r r r � test_matche s* r� c C st t dtjddt�� dd ddg �} t | �j�d�}t dtjdtjtjdtjtjtjg �}t|t �sdt�t � ||� d S ) NZaBAD_BADZ BAD_b_BADTr r g @z.*(BAD[_]+).*(BAD)F) r r r r r" r r\ � isinstance�AssertionErrorr r )r# 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 F)r r r T)r r r r r\ r r r )r r5 r r r r r r �test_match_na_kwarg� s r� c C sZ t ddddg| d�}|jjddd�}| dkr2tjnd }t d d d d g|d�}t�||� d S ) Nr~ �ABrz �ABC)r F)r r r T)r r r\ r 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 ) NrG r� r )r z.*BAD[_]+.*BADr r TF)r r r r � fullmatchr r )r rJ 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 ) NrG r� r )r z.*BAD[_]+.*BADF)r r r T)r r r r r� r r r )r rJ 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�}|jjdd d �}t�||� |jjdtjd �}t�||� d S ) Nr~ r� rz r� )r r r TF)r )rT ) r r r r r� r r rV rv )r rJ 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 )NrG r rN )r zBAD[_]*�BAD__)r r r r �findallr r )r rJ r r r r r � test_findall� s r� c C sh t dtjddt�� dd ddg �} | j�d�}t ddgtjg tjtjdgtjtjtjg �}t�||� d S ) NrG r TrN r g @zBAD[_]*r� ) r r r r r" r r� r r )rJ r r r r r �test_findall_mixed_object� s, 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�XXXX)r r �Int64�EF� r( r r ���c S s g | ]}|� d ��qS )r� )�find)�.0�vr r r � <listcomp> s z test_find.<locals>.<listcomp>� � c S s g | ]}|� d ��qS )r� )�rfind)r� r� r r r r� s c S s g | ]}|� d d��qS )r� r( )r� )r� r� r r r r� % s c S s g | ]}|� d d��qS )r� r( )r� )r� r� r r r r� + s � c S s g | ]}|� d dd��qS )r� r( r� )r� )r� r� r r r r� 1 s c S s g | ]}|� d dd��qS )r� r( r� )r� )r� r� r r r r� 7 s ) r r Zint64r r� r r r Zassert_numpy_array_equalr� )r rJ r r r r r r � test_find sB r� c C s\ t g | d�}tjtdd�� |j�d� W d Q R X tjtdd�� |j�d� W d Q R X d S )N)r z!expected a string object, not int)r\ r )r r] r^ r_ r r� r� )r rJ r r r �test_find_bad_arg_raises; 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� r( r� ) r r r Zfloat64r r� r r r� )r rJ r r r r r r � test_find_nanD s* r� c C sf | ddddg|d�}t �dd�}|j �|�}| dd d d g|d�}| tkrVt�||� n t�||� d S ) NZabcdefgZabccZcdddfgZcdefggg)r rz �cdeZcdedefgZcdeeZedddfgZedefggg)r � maketrans� translater r r Zassert_index_equal)r` r rb �tabler r r r r �test_translateh s r� c C sJ t ddddg�} t�dd�}t dddtjg�}| j�|�}t�||� d S ) Nr r r% g333333�?rz r� �d�e)r r r� r r r� r r )r5 r� r r r r r �test_translate_mixed_objectw s r� c C s dddt jd�}t|| d�}d}|jj|tjdd�}|jd �� d d d gksPt �|jj |tjd �}|d snt �|jj |tjd �}|d s�t �|jj |tjd �}|d d dks�t �|jj |tjd �}|d dks�t �d}tjt|d�� |jj|tjd �}W d Q R X |d �st �d S )Nzdave@google.comzsteve@gmail.comz rob@gmail.com)ZDaveZSteveZRobZWes)r z,([A-Z0-9._%+-]+)@([A-Z0-9.-]+)\.([A-Z]{2,4})T)rT �expandr �dave�google�com)rT )r� r� r� r z This pattern has match groups)r\ )r r r r �extractrV rv Ziloc�tolistr� r\ r� r� �countr r� � UserWarningr )r rZ r r ra r r r �test_flags_kwarg� s( r� )N);r rV �numpyr r] Zpandas�pdr r r r r$ r' �markZ parametrizeZNAr r* r6 r7 rB rC rE rF rK rO rS rX rc rk rn rp rr rs rt rw rx ry 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� I , * $ , 3 &+ $
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.07 |
proxy
|
phpinfo
|
Настройка