Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/strings/__pycache__/test_extract.cpython-37.pyc
Назад
B �5�g7e � @ s� d dl m Z d dlZd dlZd dlZd dlmZmZmZm Z m Z dd� Z dd� Z dd � Zd d � Zd d � Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zej�d dd g�d d!� �Zd"d#� Z d$d%� Z d&d'� Z d(d)� Z ej�d*d+d,d-gfd.d d-gfg�d/d0� �Z d1d2� Z!d3d4� Z"ej�d5g d6fg d7fg d8fg d9fd:d;d<gd6fd:d;d<gd9fd:d;d<gd8fd:d;d<gd9fg�d=d>� �Z#d?d@� Z$dAdB� Z%dCdD� Z&dEdF� Z'dGdH� Z(dS )I� )�datetimeN)� DataFrame�Index� MultiIndex�Series�_testingc C sB t dtjdg| d�}tjtdd�� |jjdd d� W d Q R X d S )N�fooBAD__barBAD�foo)�dtypez expand must be True or False)�matchz.*(BAD[_]+).*(BAD))�expand)r �np�nan�pytest�raises� ValueError�str�extract)�any_string_dtype�values� r �W/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/strings/test_extract.py�+test_extract_expand_kwarg_wrong_type_raises s r c C s� t dtjdg| d�}tdtjtjg| d�}|j�d�}t�||� |jjddd�}t�||� tddgtjtjgtjtjgg| d�}|jjd d d�}t�||� d S ) Nr r )r �BAD__z .*(BAD[_]+).*T)r �BADz.*(BAD[_]+).*(BAD)F)r r r r r r �tm�assert_frame_equal)r �s�expected�resultr r r �test_extract_expand_kwarg s $r c C s� t dtjddt�� dd ddg �} | jjddd �}tjtjg}td d g|d d g||||||g �}t� ||� | jjd dd �}t d tjd tjtjtjtjtjtjg �}t� ||� d S ) N�aBAD_BAD� BAD_b_BADTr � g @z.*(BAD[_]+).*(BAD)F)r �BAD_r z.*(BAD[_]+).*BAD) r r r r �todayr r r r r �assert_series_equal)Zserr �err r r r �&test_extract_expand_False_mixed_object( s " &r( c C sD t dddddg�} d}tjt|d�� | jjdd d � W d Q R X d S ) N�A1�A2�A3ZA4ZB5z,only one regex group is supported with Index)r z ([AB])([123])F)r )r r r r r r )�idx�msgr r r � test_extract_expand_index_raises; s r. c C sn | dddg|d�}d}t jt|d�� |jjddd � W d Q R X t jt|d�� |jjd dd � W d Q R X d S ) Nr) �B2�C3)r z"pattern contains no capture groups)r z [ABC][123]F)r z (?:[AB]).*)r r r r r )�index_or_seriesr �s_or_idxr- r r r �,test_extract_expand_no_capture_groups_raisesE s r3 c C sX | ddg|d�}|j jddd�}| ddgd|d �}| tkrHt�||� n t�||� d S ) Nr) r* )r z (?P<uno>A)\dF)r �A�uno)�namer )r r r r r& Zassert_index_equal)r1 r r2 r r r r r �(test_extract_expand_single_capture_groupR s r7 c C s� t dddg| d�}|jjddd�}t tjtjtjg| d�}t�||� |jjddd�}ttjtjgtjtjgtjtjgg| d�}t�||� |jjd dd�}t d d tjg| d�}t�||� |jjd dd�}td d gd dgtjtjgg| d�}t�||� |jjddd�}t d d tjgd| d�}t�||� |jjddd�}td d gd dgtjtjggddg| d�}t�||� |jjddd�}td d gd dgtjtjggddg| d�}t�||� |jjddd�}t d d tjg| d�}t�||� t dddg| d�}|jjddd�}td d gd dgtjtjgg| d�}t�||� t ddd g| d�}|jjd dd�}td d gd dgtjd ggddg| d�}t�||� t ddd g| d�}|jjd dd�}td d gd dgd tjggddg| d�}t�||� d S ) Nr) r/ r0 )r z(_)F)r z(_)(_)z ([AB])[123]r4 �Bz ([AB])([123])�1�2z(?P<letter>[AB])�letter)r6 r z!(?P<letter>[AB])(?P<number>[123])�number)�columnsr z([AB])(?P<number>[123])r z([AB])(?:[123])�A11�B22�C33z([AB])([123])(?:[123])�3z"(?P<letter>[AB])?(?P<number>[123])�Cz#(?P<letter>[ABC])(?P<number>[123])?) r r r r r r r& r r )r r r r r r r �"test_extract_expand_capture_groups^ sh ( rC c C s� dddg}t | �t |�k r$t�d� | d t |�� } t|| |d�}|jjddd�}td d tjg| |d�}t� ||� |jjd dd�}t d d gd d gdtjggddg| |d�}t� ||� d S )Nr) r/ rB zIndex too short)�indexr z(\d)F)r r9 r: z (?P<letter>\D)(?P<number>\d)?r4 r8 r; r< )r= rD r ) �lenr �skipr r r r r r r& r r )rD r �datar r r r r r �(test_extract_expand_capture_groups_index� s rH c C sH t dddgd| d�}|jjddd�}t d d d gd | d�}t�||� d S ) N�a3�b3�c2Zbob)r6 r z(?P<sue>[a-z])F)r �a�b�cZsue)r r r r r& )r r r r r r r �,test_extract_single_series_name_is_preserved� s rO c C sZ t dtjdg| d�}|jjddd�}tddgtjtjgtjtjgg| d�}t�||� d S ) Nr r )r z.*(BAD[_]+).*(BAD)T)r r r )r r r r r r r r )r r r r r r r �test_extract_expand_True� s $rP c C sn t jt jg} tdt jddt�� dd ddg �}|jjddd�}td d g| d d g| | | | | | g �}t� ||� d S ) Nr! r" Tr r# g @z.*(BAD[_]+).*(BAD))r r$ r ) r r r r r% r r r r r )r' �mixedr r r r r �%test_extract_expand_True_mixed_object� s "rR c C sn | dddg|d�}d}t jt|d�� |jjddd � W d Q R X t jt|d�� |jjd dd � W d Q R X d S ) Nr) r/ r0 )r z"pattern contains no capture groups)r z [ABC][123]T)r z (?:[AB]).*)r r r r r )r1 r r2 r- r r r �4test_extract_expand_True_single_capture_group_raises� s rS c C sT | ddg|d�}|j jddd�}| tkr,dn|}tdd d gi|d�}t�||� d S ) Nr) r* )r z (?P<uno>A)\dT)r �objectr5 r4 )r r r r r r )r1 r r2 r Zexpected_dtyper r r r �-test_extract_expand_True_single_capture_group s rU r6 � series_namec C s� t dddg| |d�}|jjddd�}ttjtjtjg|d�}t�||� |jjd dd�}ttjtjgtjtjgtjtjgg|d�}t�||� |jjd dd�}td d tjg|d�}t�||� |jjd dd�}td dgd dgtjtjgg|d�}t�||� |jjddd�}tdd d tjgi|d�}t�||� |jjddd�}td dgd dgtjtjggddg|d�}t�||� |jjddd�}td dgd dgtjtjggddg|d�}t�||� |jjddd�}td d tjg|d�}t�||� d S )Nr) r/ r0 )r6 r z(_)T)r )r z(_)(_)z ([AB])[123]r4 r8 z ([AB])([123])r9 r: z(?P<letter>[AB])r; z!(?P<letter>[AB])(?P<number>[123])r< )r= r z([AB])(?P<number>[123])r z([AB])(?:[123]))r r r r r r r r )r6 r r r r r r r �test_extract_series sB ( rW c C s� t dddg| d�}|jjddd�}tdd gd d gtjtjgg| d�}t�||� t d d dg| d�}|jjddd�}tdd gd d gtjdggddg| d�}t�||� t d d dg| d�}|jjddd�}tdd gd d gdtjggddg| d�}t�||� d S )Nr> r? r@ )r z([AB])([123])(?:[123])T)r r4 r9 r8 r: r) r/ rA z"(?P<letter>[AB])?(?P<number>[123])r; r< )r= r rB z#(?P<letter>[ABC])(?P<number>[123])?)r r r r r r r r )r r r r r r r � test_extract_optional_groupsC s& rX c C s� dddg}t | �t |�k r$t�d� | d t |�� } t|| |d�}|jjddd�}td d tjg| |d�}t � ||� |jjd dd�}td d gd d gdtjggddg| |d�}t � ||� d S )Nr) r/ rB zIndex too short)rD r z(\d)T)r r9 r: z (?P<letter>\D)(?P<number>\d)?r4 r8 r; r< )r= rD r ) rE r rF r r r r r r r r )rD r rG r r r r r r �+test_extract_dataframe_capture_groups_indexb s rY c C sJ t dddgd| d�}|jjddd�}td d d d gi| d �}t�||� d S )NrI rJ rK rV )r6 r z(?P<letter>[a-z])T)r r; rL rM rN )r )r r r r r r )r r r r r r r �'test_extract_single_group_returns_frame| s rZ c C sN dddddt jdg}ddd d d d d dg}d}dddg}t|| d�}tjddddddddgd d �}t|||| d�}|jj|tj d �}t � ||� t�d d d!d"d#d$d%g�} t|| | d&�}tjd'd(d)d*d+d,d-d.gd/d �}t|||| d�}|jj|tj d �}t � ||� t|| | d&�}d0|j _ d1|_ t|||| d�}|jj|tj d �}t � ||� d S )2Nzdave@google.comztdhock5@gmail.comzmaudelaperriere@gmail.comz'rob@gmail.com some text steve@gmail.comz%a@b.com some text c@d.com and e@f.com� )ZdaveZgoogle�com)Ztdhock5�gmailr\ )Zmaudelaperrierer] r\ )Zrobr] r\ )Zstever] r\ )rL rM r\ )rN �dr\ )�e�fr\ zY (?P<user>[a-z0-9]+) @ (?P<domain>[a-z]+) \. (?P<tld>[a-z]{2,4}) �user�domain�tld)r )r r )r# r )� r )� r )re r# )� r )rf r# )rf rd )Nr )�names)�flags)�single�Dave)ri �Toby)ri �Maude)�multiple� robAndSteve)rm �abcdef)�none�missing)rp �empty)rD r )ri rj r )ri rk r )ri rl r )rm rn r )rm rn r# )rm ro r )rm ro r# )rm ro rd )NNr )�matches� description)rs rt r )r r r r � from_tuplesr r � extractall�re�VERBOSEr r rD rg ) r rG Zexpected_tuples�patZexpected_columnsr �expected_indexr r �mir r r �test_extractall� sn r| zpat,expected_namesz"(?P<letter>[AB])?(?P<number>[123])r; r< z([AB])?(?P<number>[123])c C s` t dddg|d�}|j�| �}tdtjdftjdfgtjdd d gd d �||d �}t� ||� d S )Nr[ r) Z32)r )r4 r9 rA r: )r# r )rd r )rd r# )Nr )rg )rD r= r ) r r rv r r r r ru r r )ry Zexpected_namesr r r r r r r � test_extractall_column_names� s r} c C s� t dddgd| d�}tjdddd gd d �}|j�d �}td ddddgi|| d�}t�||� |j�d�}tddddg|| d�}t�||� d S )NrI rJ �d4c2rV )r6 r )r r )r# r )rd r )rd r# )Nr )rg z(?P<letter>[a-z])r; rL rM r^ rN )rD r z([a-z]))r r ru r rv r r r )r r rz r r r r r � test_extractall_single_group� s r c C sX t dddgd| d�}|j�d�}tddd d gtjd d d dgdd�| d�}t�||� d S )NZab3Zabc3Zd4cd2rV )r6 r z([a-z]+)�ab�abcr^ �cd)r r )r# r )rd r )rd r# )Nr )rg )rD r )r r rv r r ru r r )r r r r r r r �,test_extractall_single_group_with_quantifier s r� z data, names)N)�i1)N�i2)r� r� rI rJ r~ c sB t | �� t |�dkr*tt� �|d d�}n$� fdd�t� �D �}tj||d�}t| d||d�}tjg |d d�}|j�d �}tdg||d �}t � ||� |j�d �}tddg||d �}t � ||� |j�d �}tdg||d �}t � ||� |j�d�}tddg||d �}t � ||� |j�d�}tddg||d �}t � ||� d S )Nr# r )r6 c 3 s | ]}t |g� d �V qdS )r# N)�tuple)�.0�i)�nr r � <genexpr>) s z-test_extractall_no_matches.<locals>.<genexpr>)rg rV )r6 rD r )r z(z))r= rD r z(z)(z)z (?P<first>z)�firstz(?P<first>z)(?P<second>z)�secondz(z)(?P<second>z)) rE r �ranger ru r r rv r r r ) rG rg r rD Ztuplesr rz r r r )r� r �test_extractall_no_matches s2 r� c C s t dddgd| d�}|j�d�}tddd dgitjd d d gd d gd�| d�}t�||� | dkr�x@tdddg�tdddgdd�gD ] }|j�d�}t�||� q�W t dddgdtdddgdd�| d�}|j�d�}tddd dgitjdddgdd gd�| d�}t�||� d S )NZa1a2�b1�c1Zxxx)r6 r z[ab](?P<digit>\d)�digitr9 r: )r r )r r# )r# r r )rg )rD r rT )r6 Zs_name�XX�yy�zzZidx_name)r6 rD r )r� r )r� r# )r� r ) r r rv r r ru r r r )r r r r r, r r r �test_extractall_stringindexN s2 r� c C s> t dddgd| d�}tjtdd�� |j�d� W d Q R X d S ) NrI rJ r~ rV )r6 r zno capture groups)r z[a-z])r r r r r rv )r r r r r �(test_extractall_no_capture_groups_raisest s r� c C s� t dddgdddgdd�} | jjjd d d �}td d dg�}t�||� | jjjdd d �}dddg}t|ddgd�}t�||� d S )NrI rJ r~ r+ ZB3ZD4rV )rD r6 z([A-Z])T)r r4 r8 �Dz!(?P<letter>[A-Z])(?P<digit>[0-9]))r4 rA )r8 rA )r� �4r; r� )r= )r rD r r r r r )r �rr_ Ze_listr r r �!test_extract_index_one_two_groups| s r� c C s t dddgd| d�}d}|jj|dd�}|j�|�}|jd d d �}t�||� d }|jj|dd�}|j�|�}|jd d d �}t�||� d }|jj|dd�} |j�|�}|jd d d �}t�| |� d} |jj| dd�} |j�| �}|jd d d �}t�| |� d S )NrI rJ rK rV )r6 r z([a-z])([0-9])T)r r r )�levelz!(?P<letter>[a-z])(?P<digit>[0-9])z(?P<group_name>[a-z])z([a-z]))r r r rv �xsr r ) r r �pattern_two_noname�extract_two_nonameZhas_multi_indexZno_multi_index�pattern_two_named�extract_two_named�pattern_one_named�extract_one_named�pattern_one_noname�extract_one_nonamer r r � test_extractall_same_as_extract� s* r� c C s t jdddgdd�}tdddg|d | d �}d }|jj|d d �}|j�|�}|jddd�}t�||� d}|jj|d d �}|j�|�}|jddd�}t�||� d} |jj| d d �} |j�| �}|jddd�}t�| |� d} |jj| d d �} |j�| �}|jddd�}t�| |� d S )N)r4 r� )r8 r� )rB �third)ZcapitalZordinal)rg rI rJ rK rV )rD r6 r z([a-z])([0-9])T)r r r )r� z!(?P<letter>[a-z])(?P<digit>[0-9])z(?P<group_name>[a-z])z([a-z])) r ru r r r rv r� r r ) r r{ r r� r� Zhas_match_indexZno_match_indexr� r� r� r� r� r� r r r �-test_extractall_same_as_extract_subject_index� s0 r� ))r rw �numpyr r Zpandasr r r r r r r r r( r. r3 r7 rC rH rO rP rR rS rU �markZ parametrizerW rX rY rZ r| r} r r� r� r� r� r� r� r� r r r r �<module> sV S : U .&
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка