Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/__pycache__/test_replace.cpython-37.pyc
Назад
B �5�g�� � @ s� d dl mZ d dlmZ d dlmZ d dlZd dlZd dlZd dl m Z d dl Z d dl m Z mZmZmZmZ d dlmZ ejdd�d d ��Zejd d�d d ��ZG dd� d�ZG dd� d�ZdS )� )� annotations)�datetime)�StringION)�np_version_under1p20)� DataFrame�Index�Series� Timestamp� date_rangezdict[str, list[int | str]])�returnc C s t td��t d�d�S )N� zab..)�a�b)�list�range� r r �]/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/test_replace.py�mix_ab s r z dict[str, list[float | str]]c C s" t td��t d�ddtjdgd�S )Nr zab..r r �d)r r �c)r r �np�nanr r r r �mix_abc s r c @ s� e Zd Zdd� Zej�dddgejdgddejejgdgd d g d dd d gd �fddgddgddddgdddd gd dd d gd �fddgddgddddgdddd gd dd d gd �fg�ej�dddg�ej�d ddg�d d � ���Z d d � Z d!d"� Z d#d$� Z d%d&� Z d'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Zej�d5ejd6d7�d6ejd8�g�d9d:� �Zej�d;d<d=d>d?d@g�dAdB� �Zej�dCdDdEgdFdGdH�dIdJgfdGd6d6gdKejidGejejgfg�dLdM� �ZdNdO� ZdPdQ� ZdRdS� Z dTdU� Z dVdW� Z dXdY� Z dZd[� Z d\d]� Z!d^d_� Z"d`da� Z#dbdc� Z$ddde� Z%dfdg� Z&dhdi� Z'ej�dje(dkdldmd gi�dldne(dkdndmd gi�fe(dkdldmd giej)do�dldne(dkdndmd giej)do�fe(dkdldmd giej*do�dldne(dkdndmd giej*do�fe(dpdddgi�dde(dpdddgi�fe(dqdrdsdtgi�drdne(dqdudsdtgi�fe(dve+e,dwdxdy�e,dwdzdy�e,dwdxdy�g�i�e,dwdxdy�e,dwdzdy�e(dve+e,dwdzdy�gd �i�fe(e,d{d|d}�gd~gd��d~d�e(e,d{d|d}�gd�gd��fe(e-d�d d�d��dnejdmgd���e.d�d�d��e.d�d�d��e(e.d�d�d��e.d�d�d��e.d�d�d��gdnejdmgd���fe(dld�gdmd�gg�d�dxe(dxd�gdmd�gg�fe(dld�gdmd�gg�dldxe(dxd�gdmd�gg�fe(dld�gdmd�gg�d�d�e(dxd�gdmd�gg�fe(dld�gdmd�gg�dld�e(dxd�gdmd�gg�fg �d�d�� �Z/d�d�� Z0d�d�� Z1d�d�� Z2d�d�� Z3d�d�� Z4d�d�� Z5d�d�� Z6d�d�� Z7d�d�� Z8d�d�� Z9d�d�� Z:d�d�� Z;d�d�� Z<d�d�� Z=d�d�� Z>d�d�� Z?d�d�� Z@d�d�� ZAd�d�� ZBd�d�� ZCej�d�dnd�dldldmgdxejdzgdddFgd��fejd�dndldmgd�d�d�gdddFgd��fdGd�dndldmgdxejdzgdddFgd��fdndmgd�dldldmgdxejdzgdddFgd��fdldmgd�dndndngdxejdzgdddFgd��fd�d�dndmdmgdxejdzgdddFgd��fddFgd�dndldmgdxejdzgdddgd��fg�d�d�� �ZDej�d�dldldH�d d gdmdmggfdldmdH�d dlgdmd ggfg�d�d�� �ZEej�d�dldmd gd�dxd�gdŜdxd�dzdƜdldmd gdxd�dzgdŜfdldmd gd�d�d�gdŜd�d�d�d˜dldmd gd�d�d�gdŜfg�d�d̈́ �ZFej�d�e.dσe�Gdl�e�*dl�e�Hdl�e�Idl�g�d�dф �ZJd�dӄ ZKd�dՄ ZLej�d�d�d�d�d�d�g�ej�d�ejejg�d�dބ ��ZMej�d�ejdxg�d�d� �ZNej�d�e�Od�e�Pdndx�g�d�d� �ZQd�d� ZRejjSd�d�d�d� �ZTd�d� ZUd�d� ZVd�d� ZWd�d� ZXej�d�dlgd�gdngdngfdlgdlgdngdngfd�gd�gdngd�gfd�gdlgdngd�gfg�ej�d�eYeZej[g�d�d�� ��Z\d�S )��TestDataFrameReplacec C s� t j|d d d�< t j|d dd �< |�� }|jt jddd�}|d ksJt�t�||�d�� |}t j|jdd�|j � d�f< t j|jd d �|j � d�f< |�t jd�}|jdd �}t�||� |�� }|jt jgdgdd�}|d ks�t�t�||�d�� d S ) N�A� ���r T)�inplace� �fooi����)�value) r r �copy�replace�AssertionError�tm�assert_frame_equal�fillna�iloc�columns�get_loc)�self�datetime_frame�float_string_frame�tsframe� return_value�mf�result�expectedr r r �test_replace_inplace" s z)TestDataFrameReplace.test_replace_inplacezto_replace,values,expectedz\s*\.\s*ze|f|g�crapr r � �h�l�o)r r r z \s*(\.)\s*z(e|f|g)z\1\1z\1_crapz..Ze_crapZf_crapZg_crap�e�f�gr TF�use_value_regex_argsc C sp t td�td�td�d��}|r2|j|||d�}n|j||d|d�}|rX|d ksTt�|}t |�}t�||� d S )Nzab..�efgh�helo)r r r )r �regexr T)r> r )r r r" r# r$ r% )r* � to_replace�valuesr1 r r; �dfr0 r r r �test_regex_replace_list_obj9 s * z0TestDataFrameReplace.test_regex_replace_list_objc C s^ t |�}ddg}tjdg}ttd��td�td�d�}t |�}|j||dd �}t |d dd tjtjgd dd d gd��}t�||� ddg}ddg}|j||dd �}t |d ddddgd��}t�||� dddg}dddg}|j||dd �}t |d ddddgd��}t�||� dddg}dddg}|j||d�}t |d ddddgd��}t�||� d S )Nz\s*\.\s*r r3 r zab..Zhalo)r r r T)r> r r5 r6 r7 z \s*(\.)\s*z(a|b)z\1\1z\1_crap�a_crap�b_crapz..)r r z(b))r> r )r r r r r r" r$ r% ) r* r �dfmix�to_replace_resr@ Zmix2Zdfmix2�res�expecr r r � test_regex_replace_list_mixedq s4 z2TestDataFrameReplace.test_regex_replace_list_mixedc C s� t |�}ddg}tjdg}|�� }|j||ddd�}|d ks@t�t |d ddtjtjgd��}t�||� dd g}d d g}|�� }|j||ddd�}|d ks�t�t |d d d ddgd��}t�||� dddg}d dd g}|�� }|j||ddd�}|d k�st�t |d dd ddgd��}t�||� dddg}d dd g}|�� }|j||dd�}|d k�sbt�t |d dd ddgd��}t�||� d S )Nz\s*\.\s*r r3 T)r r> r )r r z \s*(\.)\s*z(a|b)z\1\1z\1_craprC rD z..z(b))r> r r )r r r r! r" r# r$ r% )r* r rE rF r@ rG r. rH r r r �%test_regex_replace_list_mixed_inplace� s: z:TestDataFrameReplace.test_regex_replace_list_mixed_inplacec C s� t |�}|jddidtjidd�}|�� }|jddidtjiddd�}|d ksRt�t |d ddtjtjg|d d��}t�||� t�||� |jdd idd idd�}|�� }|jdd idd iddd�}|d ks�t�t |d ddd d g|d d��}t�||� t�||� |jdd idd id �}|�� }|jdd idd idd �}|d k�sPt�t |d ddd d g|d d��}t�||� t�||� t |d tjdddg|d d��}|jddtjidd�}|�� }|jddtjiddd�}|d k�s�t�t�||� t�||� |jddtjidd�}|�� }|jddtjidd �}|d k�sHt�t |d tjdddg|d d��}t�||� t�||� d S )Nr z\s*\.\s*T)r> )r r> r r )r r r z \s*(\.)\s*z\1tyz.ty)r> r )r> r r �.)r> r )r r" r r r! r# r$ r% )r* r rE rG �res2r. rH r r r � test_regex_replace_dict_mixed� sX " z2TestDataFrameReplace.test_regex_replace_dict_mixedc C s� t |�}|jddtjiidd�}|�� }|�� }|jddtjiiddd�}|d ksVt�|jddtjiid�}|jddtjiidd�}|d ks�t�t |d ddtjtjg|d d ��}t�||� t�||� t�||� t�||� d S ) Nr z\s*\.\s*T)r> )r r> )r> r r r )r r r )r r" r r r! r# r$ r% ) r* r rE rG rL Zres4r. �res3rH r r r � test_regex_replace_dict_nested s " z3TestDataFrameReplace.test_regex_replace_dict_nestedc C sR |}t ddddgi|d�}t ddddgi|d�}|jd d id d �}t�||� d S ) N�first�abcZbcaZcab)�dtypez.bczbc.zc.br rK T)r> )r r" r$ r% )r* �any_string_dtyperR rA r1 r0 r r r �2test_regex_replace_dict_nested_non_first_character s zGTestDataFrameReplace.test_regex_replace_dict_nested_non_first_characterc C sT t dddddgdd��}t dddddgdd��}|�dddd�i�}t�||� d S ) N�Q�T� )�Type�tmpr � rX )rU rV )r r" r$ r% )r* rA r1 r0 r r r �%test_regex_replace_dict_nested_gh4115$ s z:TestDataFrameReplace.test_regex_replace_dict_nested_gh4115c C s� t |�}t |d t�tjgd �tjtjtjdgd��}|jddgtjdd�}|�� }|�� }|jddgtjddd �}|d ks�t�|jddgtjdd �}|d ks�t�t�||� t�||� t�||� d S ) Nr r r )r r r z\s*\.\s*za|bT)r> )r> r )r> r r ) r r �arrayr r" r! r# r$ r% )r* r rA rH rG rL rN r. r r r �!test_regex_replace_list_to_scalar* s" z6TestDataFrameReplace.test_regex_replace_list_to_scalarc C s� t |�}|jdddd�}|�� }|jddddd�}|d ks>t�|�� }|jdddd�}|d ksbt�t |d ddddg|d d ��}t�||� t�||� t�||� d S ) Nz\s*\.\s*r T)r> )r r> )r> r r r r r )r r r )r r" r! r# r$ r% )r* r rA rG rL r. rN rH r r r �!test_regex_replace_str_to_numericB s z6TestDataFrameReplace.test_regex_replace_str_to_numericc C s� t |�}|jddgddd�}|�� }|jddgdddd�}|d ksFt�|�� }|jddgddd�}|d ksnt�t |d ddddgddtjd gd ��}t�||� t�||� t�||� d S ) Nz\s*\.\s*r r T)r> )r> r )r> r r r r )r r r )r r" r! r# r r r$ r% )r* r rA rG rL r. rN rH r r r �(test_regex_replace_regex_list_to_numericQ s $ z=TestDataFrameReplace.test_regex_replace_regex_list_to_numericc C s� t |�}tddi�}tdtji�}|j||dd�}|�� }|j||ddd�}|d ksXt�|�� }|j||dd�}|d ks|t�t |d ddtjtjg|d d ��} t�|| � t�|| � t�|| � d S ) Nr z\s*\.\s*T)r> )r r> )r> r r r r )r r r ) r r r r r" r! r# r$ r% ) r* r rA �s1�s2rG rL r. rN rH r r r �$test_regex_replace_series_of_regexesa s " z9TestDataFrameReplace.test_regex_replace_series_of_regexesc C sV t |�}t ddddg|d |d d��}|�dd�}t�||� |jjtjksRt�d S ) Nr rZ rW r4 r r )r r r r ) r r" r$ r% r rR r Zobject_r# )r* r rA rH rG r r r �/test_regex_replace_numeric_to_object_conversions s zDTestDataFrameReplace.test_regex_replace_numeric_to_object_conversionr? � )rd �,)re rd c C sb t dddgdddgdddgd��}|j|d�}t d ddgdtjdgdddgd��}t�||� d S ) Nz1,000r �3rd r r )�col1�col2�col3)r> Z1000)r r" r r r$ r% )r* r? rA r0 r1 r r r �+test_joint_simple_replace_and_regex_replacez s z@TestDataFrameReplace.test_joint_simple_replace_and_regex_replace�metacharz[]z()z\dz\wz\sc C sB t d|dgi�}|�d|dii�}t dddgi�}t�||� d S )Nr �elseZparen)r r" r$ r% )r* rk rA r0 r1 r r r �test_replace_regex_metachar� s z0TestDataFrameReplace.test_replace_regex_metacharzdata,to_replace,expectedZxaxZxbxr r )r r ZxcxZxdxz^\s*$c C s: |}|||d�}|j |dd�}|||d�}t�||� d S )N)rR T)r> )r" r$ � assert_equal) r* �datar? r1 �frame_or_seriesrS rR �objr0 r r r � test_regex_replace_string_types� s z4TestDataFrameReplace.test_regex_replace_string_typesc C s t j|d d d�< t j|d dd �< |�t jd�}t�||�d�� t�|�dt j�|� t j|d d d�< t j|d dd �< d|d d d�< tddgd�}t�||�dd �� td t�d �fdt�d �fg�}|�d t j�}tt jt�d �fdt�d �fg�}t�||� d S ) Nr r r g �ח��Br r )�index� �-Z20150101Z20150102) r r r" r$ r% r& r �pd� to_datetime)r* r+ � zero_filledrA �df1Z expected_dfr r r � test_replace� s z!TestDataFrameReplace.test_replacec C s� t d�t d�t d�d�}t|�}ddg}tjdg}|�||�}tdd tjtjgdd d d gd dd dgd��}t�||� dd g}ddg}|�||�}tdd ddgddd d gd dd dgd��}t�||� d S )Nzab..r< r= )r r r rK r8 r3 r r r9 r: r5 r6 r7 z..)r r r r r" r$ r% )r* rq ZdfobjrF r@ rG rH r r r �test_replace_list� s$ z&TestDataFrameReplace.test_replace_listc C s� t ddgg tjdgg�}td|i�}|t kr0|}|}|�g tj�}t�||� d}tjt |j dd�d�� |�tjg i� W d Q R X tjt |j d d�d�� |�tjd d gi� W d Q R X d S ) Nr r rZ �colzwNumPy boolean array indexing assignment cannot assign {size} input values to the 1 output values where the mask is truer )�size)�matchrW �dummyZalt) r r r r r" r$ rn �pytest�raises� ValueError�format)r* rp Zserrq r1 r0 �msgr r r � test_replace_with_empty_list� s z1TestDataFrameReplace.test_replace_with_empty_listc C s� t ddd�ddd�d��}|�dddd��}t ddd�ddd�d��}t�||� |�d|�� �}t�||� t ddd�ddd�d��}tddd��}|�|ddd��}t ddd�ddd�d��}t�||� |�||�� �}t�||� d S ) Ng rZ )r r g @r )�zero�oneg �?g �?)r r" r$ r% �meanr )r* rA r0 r1 �sr r r �test_replace_series_dict� s z-TestDataFrameReplace.test_replace_series_dictc C sT t dddgdddgg�}dddd�}|�|�}ttjgd �}|j}t�||� d S )Nr �bar�bahrZ rW r4 )r r� r� )r r" r r �int64�dtypesr$ �assert_series_equal)r* rA �m�reprH rG r r r �test_replace_convert s z)TestDataFrameReplace.test_replace_convertc C s� |}t j|jdd�|j�d�f< t j|jdd �|j�d�f< |�t jd�}|jdd�}t�||� t�|�dt j�|� |�t jd�}|jdd�}t�||� t�|�dt j�|� d S ) Nr r r i����r i��)r g �ח�) r r r' r( r) r" r& r$ r% )r* r, r/ r0 r1 r r r �test_replace_mixed s z'TestDataFrameReplace.test_replace_mixedc C s� t tddgdd�tddgdd�d��}t tddgdd�td dgdd�d��}|�dd �}t�||� |jdd d d �}|d ks�t�t�||� d S ) Ng �?g @�float64)rR r rZ r� )r rs g �?T)r )r r r" r$ r% r# )r* rA r1 r0 r. r r r �&test_replace_mixed_int_block_upcasting# s z;TestDataFrameReplace.test_replace_mixed_int_block_upcastingc C s� t tddgdd�tddgdd�tddgdd�d ��}t tddgdd�td dgdd�tddgdd�d ��}|�dd �}t�||� d S ) Ng �?g @r� )rR r rZ r� rW )r rs �Cg �?)r r r" r$ r% )r* rA r1 r0 r r r �&test_replace_mixed_int_block_splitting9 s z;TestDataFrameReplace.test_replace_mixed_int_block_splittingc C s� t tddgdd�tddgdd�d��}t tdd gd d�tddgdd�d��}|�d d �}t�||� t td d gd d�tdd gd d�d��}|�dd gd d g�}t�||� d S ) Ng �?g @r� )rR r rZ r� )r rs r �objectrW r� )r r r" r$ r% )r* rA r1 r0 r r r �test_replace_mixed2M s z(TestDataFrameReplace.test_replace_mixed2c C s| t tddgdd�tddgdd�d��}|�d|�� �� �}|�� �d�}|�� }|d |jd< |d |jd < t� ||� d S ) Nr4 r r� )rR )r rs r� )r r rZ )rZ rZ ) r r r" r� �to_dictr! �astyper' r$ r% )r* rA r0 r1 r� r r r �test_replace_mixed3h s $z(TestDataFrameReplace.test_replace_mixed3c C sf t dtdd�i�}t dddddgi�}|�dddd�i�}t�||� |�ddd��}t�||� d S ) Nr} rZ r r rW r4 r )rZ r )r r r" r$ r% )r* rA r1 r0 r r r � test_replace_simple_nested_dictt s z4TestDataFrameReplace.test_replace_simple_nested_dictc C sj t dtdd�i�}t dddddgi�}|�dddd ��}t�||� |�ddddd �i�}t�||� d S ) Nr} rZ r r rW r4 r rv )���rZ r )r r r" r$ r% )r* rA r1 r0 r r r �6test_replace_simple_nested_dict_with_nonexistent_value s zKTestDataFrameReplace.test_replace_simple_nested_dict_with_nonexistent_valuec C s� |j d }|j d }tj|j d< d|j d< |jtjdid�}|jjtjdid�j}t�||� |jtjdddid�}|�� }d|j d< d|j d< |}t�||� ||j d< ||j d< d S )N)r r )rZ r rZ r )r? g �ח�)r' r r r" rV r$ r% r! )r* r+ Z orig_valueZorig2r0 r1 r- r r r �test_replace_value_is_none� s z/TestDataFrameReplace.test_replace_value_is_nonec C s� |� � �tj�}tj|d d d�< tj|d dd �< |�tjd�}t�||�d�� t�|�dtj�|� tj|d d d�< tj|d dd �< d|d d d�< |d }tj||dk< ||d< |jdd�}t�||jdd�� d S )Nr r r g �ח�rs �bfill)�method) r! r� r �float32r r" r$ r% r& )r* r+ r- ry r r0 r r r �test_replace_for_new_dtypes� s z0TestDataFrameReplace.test_replace_for_new_dtypesz"frame, to_replace, value, expected�intsrZ rW r )rR Zbools�complexy �?y @y @y Z datetime64i� r � ru i� � r r )�dt�strr� �20130101z US/Eastern)�periods�tz)r rs �20130102)r� �20130104�20130103g �?g @g @c C s t |d�||�}t�||� d S )Nr" )�getattrr$ r% )r* �framer? r r1 r0 r r r �test_replace_dtypes� s cz(TestDataFrameReplace.test_replace_dtypesc s� t jddd�� dddd��tt jdt jgdddgddd gd��}|�� ��}� �fd d �|�� D �}t�|t|�� |�dddgdddg�}tt jdt jgdddgddd gd��}t�||� dddd��tt jdt jgdddgddd gd��}|�t j��}�fd d �|�� D �}t�|t|�� t jddg� d ddg�|�� ��}|�� }x<t t � ��D ],}|j� | �| dd�}|d k�s\t ��q\W t�||� d}t j t|d�� |�� �dd � � W d Q R X d S )Nr rd )r rs r� r� �missingrW r �asdf�fdc s& i | ] \}}|� � | �| �|�qS r )r" )�.0�k�v)�to_repr@ r r � <dictcomp> s zLTestDataFrameReplace.test_replace_input_formats_listlike.<locals>.<dictcomp>c s$ i | ] \}}|� tj� | �|�qS r )r" r r )r� r� r� )r@ r r r� / s ���T)r z:Replacement lists must match in length\. Expecting 3 got 2)r rZ )r r r �infr" �itemsr$ r% r! r �lenr# r� r� r� )r* rA �filledr1 r0 �ir. r� r )r� r@ r �#test_replace_input_formats_listlike s8 $ $ $ z8TestDataFrameReplace.test_replace_input_formats_listlikec s� t tjdtjgdddgdddgd��}tjddd�� |�� d�}� fdd �|�� D �}t�|t |�� d }tj t |d �� |�� tjddg� W d Q R X tjddg� |�� d �}|� � }x4t t � ��D ]$}|j� | d d d�}|d ks�t�q�W t�||� d S )Nr rW r rd r� r� )r rs r� c s" i | ]\}}|� � | d �|�qS )r )r" )r� r� r� )r� r r r� H s zJTestDataFrameReplace.test_replace_input_formats_scalar.<locals>.<dictcomp>z.value argument must be scalar, dict, or Series)r r� T)r )r r r r� r" r� r$ r% r� r� � TypeErrorr! r r� r# )r* rA r� r1 r� r0 r� r. r )r� r �!test_replace_input_formats_scalar@ s $ z6TestDataFrameReplace.test_replace_input_formats_scalarc C s d S )Nr )r* r r r �test_replace_limitX s z'TestDataFrameReplace.test_replace_limitc C sR t dddddd��}ddd d d d �}t d dd dd d��}|�|�}t�||� d S ) NzStrongly Agree�Agree�Neutral�DisagreezStrongly Disagree)r rZ rW r4 r r rW r4 r rZ )r� r� r� zStrongly AgreezStrongly Disagree)r r" r$ r� )r* �answer�weightsr1 r0 r r r �test_replace_dict_no_regex[ s z/TestDataFrameReplace.test_replace_dict_no_regexc C sV t dddddd��}t ddd d d d ��}t d dd dd d��}|�|�}t�||� d S ) NzStrongly Agreer� r� r� zStrongly Disagree)r rZ rW r4 r r rW r4 r rZ )r� r� r� zStrongly AgreezStrongly Disagree)r r" r$ r� )r* r� r� r1 r0 r r r � test_replace_series_no_regexp s z1TestDataFrameReplace.test_replace_series_no_regexc C s� t dtjdgi�}|jtjdddid�}|jdtjfddgd�}|jdtjgddgd�}t dddgi�}t�||� t�||� t�||� d S )Nr rZ r g �ח�)r? )r? r )r r r r" r$ r% )r* rA Zres1rL rN r1 r r r �6test_replace_dict_tuple_list_ordering_remains_the_same� s zKTestDataFrameReplace.test_replace_dict_tuple_list_ordering_remains_the_samec C s4 d}t jt|�dd�}|�ddi�}t�||� d S )Nz�fol T_opp T_Dir T_Enh 0 1 0 0 vo 1 2 vr 0 0 2 2 0 0 0 3 3 0 bt 0z\s+)�sepz\DrZ )rw Zread_csvr r" r$ r% )r* �rawrA rG r r r �)test_replace_doesnt_replace_without_regex� s z>TestDataFrameReplace.test_replace_doesnt_replace_without_regexc C sF t ddgtd�d��}|�dd�}t ddg|jd��}t�||� d S )NTF�ab)r r r )r r r" r r$ r% )r* rA r0 r1 r r r � test_replace_bool_with_string� s z2TestDataFrameReplace.test_replace_bool_with_stringc C s2 t tj�dd�dk�}|�dd�}t�||� d S )NrW g �?r� Zfdsa)r r �random�randr" r$ r% )r* rA r0 r r r �(test_replace_pure_bool_with_string_no_op� s z=TestDataFrameReplace.test_replace_pure_bool_with_string_no_opc C sD t tj�dd�dk�}|�dd�}t tjdtd��}t�||� d S )NrW g �?FT)rW rW )rR ) r r r� r� r" �ones�boolr$ r% )r* rA r0 r1 r r r �test_replace_bool_with_bool� s z0TestDataFrameReplace.test_replace_bool_with_boolc C sL t ddgddgd��}|�ddd��}t ddgddgd��}t�||� d S )NTF)r rZ Zasdb�yes)r� T)r r" r$ r% )r* rA r0 r1 r r r �%test_replace_with_dict_with_bool_keys� s z:TestDataFrameReplace.test_replace_with_dict_with_bool_keysc C sV t ddgddgd��}|�ddi�}t�||� |d �ddi�}t�||d � d S ) NrZ rW r4 r )�Y0ZY1Zreplace_string�testr� )r r" r$ r% r� )r* rA r0 r r r �!test_replace_dict_strings_vs_ints� s z6TestDataFrameReplace.test_replace_dict_strings_vs_intsc C s< t dddgi�}|�tjtj gtj�}|}t�||� d S )Nr T)r r" r r� r r$ r% )r* rA �rr8 r r r �test_replace_truthy� s z(TestDataFrameReplace.test_replace_truthyc C sj t dttdd��i�}|�dtttdd�tdd���i�}|�tttdd�tdd����}t�||� d S )Nr rZ r rW � )r r r r" �dict�zipr$ r% )r* rA r0 r1 r r r �-test_nested_dict_overlapping_keys_replace_int� s $ zBTestDataFrameReplace.test_nested_dict_overlapping_keys_replace_intc C sp t �dd�}|�t�}t �dd��t�}td|i�}|�tt||���}|�dtt||��i�}t� ||� d S )NrZ r rW r� r ) r Zaranger� r� r r" r� r� r$ r% )r* r ZastrZbstrrA r0 r1 r r r �-test_nested_dict_overlapping_keys_replace_str� s zBTestDataFrameReplace.test_nested_dict_overlapping_keys_replace_strc C s� t ddddgi�}|�dddd�i�}t ddddgi�}t�||� t ddddgi�}|�dddd �i�}t ddddgi�}t�||� d S ) Nr TF�Y�N)TFr rZ )r rZ )r r" r$ r% )r* rA rG �expectr r r �test_replace_swapping_bug� s z.TestDataFrameReplace.test_replace_swapping_bugc s� dt jdddd�t jdddd�t jdddd�t jdd dd�t jdddd�t jdddd�t jd d dd�d �i� td ddddddgdgd�}t|jj�t� d �� �ks�t�td� fdd�|jjD �i�}|jd dks�t�|� � �}t � ||� d S )N�fnamei� � �M)�year�month�freqrZ i� r r i� � )zout_augmented_AUG_2011.jsonzout_augmented_JAN_2011.jsonzout_augmented_MAY_2012.jsonz out_augmented_SUBSIDY_WEEK.jsonzout_augmented_AUG_2012.jsonzout_augmented_MAY_2011.jsonzout_augmented_SEP_2013.jsonzout_augmented_AUG_2012.jsonzout_augmented_SEP_2013.jsonz out_augmented_SUBSIDY_WEEK.jsonzout_augmented_MAY_2012.jsonzout_augmented_MAY_2011.jsonzout_augmented_AUG_2011.jsonzout_augmented_JAN_2011.json)r( c s g | ]}� d | �qS )r� r )r� r� )r r r � <listcomp>� s z<TestDataFrameReplace.test_replace_period.<locals>.<listcomp>r z Period[M]) rw �Periodr �setr� r@ �keysr# r� r" r$ r% )r* rA r1 r0 r )r r �test_replace_period� s. z(TestDataFrameReplace.test_replace_periodc s� dt d�t d�t d�t d�t d�t d�t d�d �i� td d d d dddgdgd�}t|jj�t� d �� �kspt�td� fdd�|jjD �i�}|�� �}t� ||� d S )Nr� z2011-08z2011-01z2012-05z2011-04z2012-08z2011-05z2013-09)zout_augmented_AUG_2011.jsonzout_augmented_JAN_2011.jsonzout_augmented_MAY_2012.jsonz out_augmented_SUBSIDY_WEEK.jsonzout_augmented_AUG_2012.jsonzout_augmented_MAY_2011.jsonzout_augmented_SEP_2013.jsonzout_augmented_AUG_2012.jsonzout_augmented_SEP_2013.jsonz out_augmented_SUBSIDY_WEEK.jsonzout_augmented_MAY_2012.jsonzout_augmented_MAY_2011.jsonzout_augmented_AUG_2011.jsonzout_augmented_JAN_2011.json)r( c s g | ]}� d | �qS )r� r )r� r� )r r r r� s z>TestDataFrameReplace.test_replace_datetime.<locals>.<listcomp>) r r r� r� r@ r� r# r" r$ r% )r* rA r1 r0 r )r r �test_replace_datetime s* z*TestDataFrameReplace.test_replace_datetimec C s t tdddd�dtjdgd��}|�tjd�}t tdddd�tdddgd d �d��}t�||� |�d�}t�||� |�dtj�}t tdddd�tjtjdgd��}t�||� |�t d dd �t d dd ��}t t ddd �t d dd �t ddd �gdtjdgd��}t�||� |� � }tj|j d< |�dt j it d dd ��}t�||� |� � }tj|j d< |�dt j it d dd ��}t t ddd �t d dd �t ddd �gdtjdgd��}t�||� |� � }tj|j d< |�dtjit d ��}t t ddd �t d �t ddd �gdtjdgd��}t�||� d S )Nr� r4 z US/Eastern)r� r� r rW )r rs rZ r� )rR r� )r� r� r� )rZ r r z US/Pacific)r r r r r" r r$ r% r& r r! r' rw ZNaT)r* rA r0 r1 r r r �test_replace_datetimetz! s\ z,TestDataFrameReplace.test_replace_datetimetzc C sh t |�}t�||�i �� t�||�tg td��� t�||�di i�� t�||�tdi i��� d S )N)rR r )r r$ r% r" r r� )r* r rA r r r � test_replace_with_empty_dictliket s z5TestDataFrameReplace.test_replace_with_empty_dictlikez to_replace, method, expectedr� )r rs r� g @Zffill�pad)rZ rW c C sL t dddgdtjdgdddgd ��}|j|d |d �}t |�}t�||� d S ) Nr rZ rW r ru r r r )r rs r� )r? r r� )r r r r" r$ r% )r* r? r� r1 rA r0 r r r �test_replace_method} s "$z(TestDataFrameReplace.test_replace_methodzreplace_dict, final_datac C s� t ddgddggddgdd�}t�|�}tj|d d �df ddgd �}|d dkrZddgnddg}tj|d d �df |d �}t ||d ��}|�|d�}t�||� d } tj t | d �� t�||� W d Q R X |j|dd d�} | d ks�t �t�||� d S )NrZ rW r r �category)r( rR r r4 )� categories)r r zHAttributes of DataFrame.iloc\[:, 0\] \(column name=\"a\"\) are different)r T)r ) r r r\ rw Z Categoricalr" r$ r% r� r� r# ) r* Z replace_dictZ final_datarA r Zex_catr r1 r0 r� r. r r r �"test_categorical_replace_with_dict� s z7TestDataFrameReplace.test_categorical_replace_with_dictzdf, to_replace, expr r� )rg rh )r r r� �4�5�6�7)r� r� r c C s* t |�}t |�}|�|�}t�||� d S )N)r r" r$ r% )r* rA r? �expr1 r0 r r r �test_replace_commutative� s z-TestDataFrameReplace.test_replace_commutative�replacerZ20170827c C sb t �|�r.|jjdk r.|j�tjjt dd�� t dg�}|� ||d��}t |g�}t � ||� d S )Nr� z np.putmask doesn't coerce dtype)�reasonr )r r )r ZisscalarrR �itemsize�nodeZ add_markerr� �mark�xfailr r r" r$ r% )r* �requestr rA r0 r1 r r r �test_replace_replacer_dtype� s z0TestDataFrameReplace.test_replace_replacer_dtypec C sP t ddddddgidd�}|�dd �}t dd ddddgidd�}t�||� d S ) N�grprZ rW r4 r r �Int64)rR � )r r" r$ r% )r* rA r0 r1 r r r �!test_replace_after_convert_dtypes� s z6TestDataFrameReplace.test_replace_after_convert_dtypesc C sJ t dddgdddgd��}d}tjt|d �� |�d d � � W d Q R X d S ) Nr zb r zd ze zf )r� �twozZExpecting 'to_replace' to be either a scalar, array-like, dict or None, got invalid type.*)r c S s | � � S )N)�strip)�xr r r �<lambda> � zFTestDataFrameReplace.test_replace_invalid_to_replace.<locals>.<lambda>)r r� r� r� r" )r* rA r� r r r � test_replace_invalid_to_replace� s z4TestDataFrameReplace.test_replace_invalid_to_replacerR �floatr� r� r �booleanr c C s< t t�d�|d�}|jd tj tjg|d�}t�||� d S )NrW )rR )r? r )r r �eyer" r� r$ r% )r* rR r rA r0 r r r �"test_replace_no_replacement_dtypes s z7TestDataFrameReplace.test_replace_no_replacement_dtypes� replacementc C s| t dddgdddgddd gd ��}td �|_t dddgdddg|dd gd ��}td �|_|d �d|�|d < t�||� d S ) NrZ rW r4 r r r� ru r� r� )r �A1rs ZAABrs )r r r( r" r$ r% )r* r r0 r1 r r r �#test_replace_with_duplicate_columns s " z8TestDataFrameReplace.test_replace_with_duplicate_columnsz2020-01c C sF t d|gd i�}|t k r"|d }|�� }|�dd�}t�||� d S )NZPerr4 g �?g )r r! r" r$ rn )r* rp r rq r1 r0 r r r � test_replace_ea_ignore_float s z1TestDataFrameReplace.test_replace_ea_ignore_floatc C sP ddddgddddgd d d d gd dddgddddgd�}t |d��ddd��}|d jjddddgdd�|d< |d jjd dddgdd�|d < ddddgdddd gd d d d gd d ddgd dddgd�}t |d��ddd��}|d jjdddd gdd�|d< |d jjd d ddgdd�|d < |�dd �}|�dd �}|�dd �}t�||� d S )!zt Test for #23305: to ensure category dtypes are maintained after replace with direct values rZ rW r4 r r r r r g �?g @g @g @�cat1�cat2Zcat3Zcat4�obj1�obj2Zobj3Zobj4)rg rh ri �col4Zcol5)ro r� )rh r" rh T)Zorderedr" �z�catX�obj9N)r r� �catZreorder_categoriesr" r$ r% )r* � input_dict�input_df� expected_dictr1 r0 r r r � test_replace_value_category_type' s4 z5TestDataFrameReplace.test_replace_value_category_typezDcategory dtype gets changed to object type after replace, see #35268)r c C sv dgdgdgd�}t |d��dddd��}dgdgd gd�}t |d��dddd��}|�ddd d ��}t�||� d S ) zf Test to ensure category dtypes are maintained after replace with dict values r r r )rg rh ri )ro r� r# r% r$ )r r r N)r r� r" r$ r% )r* Zinput_category_dfZexpected_category_dfr' r( r) r1 r0 r r r � test_replace_dict_category_typeZ s z4TestDataFrameReplace.test_replace_dict_category_typec C sH t dddg�}t�d�}|j|didd�}t dddg�}t�||� d S )Nr r r z^a$r# T)r> )r �re�compiler" r$ r% )r* rA r> r0 r1 r r r � test_replace_with_compiled_regexv s z5TestDataFrameReplace.test_replace_with_compiled_regexc C sZ t dt�dd�t�dd�gi�}|�dt�dd�dii�}t dddgi�}t�||� d S )Nr r rZ r )r rw �Intervalr" r$ r% )r* rA r0 r1 r r r �test_replace_intervals~ s z+TestDataFrameReplace.test_replace_intervalsc C sN ddddd�i}t dt�d�i�}|�|�}t dt�d�i�}t�||� d S )NZpositiverZ r )u 正面u 中立u 负面r4 )r r r� r" r$ r% )r* Zcolumns_values_maprz r0 r1 r r r �test_replace_unicode� s z)TestDataFrameReplace.test_replace_unicodec C s8 |dg�� d�}|�� }|�d tji�}t�||� d S )Nr7 z|S)r� r! r" r r r$ rn )r* rp rq r1 r r r �test_replace_bytes� s z'TestDataFrameReplace.test_replace_bytesz!data, to_replace, value, expectedg �boxc C s0 ||�}||�}|� ||�|�}t�||� d S )N)r" r$ rn ) r* ro r? r r1 r3 rp rq r0 r r r �!test_replace_list_with_mixed_type� s z6TestDataFrameReplace.test_replace_list_with_mixed_typeN)]�__name__� __module__� __qualname__r2 r� r � parametrizer r rB rI rJ rM rO rT r[ r] r^ r_ rb rc rj rm rw ZNArr r{ r| r� r� r� r� r� r� r� r� r� r� r� r� r Zint32Zint16r 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 Zint8r� r� r r r r r r� r/ r r* r r+ r. r0 r1 r2 r �tupler\ r4 r r r r r ! st )(E$ , & "( " S $ $ . > $ ( 3r c @ s� e Zd Zej�ded�ed�d�ed�eed��d�g�ej�ddej fdg�ej�d d d g�ej�d d d g�ej�d d d g�dd� �����Z dS )�TestDataFrameReplaceRegexro zab..r< )r r r zto_replace,valuez\s*\.\s*)z \s*(\.)\s*z\1\1\1� compile_regexTF� regex_kwargr c C s� t |�}|�� }|r t�|�}|r,|} d }nd} |j|||| d�} |rV| d ksRt�|} |tjkrhtj} nd} | |j|d dkdf< t � | |� d S )NT)r r> z...r rK ) r r! r, r- r" r# r r �locr$ r% ) r* ro r? r r; r< r rA r1 r> r0 Zexpected_replace_valr r r �test_regex_replace_scalar� s" z3TestDataFrameReplaceRegex.test_regex_replace_scalarN) r5 r6 r7 r� r r8 r r r r r> r r r r r: � s r: )� __future__r r �ior r, �numpyr r� Z pandas.compatr Zpandasrw r r r r r Zpandas._testingZ_testingr$ Zfixturer r r r: r r r r �<module> s0
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.06 |
proxy
|
phpinfo
|
Настройка