Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/indexing/__pycache__/test_setitem.cpython-37.pyc
Назад
B �5�g�� � @ s@ d dl m Z d dlZd dlZd dlm mZ d dlm Z d dl m Z m Z mZ d dlmZmZmZmZ d dlZd dlmZmZmZmZmZmZm Z m Z m Z m Z m Z m!Z!m"Z"m#Z#m$Z$m%Z% d dl&m'Z( d dl)m*Z* d dl+m,Z, G d d � d �Z-G d d � d �Z.G d d� d�Z/G dd� d�Z0G dd� d�Z1G dd� d�Z2G dd� d�Z3dS )� )�datetimeN)� _registry)�is_categorical_dtype�is_interval_dtype�is_object_dtype)�CategoricalDtype�DatetimeTZDtype� IntervalDtype� PeriodDtype)� Categorical� DataFrame� DatetimeIndex�Index�Interval� IntervalIndex� MultiIndex�NaT�Period� PeriodIndex�Series� Timestamp�cut� date_range�notna� period_range)� SparseArray)�BDayc @ s� e Zd Zej�dddddg�dd� �Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zej�d ed �ed �fed!d"d#�edd$�fed%d&d'�ed&d'�fg�d(d)� �Zej�d*d+d,� ejD �d-d.g �d/d0� �Zd1d2� Z ej�d3d4d5d6d7d8d d9g�d:d;� �Z ej�d3d4d5d6d7d8d d9g�d<d=� �Z d>d?� Z d@dA� Z dBdC� Z!ej�ddDdEdFg�dGdH� �Z"ej�dIdJdJgg�dKdL� �Z#dMdN� Z$dOdP� Z%ej�dQdRdSdTgdRdRdRgg�dUdV� �Z&dWdX� Z'dYdZ� Z(d[d\� Z)d]d^� Z*d_d`� Z+ej�dadbd8d8dRgdcdddedfgdffd8dbd8dRgdcdddedfgdffdbdJdJdRgdcdddedfgdffdbdJdRgdcdddegdefdJdbdRgdedddcgdcfdbdRdJgdedddcgddfg�dgdh� �Z,didj� Z-ej�dkdldJdbd8gdme.dmdmdmdmgdmdmdmdmgdmdmdmdmggdldJdbd8gdn�fdbd8gdmdoge.dcdddmdogdedfdmdogd"dpdmdoggdldJdbd8gdn�fdldJdbge/j0dmdodqge/j1dr�e.dmdodqgdmdodqgdmdodqggdldJdbgdn�fdJdbd8gdmdodqgdsdtdugdvdwdxgge.dcdmdodqgdedsdtdugd"dvdwdxggdldJdbd8gdn�fdbdld8ge/j0dmdodqgdsdtdugdvdwdxgge/j1dr�e.dodddmdqgdtdfdsdugdwdpdvdxggdldJdbd8gdn�fdldbge.dmdogdqdsgdtduggdldbgdn�e.dmdddogdqdfdsgdtdpduggdldJdbgdn�fg�dydz� �Z2d{d|� Z3d}d~� Z4dd�� Z5d�d�� Z6d�d�� Z7d�S )��TestDataFrameSetItem�dtype�int32�int64�float32�float64c C s: t j�t|��}t j||d�||< || jj|ks6t�d S )N)r )�np�random�randn�len�arrayr �name�AssertionError)�selfr � float_frame�arr� r- �^/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/indexing/test_setitem.py�test_setitem_dtype/ s z'TestDataFrameSetItem.test_setitem_dtypec C s8 t j�t|�d�}||ddg< t�|ddg j|� d S )N� �A�B)r# r$ r% r& �tmZassert_almost_equal�values)r* r+ �datar- r- r. � test_setitem_list_not_dataframe6 s z4TestDataFrameSetItem.test_setitem_list_not_dataframec C s� t dddgdddgd�tdd d gd d �d �}tddddgtdd d dgd d �dd�}d}tjt|d�� ||d< W d Q R X t tj�ddd�dd d dgd�}d}tjt |d�� |� d d g�� � |d< W d Q R X d S ) N� r0 � �d�e�f)�barZbaz�a�b�c�foo)r( )�index�g�h�i�jZfiz)rA r( z$cannot reindex from a duplicate axis)�matchZnewcolr )� rG )�columnsz6incompatible index of inserted column with frame index�gr) r r r �pytest�raises� ValueErrorr# r$ �randint� TypeError�groupby�count)r* �df�ser�msgr- r- r. �test_setitem_error_msmgs; s z-TestDataFrameSetItem.test_setitem_error_msmgsc C sl d}d}t t|�d�}tj�|�}xt|�D ] }|||< q,W t t�||��||�t|�d�}t�||� d S )N� � )rA ) r �ranger# r$ r% �repeatZreshaper3 �assert_frame_equal)r* �N�KrQ Znew_colrD �expectedr- r- r. �test_setitem_benchmarkR s z+TestDataFrameSetItem.test_setitem_benchmarkc C s^ t tj�dd�t�d�dddgd�}|�dd|d � |�d d |d � |d �d �|d < |j}tt� d �gd t� d �g ddd ddd gd�}t � ||� |d �d �|d< |j}tt� d �gd t� d �gd ddd ddd gd�}t � ||� |d �d�|d< |j}tt� d �gd t� d �gd t� d�g ddd ddd dgd�}t � ||� d S )NrV r8 r? r> r= )rA rH r r@ r0 r< r! �xr" )rA rG r �y) r r# r$ r% �arange�insert�astype�dtypesr r r3 �assert_series_equal)r* rQ �resultr\ r- r- r. � test_setitem_different_dtype] s, " *z1TestDataFrameSetItem.test_setitem_different_dtypec C sT t dddgd�}|j|d< dddg|d< t ddddgidddgd �}t�||� d S ) Nr1 r2 �C)rA �Xr^ r_ �z)r5 rA )r rA r3 rY )r* rQ �expr- r- r. �test_setitem_empty_columns� s z/TestDataFrameSetItem.test_setitem_empty_columnsc C sF t dddd�}tt�t|��d�}||d< |d jt�d�ksBt�d S )Nz1/1/2000 00:00:00z1/1/2000 1:59:50Z10s)�freq)rA r1 zM8[ns])r r r# r` r& r r) )r* �rngrQ r- r- r. �%test_setitem_dt64_index_empty_columns� s z:TestDataFrameSetItem.test_setitem_dt64_index_empty_columnsc C sT t td�d�}tddd�|d< t tddd�ggd ddd gdgd �}t�||� d S ) Nr8 )rA Z20130101�UTC)�tz�nowr r7 r0 )rA rH )r rW r r3 rY )r* rQ r\ r- r- r. �$test_setitem_timestamp_empty_columns� s $z9TestDataFrameSetItem.test_setitem_timestamp_empty_columnsc C sp t �ddddddgdddg�}ttd�dgd �}d t|�� d t|�� d �}tjt|d �� ||d< W d Q R X d S )Nr r7 r0 r= r> r? rU r< )rH zLength of values \(z$\) does not match length of index \(z\))rF r@ )r Z from_codesr rW r& rJ rK rL )r* �catrQ rS r- r- r. �2test_setitem_wrong_length_categorical_dtype_raises� s zGTestDataFrameSetItem.test_setitem_wrong_length_categorical_dtype_raisesc C sP t dddgdddgd��}tddd g�}||d < t|d d �}t�|d |� d S ) Nr= r> r? g �?g @g @)�c_1�n_1r r7 � new_column)r( )r r r r3 rd )r* rQ Zsp_arrayr\ r- r- r. � test_setitem_with_sparse_value� s z3TestDataFrameSetItem.test_setitem_with_sparse_valuec C sh t dddgdddgd��}ttddd g�d d dgd �}||d < ttd ddg�d d �}t�|d |� d S )Nr= r> r? g �?g @g @)ru rv r r7 r0 )rA rw )r( )r r r r3 rd )r* rQ Z sp_seriesr\ r- r- r. �(test_setitem_with_unaligned_sparse_value� s z=TestDataFrameSetItem.test_setitem_with_unaligned_sparse_valuec C s� t tdddgdd�tdddgtd�tdddgtd�d��}t tg dd�tg td�tg td�d��}x>tdddg�D ],\}}t|�t|�t|�d�|j|jd < qrW t�||� d S )Nr r7 r0 r )r r8 )r= r> r? ) r r �float� enumerate�int�loc�shaper3 rY )r* r\ rQ �idxr> r- r- r. �"test_setitem_dict_preserves_dtypes� s (z7TestDataFrameSetItem.test_setitem_dict_preserves_dtypesz obj,dtypez2020-01�Mr rV )�left�rightr� z 2011-01-01z US/Eastern)rp c C sL t dddgt|gd |d�d��}t ddddgi�}||d< t�||� d S )Nr7 r0 r8 )r )r �objr r� )r r r3 rY )r* r� r r\ rQ r- r- r. � test_setitem_extension_types� s "z1TestDataFrameSetItem.test_setitem_extension_types�ea_namec C s g | ]}t |jt�s|j�qS r- )� isinstancer( �property)�.0r r- r- r. � <listcomp>� s z TestDataFrameSetItem.<listcomp>zdatetime64[ns, UTC]z period[D]c C s8 t dg�}dg||< t ddg|dgi�}t�||� d S )Nr r7 )r r3 rY )r* r� re r\ r- r- r. �test_setitem_with_ea_name� s z.TestDataFrameSetItem.test_setitem_with_ea_namec C s� t jddgdd�}t|��� }||d< tdd gdd gd�dd�}t�||� t jddgdd�}||d< tdd gdd gd�dd�}t�||� d S ) Nr7 Znatzdatetime64[ns])r �new)r r� z datetime64[s]g e��A)r# r' r Zto_framer r3 rY )r* Zdata_nsre r\ Zdata_sr- r- r. �6test_setitem_dt64_ndarray_with_NaT_and_diff_time_units� s zKTestDataFrameSetItem.test_setitem_dt64_ndarray_with_NaT_and_diff_time_units�unitrC �m�s�ms�D�Yc C s� d}t �d|� d��}t j|t jd��|�}|�d�}tdt �|�it �|�d�}|||< || jt �d�ksrt�|| j|k� � s�t�d S ) N�d zM8[�])r zdatetime64[ns]�ints)rA zM8[ns]) r# r r` r �viewrb r r) r4 �all)r* r� �nr �vals�ex_valsrQ r- r- r. �-test_frame_setitem_datetime64_col_other_units� s zBTestDataFrameSetItem.test_frame_setitem_datetime64_col_other_unitsc C s� d}t �d|� d��}t j|t jd��|�}|�d�}tdt �|�it �|�d�}t j|t jd��d�|d < ||d < |d j|k�� s�t �d S ) Nr� zM8[r� )r zdatetime64[ns]r� )rA zM8[ns]�dates) r# r r` r r� rb r r4 r� r) )r* r� r� r r� r� rQ r- r- r. �6test_frame_setitem_existing_datetime64_col_other_units s zKTestDataFrameSetItem.test_frame_setitem_existing_datetime64_col_other_unitsc C s |}|d � d�}||d< t�|d t|dd�� d|d< ||d< t�|d t|dd�� |d= |jjd }|jjd }t�||� |jj}|jj}|d ks�t |�t |�ks�t �|� � }t |j d< t |j d < |d } t�t| �td d d gdd�� t�|j|j� d S ) Nr2 r@ rg )r( r� r7 r0 )r7 r7 )r7 r0 TF)�renamer3 rd r �_mgrZarraysZ assert_extension_array_equal�_data�base�idr) �copyr �ilocr rc ) r* Ztimezone_framerQ r Zv1Zv2Zv1baseZv2base�df2re r- r- r. �test_setitem_dt64tz s( z(TestDataFrameSetItem.test_setitem_dt64tzc C s� t dddd�}ttj�dd�|d�}||d< t|d �}tj||dd � |jdksVt �|jdksdt �|� � � d�}t |j t�s�t �t�|j |� d S ) Nz1/1/2000rV rA )�periodsr( r8 )rA r F)� check_names)r r r# r$ r% r r3 �assert_index_equalr( r) Z reset_indexZ set_indexr� rA r )r* rm rQ �rsr- r- r. �test_setitem_periodindex? s z-TestDataFrameSetItem.test_setitem_periodindexc C s� t dddgdddgd��}t�ddgddgddgg�}||d d g< t dddgdddgdddgdddgd ��}|d �|j�|d < |d �|j�|d < |d j|jks�t�|d j|jks�t�t�||� d S ) N�one�two�threer7 r0 r8 )r= r> rV r? r9 )r= r> r? r9 )r r# r' rb r r) r3 rY )r* rQ r, r\ r- r- r. �'test_setitem_complete_column_with_arrayM s z<TestDataFrameSetItem.test_setitem_complete_column_with_array�f8�i8�u8c C sx t dddg|d�}ttj�dd�|d�}dddg|d < t dddd gtd�}|d krft d d d d gtd�}t�|j|� d S )Nr7 r0 r8 )r )rH r= r> r? Fr� g �?g @g @) r r r# r$ r% �objectr3 r� rH )r* r �colsrQ Z expected_colsr- r- r. �$test_setitem_bool_with_numeric_index` s z9TestDataFrameSetItem.test_setitem_bool_with_numeric_index�indexerr2 c C sb t ddgd�}t dddgi�}|||< t tjgd ddgd��}|d �d�|d< t�||� d S )Nr1 r2 )rH r7 r0 )r1 r2 r� )r r# �nanrb r3 rY )r* r� rQ �otherr\ r- r- r. �#test_setitem_frame_length_0_str_keyn s z8TestDataFrameSetItem.test_setitem_frame_length_0_str_keyc C s� dddgd }t td�|d�}d|jd< d |jd d �df< d |d< t d d ddd dgtjd dtjd dgtjd dtjd dggdd�}|r�||_|d �d�|d< |d �d�|d< n.|d �tj�|d< |d �tj�|d< ||_t� ||� d S )Nr1 r2 rg r0 r8 )rA rH )r r8 )r r1 )r7 rG )r0 rV r r7 rG rV r� )r r ) r rW r} r# r� rH rb r r3 rY )r* �using_array_managerr� rQ r\ r- r- r. �$test_setitem_frame_duplicate_columnsx s$ z9TestDataFrameSetItem.test_setitem_frame_duplicate_columnsc C s� dddgd }t td�|d�}tjtdd�� d |dg< W d Q R X |jd d �d d�f }tjtdd�� d |dg< W d Q R X d S ) Nr1 r2 rg r0 r8 )rA rH z"Columns must be same length as key)rF )r r8 rV )r rW rJ rK rL r� )r* r� rQ r� r- r- r. �2test_setitem_frame_duplicate_columns_size_mismatch� s zGTestDataFrameSetItem.test_setitem_frame_duplicate_columns_size_mismatchr� r= r> r? c C sT t dddgg|d�}t ddggddgd�}d }tjt|d �� ||d < W d Q R X d S ) Nr7 r0 r8 )rH rU � r9 r: z"Columns must be same length as key)rF r= )r rJ rK rL )r* r� rQ �rhsrS r- r- r. �#test_setitem_df_wrong_column_number� s z8TestDataFrameSetItem.test_setitem_df_wrong_column_numberc C s� t dddggdddgd�}t ddd ggdddgd�}||ddg< t ddd ggdddgd�}t�||� ||d dg< t ddd dggdddd gd�}t�||� d S ) Nr7 r0 r8 r= r> )rH rU r� � r? )r r3 rY )r* rQ r� r\ r- r- r. �/test_setitem_listlike_indexer_duplicate_columns� s zDTestDataFrameSetItem.test_setitem_listlike_indexer_duplicate_columnsc C s^ t dddggdddgd�}t ddggddgd�}d }tjt|d �� ||ddg< W d Q R X d S ) Nr7 r0 r8 r= r> )rH rU r� z"Columns must be same length as key)rF )r rJ rK rL )r* rQ r� rS r- r- r. �@test_setitem_listlike_indexer_duplicate_columns_not_equal_length� s zUTestDataFrameSetItem.test_setitem_listlike_indexer_duplicate_columns_not_equal_lengthc C s� t dtd�i�}t|d d�}t|jjt�s0t�||d< t� |�|d< |j |d< t� |j �|d< |� t �|d< t |d j�s�t�t|d jj�s�t�t |d j�s�t�t|d jj�s�t�t|d �s�t�t|d �s�t�t|d �s�t�d d � }t�||j�||j�� tj||j�||j�d d � tj||j�||j�d d � tj||j�||j�d d � t�|d |d � tj|d |d d d � t�|d |d � tj|d |d d d � d S ) Nr1 rU rV r2 rg r� �E�Fc S s t t�| ��S )N)r r# r' )r^ r- r- r. �<lambda>� � z=TestDataFrameSetItem.test_setitem_intervals.<locals>.<lambda>F)r� )r rW r r� rs � categoriesr r) r# r' r4 rb r� r r r r r3 r� r2 rg r� rd )r* rQ rR r? r- r- r. �test_setitem_intervals� s0 z+TestDataFrameSetItem.test_setitem_intervalsc C s\ t dttd���d�i�}|jj�ddg�|_t dtddg��ddg�i�}t� ||� d S )NrC Zmn�categoryr� r� ) r r �listrb rC rs Zreorder_categoriesr r3 rY )r* rQ r\ r- r- r. �test_setitem_categorical� s z-TestDataFrameSetItem.test_setitem_categoricalc C sD t g dd�}tdg|d�}g |d< tdg|d�}t�|j|j� d S )Nr )r( r1 )rH rA )r r r3 r� rA )r* rA re r\ r- r- r. � test_setitem_with_empty_listlike� s z5TestDataFrameSetItem.test_setitem_with_empty_listlikezcols, values, expectedrg r7 r0 r8 rG c C s8 t |g|d�}|d |d< |d jd }||ks4t�d S )N)rH r= r )r r4 r) )r* r� r4 r\ rQ re r- r- r. �test_setitem_same_column� s z-TestDataFrameSetItem.test_setitem_same_columnc C s$ dddgddgdddgf}t �|�}td d d �}tj�d d t|�t|�f�}t|||d�}|jj � � |d d d � }}tj� |� |d j |d d d�f |d< t �|d |d � tj� |� |d j ||f |d< t �|d |d � tj� |� |d j ||f |d< t �|d |d � d S )NZjim�joe�jolie�first�lastr� �centerr� Z20141006� )r� r7 i� )rH rA ���)r� r� )r� r� )r� r� )r� r� )r Z from_productr r# r$ rM r& r rA r4 r� �shuffler} r3 rY )r* �itr� rA r� rQ rD rE r- r- r. �test_setitem_multi_index s z-TestDataFrameSetItem.test_setitem_multi_indexzcolumns,box,expectedr1 � )rH � � � )r rU r� r� � � � c C s: t ddgddgddggddgd �}|||< t�||� d S ) Nr7 r0 r8 rG rV r� r1 r2 )rH )r r3 rY )r* rH �boxr\ rQ r- r- r. �!test_setitem_list_missing_columns' s 3"z6TestDataFrameSetItem.test_setitem_list_missing_columnsc C sF t t|d |d ��}||d< |d }t||jdd�}t�||� d S )Nr1 r2 �tuples)rA r( )r� �zipr rA r3 rd )r* r+ r� re r\ r- r- r. �test_setitem_list_of_tuples^ s z0TestDataFrameSetItem.test_setitem_list_of_tuplesc C s\ t dddgdddgd��}dd � d D �}d|j|< t dddgdddgd��}t�||� d S ) Nr7 r0 r8 rG rV r� )r= r> c s s | ] }|V qd S )Nr- )r� r^ r- r- r. � <genexpr>i s zCTestDataFrameSetItem.test_setitem_iloc_generator.<locals>.<genexpr>)r7 r0 )r r� r3 rY )r* rQ r� r\ r- r- r. �test_setitem_iloc_generatorf s z0TestDataFrameSetItem.test_setitem_iloc_generatorc C s` t dddgdddgd��}dd � d D �}d|j|df< t dddgdddgd��}t�||� d S ) Nr7 r0 r8 rG rV r� )r= r> c s s | ] }|V qd S )Nr- )r� r^ r- r- r. r� p s zSTestDataFrameSetItem.test_setitem_iloc_two_dimensional_generator.<locals>.<genexpr>)r7 r0 )r r� r3 rY )r* rQ r� r\ r- r- r. �+test_setitem_iloc_two_dimensional_generatorn s z@TestDataFrameSetItem.test_setitem_iloc_two_dimensional_generatorc C s� t ddd�}t|d�}t d|tjd�|d< t d|dd�|d< t d |dd�|d < t d |tjd�|d < |j}t tjtttjgtd �d�}t� ||� d S ) Nr� ZS24)r( r )rA r= )r( rA r r> ZS64r? r9 �abcd) r r r# Zuint32Zuint8rc r� r� r3 rd )r* rA rQ re r\ r- r- r. �(test_setitem_dtypes_bytes_type_to_objectu s z=TestDataFrameSetItem.test_setitem_dtypes_bytes_type_to_objectc C s� t ddgddgd���ddd��}td|jd �}|d |j|d f< |d |j|d f< t ddgddgd���ddd��}t�||� d S ) Nr8 rG rV r� )r= r> r ZInt64F)rA r= r> )r rb r rA r} r3 rY )r* re �maskr\ r- r- r. � test_boolean_mask_nullable_int64� s z5TestDataFrameSetItem.test_boolean_mask_nullable_int64N)8�__name__� __module__� __qualname__rJ �mark� parametrizer/ r6 rT r] rf rk rn rr rt rx ry r� r r r r r r r� � ea_registryrc 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� r� r� r- r- r- r. r . s� # $$ & - " * & 6 r c @ s@ e Zd Zejdd� �Zejdd� �Zdd� Zdd� Zd d � Z d S ) �TestSetitemTZAwareValuesc C s t ddgdd�}|�d�}|S )Nz2013-1-1 13:00z2013-1-2 14:00r2 )r( z US/Pacific)r � tz_localize)r* Znaiver r- r- r. r � s z TestSetitemTZAwareValues.idxc C s. t tj|�� dd�dd�}|j|jks*t�|S )Nr� )r r2 )r( )r r# r' �tolistr r) )r* r r\ r- r- r. r\ � s z!TestSetitemTZAwareValues.expectedc C s� t tj�dd�dgd�}||d< t�t�� }|jdddgd�|d< W d Q R X d }|t|d j �ksft �|d }t |� d �� d �dd �}t�||� d S ) Nr0 r7 r1 )rH r2 Fr )Zkeep_tzrA z(do 'idx.tz_convert(None)' before callingro )r( )r r# r$ r% r3 �assert_produces_warning� FutureWarningZ to_series�str�messager) r Z tz_convertr� rd )r* r r\ rQ r� rS re �compr- r- r. �test_setitem_dt64series� s z0TestSetitemTZAwareValues.test_setitem_dt64seriesc C s8 t tj�dd�dgd�}||d< |d }t�||� d S )Nr0 r7 r1 )rH r2 )r r# r$ r% r3 rd )r* r r\ rQ re r- r- r. �test_setitem_datetimeindex� s z3TestSetitemTZAwareValues.test_setitem_datetimeindexc C s< t tj�dd�dgd�}|�� |d< |d }t�||� d S )Nr0 r7 r1 )rH r2 )r r# r$ r% Z to_pydatetimer3 rd )r* r r\ rQ re r- r- r. �.test_setitem_object_array_of_tzaware_datetimes� s zGTestSetitemTZAwareValues.test_setitem_object_array_of_tzaware_datetimesN) r� r� r� rJ Zfixturer r\ r� r� r� r- r- r- r. r� � s r� c @ sJ e Zd Zejdd� �Zdd� Zdd� Zdd� Zd d � Z d d � Z d d� Z dS )�!TestDataFrameSetItemWithExpansionc C sr t dddgdddgd��}|d }t�dd gd d gd d gg�|d dg< d|jd< tdddgdd�}t�||� d S )Nr7 r0 r8 rG r� )r= r> r= g�������?g�������?g333333�?g�������?g �?r? r9 r� )r r )r( )r r# r' r� r r3 rd )r* rQ rR r\ r- r- r. �test_setitem_listlike_views� s $ z=TestDataFrameSetItemWithExpansion.test_setitem_listlike_viewsc C sR t ddgddgg�}ddg|d< t dddgdddggdddgd �}t�||� d S ) Nr7 r0 r8 rG rV r� z0 - Namer )rH )r r3 rY )r* rQ r\ r- r- r. �.test_setitem_string_column_numpy_dtype_raising� s "zPTestDataFrameSetItemWithExpansion.test_setitem_string_column_numpy_dtype_raisingc C sf t dddgdd�}ttd��|jd d �df< t dtjtjgdtjtjggdddgd�}t�||� d S ) Nr= r> r" )rH r r0 r r7 )rH )r r� rW r} r# r� r3 rY )r* rQ r\ r- r- r. �'test_setitem_empty_df_duplicate_columns� s (zITestDataFrameSetItemWithExpansion.test_setitem_empty_df_duplicate_columnsc C s> t dtjtj�ddd�dd�i�}tdd� tddd �D ��}|jdgd d �}t|j tdd d �d |d�}|j }||d< t |� |j }t t�d�t|d d�gddgd�}t�||� ||d< t |� |j }t t�d�t|d d�t|d d�gdddgd�}t�||� |d }|d }t�|jj|� d|_t�|�� |�� � d S )N�valuer i' r� r )r c S s g | ]}|� d |d � ��qS )z - i� r- )r� rD r- r- r. r� � s zcTestDataFrameSetItemWithExpansion.test_setitem_with_expansion_categorical_dtype.<locals>.<listcomp>i� T)ZbyZ ascendingi) F)r� �labelsr� )r� Zordered)rA r� )r r# r' r$ rM r rW � sort_valuesr r r4 r� rc r r r r3 rd Zassert_categorical_equalr� r( Z sort_index) r* rQ r rR rs re r\ Zresult1Zresult2r- r- r. �-test_setitem_with_expansion_categorical_dtype� s6 zOTestDataFrameSetItemWithExpansion.test_setitem_with_expansion_categorical_dtypec C s2 t � }d|d< t dgd��tj�}t�||� d S )Nr7 r@ )rH )r rb r# r r3 rY )r* rQ r\ r- r- r. � test_setitem_scalars_no_index s z?TestDataFrameSetItemWithExpansion.test_setitem_scalars_no_indexc C sL d|j kst�|d |d< d|j ks(t�|d }|d }tj||dd� d S )N)r1 r2 r1 F)r� )rH r) r3 rd )r* r+ re r\ r- r- r. � test_setitem_newcol_tuple_key s z?TestDataFrameSetItemWithExpansion.test_setitem_newcol_tuple_keyc C sX t ddt� d�}t|td�d�}tddd�}t|�}tj||< t�|| �� � sTt �d S ) Nz1/1/2012z2/1/2012)�start�endrl rU )rH rA i� r� r7 ) r r r rW r r r# r� �isnanr� r) )r* rH r5 �t�tsr- r- r. �#test_frame_setitem_newcol_timestamp s zETestDataFrameSetItemWithExpansion.test_frame_setitem_newcol_timestampN) r� r� r� �tdZ&skip_array_manager_not_yet_implementedr� r� r r r r r r- r- r- r. r� � s - r� c @ s� e Zd Zdd� Zej�dejej g�ej�de e j e ej g�ej�ddddg�d d � ���Zej�de e j e ej g�ej�ddddg�d d � ��Zej�dejej g�ej�de e j e ej g�ej�ddddg�d d� ���ZdS )�TestDataFrameSetItemSlicingc C sH t t�d��}d|dd �< t�d�}d|dd �< t |�}t�||� d S )N)r� r7 r7 ���)r r# Zzerosr3 rY )r* rQ r, r\ r- r- r. �test_setitem_slice_position+ s z7TestDataFrameSetItemSlicing.test_setitem_slice_positionr� r� r� r7 r0 r8 c C s~ t dddggdddgg| ddd gd �}|d d d g�||�dd �< t dddggd d d gg| ddd gd �}t�||� d S )Nr7 r8 rV r0 rG r� r= r> r? )rH rU r� r� )r r3 rY )r* r� r� r� rQ r\ r- r- r. �+test_setitem_slice_indexer_broadcasting_rhs4 s **zGTestDataFrameSetItemSlicing.test_setitem_slice_indexer_broadcasting_rhsc C s� t dddggdddgg| ddd gd �}|d d d g�|jttd|d ��< t dddggd d d gg| ddd gd �}t�||� d S )Nr7 r8 rV r0 rG r� r= r> r? )rH rU r� r� )r r� r� rW r3 rY )r* r� r� rQ r\ r- r- r. �*test_setitem_list_indexer_broadcasting_rhs> s *"*zFTestDataFrameSetItemSlicing.test_setitem_list_indexer_broadcasting_rhsc C s� t dddgdddggddd gg| d d d gd �}|dddg�||�dd �< t dddggdddgg|d d d d gdd�}t�||� d S )Nr7 r8 rV r^ r_ ri r0 rG r� r= r> r? )rH rU r� r� r� )rH r )r r3 rY )r* r� r� r� rQ r\ r- r- r. �0test_setitem_slice_broadcasting_rhs_mixed_dtypesG s 0 zLTestDataFrameSetItemSlicing.test_setitem_slice_broadcasting_rhs_mixed_dtypesN)r� r� r� r rJ r� r� r3 �setitemr� r r# r' r� �pdr r r r- r- r- r. r * s r c @ s e Zd Zdd� Zdd� ZdS )� TestDataFrameSetItemCallablec C s` t ddddgddddgd ��}d d d d g|dd� < t d d d d gddddgd ��}t�||� d S )Nr7 r0 r8 rG rV r� r� r� )r1 r2 r� r� r� r� c S s dS )Nr1 r- )r^ r- r- r. r� \ r� zDTestDataFrameSetItemCallable.test_setitem_callable.<locals>.<lambda>)r r3 rY )r* rQ rj r- r- r. �test_setitem_callableY s z2TestDataFrameSetItemCallable.test_setitem_callablec C sL dd� }t ddgddgg�}|||dk< t d|g|dgg�}t�||� d S )Nc S s | d S )Nr7 r- )r^ r- r- r. �incc s zETestDataFrameSetItemCallable.test_setitem_other_callable.<locals>.incr� r7 r )r r3 rY )r* r rQ r\ r- r- r. �test_setitem_other_callablea s z8TestDataFrameSetItemCallable.test_setitem_other_callableN)r� r� r� r r r- r- r- r. r X s r c @ s� e Zd Zejejjddd� dd� gddgd�dd � ��Zej�d e j e j g�d d � �Z d d� Z ej�dddg�ej�di ddgiddgig�dd� ��Zdd� Zdd � Zd S ) � TestDataFrameSetItemBooleanMask� mask_typec C s | t �| �d kS )Nr0 )r# �abs)rQ r- r- r. r� q r� z(TestDataFrameSetItemBooleanMask.<lambda>c C s | t �| �d kjS )Nr0 )r# r r4 )rQ r- r- r. r� q r� Z dataframer' )Zidsc C sL |� � }||�}|� � }tj||< |� � }tj|jt�|�< t�||� d S )N)r� r# r� r4 r' r3 rY )r* r r+ rQ r� re r\ r- r- r. �test_setitem_boolean_maskn s z9TestDataFrameSetItemBooleanMask.test_setitem_boolean_maskr� c C s\ t ddddgddddgd ��}|�� }|d dk}||�| �d �||�|< t�||� d S ) Nr7 rG r0 r8 rV r� r� r� )r= r> r= )r r� r r3 rY )r* r� rQ r\ r� r- r- r. �"test_setitem_boolean_mask_aligning� s zBTestDataFrameSetItemBooleanMask.test_setitem_boolean_mask_aligningc C s" t dddddddgddgd�}tdddddd d g�}d d d d d d d g}t||d �|d�}t dddddddgdddgd�}tdddddd d g�}d d ddd d d g}t||d �|d�}|�� } tjtdd�� | d jjdddgdd�} W d Q R X | d ks�t �|d dk} dd g|| < t� || � d S )Nr= r> )r� rC rD rE �k�lr� r� r7 r0 )�catsr4 )rA r? r8 F)Zcheck_stacklevelr! T)Zinplace) r r r r� r3 r� r� rs Zset_categoriesr) rY ) r* Zcats2Zidx2Zvalues2Z exp_multi_rowZcatsfZidxfZvaluesfrQ Z exp_fancyZ return_valuer� r- r- r. � test_setitem_mask_categorical� s" z=TestDataFrameSetItemBooleanMask.test_setitem_mask_categoricalr rz r �kwargsrA r7 rH r1 c C s6 ||d< t f |�}|�� }d|||k< t�||� d S )Nr �/ )r r� r3 rY )r* r r# rQ r� r- r- r. �%test_setitem_empty_frame_with_boolean� s zETestDataFrameSetItemBooleanMask.test_setitem_empty_frame_with_booleanc C s� t td��}dddg}t||tjdddgdd d gd d d ggtd�d�}t||t�t|�t|�f�d�}t||tjdddgdd dgdddggtd�d�}d||d | k< t� ||� t j t dd�� d||j d d� dk< W d Q R X d S )Nr8 r1 r2 rg g g �?g �?g �?g @g @g @g @g @)r )rA rH r5 r� zItem wrong length)rF r0 )r� rW r r# r' rz Zonesr& r3 rY rJ rK rL rA )r* r r� Zdf1r� r\ r- r- r. � test_setitem_boolean_indexing� s & * z=TestDataFrameSetItemBooleanMask.test_setitem_boolean_indexingc C sd t ddgddgdd��}|�� }tddgdd �}t d d d gid dgd�|j|d gf< t�||� d S )Nr7 r0 r8 rG r= )r= r> r? Fr? )r( r> rV r� r )rA )r r� r r} r3 rY )r* rQ r\ r� r- r- r. �-test_loc_setitem_all_false_boolean_two_blocks� s $zMTestDataFrameSetItemBooleanMask.test_loc_setitem_all_false_boolean_two_blocksN)r� r� r� r Z skip_array_manager_invalid_testrJ r� r� r r3 r r} r r" r% r& r' r- r- r- r. r m s * r c @ s� e Zd Zdd� Zdd� Zdd� Zej�dde j dgdggd d �dgdggg�d d � �Z ej�dd e � d gd gg�d gd ggg�dd� �Z dS )�%TestDataFrameSetitemCopyViewSemanticsc C sP d|j kst�|d �� }||d< tj|d dd�< t|dd� ��� sLt�d S )Nr� r1 rV rU )rH r) r� r# r� r r� )r* r+ r� r- r- r. �test_setitem_always_copy� s z>TestDataFrameSetitemCopyViewSemantics.test_setitem_always_copyc C s� t ddddgddddgd �d d d d gd�}|�d dtj� |d }d|j|jd d � df< ttjtjddg|jdd�}|d |k s�t�t� |d |� d S )Ng�������?g������ @g������@gffffff@gffffff@gffffff@gffffff @g333333 @)r^ r_ r r7 r0 r8 )rA ri �* )rA r( ) r ra r# r� r} rA r r) r3 rd )r* rQ r@ r\ r- r- r. �test_setitem_clear_caches� s ( z?TestDataFrameSetitemCopyViewSemantics.test_setitem_clear_cachesc C sj ddgd }t ddg|d�}|�� }|d d � }d|d< t ddddgg|d �}t�||� t�||� d S ) Nr1 r2 r0 g r )rA rH )r0 rV rV )rH )r r� r3 rY )r* r� rQ �df_copy�df_viewr\ r- r- r. �*test_setitem_duplicate_columns_not_inplace� s zPTestDataFrameSetitemCopyViewSemantics.test_setitem_duplicate_columns_not_inplacer r7 r )r c C s� |s t jjdd�}|j�|� ddg}tdddg|d�}|�� }|d d � }||dg< tddgddgg|d�} t�|| � t�||� d S ) Nz.Setitem with same dtype still changing inplace)�reasonr1 r2 r r7 )rA rH )rH ) rJ r� Zxfail�nodeZ add_markerr r� r3 rY ) r* r r� �requestr� r� rQ r, r- r\ r- r- r. �#test_setitem_same_dtype_not_inplace� s zITestDataFrameSetitemCopyViewSemantics.test_setitem_same_dtype_not_inplaceg �?c C sl ddg}t dddg|d�}|�� }|d d � }||dg< t ddgddgg|d�}t�||� t�||� d S )Nr1 r2 r r7 )rA rH g �?)rH )r r� r3 rY )r* r r� rQ r, r- r\ r- r- r. �2test_setitem_listlike_key_scalar_value_not_inplace s zXTestDataFrameSetitemCopyViewSemantics.test_setitem_listlike_key_scalar_value_not_inplaceN) r� r� r� r) r+ r. rJ r� r� r# r' r2 r3 r- r- r- r. r( � s 0r( )4r �numpyr# rJ Z pandas.util._test_decorators�utilZ_test_decoratorsr Zpandas.core.dtypes.baser r� Zpandas.core.dtypes.commonr r r Zpandas.core.dtypes.dtypesr r r r Zpandasr r r r r r r r r r r r r r r r r Zpandas._testingZ_testingr3 Zpandas.core.arraysr Zpandas.tseries.offsetsr r r� r� r r r r( r- r- r- r. �<module> s, H e.m.e
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка