Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexing/__pycache__/test_iloc.cpython-37.pyc
Назад
B �5�g � � @ s d Z ddlmZ ddlZddlmZmZ ddlZddlZddl m m Z ddl mZmZmZmZmZmZmZmZmZmZmZmZ ddlmZ ddl m Z ddl m Z ddl!m"Z" e�#d �Z$G d d � d e"�Z%G d d � d �Z&G dd� d�Z'G dd� d�Z(G dd� d�Z)G dd� d�Z*dS )z* test positional based indexing with iloc � )�datetimeN)�catch_warnings� simplefilter) �NA� Categorical�CategoricalDtype� DataFrame�Index�Interval�NaT�Series�array�concat� date_range�isna)� is_scalar)� IndexingError)�Basezuonly integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`) and integer or boolean arrays are valid indicesc @ s. e Zd Zej�ddddddgg�dd� �ZdS ) �TestiLoc�key� ���r � c C s | j d|dddddgtd� d S )N�iloc�labels�mixed�tsZfloats�empty)ZtypsZfails)Z check_result� IndexError)�selfr � r �U/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexing/test_iloc.py�"test_iloc_getitem_int_and_list_int* s z+TestiLoc.test_iloc_getitem_int_and_list_intN)�__name__� __module__� __qualname__�pytest�mark� parametrizer" r r r r! r ) s r c @ sj e Zd ZdZej�ded�ed�ed�dddge ed��e � dddg�g�ej�de j e jg�d d � ��Zejej�d eeg�d d � ��Zdd� Zdd� Zej�de �d�ed�fg�ej�ded�ddggddged�feddd�ged�fg�dd � ��Zej�d ddg�d d � �Zd d!� Z d"d#� Z d$d%� Z d&d'� Z d(d)� Z ej�d*d+d,gd+d,d+d,gg�d-d.� �Z!d/d0� Z"d1d2� Z#d3d4� Z$d5d6� Z%d7d8� Z&d9d:� Z'd;d<� Z(d=d>� Z)d?d@� Z*dAdB� Z+dCdD� Z,dEdF� Z-ej�ddgeddd�e �dg�g�ej�dGdHge �dHg�g�dIdJ� ��Z.dKdL� Z/dMdN� Z0dOdP� Z1dQdR� Z2dSdT� Z3dUdV� Z4ejj5dWdX�dYdZ� �Z6d[d\� Z7d]d^� Z8d_d`� Z9ej�dGde:e j;g�dadb� �Z<dcdd� Z=ej�de j e jg�dedf� �Z>dgdh� Z?ejdidj� �Z@ej�dkee jg�dldm� �ZAej�ddgedd�g�dndo� �ZBdpdq� ZCdrds� ZDdtdu� ZEdvdw� ZFdxdy� ZGdzd{� ZHej�d|ee jg�ej�d}ee jg�d~d� ��ZIdS )��TestiLocBaseIndependentz Tests Independent Of Base Classr N� r r r �indexerc C sd t dtd�itd�}tdddg�}|s<|jjd �|�s<t�|�� }|j }|||�|df< t |t �on|t d �k}|sx|r�t d|i�} t � |j |�r�t�n,t d|i��t�} |s�t � |d j |�s�t�t�|| � d|jd< |r�|d dks�t�n|d dk�st�t t jddd gtd�td�d ��}|�� }|j }|||�|df< t |td�d ��} t�|| � d S ) Nr r* )�dtype�alpha�beta�gamma)r r r r )r r )r �range�objectr �_mgr�blocksZ_can_hold_element�AssertionError�copy�values� isinstance�slice�npZ shares_memory�astype�tm�assert_frame_equalr r ) r r+ r �using_array_manager�frame�cat�dfZ orig_vals� overwrite�expectedr r r! �%test_iloc_setitem_fullcol_categorical: s2 "z=TestiLocBaseIndependent.test_iloc_setitem_fullcol_categorical�boxc C s� t ddddg�}||�d��}|tks*|s2|j}n |d j}|tkr`||dd � �|jd d�< n ||dd � �|jd d�df< |tj ddddgdd��}t�||� |tkr�|j|ks�t�n4|r�|d j|ks�t�n |jj |j kr�|j d k s�t�d S )Nr r r* � �i8r )r, ) r Zto_numpyr r6 r r9 r; Z assert_equalr4 �base)r �frame_or_seriesrD r= �arr�objr6 rB r r r! � test_iloc_setitem_ea_inplaceq s z4TestiLocBaseIndependent.test_iloc_setitem_ea_inplacec C sJ t dddg�}ttd�|d�}|j�d�s.t�|�� }|j�d�sFt�d S )Nr r r* )�index)r )r r )r r r0 r Z_is_scalar_accessr4 �to_frame)r rL �serr@ r r r! �test_is_scalar_access� s z-TestiLocBaseIndependent.test_is_scalar_accessc C s� t tj�d�td�d�}d}tjt|d��$ |jd d �dddd d d gf W d Q R X tjt|d�� |jdd g W d Q R X tjt|d�� |jdd g W d Q R X tjt|d�� |jdg W d Q R X |d }tjt|d�� |jdg W d Q R X tjt|d�� |jdg W d Q R X d}tjt|d�� |jd W d Q R X tjt|d�� |jd W d Q R X tjt|d�� |jd W d Q R X tjt|d�� |jd W d Q R X |jd d �d d�f }|jd d �d d �f }t � ||� |jd d �dd�f }|jd d �d d�f }t � ||� |jd d �dd d�f }|jd d �d d d�f }t � ||� |jd d �d dd�f }|jd d �d d d�f }t � ||� |jd d �dd �f }|jd d �d d �f }t � ||� |jd d �dd �f }|jd d �d d�f }t � ||� |jd d �ddd�f }|jd d �d d�f }t � ||� |jd d �dd�f }|jd d �d d�f }t � ||� |jdd � }|jdd � }t � ||� |jd d � }|jd d� }t � ||� |jd d d� }|jd d d� }t � ||� dd� }t tj� d d�td�d�}||jd d �dd �f t |j d �� ||jd d �dd �f |jd d �dgf � ||jd d � |jd g � d}tjt|d�� |jd d d g W d Q R X d}tjt|d�� |jd d �d f W d Q R X d S ) N)� � �ABCDE)�columnsz%positional indexers are out-of-bounds)�matchr r r r* rE rQ � i���d �Ai����z*single positional indexer is out-of-bounds� ���i����r i����� � c S s t | � | j t�| |� d S )N)�str�dtypesr; r< )�resultrB r r r! �check� s z?TestiLocBaseIndependent.test_iloc_exceeds_bounds.<locals>.checkZAB)rL � )r r9 �randomZ random_sample�listr& �raisesr r r; r< �assert_series_equal�randnrL )r r@ �msg�sr^ rB r_ Zdflr r r! �test_iloc_exceeds_bounds� s� ( $, z0TestiLocBaseIndependent.test_iloc_exceeds_boundsz index,columnsrP rR zindex_vals,column_valsrW �D�1�2i� c C sP t tj�t|�t|��||d�}d}tjt|d�� |j||f W d Q R X d S )N)rL rS z$.iloc requires numeric indexers, got)rT ) r r9 ra re �lenr& rc r r )r rL rS Z index_valsZ column_valsr@ rf r r r! � test_iloc_non_integer_raises� s z4TestiLocBaseIndependent.test_iloc_non_integer_raises�dimsc C sV |dkrt t�d��}ntt�d��dd��}tjtdd�� |jd W d Q R X d S )Nr rX rV z Cannot index by location index)rT �a) r r9 �aranger �reshaper& rc � TypeErrorr )r rn rg r r r! � test_iloc_getitem_invalid_scalar s z8TestiLocBaseIndependent.test_iloc_getitem_invalid_scalarc C s| t �dddg�}|�� }tdddgddd gd d d gd �dddgd�}|j| t�||� |jd d �|f t�||� d S )Nr r r rV �e �f �g �h �i �j �k �l )rW �B�Cr* )rL )r9 r r5 r r r; Zassert_numpy_array_equal)r Zarray_with_neg_numbersZ array_copyr@ r r r! �-test_iloc_array_not_mutating_negative_indices s zETestiLocBaseIndependent.test_iloc_array_not_mutating_negative_indicesc C s� t dddgdddgd��}|d }|jd }|jd }t�||� |jd g }|jd g }t�||� |jd }|jd }||ks�t�|jd g }|jd g }t�||� td gdgd �}|jd g }t�||� d S )Nr r* rQ � rZ � )rW r| rW r ���ro )rL r )r r r; rd r< r4 r )r r@ rg rB r^ r r r! �/test_iloc_getitem_neg_int_can_reach_first_index+ s" zGTestiLocBaseIndependent.test_iloc_getitem_neg_int_can_reach_first_indexc C s� t d dd�ddd�g�}t ddd�ddd�g�}t||gdd�}|jd }t|�sVt�|jdd d �f }ttjdddgd d d d gdd �}t� ||� d S ) Nr )rW r| r r* rE )�axis)r r r rW r| )rL �name) r r r r r4 r r9 �nanr; rd )r �df1�df2r@ r^ rB r r r! �test_iloc_getitem_dupsF s z.TestiLocBaseIndependent.test_iloc_getitem_dupsc C s� t dddd�dddd�dd d d�g�}t dddd�g�}t�|jd g |� t dddd�dddd�g�}t�|jd dg |� t ddd �d d d �gd dgd �}|jd dgddgf }t�||� d S )Nr r r* )rW r| r} rV �� i, i� i� i� r )r| r} )rL )r r; r< r )r r@ rB r^ r r r! �test_iloc_getitem_arrayT s z/TestiLocBaseIndependent.test_iloc_getitem_arrayc C s� t dddd�dddd�dd d d�g�}t dddd�dddd�g�}|jd d d g }t�||� t dddd�dd d d�gd dgd�}|jdd� }t�||� d S )Nr r r* )rW r| r} rV r� i, i� i� i� TFr )rL c S s | j d dkS )Nr r )rL )�xr r r! �<lambda>w � z@TestiLocBaseIndependent.test_iloc_getitem_bool.<locals>.<lambda>)r r r; r< )r r@ rB r^ r r r! �test_iloc_getitem_boolg s "z.TestiLocBaseIndependent.test_iloc_getitem_boolrL TFc C sN t dddg�}dt|�� dt|�� �}tjt|d�� |j| W d Q R X d S )Nr r r* z Boolean index has wrong length: z instead of )rT )r rl r& rc r r )r rL rg rf r r r! � test_iloc_getitem_bool_diff_lenz s z7TestiLocBaseIndependent.test_iloc_getitem_bool_diff_lenc C s� t dddd�dddd�dd d d�g�}t dddd�dddd�g�}|jd d� }t�||� t ddd �gdgd �}|jdd�d d�f }t�||� t ddd�ddd�dd d�g�}|jd d �dd� f }t�||� d S )Nr r r* )rW r| r} rV r� i, i� i� i� )rW r| )rL r )rW r} c S s ddgS )Nr r r )r@ r r r! r� � r� zATestiLocBaseIndependent.test_iloc_getitem_slice.<locals>.<lambda>)r r r; r< )r r@ rB r^ r r r! �test_iloc_getitem_slice� s z/TestiLocBaseIndependent.test_iloc_getitem_slicec C s� t tj�dd�ddddgd�}t tjjdddd��dd �dd gd�}t||gd d �}t�|j d d �d d�f |� t�|j d d �dd �f |� t||gd d �}t�|j d d �d d �f |� t�|j d d �d d �f |� t||j d d �dgf gd d �}t�|j d d �dd �f |� t||gdd �}t�|j dd�d d �f |� t�|j dd�d d �f |� t�|j dd �d d �f |� t�|j dd �d d �f |� d S )NrX rE rW r| )rS r rP )�sizer r} r )r� r* ) r r9 ra re �randintrq r r; r< r )r r� r� r@ �expr r r! � test_iloc_getitem_slice_dups� s $ z4TestiLocBaseIndependent.test_iloc_getitem_slice_dupsc C s� t tj�dd�t�ddd�t�ddd�d�}d|jd < |jd }|dksLt�d|jd d �dd�f< |jd d �dd�f }|jd d �dd�f }t�||� t ddd d gd �}|jdd� d7 < t dddgdd d gd �}t� ||� d S ) NrE r � r � r* )rL rS r )r r rQ r` )rL ) r r9 ra re rp r r4 r; r< r rd )r r@ r^ rB rg r r r! �test_iloc_setitem� s * z)TestiLocBaseIndependent.test_iloc_setitemc C s� t t�d��d�dddgdddgd�}|jddgdd gf |jddgdd gf d 7 < t t�dd d d dddddg ��d�dddgdddgd�}t�||� d S )N� )r* r* rW r| r} )rL rS r r r rV rt ru r* rw rx r` r r� )r r9 rp rq r r r; r< )r r@ rB r r r! �test_iloc_setitem_list� s $ z.TestiLocBaseIndependent.test_iloc_setitem_listc C sx t ddddg�}t ddddg�}|�� }ddg|jt ddg�< t�||� |�� }ddg|jtddg�< t�||� d S )Nr r r r* r ���)r r5 r r; rd r )r Zs_origrB rg r r r! � test_iloc_setitem_pandas_object� s z7TestiLocBaseIndependent.test_iloc_setitem_pandas_objectc C sf t d dd�ddd�g�}t ddd�ddd�g�}t||gdd�}|�d�}t�|jd d �df �}|| j}|j|df |j|df< t�||� t ddgddgd��}d d g|_ |d = t�||� |jddgddgf |jddgddgf< t�||� |jddgddgf j d d �|jddgddgf< |jddgddgf j d d �|jddgddgf< t�||� d S ) Nr )rW r| r r* rE )r� r )r r r| rW T)Zdrop) r r Zfillnar9 �isnanr rL r; r< rS Z reset_index)r r� r� r@ rB Zinds�maskr r r! �test_iloc_setitem_dups� s" ( 00z.TestiLocBaseIndependent.test_iloc_setitem_dupsc C s� t ddgddggddgd�}|jd d �df �d�|jd d �df< |sZt|jj�dksZt�|�� }|jddgddgf |jddgddgf< t� ||� d S )Nr r r r* r| )rS Zf8) r r r: rl r2 r3 r4 r5 r; r< )r r= r@ rB r r r! �9test_iloc_setitem_frame_duplicate_columns_multiple_blocks s &(zQTestiLocBaseIndependent.test_iloc_setitem_frame_duplicate_columns_multiple_blocksc C s� t tj�dd�tddd�tddd�d�}|jd }|jd }t�||� |jd }|jd }||ksht �|jdd� }|jdd � }t� ||� |jd d �dd �f }|jd d �dd �f }t� ||� |jdd d g }|jdddg }t� ||� |jdd d gdd gf }|jdddgddgf }t� ||� |jdd d gdd gf }|jdddgddgf }t� ||� |jddd d gdd gf }|jddddgddgf }t� ||� t td d �t d�}|j|j }|jddddg }t� ||� d S )NrX rE r rP r r� )rL rS )r r )rE rE � r* rQ r r` r r[ )rL r, )r r9 ra re r0 r �locr; rd r4 r< r r1 rL )r r@ r^ r� rB rg r r r! �test_iloc_getitem_frame s< & z/TestiLocBaseIndependent.test_iloc_getitem_framec C s� t tj�dd�td�td�d�}|jd }|jd }||ks@t�|jd d �dd �f }|jd d �d gf }t� ||� |jd }|jd }||ks�t�d }t j t |d�� |jd W d Q R X d}t j t |d�� |jd W d Q R X d S )NrX rE � abcdefghij�ABCD)rL rS )r r )�br| r r* r} )r r )�jri z*single positional indexer is out-of-bounds)rT )rX rQ z�Location based indexing can only have \[integer, integer slice \(START point is INCLUDED, END point is EXCLUDED\), listlike of integers, boolean array\] types)r r9 ra re rb r r� r4 r; r< r& rc r � ValueError)r r@ r^ r� rB rf r r r! � test_iloc_getitem_labelled_frameC s" z8TestiLocBaseIndependent.test_iloc_getitem_labelled_framec C s� t j�dd�}tddd�}td�}t|||d�}|�� |jdd�d d �f }t|� |j t|dd�d d �f |dd� |d d � d�}t � ||� td �|_ |jdd�d d �f }t|� |j t|dd�d d �f |dd� td �d�}t � ||� t j�dd�}tt d d d ��}tt d dd ��}t|||d�}|�sJ|jjd j |jdd�d d�f }t|� |j t|dd�d d�f |dd� |d d� d�}t � ||� d S )Nr` rE Z20130101)Zperiodsr� )rL rS r* rQ r r ZaaaaZaar� r� r )r9 ra re r rb r Zdescriber r\ r] r; r< rS r0 r2 r3 Zmgr_locs)r r= rI rL rS r@ r^ rB r r r! �test_iloc_getitem_doc_issued s6 . * .z3TestiLocBaseIndependent.test_iloc_getitem_doc_issuec C sb t tj�dd�td�td�d�}d|jd< |jd }|dks@t�d|jd d �d d �f< |jd d �d d �f }|jd d �d d �f }t�||� t tj�d�t dd d �d �}d|jd< |jd }|dks�t�d|jd d�< |jd d� }|jd d� }t� ||� t d gd �}dd dg|jdd d �< dd dg|jdd d �< |}t ddd d ddg�}t� ||� d S )NrX rE r� r� )rL rS r )r r r r r* rP )rL r r` rQ ) r r9 ra re rb r r4 r; r< r r0 rd )r r@ r^ rB rg r r r! �test_iloc_setitem_series� s. z0TestiLocBaseIndependent.test_iloc_setitem_seriesc C s� t tjddd�tjdddd�d��}ddgddgg|jd d �< t d d ddd gdd dddgd��}t�||� t dddddgtjdddd�d��}ddgddgg|jd d �< t dddddgdd dddgd��}t�||� d S )NrQ �int64)r, rX )rW r| rZ r� r� r rE r r r` r� ro r� �c�d�er� �y)r r9 rp r r; r< )r r@ rB r r r! � test_iloc_setitem_list_of_lists� s "" ""z7TestiLocBaseIndependent.test_iloc_setitem_list_of_lists�value�Zc C sL t ddgddggddgd�}||jd|f< |jd }t|�rD|d ksHt�d S ) Nr r r* rE rW r| )rS r )r r r� )r r r r4 )r r+ r� r@ r^ r r r! �#test_iloc_setitem_with_scalar_index� s z;TestiLocBaseIndependent.test_iloc_setitem_with_scalar_indexc C s t ttd��td�dgd�}|jd dk}d}tjt|d�� |j| W d Q R X tt|��|_ d }tjt |d�� |j| W d Q R X |jt j d gt|� t d � }t�||� t �d �}d| }d d� |D �}t ||d�|�}ddddddddd d� }td d��� tdt� x�dD ]�} |jdkj}| �rFt|ttt|| ����}x�dD ]�} y:| �rlt|| dd � �} n|} tt| | d �� ��} W n2 tt t fk �r� } z t| �} W d d } ~ X Y nX | | f}|� |�}|| k�rLt d|� d| � d |� d ����qLW �qW W d Q R X d S ) NrQ rR ro )rL rS r r zBiLocation based boolean indexing cannot use an indexable as a mask)rT zDiLocation based boolean indexing on an integer type is not availableT)r, rE c S s g | ] }t |��qS r )�bin)�.0�numr r r! � <listcomp>� s z:TestiLocBaseIndependent.test_iloc_mask.<locals>.<listcomp>)�locs�numsZ0b1100Z0b11ztUnalignable boolean Series provided as indexer (index of the boolean Series and of the indexed object do not match).) )N� )Nz.loc)Nz.iloc)rL r� )rL z.loc)rL z.iloc)r� r� )r� z.loc)r� z.iloc)�record�ignore)NrL r� )r� z.locz.ilocr r� �[z] does not match [z ], received [�]) r rb r0 ro r& rc r� r rl rL �NotImplementedErrorr9 r �boolr; r< rp r r � UserWarningr� r6 r �reversed�getattrr\ r� �sumr �getr4 )r r@ r� rf r^ r� r� ZrepsrB �idx�method�accessorZanswerr� r �rr r r! �test_iloc_mask� sX z&TestiLocBaseIndependent.test_iloc_maskc C s� t dgd dgd d��}t�d�d }|j| }t|d| d| g�}|j| }t�||� t dgd dgd d��}t|d| d| g�}tjt d d �� |j | W d Q R X d S ) Ng�������?i� r )rW r| rU �c r r* i� z not in index)rT ) r r9 rp r r r; r< r& rc �KeyErrorr� )r r@ r� rB Zdf3r^ r� r r r! � test_iloc_non_unique_indexing s z5TestiLocBaseIndependent.test_iloc_non_unique_indexingc C s� t �dd�}t j|jd d �g f |jd d �d d�f ddd� t j|jg d d �f |jd d�d d �f ddd� t j|jg |jd d�d d �f ddd� d S )NrQ r r T)Zcheck_index_typeZcheck_column_type)r; ZmakeCustomDataframer< r )r r@ r r r! �"test_iloc_empty_list_indexer_is_ok' s z:TestiLocBaseIndependent.test_iloc_empty_list_indexer_is_okc C s� t ddddgi�}|jd d � }||k s,t�dddg|d< |rT|d dk�� rht�n|d dk�� sht�tddddddg�}|jd d � }||k s�t�dd d g|d d�< t|d d� dd d gk�s�t�d S ) Nro r r r* rE rQ r` r r� r� )r r r4 �allr )r r= Z original_dfZ sliced_dfZoriginal_seriesZ sliced_seriesr r r! �&test_identity_slice_returns_new_object= s z>TestiLocBaseIndependent.test_identity_slice_returns_new_objectc C sD t ddgddgg�}|jt�d� }tddgdd�}t�||� d S )Nr r r* rE r )r� )r r r9 r r r; rd )r r@ r^ rg r r r! � test_indexing_zerodim_np_arrayT s z6TestiLocBaseIndependent.test_indexing_zerodim_np_arrayc C s, t ddg�}|jt�d� }|dks(t�d S )Nr r r )r r r9 r r4 )r rg r^ r r r! �%test_series_indexing_zerodim_np_array[ s z=TestiLocBaseIndependent.test_series_indexing_zerodim_np_arrayz1https://github.com/pandas-dev/pandas/issues/33457)�reasonc C s\ t dddg�}t|dddgd��}|d d d� |jd d �d f< t dddg�}t�||� d S ) NrW r| r} r r r* )r r r r )r r r r; Zassert_categorical_equal)r r? r@ rB r r r! �-test_iloc_setitem_categorical_updates_inplacea s zETestiLocBaseIndependent.test_iloc_setitem_categorical_updates_inplacec C sB t ddgddgddgdtjgg�}|j|jdk d9 < t ddgddgdd gdtjgg�}t�||� |j|jdk d9 < t ddgddgdd gd tjgg�}t�||� |jd d d d g d9 < t ddgdd gdd gd tjgg�}t�||� |jd d d d g d < t d dgddgddgdtjgg�}t�||� d S )Nr r r r* rE rQ r` r� rX r� TFg g @g @g (@g @g @)r r9 r� r rL r; r< )r r^ rB r r r! � test_iloc_with_boolean_operationm s "" " " "z8TestiLocBaseIndependent.test_iloc_with_boolean_operationc C sX t dtd�� �i�}|jd }tdgdddddgd �}t|dgdd d �}t�||� d S ) Nr� z a b c d er ro r� r� r� r� )� categories�category)rL r� r, )r r �splitr r r; rd )r r@ r^ Zraw_catrB r r r! �?test_iloc_getitem_singlerow_slice_categoricaldtype_gives_series� s zWTestiLocBaseIndependent.test_iloc_getitem_singlerow_slice_categoricaldtype_gives_seriesc C s� t dddg��d�}|jdd� }t ddg��tdddg��}t�||� |jddg }t ddg��tdddg��}t�||� |jdddg }t dg��tdddg��}t�||� d S )Nr r r* r� r TF)r r: r r r; rd )r rN r^ rB r r r! �$test_iloc_getitem_categorical_values� s z<TestiLocBaseIndependent.test_iloc_getitem_categorical_valuesc C s> t dddgdd�}||jd< t tddgdd�}t�||� d S )Nr r r ztimedelta64[ns])r, )r r r r; rd )r r� ZseriesrB r r r! �%test_iloc_setitem_td64_values_cast_na� s z=TestiLocBaseIndependent.test_iloc_setitem_td64_values_cast_nac C sl t g �}ttj�t|�t|��||d�}tjjddd�}d|j� �}tj t |d�� d|j |< W d Q R X d S )N)rL rS rQ )r r r )r� z Cannot set values with ndim > )rT r ) r r r9 ra re rl r� �ndimr& rc r� r )r r� rJ Znd3rf r r r! �4test_iloc_setitem_empty_frame_raises_with_3d_ndarray� s zLTestiLocBaseIndependent.test_iloc_setitem_empty_frame_raises_with_3d_ndarrayc C s� t �d�}t|�}t �d�}|jdd� t|�}t�||�dddg ||�dddg � t�||�dg ||�dg � t�||�d ||�d � t�||�dd� ||�dd� � d S )NrX F)�writer r r* )r9 �eyer Zsetflagsr; r< rd )r r+ Zrw_arrayZrw_dfZro_arrayZro_dfr r r! �"test_iloc_getitem_read_only_values� s ( z:TestiLocBaseIndependent.test_iloc_getitem_read_only_valuesc C s� t dtjddd�i�}t�dddg�}d|j_|j| }|jdddg }t� ||� |d j| }|d jdddg }t� ||� d S ) N�datarV �float64)r, r r* r` F) r r9 Zonesr �flagsZ writeabler r� r; r< rd )r r@ �indicesr^ rB r r r! � test_iloc_getitem_readonly_key� s z6TestiLocBaseIndependent.test_iloc_getitem_readonly_keyc C sV t dgdgd�}tdddg�|jd< t dtdddg�gidgdgd�}t�||� d S )Nro r )rS rL r r r* )r r )r r r r; r< )r r@ rB r r r! �"test_iloc_assign_series_to_df_cell� s "z:TestiLocBaseIndependent.test_iloc_assign_series_to_df_cell�klassc C sn t dddgdddgd��}|dd d g�}|j|df d |j|df< t dddgd ddgd��}t�||� d S ) Nr� r� �zr r* rE )�flagr� TFr )r r r; r< )r r� r@ r+ rB r r r! � test_iloc_setitem_bool_indexer� s z6TestiLocBaseIndependent.test_iloc_setitem_bool_indexerc C s� t dddgdddgd��}t dd d gd d d gdddgd��}|jd d �dgf |jd d �|f< t dd d gdddgdddgd��}t�||� d S )NrZ r� r� r� � � )Za2�b2r r r* rE rQ r` r r� r� )ro r� r� r )r r r; r< )r r+ r� r� rB r r r! �%test_iloc_setitem_pure_position_based� s """z=TestiLocBaseIndependent.test_iloc_setitem_pure_position_basedc C s� t ddgddgd��}ddd�}||jd< t ddgddgd��}t�||� t ddgddgd��}||jd< t ddgddgd��}t�||� d S )Nr r )r� r� r� r� g @g �X@)r r r; r< )r r@ �rhsrB r r r! �"test_iloc_setitem_dictionary_value� s z:TestiLocBaseIndependent.test_iloc_setitem_dictionary_valuec C st t tj�dd�dddgtd�d�}|jdd � }t�|jd |� |jdd �df }t� |jd |� dddg|_ |jdd � }t�|jd |� |jdd �df }t� |jd |� t tj�d d�ddddgtd�d�}|jdd � }t�|jd |� |jdd �df }t� |jd |� ddd dg|_ |jdd g }t�|jd |� |jdd gdf }t� |jd |� d S ) Nr* g�������?g�������?�abc)rL rS r r )g�������?ro rE r r ) r r9 ra re rb r r; r< r� rd rL )r r@ �expectr r r! �"test_iloc_getitem_float_duplicates� s, "z:TestiLocBaseIndependent.test_iloc_getitem_float_duplicatesc C s� G dd� d�}t ddgdgd�}|d�|jd< |d�|jd< t ddgdgd�}|d�|jd< t�||� t ddgdgd�}|d�|jd< tj|jd< t ddgdgd�}t�||� d S )Nc @ s< e Zd Zdd� Zed�dd�ZeZed�dd�Zdd � Z d S ) zCTestiLocBaseIndependent.test_iloc_setitem_custom_object.<locals>.TOc S s || _ d S )N)r� )r r� r r r! �__init__ s zLTestiLocBaseIndependent.test_iloc_setitem_custom_object.<locals>.TO.__init__)�returnc S s d| j � d�S )Nr� r� )r� )r r r r! �__str__ s zKTestiLocBaseIndependent.test_iloc_setitem_custom_object.<locals>.TO.__str__c S s | j |j kS )N)r� )r �otherr r r! �__eq__# s zJTestiLocBaseIndependent.test_iloc_setitem_custom_object.<locals>.TO.__eq__c S s | S )Nr )r r r r! �view& s zHTestiLocBaseIndependent.test_iloc_setitem_custom_object.<locals>.TO.viewN) r# r$ r% r� r\ r� �__repr__r� r� r� r r r r! �TO s r� r r )rL rS )r r r )r r r; r< r9 r� )r r� r@ r^ r r r! � test_iloc_setitem_custom_object s z7TestiLocBaseIndependent.test_iloc_setitem_custom_objectc C s� t tj�dd�td�td�d�}|jd }t|t�s8t�t � |j |j d � |j jd d �df }t|t�snt�t � |j |j d � d S )Nr* �ABCZaab)rS rL r ) r r9 ra Zrandrb r r7 r r4 r; �assert_almost_equalr6 �T)r r@ r^ r r r! �!test_iloc_getitem_with_duplicates: s z9TestiLocBaseIndependent.test_iloc_getitem_with_duplicatesc C sV t dddgdddggdddgd�}|jd d �dgf }|jdgdd �}t�||� d S ) Nr r r* rE rQ r` )rS r )r� )r r Ztaker; r< )r r@ r^ rB r r r! �"test_iloc_getitem_with_duplicates2F s "z:TestiLocBaseIndependent.test_iloc_getitem_with_duplicates2c C s� t tdd�ddgi�}|jd }ttdd�didd�}t�||� |jd d �df }tddgtdd�d�}t�||� |�� }|jd d �df d7 < t tdd�ddgi�}t�||� d S )Nr r r )r� r* )r r r r r; rd r5 r< )r r@ r^ rB r r r! �test_iloc_intervalM s z*TestiLocBaseIndependent.test_iloc_interval� indexing_func�rhs_funcc C s� t dddg�}|ddg�|j|dddg�< t dddg�}t�||� tddddgi�}|dgdgg�|j|dddg�< tddddgi�}t�||� d S ) Nr r r rQ rX TFro )r r r; rd r r< )r r r rN rB r@ r r r! � test_loc_setitem_boolean_list^ s z5TestiLocBaseIndependent.test_loc_setitem_boolean_list)Jr# r$ r% �__doc__r& r' r( r8 r0 r r9 Zasarrayr; r� r rC �td�&skip_array_manager_not_yet_implementedr r rK rO rh rp rb r rm rs 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� Zxfailr� r� r� r� r r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r! r) 7 s� , f $ & /!+ $& G " !" r) c @ s* e Zd Zdd� Zdd� Zejdd� �ZdS )�TestILocErrorsc C sd |}|t kr|�� }d}tjt|d�� |jd W d Q R X tjttd�� d|jd< W d Q R X d S )Nz5Cannot index by location index with a non-integer key)rT g @r )r rM r& rc rr r r �_slice_iloc_msg)r Zseries_with_simple_indexrH rJ rf r r r! �test_iloc_float_raisesp s z%TestILocErrors.test_iloc_float_raisesc C sp t jtdd��" |jd d �d d �d d �f W d Q R X t jtdd��" d|jd d �d d �d d �f< W d Q R X d S )NzToo many indexers)rT ztoo many indices for arrayr )r& rc r r r )r Z float_framer r r! �*test_iloc_getitem_setitem_fancy_exceptions� s &z9TestILocErrors.test_iloc_getitem_setitem_fancy_exceptionsc C sp t ddddgi�}t ddddgi�}t�t�� d|j|< W d Q R X d}tjt|d�� |j| W d Q R X d S ) Nro r r r* TFzWDataFrame indexer is not allowed for .iloc Consider using .loc for automatic alignment.)rT )r r; Zassert_produces_warning� FutureWarningr r& rc r )r r@ r+ rf r r r! �test_iloc_frame_indexer� s z&TestILocErrors.test_iloc_frame_indexerN)r# r$ r% r r r r r r r r r! r m s r c @ sD e Zd Zdd� Zdd� Zdd� Zej�ddd gd d g�d d � �Z dS )� TestILocSetItemDuplicateColumnsc C s� t d dd�ddd�g�}t ddd�ddd�g�}t||gdd�}d|jd< |jd dks\t�|jd dksnt�|jjd tjks�t�d S ) Nr )rW r| r r* rE )r� r )r r )r r )r r r r4 r] r9 r� )r r� r� r@ r r r! �*test_iloc_setitem_scalar_duplicate_columns� s zJTestILocSetItemDuplicateColumns.test_iloc_setitem_scalar_duplicate_columnsc C sX t dddggdddgd�}dg|jd d �df< t dddggdddgd�}t�||� d S ) Nr r\ �str2ro r� )rS Zstr3r )r r r; r< )r r@ rB r r r! �(test_iloc_setitem_list_duplicate_columns� s zHTestILocSetItemDuplicateColumns.test_iloc_setitem_list_duplicate_columnsc C sj t tjdtjd��dd�ddddgd�}|jd d �df �tj�|jd d �df< |jjd tjksft �d S ) Nr� )r, r rE rW r| )rS r ) r r9 rp r� rq r r: r� r] r4 )r r@ r r r! �*test_iloc_setitem_series_duplicate_columns� s &(zJTestILocSetItemDuplicateColumns.test_iloc_setitem_series_duplicate_columnsr] � init_value�expected_value)r� �0r )�floatz1.2g333333�?c C sj t |ddggdddgd�}|jd d �df �|�|jd d �df< t |ddggdddgd�}t�||� d S )Nr\ r ro r� )rS r )r r r: r; r< )r r] r r r@ Z expected_dfr r r! �*test_iloc_setitem_dtypes_duplicate_columns� s &zJTestILocSetItemDuplicateColumns.test_iloc_setitem_dtypes_duplicate_columnsN) r# r$ r% r r r r& r' r( r r r r r! r � s r c @ s e Zd Zdd� Zdd� ZdS )�TestILocCallablec C s� t ddddgtd�d�td�d�}|jd d � }t�||jddg � |jd d � d d �f }t�||jddgd d �f � |jd d � d d � f }t�||jddgdf � |jdd � dd � f }t�||jddgdgf � |jddgdd � f }t�||jddgdf � |jddgdd � f }t�||jddgdgf � |jdd � df }t�||jddgdf � |jdd � dgf }t�||jddgdgf � d S )Nr r r* rE �aabb)�X�Yr� )rL c S s ddgS )Nr r* r )r� r r r! r� � r� zCTestILocCallable.test_frame_iloc_getitem_callable.<locals>.<lambda>c S s ddgS )Nr r* r )r� r r r! r� � r� c S s ddgS )Nr r* r )r� r r r! r� � r� c S s dS )Nr r )r� r r r! r� � r� r c S s ddgS )Nr r* r )r� r r r! r� � r� c S s dgS )Nr r )r� r r r! r� � r� c S s dS )Nr r )r� r r r! r� � r� c S s dgS )Nr r )r� r r r! r� � r� c S s ddgS )Nr r* r )r� r r r! r� � r� c S s ddgS )Nr r* r )r� r r r! r� � r� )r rb r r; r< rd )r r@ �resr r r! � test_frame_iloc_getitem_callable� s" " z1TestILocCallable.test_frame_iloc_getitem_callablec C sF t ddddgtd�d�td�d�}|�� }d |jd d � < |�� }d |jddg< t�||� |�� }d |jd d � d d �f< |�� }d |jddgd d �f< t�||� |�� }d|jdd � dd � f< |�� }d|jddgd f< t�||� |�� }d|jdd � dd � f< |�� }d|jddgd gf< t�||� |�� }d|jddgdd � f< |�� }d|jddgd f< t�||� |�� }d|jddgdd � f< |�� }d|jddgd gf< t�||� |�� }d|jdd � d f< |�� }d|jddgd f< t�||� |�� }ddg|jdd � d gf< |�� }ddg|jddgd gf< t�||� d S )Nr r r* rE r )r r r� )rL r c S s ddgS )Nr r* r )r� r r r! r� � r� zCTestILocCallable.test_frame_iloc_setitem_callable.<locals>.<lambda>r c S s ddgS )Nr r* r )r� r r r! r� � r� rQ c S s ddgS )Nr r* r )r� r r r! r� � r� c S s dS )Nr r )r� r r r! r� � r� � c S s ddgS )Nr r* r )r� r r r! r� � r� c S s dgS )Nr r )r� r r r! r� � r� r� c S s dS )Nr r )r� r r r! r� r� ���c S s dgS )Nr r )r� r r r! r� r� rX c S s ddgS )Nr r* r )r� r r r! r� r� c S s ddgS )Nr r* r )r� r r r! r� r� )r rb r5 r r; r< )r r@ r r� r r r! � test_frame_iloc_setitem_callable� sR " z1TestILocCallable.test_frame_iloc_setitem_callableN)r# r$ r% r r! r r r r! r � s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestILocSeriesc C s� t tj�d�ttddd��d�}x8tt|��D ](}|j| }||j| }t � ||� q.W |jt dd� }|j dd� }t � ||� d|d d �< |dd� dk�� s�t�|jddddd g }|�|jddddd g �}t � ||� d S ) NrX r rP r )rL r r* rE rQ )r r9 ra re rb r0 rl r rL r; r� r8 r� rd r� r4 Zreindex)r rN �ir^ r� rB r r r! � test_iloc s zTestILocSeries.test_ilocc C s. t dddgdddgd�}|jd dks*t�d S )Nr r r )rL )r r r4 )r rN r r r! �test_iloc_getitem_nonunique5 s z*TestILocSeries.test_iloc_getitem_nonuniquec C s\ t dddg�}t dddgdddgd�}|jdd� |jdd�< t dddg�}t�||� d S ) Nr r r* rE rQ r` r )rL )r r r; rd )r Zser1Zser2rB r r r! r� 9 s z4TestILocSeries.test_iloc_setitem_pure_position_basedc C sd t dgtjgd��}|jd d �df �d�|jd d �df< t dgttgdd�d��}t�||� d S )N�test)ro r� r� ZInt64)r, ) r r9 r� r� r: r r r; r< )r r^ rB r r r! �#test_iloc_nullable_int64_size_1_nanA s &z2TestILocSeries.test_iloc_nullable_int64_size_1_nanN)r# r$ r% r$ r% r� r' r r r r! r" s r" )+r r �re�warningsr r �numpyr9 r& Z pandas.util._test_decorators�utilZ_test_decoratorsr Zpandasr r r r r r r r r r r r Zpandas._testingZ_testingr; Zpandas.api.typesr Zpandas.core.indexingr Z pandas.tests.indexing.commonr �escaper r r) r r r r" r r r r! �<module> s6 8 >/,V
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка