Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexing/__pycache__/test_partial.cpython-37.pyc
Назад
B �5�g][ � @ sp d Z ddlZddlZddlm mZ ddlZ ddlm Z m Z m Z m Z mZmZmZ ddlmZ G dd� d�ZdS )zx test setting *parts* of objects both positionally and label based TODO: these should be split among the indexer tests � N)� DataFrame�Index�Period�Series� Timestamp� date_range� period_rangec @ s e Zd Zdd� Zejdd� �Zdd� Zdd� Ze j � d d d g�d d � �Z dd� Z dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Ze j � d$ed%d&d'd(�d)d*d+ged)d'd,�ed*d'd,�ed+d'd,�gfed%d&d'd(�d)d*d+ged)d'd,�ed*d'd,�ed+d'd,�gfe j d-d&d.�d/d0d1ge � d2�e � d3�e � d4�gfg�d5d6� �Z e j � d7ed%d&d'd(�d)d8gfed%d&d'd(�d)d8gfe j d-d&d.�d9d:gfg�d;d<� �Z e j � d=ed%d&d'd(�d/d0gd>fed%d&d'd(�d/d0gd>fe j d-d&d.�d)d*gd?fg�d@dA� �Z!dBdC� Z"dDdE� Z#dFS )G�TestPartialSettingc C s t dddg�}|�� }d|d< t ddddgddddgd�}t�||� |�� }d|jd< t ddddgddddgd�}t�||� |�� }d|d< t ddddgddddgd�}t�||� |�� }d|jd< t ddddgddddgd�}t�||� |�� }d}tjt|d �� d|jd< W d Q R X d }tjt|d �� d|j d< W d Q R X t t � d �� dd�d d gdd�}|�� }d}tjt|d �� d|jd< W d Q R X d}tjt|d �� d|j d< W d Q R X t tdddgdddgd���}|�� }|jd |jd< t�||� t tdddgdddgd���}|�� }|jd |jd< t�||� t tddddgddddgd���}|�� }|jd |jd< t�||� t tdddgdddgd���}|�� }|jd d �d f |jd d �d f< t�||� t tdddgt dddg�d���}|�� }|d �t j�|d < |jd d �d f |jd d �d f< t�||� |�� }|d |d< |�� }|jd d �d f |jd d �df< t�||� |�� }|d |d< |�� }|jd d �d f |jd d �df< t�||� tddd�}t t j�dd�|d d ddgd�}tj|t d di|dd � |j d�gd d �}|�� }d|j|d |j d f< t�||� |�� }d|j|d |j d f< t�||� t ddi|dd � |j d�}tj||gdd �}|�� }d|j|d |j df< t�||� |�� }d|j|d |j df< t�||� d S ) N� � � � r )�indexg @z%iloc cannot enlarge its target object)�matchz/index 3 is out of bounds for axis 0 with size 3� �A�B�int64)�columns�dtype)� r z/index 2 is out of bounds for axis 0 with size 2r )r r �Cz1/1/2000� )�periods�D)r r � ���T)�sort)�axis)r �copy�tm�assert_series_equal�loc�pytest�raises� IndexError�ilocZiatr �np�arangeZreshape�dict�assert_frame_equal�astype�float64r �randomZrandn�pd�concat�freq�at) �selfZs_orig�s�expected�msgZdf_orig�df�datesZ exp_other� r8 �X/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexing/test_partial.py�test_partial_setting s� " " ( z'TestPartialSetting.test_partial_settingc C sh t ddgddggddgd�}|jd �� }d|_|�|�}|jd |jd< t�||� t dd gd�}tdtd �d �|jd < t�|t dd gd gd �� t dd gd�� � }tdd gd �|jd < t t j dggdd gd gdd�}t�||� t dd gd�}d}t j t|d�� dddg|jd < W d Q R X t dd gd�}ddg|jd< t ddggdgdd gdd�}t�||� d S )NTr Fr ZfemaleZfitness)r r r r )r r )r r r, )r r r z(cannot set a row with mismatched columns)r r r r �object)r r r )r r" r �name�appendr r* r �rangeZ _consolidater' �nanr# r$ � ValueError)r2 r6 r3 r4 �expr5 r8 r8 r9 � test_partial_setting_mixed_dtype� s* z3TestPartialSetting.test_partial_setting_mixed_dtypec C s� t ddgddgd�}t tjdtjgdddgd�}tjtdd�� |jdddg W d Q R X |�dddg�}tj ||d d � t tjdtjtjgdddd gd�}tjtdd�� |jdddd g W d Q R X |�dddd g�}tj ||d d � t dddgdddgd�}|jdddg }tj ||d d � t ddtjdgddd dgd�}tjtdd�� |jddd dg W d Q R X |�ddd dg�}tj ||d d � d }tjt|d�� |jdddg W d Q R X t ddtjgdddgd�}tjtdd�� |jdddg W d Q R X |�dddg�}tj ||d d � t ddd gdddgd�}t d tjtjgdddgd�}tjtdd�� |jdddg W d Q R X |�dddg�}tj ||d d � t ddd dgddddgd�}t tjd d gdddgd�}tjtdd�� |jdddg W d Q R X |�dddg�}tj ||d d � t ddd dgddddgd�}t tjddgdddgd�}tjtdd�� |jdddg W d Q R X |�dddg�}tj ||d d � t ddd dgddddgd�}t dtjtjgdddgd�}tjtdd�� |jdddg W d Q R X |�dddg�}tj ||d d � t ddd dgddddgd�}t dtjtjgdddgd�}tjtdd�� |jdddg W d Q R X |�dddg�}tj ||d d � t ddddgddddgd�}|j ddddg }tj ||d d � d S )Ng�������?g�������?r r )r r z not in index)r T)�check_index_type�xzM\"None of \[Int64Index\(\[3, 3, 3\], dtype='int64'\)\] are in the \[index\]\"g333333�?r g�������?r r r r ) r r' r? r# r$ �KeyErrorr" �reindexr r! r& )r2 �serr4 �resultr5 r3 r8 r8 r9 �test_series_partial_set� sx " z*TestPartialSetting.test_series_partial_setc C s. t ddgddd�}tddg|dd �}tjtd d �� |jd dd g W d Q R X tjtd d �� |jd dd dg W d Q R X t dddgddd�}tdddg|dd �}|jdddg }tj||dd� tjtdd �� |jddddg W d Q R X d}tjt|d �� |jd d d g W d Q R X tjtd d �� |jddd g W d Q R X t ddd gddd�}tjtd d ��& tdddg|dd �jd ddg W d Q R X t ddd dgddd�}tjtd d ��( tddddg|dd �jdd d g W d Q R X t ddd dgddd�}tjtd d ��( tddddg|dd �jdddg W d Q R X t ddddgddd�}tjtd d ��( tddddg|dd �jdddg W d Q R X t ddd dgddd�}tjtd d ��( tddddg|dd �jdddg W d Q R X t ddddgddd�}tddddg|dd �}|jddddg }tj||dd� d S )Nr r r �idx)r r< g�������?g�������?r3 )r r< z\[3\] not in index)r r z not in indexrD T)rC z\['x'\] not in indexzY\"None of \[Int64Index\(\[3, 3, 3\], dtype='int64', name='idx'\)\] are in the \[index\]\"g333333�?r g�������?r r r r ) r r r# r$ rE r" r r! r& )r2 rJ rG Zexp_idxr4 rH r5 r8 r8 r9 �!test_series_partial_set_with_name sH *,,,,z4TestPartialSetting.test_series_partial_set_with_name�key�d g Y@c C s� t �� }|�� }|jd |j|d d �f< tt|j�|g t|jj d�}t j |j |jdg j gdd�}t |||jd�}t �||� d S )Nr )r r< )r )r r )r �makeTimeDataFramer r& r" r �listr r; r< r' Z concatenate�valuesr r r* )r2 rL �origr6 Zex_indexZex_datar4 r8 r8 r9 �6test_setitem_with_expansion_numeric_into_datetimeindexX s zITestPartialSetting.test_setitem_with_expansion_numeric_into_datetimeindexc C s~ t �� }|�� }|jd |jdd d �f< |�t|jd dd��}t �||� t �|j t |j � � dg �� |j j dkszt �d S )Nr �a)r< r; )r rN r r& r" r= r r* �assert_index_equalr r �tolistr �AssertionError)r2 rQ r6 rA r8 r8 r9 �test_partial_set_invalidd s z+TestPartialSetting.test_partial_set_invalidc C s� t � }d}tjt|d�� d|jd< W d Q R X tjt|d�� tdgdgd�|jd< W d Q R X d}tjt|d�� d|jd d �df< W d Q R X d S )Nz*cannot set a frame with no defined columns)r r �foo)r z5cannot set a frame with no defined index and a scalar)r r# r$ r@ r" r )r2 r6 r5 r8 r8 r9 � test_partial_set_empty_frames s z/TestPartialSetting.test_partial_set_empty_framec C s� t dgtg dd�d�}t tg dd�d�}tg dd�|d< t�||� t � }t|j�|d< t�||� t � }|j|d< t�||� d S )NrX r; )r )r r )r )r r r r r* r )r2 r4 r6 r8 r8 r9 � test_partial_set_empty_frame2� s z0TestPartialSetting.test_partial_set_empty_frame2c C s� t dgtg dd�d�}|d �d�|d< t tg dd�d�}g |d< t�||� t tg dd�d�}tt�t|��dd�|d< t�||� d S )NrX r )r )r r r, )r ) r r r+ r r* r r' r( �len)r2 r4 r6 r8 r8 r9 � test_partial_set_empty_frame3� s z0TestPartialSetting.test_partial_set_empty_frame3c C sZ t tg dd�d�}tt|��|d< t dgtg dd�d�}|d �d�|d< t�||� d S )Nr )r )r rX )r r )r r r> r[ r+ r r* )r2 r6 r4 r8 r8 r9 � test_partial_set_empty_frame4� s z0TestPartialSetting.test_partial_set_empty_frame4c C s� t � }t�|jtg td�� t � }tdgdgd�|d< tdgdgd�|jd d �df< t�|t dggdgdgd�� t�||� d S )N)r r rX )r )r r ) r r rT r r r; r r" r* )r2 r6 Zdf2r8 r8 r9 � test_partial_set_empty_frame5� s z0TestPartialSetting.test_partial_set_empty_frame5c C s� t dtdtd�d�idddgd�}t ddgd�}tdtd�d�|d< |j t|� t�||� t ddgd�}tdtd�d�|jd d �df< |j t|� t�||� d S )Nr r r )r r r )r )r r r> Zdtypes�strr r* r" )r2 r4 r6 r8 r8 r9 �%test_partial_set_empty_frame_no_index� s " z8TestPartialSetting.test_partial_set_empty_frame_no_indexc C s� t dddgtg dd�d�}|d �d�|d< |d �d�|d< |d �d�|d< t dd d gd d d gd��}||jdk }tj|d< t�||� t ddddgd�}|d �d�|d< t dddgd�}d |d< t�||� t�|d t dt d�� t dd d gd d d gd��}||jdk }|j dddgd�}t dddgtg dd�d�}|d �d�|d< |d �d�|d< |d �d�|d< t�||� d S )Nr r ZNewr )r )r r r, r r r g333333�?g������@g������@)r r r rS �bzc c�d)r )r< r r ) r r r+ r r' r? r r* r! r r; rF )r2 r4 r6 �yrH r8 r8 r9 � test_partial_set_empty_frame_row� s, z3TestPartialSetting.test_partial_set_empty_frame_rowc C s^ t ttd��}t dttd�i�}t�||� t tdtd��}t dttd�i�}t�||� d S )N)r r rX )r< r )r r r; r r* )r2 r6 r4 r8 r8 r9 �'test_partial_set_empty_frame_set_series� s z:TestPartialSetting.test_partial_set_empty_frame_set_seriesc C s> t dgd�}|�� }d|d< t ddgdgd�}t�||� d S )Nr )r rS )r r )r r r r* )r2 r6 r4 r8 r8 r9 �2test_partial_set_empty_frame_empty_copy_assignment� s zETestPartialSetting.test_partial_set_empty_frame_empty_copy_assignmentc C s� t ddgd�}ddg|d< t ddgtjtjgd��}tj||dd� t ddgd�}d d g|d< t d d gtjtjgd�td �}t�||� t ddgd�}d|jd < t dgtjgd��}tj||dd� d S ) NrD rc )r r r )rD rc F)Z check_dtype�1�2)r )r rD )r r' r? r r* r; r" )r2 r6 r4 r8 r8 r9 �0test_partial_set_empty_frame_empty_consistencies� s zCTestPartialSetting.test_partial_set_empty_frame_empty_consistencieszidx,labels,expected_idx�2000� r )�startr r0 z 2000-01-04z 2000-01-08z 2000-01-12)r0 z1 day)rl r Z4DZ8DZ12Dz4 dayz8 dayz12 dayc C sR |t d�|d�}dddg}|||�}t�||j| � |tkrNt�||| � d S )Nrk )r r r � )r> r Z assert_equalr" r r! )r2 rJ �labelsZ expected_idxZframe_or_series�objZexpected_valuer4 r8 r8 r9 �4test_loc_with_list_of_strings_representing_datetimes s zGTestPartialSetting.test_loc_with_list_of_strings_representing_datetimesz idx,labelsz 2000-01-30z3 dayz30 dayc C s� t td�|d�}ttd�|d�}d}tjt|d�� |j| W d Q R X tjt|d�� || W d Q R X tjt|d�� |j| W d Q R X d S )Nrk )r z not in index)r )r r> r r# r$ rE r" )r2 rJ rn r3 r6 r5 r8 r8 r9 �Btest_loc_with_list_of_strings_representing_datetimes_missing_value8 s zUTestPartialSetting.test_loc_with_list_of_strings_representing_datetimes_missing_valuezidx,labels,msgzHNone of \[Index\(\['4D', '8D'\], dtype='object'\)\] are in the \[index\]zXNone of \[Index\(\['2000-01-04', '2000-01-08'\], dtype='object'\)\] are in the \[index\]c C s� t td�|d�}ttd�|d�}tjt|d�� |j| W d Q R X tjt|d�� || W d Q R X tjt|d�� |j| W d Q R X d S )Nrk )r )r )r r> r r# r$ rE r" )r2 rJ rn r5 r3 r6 r8 r8 r9 �Etest_loc_with_list_of_strings_representing_datetimes_not_matched_typeU s !zXTestPartialSetting.test_loc_with_list_of_strings_representing_datetimes_not_matched_typec C s� t i tjddd�d�}tdtjddd�d�}||d< t ddgd itjddd�d�}t�||� t � }tdtjddd�d�}||d< t ddgd itjddd�d�}t�||� d S ) Nr Zdf_index)r< )r g�G�z��?r Z series_index�series)r r. Z RangeIndexr r r* )r2 r6 rs r4 r8 r8 r9 �test_index_name_empty� s z(TestPartialSetting.test_index_name_emptyc C sf t �dddd g�}ttt|��|d�}ttt|d d� ��|d d� d�}|dd� }t�||� d S )Nz 2012-01-01z 2012-01-02z 2012-01-03)r r z 2012-01-04)r. � to_datetimer r> r[ r r* )r2 r r6 r4 rH r8 r8 r9 �,test_slice_irregular_datetime_index_with_nan� s $ z?TestPartialSetting.test_slice_irregular_datetime_index_with_nanN)$�__name__� __module__� __qualname__r: �tdZ&skip_array_manager_not_yet_implementedrB rI rK r# �markZ parametrizerR rW rY rZ r\ r] r^ r` rd re rf ri r r r r r. Ztimedelta_rangeZ Timedeltarp rq rr rt rv r8 r8 r8 r9 r sh )[: * " r )�__doc__�numpyr' r# Z pandas.util._test_decorators�utilZ_test_decoratorsrz Zpandasr. r r r r r r r Zpandas._testingZ_testingr r r8 r8 r8 r9 �<module> s $
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.1 |
proxy
|
phpinfo
|
Настройка