Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/extension/base/__pycache__/setitem.cpython-37.pyc
Назад
B �5�gQ/ � @ sD d dl Zd dlZd dlZd dlmZ d dlm Z G dd� de �Z dS )� N)�BaseExtensionTestsc @ s� e Zd Zejdd� dd� dd� dd� dd� dd� dd� gd d d d d ddgd�dd� �Zdd� Zdd� Zdd� Zdd� Z dd � Z ej � d d d g�d d!� �Z d"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zej j de�d.d.d.d/d/g�ejd.d.d.d/d/gd0d1�ejd.d.d.ejejgd0d1�gd2d3d4gd5�d6d7� �Zd8d9� Zd:d;� Zej j d<d=d>d?gejd=d>d?gd@d1�e�d=d>d?g�gdAdBd2gd5�dCdD� �Zej j dEd=d>d?ejgd/fej d=d>d?ejgd.ej j dFdG�dH�ejd=d>d?ejgd@d1�d/fejd=d>d?ejgd@d1�d/fgdIdJdKdLgd5�dMdN� �Z ej � dOd.d/g�ej � d d dPg�dQdR� ��Z ej � d d dPg�dSdT� �Z dUdV� Z!dWdX� Z"dYdZ� Z#d[d\� Z$d]d^� Z%d_d`� Z&dadb� Z'dcdd� Z(dedf� Z)dgdh� Z*didj� Z+dkdl� Z,dmdn� Z-dPS )o�BaseSetitemTestsc C s | j S )N)�index)�x� r �Y/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/extension/base/setitem.py�<lambda> � zBaseSetitemTests.<lambda>c C s t | j�S )N)�listr )r r r r r r c C s t d �S )N)�slice)r r r r r r c C s t dt| ��S )Nr )r �len)r r r r r r c C s t t| ��S )N)�ranger )r r r r r r c C s t tt| ���S )N)r r r )r r r r r r c C s t jt| �td�S )N)�dtype)�npZonesr �bool)r r r r r r r z list[index]Z null_sliceZ full_slicer z list(range)�mask)�params�idsc C s |j S )z� Fixture for an indexer to pass to obj.loc to get/set the full length of the object. In some cases, assumes that obj.index is the default RangeIndex. )�param)�self�requestr r r � full_indexer s z BaseSetitemTests.full_indexerc C s2 |rt �|�}|d |d< |d |d ks.t�d S )N� r )�pd�Series�AssertionError)r �data� box_in_seriesr r r �test_setitem_scalar_series' s z+BaseSetitemTests.test_setitem_scalar_seriesc C sZ |rt �|�}|�� }|d |d g|ddg< |d |d ksBt�|d |d ksVt�d S )Nr r )r r �copyr )r r r �originalr r r �test_setitem_sequence- s z&BaseSetitemTests.test_setitem_sequencec C s� t �|�}|�� }|d g}|r*|�|�}d}tjt|�d�d�� ||ddg< W d Q R X | �||� tjt|�d�d�� ||t d�< W d Q R X | �||� d S )Nr z5cannot set using a {} indexer with a different lengthz list-like)�matchr r � ) r r r Z_from_sequence�pytest�raises� ValueError�format�assert_series_equalr )r r Zas_array�serr �value�xprr r r �.test_setitem_sequence_mismatched_length_raises6 s z?BaseSetitemTests.test_setitem_sequence_mismatched_length_raisesc C s8 |rt �|�}|�� }g |tjg td�< | �||� d S )N)r )r r r r �array�int� assert_equal)r r r r r r r �test_setitem_empty_indexerG s z+BaseSetitemTests.test_setitem_empty_indexerc C sJ |rt �|�}|d |ddg< |d |d ks2t�|d |d ksFt�d S )N� r r )r r r )r r r r r r � test_setitem_sequence_broadcastsN s z1BaseSetitemTests.test_setitem_sequence_broadcasts�setter�loc�ilocc C s8 t �|�}t||�}|d |d< |d |d ks4t�d S )Nr r )r r �getattrr )r r r3 �arrr r r �test_setitem_scalarU s z$BaseSetitemTests.test_setitem_scalarc C sB t �t�t|��|d��}|d |jd< |jd |d ks>t�d S )N)�A�Br )r r: )r � DataFramer �aranger r4 r )r r �dfr r r � test_setitem_loc_scalar_mixed\ s z.BaseSetitemTests.test_setitem_loc_scalar_mixedc C s6 t �d|i�}|d |jd< |jd |d ks2t�d S )Nr: r )� r: )r r; r4 r )r r r= r r r � test_setitem_loc_scalar_singlea s z/BaseSetitemTests.test_setitem_loc_scalar_singlec C s8 t �||d��}|d |jd< |jd |d ks4t�d S )N)r9 r: r )r? r: )r r; r4 r )r r r= r r r �,test_setitem_loc_scalar_multiple_homogoneousf s z=BaseSetitemTests.test_setitem_loc_scalar_multiple_homogoneousc C sB t �t�t|��|d��}|d |jd< |jd |d ks>t�d S )N)r9 r: r )r r )r r: )r r; r r< r r5 r4 r )r r r= r r r � test_setitem_iloc_scalar_mixedk s z/BaseSetitemTests.test_setitem_iloc_scalar_mixedc C s6 t �d|i�}|d |jd< |jd |d ks2t�d S )Nr: r )r? r )r? r: )r r; r5 r4 r )r r r= r r r � test_setitem_iloc_scalar_singlep s z0BaseSetitemTests.test_setitem_iloc_scalar_singlec C s8 t �||d��}|d |jd< |jd |d ks4t�d S )N)r9 r: r )r? r )r? r: )r r; r5 r4 r )r r r= r r r �-test_setitem_iloc_scalar_multiple_homogoneousu s z>BaseSetitemTests.test_setitem_iloc_scalar_multiple_homogoneousTF�boolean)r z numpy-arrayz boolean-arrayzboolean-array-na)r c C sX |d d� � � }|�dddddg�}|r<t�|�}t�|�}|d ||< | �||� d S )N� r r# � )r �taker r r/ )r r r r r7 �expectedr r r �test_setitem_maskz s z"BaseSetitemTests.test_setitem_maskc C sz t �ddg�}|r t�|�}tjtdd�� |d ||< W d Q R X tj|dd�}tjtdd�� |d ||< W d Q R X d S )NTFz wrong length)r" r rE )r )r r- r r r$ r% � IndexError)r r r r r r r �test_setitem_mask_raises� s z)BaseSetitemTests.test_setitem_mask_raisesc C sn t jtj|jdd�dd�}d|d d�< t j|dd�< |rBt �|�}|d ||< |d d� |d k�� sjt�d S )Nr )r rE Tr# rF r ) r r- r �zeros�shape�NAr �allr )r r r r r r r �'test_setitem_mask_boolean_array_with_na� s z8BaseSetitemTests.test_setitem_mask_boolean_array_with_na�idxr r r1 ZInt64r z integer-arrayc C sX |d d� � � }|�dddddg�}|r<t�|�}t�|�}|d ||< | �||� d S )NrF r r# rG )r rH r r r/ )r r rR r r7 rI r r r �test_setitem_integer_array� s z+BaseSetitemTests.test_setitem_integer_arrayzidx, box_in_serieszGH-31948)�reason)Zmarksz list-Falsez list-Truezinteger-array-Falsezinteger-array-Truec C sZ |� � }|r,tj|dd� tt|��D �d�}d}tjt|d�� |d ||< W d Q R X d S )Nc S s g | ]}t �d ��qS )rG )�tmZrands)�.0�_r r r � <listcomp>� s zMBaseSetitemTests.test_setitem_integer_with_missing_raises.<locals>.<listcomp>)r z9Cannot index with an integer indexer containing NA values)r" r )r r r r r r$ r% r& )r r rR r r7 �msgr r r �(test_setitem_integer_with_missing_raises� s z9BaseSetitemTests.test_setitem_integer_with_missing_raises� as_callableNc s� t �|�}tjt|�td�� d� d d�< |r:� fdd�}n� }|rNt||�}n|}|dd� ||< |dd� ||< |d |d ks�t�|d |d ks�t�d S ) N)r Tr1 c s � S )Nr )r )r r r r � r z<BaseSetitemTests.test_setitem_mask_aligned.<locals>.<lambda>rF � r r � )r r r rM r r r6 r )r r r[ r3 r) Zmask2�targetr )r r �test_setitem_mask_aligned� s z*BaseSetitemTests.test_setitem_mask_alignedc C st t �|�}tjt|�td�}d|d d�< |r8t||�}n|}|d ||< |d |d ks\t�|d |d kspt�d S )N)r Tr1 r? r r )r r r rM r r r6 r )r r r3 r) r r^ r r r �test_setitem_mask_broadcast� s z,BaseSetitemTests.test_setitem_mask_broadcastc C s� t �d|i�}|�� }d|d< t �|dgt|� d��}| �||� |�� }d|jd d �df< | �||� ||d< t �||d��}| �||� d S )Nr9 r r: )r9 r: )r r; r r �assert_frame_equalr4 )r r r= �resultrI r r r �test_setitem_expand_columns� s z,BaseSetitemTests.test_setitem_expand_columnsc C sx t �ddgt|� i�}|�� }||d< t �dgt|� |d��}| �||� |�� }||jd d �df< | �||� d S )Nr9 r r: )r9 r: )r r; r r ra r4 )r r r= rb rI r r r �"test_setitem_expand_with_extension s z3BaseSetitemTests.test_setitem_expand_with_extensionc C sh t �ddgt|� i�}dt|d d� �� dt|�� d�}tjt|d�� |d d� |d< W d Q R X d S ) Nr9 r zLength of values \(rF z$\) does not match length of index \(z\))r" r: )r r; r r$ r% r& )r r r= r+ r r r �!test_setitem_frame_invalid_length s "z2BaseSetitemTests.test_setitem_frame_invalid_lengthc C sP t j|d d� ddgd�}t j|�ddg�|jd�}|d |d< | �||� d S )Nr1 )r r )r r )r r )r r rH r r( )r r r) rI r r r �test_setitem_tuple_index s z)BaseSetitemTests.test_setitem_tuple_indexc C s\ |d d� � � }|�dddddg�}|r<t�|�}t�|�}|d |d d�< | �||� d S )NrF r r# rG )r rH r r r/ )r r r r7 rI r r r �test_setitem_slice s z#BaseSetitemTests.test_setitem_slicec C s� |d d� � � }tj|dddddgd�}tj|�dddd d g�|jd�}|� � }|d |jd d �< | �||� |� � }|d |jd d�< | �||� d S ) NrF �a�b�c�d�e)r r r# rG )r r r rH r r5 r/ r4 )r r r7 �srI rb r r r �test_setitem_loc_iloc_slice* s z,BaseSetitemTests.test_setitem_loc_iloc_slicec C s: |d d� }t �t�� |d d� |d d�< W d Q R X d S )NrF r1 r )r$ r% r& )r r r7 r r r �)test_setitem_slice_mismatch_length_raises7 s z:BaseSetitemTests.test_setitem_slice_mismatch_length_raisesc C s< |d d� � � }|dd � |d d�< | �||dd � � d S )NrF ���)r Z assert_extension_array_equal)r r r7 r r r �test_setitem_slice_array< s z)BaseSetitemTests.test_setitem_slice_arrayc C s: |d d� � � }t�t�� |ddg |d< W d Q R X d S )NrF r r )r r$ r% r& )r r r7 r r r �&test_setitem_scalar_key_sequence_raiseA s z7BaseSetitemTests.test_setitem_scalar_key_sequence_raisec C sL |� � }|d d � }|d |d< |d |d ks4t�|d |d ksHt�d S )Nr r )�viewr )r r Zview1Zview2r r r � test_setitem_preserves_viewsF s z-BaseSetitemTests.test_setitem_preserves_viewsc C sP t �dt �|�i� }}t j|jd�}||�}|d |j|df< | �||� d S )Nr )r )r r; r r r4 ra )r r r r= rI rb �keyr r r �,test_setitem_with_expansion_dataframe_columnO s z=BaseSetitemTests.test_setitem_with_expansion_dataframe_columnc C s\ t j|dd�}t j|jtdd�}||�}||j|< t j|�t�|jdtd�}| �||� d S )Nr )�name)r r rw )r rw r )r r r �objectr4 Zastyper( )r r r r) rb ru rI r r r �test_setitem_seriesY s z$BaseSetitemTests.test_setitem_seriesc C sF t j|dd�}t�t|��}t�|d�}|| }|d= | �||� d S )Nr )rw r )r r r r< r �deleter( )r r r) ZtakerrI r r r �test_delitem_seriesh s z$BaseSetitemTests.test_delitem_series).�__name__� __module__� __qualname__r$ Zfixturer r r! r, r0 r2 �markZ parametrizer8 r> r@ rA rB rC rD r r- r rO rJ rL rQ rS r ZxfailrZ r_ r` rc rd re rf rg rn ro rq rr rt rv ry r{ r r r r r s~ * r ) �numpyr r$ Zpandasr Zpandas._testingZ_testingrU Z pandas.tests.extension.base.baser r r r r r �<module> s
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка