Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/series/indexing/__pycache__/test_setitem.cpython-37.pyc
Назад
B �5�g s � @ sx d dl mZm Z d dlZd dlZd dlmZmZmZm Z m Z m Z m Z m Z mZmZmZ d dlmZ d dlmZ d dlmZ G dd� d�ZG dd � d �ZG d d � d �ZG d d � d �ZG dd� d�Z G dd� d�Z G dd� d�Z dd� Z dd� Z dd� Z!dd� Z"G d d � d �Z#G d d � d �Z$ej%�&d ej'e d!d"d#d$d%d&d'd(d)g �e ej(d"ej(d$ej(d&ej(d(ej(g �e)ddd!�d*d+�ej'e d,d,d-d-g�e ej(d,ej(d-ge*d.�e)ddd!�d/d+�ej'e e�+d)��e ej(ej(ej(ej(ej(d$d%d&d'd(g �e)dd$�d0d+�ej'e d1d!d"g�e ej(d!d"g�d d2d+�ej'e d-g�e ej(ge*d.�d d3d+�ej'e d-d,g�e ej(d,ge*d.�d d4d+�g�G d5d6� d6e$��Z,G d7d8� d8e$�Z-G d9d:� d:e$�Z.G d;d<� d<e$�Z/G d=d>� d>e$�Z0G d?d@� d@e$�Z1ej%�&dAe d1d!d"g�e ej(d!d"g�fe dBdCdDg�e ej(dCdDg�fe e dEd1d1�e dEd1d!�e dEd1d"�g�e e e dEd1d!�e dEd1d"�g�fe dFdGdHg�e ddGdHg�fg�G dIdJ� dJe$��Z2dKdL� Z3dS )M� )�date�datetimeN) � Categorical� DataFrame� DatetimeIndex�Index� MultiIndex�NaT�Series� Timedelta� Timestamp� date_range� period_range)� IndexingError)�BDayc @ sP e Zd Zdd� Zdd� Zdd� Zdd� Zej� d d d d g�d d� �Z dd� Z dS )�TestSetitemDT64Valuesc C s� t tddd��}d |d< |d tks(t�d |dd�< |d tksDt�tj|d< |d tks^t�tj|dd�< |d tks|t�d S ) Nz1/1/2000� )�periods� � � � � )r r r �AssertionError�np�nan)�self�series� r �_/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/series/indexing/test_setitem.py�test_setitem_none_nan s z+TestSetitemDT64Values.test_setitem_none_nanc C s@ t �ddg�}tddg|d�}|�� }d|jg < t�||� d S )N)�a� )�b� r" r$ )�indexr )r � from_tuplesr �copy�loc�tm�assert_series_equal)r �idx�result�expectedr r r �#test_setitem_multiindex_empty_slice+ s z9TestSetitemDT64Values.test_setitem_multiindex_empty_slicec C sN t dddgdddgd�}t�� |d< |jt�� ks6t�|d t�� ksJt�d S )Nr" r$ r �Dater# �other)r% )r r �todayr/ r )r �serr r r � test_setitem_with_string_index3 s z4TestSetitemDT64Values.test_setitem_with_string_indexc C sT t dddd�}ddddg}t||d �}|�� }tj|d< tj|jd < t�||� d S ) NZ2017r z US/Eastern)r �tz)r r" )r r$ )r r )r r )r% r )r r r' r r �ilocr) r* )r �arrr% r, r- r r r �)test_setitem_tuple_with_datetimetz_values: s z?TestSetitemDT64Values.test_setitem_tuple_with_datetimetz_valuesr4 z US/Eastern�UTCz Asia/Tokyoc C s t tddd|d��}|jd|� d�ks*t�t td|d�td |d�td |d�g�}|�� }td |d�||�d < t�||� t td |d�td |d�gd dgd�}|jd|� d�ks�t�t td|d�td |d�td|d�g�}|�� }|||�d dg< t�||� d S )Nz 2016-01-01�Hr )�freqr r4 zdatetime64[ns, �]z2016-01-01 00:00)r4 z2011-01-01 00:00z2016-01-01 02:00z 2011-01-01r" z 2012-01-01r$ )r% z2012-01-01 00:00)r r �dtyper r r' r) r* )r r4 � indexer_sli�orig�expr2 �valsr r r �test_setitem_with_tzD s( z*TestSetitemDT64Values.test_setitem_with_tzc C s d}t tddd|d��}|jd|� d�ks.t�t td|d �td |d �td |d �g�}|�� }td |d �||�d < t�||� t td |d �td|d �gd dgd�}|jd|� d�ks�t�t td|d �td|d �td|d �g�}|�� }|||�d dg< t�||� d S )Nz US/Easternz 2016-11-06r9 r )r: r r4 zdatetime64[ns, r; z2016-11-06 00:00-04:00)r4 z2011-01-01 00:00-05:00z2016-11-06 01:00-05:00z 2011-01-01r" z 2012-01-01r$ )r% z2016-11-06 00:00z2011-01-01 00:00z2012-01-01 00:00)r r r< r r r' r) r* )r r= r4 r> r? r2 r@ r r r �test_setitem_with_tz_dsti s* z.TestSetitemDT64Values.test_setitem_with_tz_dstN) �__name__� __module__� __qualname__r r. r3 r7 �pytest�mark� parametrizerA rB r r r r r s %r c @ sd e Zd Zdd� Zej�dejej g�ej�dddg�dd� ��Z ej�d d de j fg�d d � �Z d S )�TestSetitemScalarIndexerc C sF t t�dd�t�dd�d�}d}tjt|d�� d|d< W d Q R X d S )Nr r )r% z2index -11 is out of bounds for axis 0 with size 10)�match�fooi����)r r) Z rands_arrayrF �raises� IndexError)r r2 �msgr r r �#test_setitem_negative_out_of_bounds� s z<TestSetitemScalarIndexer.test_setitem_negative_out_of_bounds�indexer� ser_indexr r" c C sX t ddgdd�}||�}t dg|gd�|d< t t dg|gd�dgdd�}t�||� d S )Nr �object)r< �* )r% )r r) r* )r rP rQ r2 Zidxrr- r r r � test_setitem_series_object_dtype� s z9TestSetitemScalarIndexer.test_setitem_series_object_dtypezindex, exp_value)r rS c C s> t ddg�}t dg|gd�|jd< t |dg�}t�||� d S )Nr rS )r% )r r( r) r* )r r% Z exp_valuer2 r- r r r �test_setitem_series� s z,TestSetitemScalarIndexer.test_setitem_seriesN)rC rD rE rO rF rG rH r) r( �atrT r r rU r r r r rI � s rI c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �TestSetitemSlicesc C sd d}t jt|jdd�d�� d|dd�< W d Q R X t jt|jdd�d�� d|d d�< W d Q R X d S ) NzUcannot do slice indexing on DatetimeIndex with these indexers \[{key}\] of type floatz4\.0)�key)rJ r g @g $@z4\.5g @)rF rL � TypeError�format)r �datetime_seriesrN r r r � test_setitem_slice_float_raises� s z1TestSetitemSlices.test_setitem_slice_float_raisesc C sT t td�ttd��d�}d|dd �< |dk�� s4t�d|d d�< |dk�� sPt�d S )Nr )r% r i�r )r �range�list�allr )r r2 r r r �test_setitem_slice� s z$TestSetitemSlices.test_setitem_slicec C sd t tj�d�ddddddddgd �}d |d d�< |d d� d k�� sHt�|dd � d k�� r`t�d S ) N� r$ r r r � � � )r% r )r r �randomZrandnr_ r �any)r r2 r r r �test_setitem_slice_integers� s $ z-TestSetitemSlices.test_setitem_slice_integersc C sJ t t�d��t�t�d�d�}d|d d d�< |d d d� dk�� sFt�d S )N� )r% r r$ )r r) Z makeIntIndex�astype�floatr_ r )r r r r r �test_setitem_slicestep� s z(TestSetitemSlices.test_setitem_slicestepc C sl t �ddgtd�f�}ttjdtjd�|�}d||�d d d�< tdddd dd d d dd g |�}t�||� d S )Nr r" Zabcder )r< �d r r$ r r r r � ) r Z from_productr^ r r �arangeZint64r) r* )r r= �mir, r- r r r � test_setitem_multiindex_slice� s z/TestSetitemSlices.test_setitem_multiindex_sliceN)rC rD rE r\ r` rg rk rp r r r r rW � s rW c @ sZ e Zd Zdd� Zdd� Zdd� Zej�de e j e g�dd � �Z d d � Zd d � Zdd� ZdS )�TestSetitemBooleanMaskc C s� ||� � k}|�� }|d ||< |d }t�|| || � |�� }|d dd� ||< |d dd� �|�}|| || < t�|| || � d S )Nr$ r r )�medianr' r) r* Z reindex_like)r � string_series�maskr, r- r r r �test_setitem_boolean� s z+TestSetitemBooleanMask.test_setitem_booleanc C sj |}|j dt� d�|�� k}d}tjt|d�� d||< W d Q R X tjt|d�� d|j|< W d Q R X d S )Nr" )r: zsUnalignable boolean Series provided as indexer \(index of the boolean Series and of the indexed object do not match)rJ )�shiftr rr rF rL r r( )r r[ �tsZ mask_shiftedrN r r r �test_setitem_boolean_corner� s z2TestSetitemBooleanMask.test_setitem_boolean_cornerc C s@ |� � }|�� }d||dk< |�� }d||dk< t�||� d S )Nr )Z sort_valuesr' r) r* )r rs Zorderedr' r- r r r �$test_setitem_boolean_different_order� s z;TestSetitemBooleanMask.test_setitem_boolean_different_order�funcc C sF t d dd g�}|dddg�}ddg||< t dddg�}t�||� d S )Nr# TFr! �c)r r) r* )r rz r2 rt r- r r r � test_setitem_boolean_python_list s z7TestSetitemBooleanMask.test_setitem_boolean_python_listc C s� t ddddg|d�}t td�|d�||dk< t ddddg|d�}t�||� t ddddg|d�}t td�|d�|j|dk< t�||� t ddddg|d�}t td�|d�}|j|d k |j|dk< t�||� d S ) Nr r r ra )r< r r$ r r" )r r] r) r* r( )r Zany_nullable_numeric_dtyper2 r- Zloc_serr r r �'test_setitem_boolean_nullable_int_types s z>TestSetitemBooleanMask.test_setitem_boolean_nullable_int_typesc C sv t d gd �}dgd dgd dgd }td�||< |}t d gd ttd�� d gd ��d�}t�||� d S )Nr Fr Tr r$ rR )r r] r^ ri r) r* )r r2 rt r, r- r r r �Atest_setitem_with_bool_mask_and_values_matching_n_trues_in_length s *zXTestSetitemBooleanMask.test_setitem_with_bool_mask_and_values_matching_n_trues_in_lengthc C s< t dddg�}tj|d< t tjddgtd�}t�||� d S )NTFr )r< )r r r rR r) r* )r r, r- r r r �test_setitem_nan_with_bool" s z1TestSetitemBooleanMask.test_setitem_nan_with_boolN)rC rD rE ru rx ry rF rG rH r^ r �arrayr r| r} r~ r r r r r rq � s rq c @ s e Zd Zdd� Zdd� ZdS )� TestSetitemViewCopySemanticsc C s� t dddd�}|d }t|�}|j|k s,t�|jjj|jjjk sDt�|jdksRt�t|jd< |jjd kslt�|j|k szt�|jjj|jjjk s�t�|d |ks�t�|jdks�t�d S )NZ20130101r z US/Eastern)r r4 r" �D) r r �_valuesr �_data�baser: r r5 )r �dtirw r2 r r r �,test_setitem_invalidates_datetime_index_freq+ s zITestSetitemViewCopySemantics.test_setitem_invalidates_datetime_index_freqc C s� t dddd�}|d }t|�}|j|k s,t�|jjj|jjjk sDt�|jjd |k sXt�|jjd jj|jjjk svt�t|d d d�< |d tks�t�|d |ks�t�d S )Nz 2016-01-01r z US/Pacific)r r4 r r ) r r r� r r� r� Z_mgrZarraysr )r r� rw r2 r r r �'test_dt64tz_setitem_does_not_mutate_dti> s zDTestSetitemViewCopySemantics.test_dt64tz_setitem_does_not_mutate_dtiN)rC rD rE r� r� r r r r r� * s r� c @ s e Zd Zdd� Zdd� ZdS )�TestSetitemCallablec C sL t ddddgtd�d�}d|dd � < t ddddgtd�d�}t�||� d S ) Nr" r$ r r ZABCD)r% ���c S s dS )N�Ar )�xr r r �<lambda>Q � z?TestSetitemCallable.test_setitem_callable_key.<locals>.<lambda>)r r^ r) r* )r r2 r- r r r �test_setitem_callable_keyN s z-TestSetitemCallable.test_setitem_callable_keyc C sD dd� }t ddddg�}|||dk < t dd|dg�}t�||� d S )Nc S s | d S )Nr" r )r� r r r r� X r� zATestSetitemCallable.test_setitem_callable_other.<locals>.<lambda>r" r$ r� r r )r r) r* )r �incr2 r- r r r �test_setitem_callable_otherV s z/TestSetitemCallable.test_setitem_callable_otherN)rC rD rE r� r� r r r r r� M s r� c @ sd e Zd Zdd� Zdd� Zdd� Zej�de d�e d�� � e d�� � g�d d � �Z d d � Z d d� ZdS )�TestSetitemWithExpansionc C s6 t d�}ttd�}d||< td|g�}t�||� d S )Nz 2012-01-01)r< �/ )r r rR r) r* )r rX r r- r r r �test_setitem_empty_seriesb s z2TestSetitemWithExpansion.test_setitem_empty_seriesc C s^ t g tg dd�td�}td�}d||< t dt|gdd��}t�||� |jj|jjksZt�d S )Nr� )r: )r< z 2012-01-01r� ) r r rR r r) r* r% r: r )r r rX r- r r r �6test_setitem_empty_series_datetimeindex_preserves_freqj s zOTestSetitemWithExpansion.test_setitem_empty_series_datetimeindex_preserves_freqc C sV t d�}t|gdgtd�}|d }tg td�}d|d< ||d< |d }||ksRt�d S )Nl P�i�5� � timestamp)r% r< )r< g �r@Zanything)r r rR r )r r� r r- r, r r r �3test_setitem_empty_series_timestamp_preserves_dtypes s zLTestSetitemWithExpansion.test_setitem_empty_series_timestamp_preserves_dtype�tdz9 daysc C s� t d|gddgtd�}t dg�}||d< t�||� t|d t�sFt�t dg�}td�|jd< t�||� t|d t�s|t�d S )Nr� r r� )r% r< z9 days)r rR r) r* � isinstancer r r( )r r� r- r2 r r r �#test_append_timedelta_does_not_cast s z<TestSetitemWithExpansion.test_append_timedelta_does_not_castc C sR t td�}td�|d< d|d< d|d< t td�ddgdddgd�}t�||� d S ) N)r< z 2016-01-01r! g @r# rK r{ )r% )r rR r r) r* )r r2 r- r r r �*test_setitem_with_expansion_type_promotion� s zCTestSetitemWithExpansion.test_setitem_with_expansion_type_promotionc C sJ |� � }d|jkst�d|d< tdgdgdd�}|�|�}t�||� d S )NZfoobarr" r )r% �name)r' r% r r �appendr) r* )r rs r2 Zappr- r r r �test_setitem_not_contained� s z3TestSetitemWithExpansion.test_setitem_not_containedN)rC rD rE r� r� r� rF rG rH r Zto_timedelta64Zto_pytimedeltar� r� r� r r r r r� a s r� c C sj t �d�} d| j_t| �}xJtt|��D ]:}d}tjt |d�� d||< W d Q R X | | dks(t �q(W d S )Nr Fz#assignment destination is read-only)rJ r" r ) r �zeros�flags� writeabler r] �lenrF rL � ValueErrorr )r� r �nrN r r r �.test_setitem_scalar_into_readonly_backing_data� s r� c C sT t �d�} d| j_t| �}d}tjt|d�� d|dd�< W d Q R X | �� rPt �d S )Nr Fz#assignment destination is read-only)rJ r" r ) r r� r� r� r rF rL r� rf r )r� r rN r r r �-test_setitem_slice_into_readonly_backing_data� s r� c C s@ t tddgddgd��} | �� }d|d d �< t tddgddgd��}t�||� | �� }d|d< t tddgddgd��}t�||� | �� }d||jdk< t tddgddgd��}t�||� | �� }d|ddg< t tddgddgd��}t�||� | �� }dd g|_d|d < t tddgddgd�dd gd �}t�||� d S ) Nr# r! )� categoriesr" r FTr� �y)r% )r r r' r) r* r% )r> r2 r? r r r �&test_setitem_categorical_assigning_ops� s, r� c C sJ t tdddg��} t tdtjdgdddgd��}tj| d< t�| |� d S )Nr" r$ r )r� )r r r r r) r* )r2 r? r r r �!test_setitem_nan_into_categorical� s r� c @ s> e Zd Zej�dddg�ejjddddged�d d � ��Zd S ) �TestSetitemCasting�uniqueTF�valr g @�3)Zidsc C s� t ddg�}|sddg|_|||�d< t|jd �t|�ks@t�t d|gt|jd�}|sx|tjk rxt ||gtddgd�}t�||� d S )NTFr" )r< r% )r r% �typer5 r rR r) r* )r r� r= r� r2 r- r r r � test_setitem_non_bool_into_bool� s z2TestSetitemCasting.test_setitem_non_bool_into_boolN)rC rD rE rF rG rH r� r� r r r r r� � s r� c @ s^ e Zd ZdZejdd� �Zdd� Zdd� Zdd � Z d d � Z d d � Z dd� Z dd� Z dd� ZdS )�SetitemCastingEquivalentsz� Check each of several methods that _should_ be equivalent to `obj[key] = val` We assume that - obj.index is the default Index(range(len(obj))) - the setitem does not expand the obj c C s dS )z\ Indicate that we are not (yet) checking whether or not setting is inplace. Nr )r r r r � is_inplace s z$SetitemCastingEquivalents.is_inplacec C s> |}|� � }|j}|||�|< t�||� | �||||� d S )N)r' r� r) r* �_check_inplace) r �objrX r- r� rP r� r> r6 r r r � check_indexer s z'SetitemCastingEquivalents.check_indexerc C sP |d kr nB|r>|j jdkr.|j|jjks<t�qL|j|ksLt�nt�||j� d S )N)�m�M)r< �kindr� r� r r) Z assert_equal)r r� r> r6 r� r r r r� s z(SetitemCastingEquivalents._check_inplacec C s t |t�sd S | �||||||� |tjkrD| �||||tj|� n |tjkrd| �||||tj|� t||d �}| �||||||� |tjk r�t ||d �}| �||||||� |g} | �|| ||||� t � | �} | �|| ||||� dd� |gD �} | �|| ||||� d S )Nr" c s s | ] }|V qd S )Nr )�.0r� r r r � <genexpr>? s z9SetitemCastingEquivalents.test_int_key.<locals>.<genexpr>) r� �intr� r) r( rV r5 Ziatr] �slicer r� ) r r� rX r- r� r= r� �rngZslc�ilkey�indkey�genkeyr r r � test_int_key& s$ z&SetitemCastingEquivalents.test_int_keyc C s� t |t�sd S |tjk r,| �||||||� ttt|���| }| �||||||� t� |�}| �||||||� dd� |D �} | �|| ||||� d S )Nc s s | ] }|V qd S )Nr )r� r� r r r r� P s z;SetitemCastingEquivalents.test_slice_key.<locals>.<genexpr>) r� r� r) r( r� r^ r] r� r r� ) r r� rX r- r� r= r� r� r� r� r r r �test_slice_keyB s z(SetitemCastingEquivalents.test_slice_keyc C s< t j|jtd�}d||< |�� }|||�|< t�||� d S )N)r< T)r r� �shape�boolr' r) r* )r r� rX r- r� r= rt r r r � test_mask_keyS s z'SetitemCastingEquivalents.test_mask_keyc C sX t j|jtd�}d||< |}|�� }|j}|�| |�} t�| |� | � ||||� d S )N)r< T) r r� r� r� r' r� �wherer) r* r� ) r r� rX r- r� r� rt r> r6 �resr r r �test_series_where\ s z+SetitemCastingEquivalents.test_series_wherec C s� t |�j|jkrt�d� tj|jtd�}d||< |jtkrZd}tjj |d�}|j � |� t |�� | |�} t �| t |�� d S )Nz"test not applicable for this dtype)r< Tz3Index/Series casting behavior inconsistent GH#38692)�reason)r r< rF �skipr r� r� r� rG Zxfail�nodeZ add_markerr� r) �assert_index_equal) r r� rX r- r� �requestrt rN rG r� r r r �test_index_wherei s z*SetitemCastingEquivalents.test_index_wherec C sV t |�j|jkrt�d� tj|jtd�}d||< t |��||�}t � |t |�� d S )Nz"test not applicable for this dtype)r< T) r r< rF r� r r� r� r� Zputmaskr) r� )r r� rX r- r� rt r� r r r �test_index_putmaskx s z,SetitemCastingEquivalents.test_index_putmaskN)rC rD rE �__doc__rF �fixturer� r� r� r� r� r� r� r� r� r r r r r� � s r� zobj,expected,keyr$ r r r r r ra rm r Zint_series_slice_key_step)�idTF)r< Zbool_series_slice_key_stepZint_series_slice_keyr" Zint_series_int_keyZ bool_series_int_key_change_allZbool_series_int_keyc @ s, e Zd Zejeje�d�gd�dd� �ZdS )� TestSetitemCastingEquivalents�NaN)�paramsc C s |j S )zi One python float NaN, one np.float64. Only np.float64 has a `dtype` attribute. )�param)r r� r r r r� � s z!TestSetitemCastingEquivalents.valN) rC rD rE rF r� r r Zfloat64r� r r r r r� � s 0r� c @ sl e Zd Zejdd� �Zejeeegd�dd� �Z ejdd� �Z ejdd � �Z ejd d � �Z ejd d � �Z dS )�!TestSetitemTimedelta64IntoNumericc C s t �dd�}|S )Nr �ns)r � timedelta64)r r� r r r r� � s z%TestSetitemTimedelta64IntoNumeric.val)r� c C s |j S )N)r� )r r� r r r r< � s z'TestSetitemTimedelta64IntoNumeric.dtypec C s t �d��|�}t|�}|S )Nr )r rn ri r )r r< r6 r2 r r r r� � s z%TestSetitemTimedelta64IntoNumeric.objc C s8 t �d��|�}t|�}|�t�}t �dd�|jd< |S )Nr r r� r )r rn ri r rR r� �values)r r< r6 r2 r r r r- � s z*TestSetitemTimedelta64IntoNumeric.expectedc C s dS )Nr r )r r r r rX � s z%TestSetitemTimedelta64IntoNumeric.keyc C s dS )zM Indicate we do _not_ expect the setting to be done inplace. Fr )r r r r r� � s z,TestSetitemTimedelta64IntoNumeric.is_inplaceN)rC rD rE rF r� r� �complexr� rj r< r� r- rX r� r r r r r� � s r� c @ s� e Zd Zejddgd�dd� �Zejdd� �Zejdd � �Zejd d � �Zejd d � �Z ejde e j gd�dd� �Z ejdd� �ZdS )�TestSetitemDT64IntoIntzM8[ns]zm8[ns])r� c C s |j S )N)r� )r r� r r r r< � s z TestSetitemDT64IntoInt.dtypec C s t �dd�}|dkr || }|S )Nz2021-01-18 13:25:00r� zm8[ns])r � datetime64)r r< r� r r r �scalar� s z TestSetitemDT64IntoInt.scalarc C s, t ||dgtd�}t|d t|��s(t�|S )Nr )r< r )r rR r� r� r )r r� r- r r r r- � s z TestSetitemDT64IntoInt.expectedc C s t dddg�S )Nr" r$ r )r )r r r r r� � s zTestSetitemDT64IntoInt.objc C s t d d�S )Nr� )r� )r r r r rX � s zTestSetitemDT64IntoInt.keyNc C s |j }|d kr|S |||g�S )N)r� )r r� r� �boxr r r r� s zTestSetitemDT64IntoInt.valc C s dS )NFr )r r r r r� s z!TestSetitemDT64IntoInt.is_inplace)rC rD rE rF r� r< r� r- r� rX r^ r r� r� r� r r r r r� � s r� c @ sn e Zd Zejdd� �Zejdd� �Zejdedd�gd�dd � �Zejd e j gd�d d � �Z ejd d� �Z d S )�TestSetitemNAPeriodDtypec C sN t tdddd��}tj|j�d�|< || tksJtdd� || D ��sJt�|S )Nz 2000-01-01r r� )r r: �i8c s s | ]}|t kV qd S )N)r )r� r� r r r r� s z4TestSetitemNAPeriodDtype.expected.<locals>.<genexpr>)r r r �valuer� �viewr_ r )r rX r? r r r r- s &z!TestSetitemNAPeriodDtype.expectedc C s t tdddd��S )Nz 2000-01-01r r� )r r: )r r )r r r r r� s z TestSetitemNAPeriodDtype.objr r )r� c C s |j S )N)r� )r r� r r r rX s z TestSetitemNAPeriodDtype.keyNc C s |j S )N)r� )r r� r r r r� s z TestSetitemNAPeriodDtype.valc C s dS )NTr )r r r r r� " s z#TestSetitemNAPeriodDtype.is_inplace) rC rD rE rF r� r- r� r� rX r r r� r� r r r r r� s r� c @ s� e Zd Zejddddgd�dd� �Zejdd � �Zejd eje e� d d �e� d d �gd�d d� �Z ejdd� �Z ejdd� �Zejdd� �Zd S )� TestSetitemNADatetimeLikeDtypezm8[ns]zM8[ns]zdatetime64[ns, UTC]zdatetime64[ns, US/Central])r� c C s |j S )N)r� )r r� r r r r< - s z$TestSetitemNADatetimeLikeDtype.dtypec C s0 t ddd�j}t||d�}|j|ks(t�t|�S )Nz 2016-01-01r )r )r< )r Zasi8r r< r r )r r< Zi8valsr+ r r r r� 3 s z"TestSetitemNADatetimeLikeDtype.objNr r� c C s |j S )N)r� )r r� r r r r� : s z"TestSetitemNADatetimeLikeDtype.valc C s& |t kp$|d kp$|tjkp$|j|jkS )N)r r r r< )r r� r� r r r r� F s z)TestSetitemNADatetimeLikeDtype.is_inplacec C s0 |r |j nt}t|gt|dd � � |d�}|S )Nr" )r< )r< rR r r^ )r r� r� r� r< r- r r r r- M s z'TestSetitemNADatetimeLikeDtype.expectedc C s dS )Nr r )r r r r rX S s z"TestSetitemNADatetimeLikeDtype.key)rC rD rE rF r� r< r� r r r r� r� r� r� r- rX r r r r r� ' s r� c @ sD e Zd Zejdd� �Zejdd� �Zejdd� �Zejdd� �Zd S ) �$TestSetitemMismatchedTZCastsToObjectc C s t tdddd��S )N�2000r$ z US/Central)r r4 )r r )r r r r r� Z s z(TestSetitemMismatchedTZCastsToObject.objc C s t ddd�S )Nr� z US/Eastern)r4 )r )r r r r r� ^ s z(TestSetitemMismatchedTZCastsToObject.valc C s dS )Nr r )r r r r rX b s z(TestSetitemMismatchedTZCastsToObject.keyc C s$ t tddd�tddd�gtd�}|S )Nz2000-01-01 00:00:00-05:00z US/Eastern)r4 z2000-01-02 00:00:00-06:00z US/Central)r< )r r rR )r r- r r r r- f s z-TestSetitemMismatchedTZCastsToObject.expectedN) rC rD rE rF r� r� r� rX r- r r r r r� X s r� z obj,expectedg �?g @g @i� rK �barZbazc @ s6 e Zd Zejdd� �Zejdd� �Zejdd� �ZdS )�TestSeriesNoneCoercionc C s dS )Nr r )r r r r rX � s zTestSeriesNoneCoercion.keyc C s d S )Nr )r r r r r� � s zTestSeriesNoneCoercion.valc C s |j jdkS )N�i)r< r� )r r� r r r r� � s z!TestSeriesNoneCoercion.is_inplaceN)rC rD rE rF r� rX r� r� r r r r r� r s r� c C s� t dddgdddgd��} | �d�}d |d d d g< td ddgdd �}t�||� d | jd d d gdf< t dd ddgi�}t�| |� d S )Nr" r$ r r r r )r! r# r# rm TF)�datar� r r! )r �popr r) r* r( Zassert_frame_equal)Zdfr, r- r r r � test_setitem_with_bool_indexer� s r� )4r r �numpyr rF Zpandasr r r r r r r r r r r Zpandas._testingZ_testingr) Zpandas.core.indexingr Zpandas.tseries.offsetsr r rI rW rq r� r� r� r� r� r� r� r� r� rG rH r� r r� rR rn r� r� r� r� r� r� r� r� r r r r �<module> s� 4 t+T#H " $ ((1 $ (
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка