Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexing/__pycache__/test_coercion.cpython-37.pyc
Назад
B �5�g� � @ s� d dl mZ d dlmZ d dlZd dlZd dlZd dlm Z m Z d dl Z d dl mZ ejddd�dd � �ZG d d � d �ZG d d � d e�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZdS )� )� annotations)� timedeltaN)�IS64�is_platform_windowsT�class)Zautouse�scopec # sz � j }t�|j|j|jg�}� fdd�}� jj}|js<|j rDd V n2x*|D ]"}||�sJt d|j � d|� ���qJW d V d S )Nc s. | \�� ��j jj}t� ��fdd�|D ��S )Nc 3 s, | ]$}�|j ko"� |j ko"�|j kV qd S )N)�name)�.0�x)�dtype�klass�method� �Y/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexing/test_coercion.py� <genexpr>! s z<check_comprehensiveness.<locals>.has_test.<locals>.<genexpr>)�node�session�items�any)�comboZ cls_funcs)�request)r r r r �has_test s z)check_comprehensiveness.<locals>.has_testz test method is not defined: z, ) �cls� itertools�product�klasses�dtypesr �config�option�lf�keyword�AssertionError�__name__)r r Zcombosr �optsr r )r r �check_comprehensiveness s r$ c @ s6 e Zd ZddgZddddddd d d g Zed d � �ZdS )� CoercionBase�index�series�object�int64�float64� complex128�bool� datetime64� datetime64tz� timedelta64Zperiodc C s t | ��d S )N)�NotImplementedError)�selfr r r r D s zCoercionBase.methodN)r" � __module__� __qualname__r r �propertyr r r r r r% 5 s r% c @ s e Zd ZdZdd� Zej�ddefdefdefdefg�d d � �Z ej�dde j fde j fde j fdefg�d d � �Zej�de �d�e jfe �d �e jfg�dd� �Zej�dde j fde j fde j fdefg�dd� �Zej�dde j fde j fde j fdefg�dd� �Zej�ddefdefdefdefdefde jfg�dd� �Zej�de�d�dfdefdefg�dd � �Zej�dejdd d �d fejdd d �efe�d�efdefg�d!d"� �Zej�de�d#�d$fdefdefg�d%d&� �Z d'd(� Z d)d*� Z ej�ddefd+e fdefg�d,d-� �Z ej�dd+e j fde j fdefg�d.d/� �Z!ej�dd+e fd0e j fdefg�d1d2� �Z"ejj#d3d4�d5d6� �Z$ejj#d3d4�d7d8� �Z%ejj#d3d4�d9d:� �Z&ejj#d3d4�d;d<� �Z'ejj#d3d4�d=d>� �Z(ejj#d3d4�d?d@� �Z)ejj#d3d4�dAdB� �Z*dCS )D�TestSetitemCoercion�setitemc C s. |� � }||d< t�||� |j|ks*t�dS )z4test series value's coercion triggered by assignment� N)�copy�tm�assert_series_equalr r! )r1 �original_seriesZ loc_valueZexpected_series�expected_dtype�tempr r r �!_assert_setitem_series_conversionM s z5TestSetitemCoercion._assert_setitem_series_conversionz val,exp_dtyper7 g�������?y �? �?Tc C sB t �td��}|jtks t�t �d|ddg�}| �||||� d S )N�abcd�a�c�d)�pd�Series�listr r( r! r> )r1 �val� exp_dtype�obj�expr r r �test_setitem_series_object] s z.TestSetitemCoercion.test_setitem_series_objectc C s� t �ddddg�}|jtjks"t�|tjkrjt �ddddg�}| �|d|tj� tj j dd�}|j � |� t �d|ddg�}| �||||� d S )Nr7 � � � g�������?z GH12747 The result must be float)�reason) rC rD r �npr) r! r* r> �pytest�mark�xfailr � add_marker)r1 rF rG r rH rI rQ r r r �test_setitem_series_int64g s z-TestSetitemCoercion.test_setitem_series_int64i c C s� t jddddgtjd�}|jtjks(t�|tjkrvt jddddgtjd�}| �|||tj� tj j dd�}|j � |� |tjkr�d nt }d }tj||d �� t jd|ddgtjd�}W d Q R X | �||||� d S ) Nr7 rK rL rM )r r z6BUG: it must be pd.Series([1, 1, 3, 4], dtype=np.int16)rN z2Values are too large to be losslessly cast to int8)�match)rC rD rO �int8r r! �int16r> rP rQ rR r rS � FutureWarningr9 �assert_produces_warning) r1 rF rG r rH rI rQ �warn�msgr r r �test_setitem_series_int8x s "z,TestSetitemCoercion.test_setitem_series_int8c C sH t �ddddg�}|jtjks"t�t �d|ddg�}| �||||� d S )Ng�������?g������@gffffff @g������@)rC rD r rO r* r! r> )r1 rF rG rH rI r r r �test_setitem_series_float64� s z/TestSetitemCoercion.test_setitem_series_float64c C sH t �ddddg�}|jtjks"t�t �d|ddg�}| �||||� d S )Ny �? �?y @ @y @ @y @ @)rC rD r rO r+ r! r> )r1 rF rG rH rI r r r � test_setitem_series_complex128� s z2TestSetitemCoercion.test_setitem_series_complex128�3rL c C sL t �ddddg�}|jtjks"t�t jd|ddg|d�}| �||||� d S )NTF)r )rC rD r rO �bool_r! r> )r1 rF rG rH rI r r r �test_setitem_series_bool� s z,TestSetitemCoercion.test_setitem_series_boolz 2012-01-01zdatetime64[ns]r c C sp t �t �d�t �d�t �d�t �d�g�}|jdks8t�t �t �d�|t �d�t �d�g�}| �||||� d S )Nz 2011-01-01z 2011-01-02z 2011-01-03z 2011-01-04zdatetime64[ns])rC rD � Timestampr r! r> )r1 rF rG rH rI r r r � test_setitem_series_datetime64� s z2TestSetitemCoercion.test_setitem_series_datetime64z US/Eastern)�tzzdatetime64[ns, US/Eastern]z US/Pacificc C s� d}t �t jd|d�t jd|d�t jd|d�t jd|d�g�}|jdksLt�t �t jd|d�|t jd|d�t jd|d�g�}| �||||� d S )Nz US/Easternz 2011-01-01)rd z 2011-01-02z 2011-01-03z 2011-01-04zdatetime64[ns, US/Eastern])rC rD rb r r! r> )r1 rF rG rd rH rI r r r � test_setitem_series_datetime64tz� s z4TestSetitemCoercion.test_setitem_series_datetime64tzz12 dayztimedelta64[ns]c C sp t �t �d�t �d�t �d�t �d�g�}|jdks8t�t �t �d�|t �d�t �d�g�}| �||||� d S )Nz1 dayz2 dayz3 dayz4 dayztimedelta64[ns])rC rD � Timedeltar r! r> )r1 rF rG rH rI r r r � test_setitem_series_timedelta64� s z3TestSetitemCoercion.test_setitem_series_timedelta64c C s> t �ddg�}t|j�|d d �< t �ddg�}t�||� d S )Nr@ r7 )rC rD rE �valuesr9 r: )r1 Zser�expectedr r r �0test_setitem_series_no_coercion_from_values_list s zDTestSetitemCoercion.test_setitem_series_no_coercion_from_values_listc C s� |� � }d||< tjdddddg|d�}t�||� |jj|ksDt�|� � }d|j|< tjdddddg|d�}t�||� |jj|ks�t�dS )z-test index's coercion triggered by assign key� r7 rK rL rM )r& N) r8 rC rD r9 r: r& r r! �loc)r1 r; Zloc_keyZexpected_indexr< r= rI r r r � _assert_setitem_index_conversion s z4TestSetitemCoercion._assert_setitem_index_conversionrk c C s� t jddddgtd�d�}|jjtks*t�|tkrb|�� }d}t j ||d�� d |d < W d Q R X n$t � td�|g �}| � ||||� d S ) Nr7 rK rL rM r? )r& z/index 5 is out of bounds for axis 0 with size 4)rU rk ) rC rD rE r& r r( r! � IndexErrorr8 rP �raises�Indexrm )r1 rF rG rH r= r[ � exp_indexr r r �test_setitem_index_object! s z-TestSetitemCoercion.test_setitem_index_objectc C sL t �ddddg�}|jjtjks$t�t �dddd|g�}| �||||� d S )Nr7 rK rL rM r ) rC rD r& r rO r) r! rp rm )r1 rF rG rH rq r r r �test_setitem_index_int641 s z,TestSetitemCoercion.test_setitem_index_int64gffffff@c C s� t jddddgddddgd �}|jjtjks0t�|tkr�|�� }d }t j ||d �� d |d < W d Q R X t j j d d�}|j �|� t �dddd|g�}| �||||� d S )Nr7 rK rL rM g�������?g������ @g������@gffffff@)r& z/index 5 is out of bounds for axis 0 with size 4)rU rk z%TODO_GH12747 The result must be float)rN )rC rD r& r rO r* r! rn r8 rP ro rQ rR r rS rp rm ) r1 rF rG r rH r= r[ rQ rq r r r �test_setitem_index_float64; s z.TestSetitemCoercion.test_setitem_index_float64zTest not implemented)rN c C s t �d S )N)r0 )r1 r r r �test_setitem_series_periodM s z.TestSetitemCoercion.test_setitem_series_periodc C s t �d S )N)r0 )r1 r r r � test_setitem_index_complex128Q s z1TestSetitemCoercion.test_setitem_index_complex128c C s t �d S )N)r0 )r1 r r r �test_setitem_index_boolU s z+TestSetitemCoercion.test_setitem_index_boolc C s t �d S )N)r0 )r1 r r r � test_setitem_index_datetime64Y s z1TestSetitemCoercion.test_setitem_index_datetime64c C s t �d S )N)r0 )r1 r r r � test_setitem_index_datetime64tz] s z3TestSetitemCoercion.test_setitem_index_datetime64tzc C s t �d S )N)r0 )r1 r r r � test_setitem_index_timedelta64a s z2TestSetitemCoercion.test_setitem_index_timedelta64c C s t �d S )N)r0 )r1 r r r �test_setitem_index_periode s z-TestSetitemCoercion.test_setitem_index_periodN)+r" r2 r3 r r> rP rQ � parametrizer( rJ rO r) r* r+ rT Zint32rV rW r\ r] r^ r` ra rC rb rc re rf rg rj rm rn rr rs rt rR ru rv rw rx ry rz r{ r r r r r5 I sj ( ,,, & & "& $r5 c @ s� e Zd ZdgZdZdd� Zej�ddde fdde fdde fd d e fg�d d � �Z ej�ddde j fdde j fdde fd d e fg�d d � �Zej�ddde j fdde j fdde fd d e fg�dd� �Zejjde�d�dfejddd�dfgddgd�ej�de�d�ejddd�dg�d d � ��Zd d � Zej�dejd d!d"�d d#fe�d�e�d�e fdde fd d e fg�d$d%� �Zejjd&d'�d(d)� �Zejjd&d'�d*d+� �Zd,S )-�TestInsertIndexCoercionr& �insertc C s2 |� � }|�d|�}t�||� |j|ks.t�dS )z!test coercion triggered by insertr7 N)r8 r~ r9 �assert_index_equalr r! )r1 �original�valueri r< �target�resr r r �_assert_insert_conversiono s z1TestInsertIndexCoercion._assert_insert_conversionz"insert, coerced_val, coerced_dtyper7 g�������?Fr c C sD t �td��}|jtks t�t �d|dddg�}| �||||� d S )Nr? r@ �brA rB )rC rp rE r r( r! r� )r1 r~ � coerced_val� coerced_dtyperH rI r r r �test_insert_index_objectv s z0TestInsertIndexCoercion.test_insert_index_objectc C sJ t �ddddg�}|jtjks"t�t �d|dddg�}| �||||� d S )Nr7 rK rL rM )rC Z Int64Indexr rO r) r! rp r� )r1 r~ r� r� rH rI r r r �test_insert_index_int64� s z/TestInsertIndexCoercion.test_insert_index_int64g �?c C sJ t �ddddg�}|jtjks"t�t �d|dddg�}| �||||� d S )Ng �?g @g @g @)rC Z Float64Indexr rO r* r! rp r� )r1 r~ r� r� rH rI r r r �test_insert_index_float64� s z1TestInsertIndexCoercion.test_insert_index_float64zfill_val,exp_dtypez 2012-01-01zdatetime64[ns]z US/Eastern)rd zdatetime64[ns, US/Eastern]r- r. )�ids� insert_valuez Asia/Tokyoc C s� t jddddg|jd�}|j|ks&t�t jd|�� dddg|jd�}| �||||� |jr�t �d�}|�d|�}|� t ��d|�} | jt ks�t�t � || � t jddd�}|�d|�}|� t ��d|�} | jt ks�t�t � || � nHt jddd�}|�d|�}|� t ��d|�} | jt k�s t�t � || � d} |�d| �}|� t ��d| �} | d | k�s`t�| jt k�spt�t � || � d S ) Nz 2011-01-01z 2011-01-02z 2011-01-03z 2011-01-04)rd z 2012-01-01r7 z Asia/Tokyo) rC � DatetimeIndexrd r r! �dater� rb r~ �astyper( r9 r ) r1 r �fill_valrG r� rH rI �ts�resultri �itemr r r �test_insert_index_datetimes� s: z3TestInsertIndexCoercion.test_insert_index_datetimesc C s� t �ddddg�}|jdks t�t �dddddg�}| �|t �d�|d� xNt �d�dgD ]<}|�d|�}|�t ��d|�}|jt ks�t�t � ||� qZW d S ) Nz1 dayz2 dayz3 dayz4 dayztimedelta64[ns]z10 dayz 2012-01-01r7 ) rC �TimedeltaIndexr r! r� rf rb r~ r� r( r9 r )r1 rH rI r� r� ri r r r � test_insert_index_timedelta64� s z5TestInsertIndexCoercion.test_insert_index_timedelta64z2012-01�M)�freqz period[M]c C sJ t jddddgdd�}|jdks$t�t jddd�|t jddd�t jddd�t jddd�g}t|t j�r�t j|dd�}| �||||� | �|t|�||� n�|�d|�}|� t ��d|�}t � ||� t|t j ��s|�dt|��}|� t ��dt|��}t � ||� d } tjt| d ��* t �t�� t j|dd� W d Q R X W d Q R X d S ) Nz2011-01z2011-02z2011-03z2011-04r� )r� z period[M]r z%Unexpected keyword arguments {'freq'})rU )rC � PeriodIndexr r! �Period� isinstancer� �strr~ r� r( r9 r rb rP ro � TypeErrorrY rX rp ) r1 r~ r� r� rH �datarI r� ri r[ r r r �test_insert_index_period� s, z0TestInsertIndexCoercion.test_insert_index_periodzTest not implemented)rN c C s t �d S )N)r0 )r1 r r r � test_insert_index_complex128 s z4TestInsertIndexCoercion.test_insert_index_complex128c C s t �d S )N)r0 )r1 r r r �test_insert_index_bool s z.TestInsertIndexCoercion.test_insert_index_boolN)r" r2 r3 r r r� rP rQ r| r( r� rO r) r* r� r� rC rb r� r� r� r� rR r� r� r r r r r} j sL &.(r} c @ s* e Zd ZdZdd� Zej�ddefdefdefdefg�d d � �Z ej�dde j fde j fde j fdefg�d d � �Zej�d de j fde j fde j fdefg�dd� �Zej�dde j fde j fde j fdefg�dd� �Zej�ddefdefdefde jfg�dd� �Zejjde�d�dfejddd�efgddgd�dd � �Zej�d e�d�e�d��� e�d��� g�d d � �Zejjd d!�d"d#� �Zejjd$d!�d%d&� �Zejjd$d!�d'd(� �Z ejjd$d!�d)d*� �Z ejjd$d!�d+d,� �Z ej�d-ej d.d/�e d.d/�e �!d.d0�g�d1d2� �Z"d3d4� Z#d5S )6�TestWhereCoercion�wherec C s2 |� � }|�||�}t�||� |j|ks.t�dS )z test coercion triggered by whereN)r8 r� r9 � assert_equalr r! )r1 r� �condrh ri r< r� r� r r r �_assert_where_conversion) s z*TestWhereCoercion._assert_where_conversionzfill_val,exp_dtyper7 g�������?y �? �?Tc s� |}|t d��}|jtks t�|ddddg�}� dkrF|tjkrFd}n� }|d|d|g�}| �||� ||� � dkr�|ddddg�} n|� fdd�d D ��} |d| d d| d g�}| �||| ||� d S ) Nr? TFr7 r@ rA c 3 s | ]}|� V qd S )Nr )r r )r� r r r G s z6TestWhereCoercion.test_where_object.<locals>.<genexpr>)rk � � � rL )rE r r( r! rC rD r� ) r1 �index_or_seriesr� rG r rH r� Zret_valrI rh r )r� r �test_where_object2 s z#TestWhereCoercion.test_where_objectc s� |}|t jkr"|tjkr"t�d� |ddddg�}|jtjksBt�|ddddg�}|d� d� g�}| � ||� ||� � dkr�|ddddg�}n|� fdd �d D ��}|d|d d|d g�}| � |||||� d S ) NzComplex Index not supportedr7 rK rL rM TFc 3 s | ]}|� V qd S )Nr )r r )r� r r r ^ s z5TestWhereCoercion.test_where_int64.<locals>.<genexpr>)rk r� r� r� ) rC rp rO r+ rP �skipr r) r! r� ) r1 r� r� rG r rH r� rI rh r )r� r �test_where_int64L s z"TestWhereCoercion.test_where_int64zfill_val, exp_dtypec s� |}|t jkr"|tjkr"t�d� |ddddg�}|jtjksBt�|ddddg�}|d� d� g�}| � ||� ||� � dkr�|ddddg�}n|� fdd �d D ��}|d|d d|d g�}| � |||||� d S ) NzComplex Index not supportedg�������?g������@gffffff @g������@TFc 3 s | ]}|� V qd S )Nr )r r )r� r r r t s z7TestWhereCoercion.test_where_float64.<locals>.<genexpr>)rk r� r� r� r7 rL ) rC rp rO r+ rP r� r r* r! r� ) r1 r� r� rG r rH r� rI rh r )r� r �test_where_float64b s z$TestWhereCoercion.test_where_float64c s� t j}|ddddg�}|jtjks&t�|ddddg�}|d� d� g�}| �||� ||� � dkrr|ddddg�}n|� fdd�d D ��}|d|d d|d g|d �}| �|||||� d S ) Ny �? �?y @ @y @ @y @ @TFc 3 s | ]}|� V qd S )Nr )r r )r� r r r � s zATestWhereCoercion.test_where_series_complex128.<locals>.<genexpr>)rk r� r� r� r7 rL )r )rC rD r rO r+ r! r� )r1 r� rG r rH r� rI rh r )r� r � test_where_series_complex128x s z.TestWhereCoercion.test_where_series_complex128c s� t j}|ddddg�}|jtjks&t�|ddddg�}|d� d� g�}| �||� ||� � dkrr|ddddg�}n|� fdd�dD ��}|d|d d|d g�}| �|||||� d S )NTFc 3 s | ]}|� V qd S )Nr )r r )r� r r r � s z;TestWhereCoercion.test_where_series_bool.<locals>.<genexpr>)rk r� r� r� r7 rL )rC rD r rO r` r! r� )r1 r� rG r rH r� rI rh r )r� r �test_where_series_bool� s z(TestWhereCoercion.test_where_series_boolz 2012-01-01zdatetime64[ns]z US/Eastern)rd r- r. )r� c C s t �t �d�t �d�t �d�t �d�g�}|jdks8t�t �ddddg�}t �t �d�|t �d�|g�}| �|||||� t �t j|dd ��}|jr�t �t �d�t jd d d �t �d�t jd d d �g�}| �|||||� t �t �d�|d t �d�|d g�}| �|||||� d S )Nz 2011-01-01z 2011-01-02z 2011-01-03z 2011-01-04zdatetime64[ns]TFrM )�periodsz2012-01-02 00:00z US/Eastern)rd z2012-01-04 00:00r7 rL )rC rD rb r r! r� � date_rangerd )r1 r� rG rH r� rI rh r r r � test_where_series_datetime64� s0 z.TestWhereCoercion.test_where_series_datetime64r� c C s� d}t �t �d�t �d�t �d�t �d�g�}|jdks<t�t �ddddg�}|�||�}t �|d ||d |g�}t�||� t �t j |d d ��}t �t �d�t �d �t �d�t �d �g�}| � |||||� d S )Nzdatetime64[ns]z 2011-01-01z 2011-01-02z 2011-01-03z 2011-01-04TFr rK rM )r� z 2012-01-02z 2012-01-04) rC rp rb r r! r� r� r9 r r� r� ) r1 r� rG rH r� r� ri rh rI r r r �test_where_index_datetime� s$ z+TestWhereCoercion.test_where_index_datetimez0GH 22839: do not ignore timezone, must be object)rN c C s� t jddd�}t}t �t �d�t �d�t �d�t �d�g�}|jdksJt�t �d d d d g�}d }tjt|d �� |� ||� W d Q R X t �t j |d d��}t jt �d�t jddd�t �d�t jddd�g|d�}| � |||||� d S )Nz 2012-01-01z US/Eastern)rd z 2011-01-01z 2011-01-02z 2011-01-03z 2011-01-04zdatetime64[ns]TFz=Index\(\.\.\.\) must be called with a collection of some kind)rU rM )r� z 2012-01-02z 2012-01-04)r ) rC rb r( rp r r! rP ro r� r� r� r� )r1 r� rG rH r� r[ rh rI r r r � test_where_index_datetime64tz� s( z/TestWhereCoercion.test_where_index_datetime64tzzTest not implementedc C s t �d S )N)r0 )r1 r r r �test_where_index_complex128 s z-TestWhereCoercion.test_where_index_complex128c C s t �d S )N)r0 )r1 r r r �test_where_index_bool s z'TestWhereCoercion.test_where_index_boolc C s t �d S )N)r0 )r1 r r r � test_where_series_timedelta64" s z/TestWhereCoercion.test_where_series_timedelta64c C s t �d S )N)r0 )r1 r r r �test_where_series_period& s z*TestWhereCoercion.test_where_series_periodr� � )�days�Dc C s� t jddd�}t�ddddg�}t �d||dg�}|�||�}t�||� t�dd�}t j |d |||d gt d �}|d |ks�t �|�||�}t�||� d S ) Nz1 DayrM )r� TFz4 Days�NaT�nsr rL )r r7 ) rC Ztimedelta_rangerO �arrayr� r� r9 r r- rp r( r! )r1 r� Ztdir� ri r� Zdtnatr r r � test_where_index_timedelta64* s z.TestWhereCoercion.test_where_index_timedelta64c C s0 t jdddd�}|�d�}t�dddg�}|d |jd }t �||d |g�}|�||�}t� ||� tj ||jd t d �}|�||�}t �|d |d |d g�}t� ||� t j dd�}t j ||d |gt d �}|�||�}t� ||� t �dd�} t j | |d | gt d �}|�|| �}t� ||� d S )Nz 2016-01-01rL ZQS)r� r� �QFT���� r7 )r r rK rM )r� z 2020-04-21r� )rC r� Z to_periodrO r� r� r� r� r9 r Zasarrayr( rf rp r� ) r1 Zdti�pir� r� ri r� �other�tdZperr r r �test_where_index_period= s& z)TestWhereCoercion.test_where_index_periodN)$r" r2 r3 r r� rP rQ r| r( r� rO r) r* r+ r� r� r� r` r� rC rb r� Z to_datetime64Z to_pydatetimer� rR r� r� r� r� r� rf r r/ r� r� r r r r r� % sL &,,( ) ,r� c @ s e Zd ZdZejjdd�dd� �Zdd� Zej� dd e fd e fd e fd e fg�d d� �Z ej� dd e j fd e j fd e jfd e fg�dd� �Zej� dd e jfd e jfd e jfd e fg�dd� �Zejj de�d�dfejddd�e fd e fde fgddddgd �d d � �Zej� dejddd�d fe�d�e fejdd d�e fd e fde fg�d!d"� �Zejjdd�d#d$� �Zejjdd�d%d&� �Zejjdd�d'd(� �Zejjdd�d)d*� �Zejjdd�d+d,� �Zejjdd�d-d.� �Zejjdd�d/d0� �Zejjdd�d1d2� �Z d3S )4�TestFillnaSeriesCoercion�fillnazTest not implemented)rN c C s t �d S )N)r0 )r1 r r r � test_has_comprehensive_testsa s z5TestFillnaSeriesCoercion.test_has_comprehensive_testsc C s0 |� � }|�|�}t�||� |j|ks,t�dS )z!test coercion triggered by fillnaN)r8 r� r9 r� r r! )r1 r� r� ri r< r� r� r r r �_assert_fillna_conversione s z2TestFillnaSeriesCoercion._assert_fillna_conversionzfill_val, fill_dtyper7 g�������?y �? �?Tc C sH |}|dt jddg�}|jtks$t�|d|ddg�}| �||||� d S )Nr@ rA rB )rO �nanr r( r! r� )r1 r� r� � fill_dtyper rH rI r r r �test_fillna_objectl s z+TestFillnaSeriesCoercion.test_fillna_objectzfill_val,fill_dtypec C sb |}|dt jddg�}|jt jks&t�|d|ddg�}|t jkrN|tjkrNt}| � ||||� d S )Ng�������?gffffff @g������@) rO r� r r* r! r+ rC rp r( r� )r1 r� r� r� r rH rI r r r �test_fillna_float64x s z,TestFillnaSeriesCoercion.test_fillna_float64c C sJ t �dtjddg�}|jtjks$t�t �d|ddg�}| �||||� d S )Ny �? �?y @ @y @ @)rC rD rO r� r r+ r! r� )r1 r� r� rH rI r r r � test_fillna_series_complex128� s z6TestFillnaSeriesCoercion.test_fillna_series_complex128z 2012-01-01zdatetime64[ns]z US/Eastern)rd r r- r. r( )r� c C sl |}|t �d�t jt �d�t �d�g�}|jdks6t�|t �d�|t �d�t �d�g�}| �||||� d S )Nz 2011-01-01z 2011-01-03z 2011-01-04zdatetime64[ns])rC rb r� r r! r� )r1 r� r� r� r rH rI r r r �test_fillna_datetime� s z-TestFillnaSeriesCoercion.test_fillna_datetimezdatetime64[ns, US/Eastern]z Asia/Tokyoc C s� |}d}|t jd|d�t jt jd|d�t jd|d�g�}|jdksFt�|t jd|d�|t jd|d�t jd|d�g�}| �||||� d S )Nz US/Easternz 2011-01-01)rd z 2011-01-03z 2011-01-04zdatetime64[ns, US/Eastern])rC rb r� r r! r� )r1 r� r� r� r rd rH rI r r r �test_fillna_datetime64tz� s z1TestFillnaSeriesCoercion.test_fillna_datetime64tzc C s t �d S )N)r0 )r1 r r r �test_fillna_series_int64� s z1TestFillnaSeriesCoercion.test_fillna_series_int64c C s t �d S )N)r0 )r1 r r r �test_fillna_index_int64� s z0TestFillnaSeriesCoercion.test_fillna_index_int64c C s t �d S )N)r0 )r1 r r r �test_fillna_series_bool� s z0TestFillnaSeriesCoercion.test_fillna_series_boolc C s t �d S )N)r0 )r1 r r r �test_fillna_index_bool� s z/TestFillnaSeriesCoercion.test_fillna_index_boolc C s t �d S )N)r0 )r1 r r r � test_fillna_series_timedelta64� s z7TestFillnaSeriesCoercion.test_fillna_series_timedelta64c C s t �d S )N)r0 )r1 r r r �test_fillna_series_period� s z2TestFillnaSeriesCoercion.test_fillna_series_periodc C s t �d S )N)r0 )r1 r r r � test_fillna_index_timedelta64� s z6TestFillnaSeriesCoercion.test_fillna_index_timedelta64c C s t �d S )N)r0 )r1 r r r �test_fillna_index_period� s z1TestFillnaSeriesCoercion.test_fillna_index_periodN) r" r2 r3 r rP rQ rR r� r� r| r( r� rO r* r+ r� r� rC rb r� r� r� r� r� r� r� r� r� r� r r r r r� [ sJ & , r� c @ s� e Zd ZU dgZdZi Zded< ddged< dd ged < d d ged < ddged< ddged< e�d�e�d�ged< x8dD ]0Z de � d�Z ejde d�ejde d�gee < qzW e� d�e� d �ged < e j d dgd �d d!� �Ze j dd d dddd"d#d g d �d$d%� �Ze j dd d dddd"d#d g dd d ddd&d'd'd(g d)�d*d+� �Ze j d,d-� �Zd.d/� Ze jjd0d dddd d gdd1�e jjd2d"d#gdd1�d3d4� ��Ze jjd0dd"d#gdd1�e jjd2dd"d#gdd1�d5d6� ��Ze jjd7d8�d9d:� �Zd;S )<�TestReplaceSeriesCoercionr' �replacezdict[str, list]�repr@ r� r( rM rk r) g�������?g������@r* y �? �?y @ @r+ TFr, z 2011-01-01z 2011-01-03zdatetime64[ns])�UTCz US/Easternzdatetime64[ns, �])rd z1 dayz2 dayztimedelta64[ns]�dict)�paramsc C s |j S )N)�param)r1 r r r r �how s z TestReplaceSeriesCoercion.howzdatetime64[ns, UTC]zdatetime64[ns, US/Eastern]c C s |j S )N)r� )r1 r r r r �from_key s z"TestReplaceSeriesCoercion.from_keyr- r. r/ )r� r� c C s |j S )N)r� )r1 r r r r �to_key' s z TestReplaceSeriesCoercion.to_keyc C sP |dkr$t t| j| | j| ��}n(|dkrHtj| j| | j| d�}nt�|S )z9 Object we will pass to `Series.replace` r� r' )r& )r� �zipr� rC rD � ValueError)r1 r� r� r� �replacerr r r r� B s z"TestReplaceSeriesCoercion.replacerc C s� t jddgdd�}t j| j| |dd�}|j|ks6t�|�d�rN|�d�rNd S |dkrZd S |�|�}|d krt|d ks�|d kr�|d kr�tr�t � r�t � d |� d|� �� t j| j| |d|d�}n$t j| j| |dd�}|j|ks�t�t � ||� d S )NrL rM Zxxx)r �yyy)r& r �datetime)zdatetime64[ns, US/Eastern]zdatetime64[ns, UTC]r* r) r+ )r) r* z32-bit platform buggy: z -> )r& r r )rC rp rD r� r r! � startswithr� r r rP r� r9 r: ) r1 r� r� r� r� r& rH r� rI r r r �test_replace_seriesO s z-TestReplaceSeriesCoercion.test_replace_seriesr� )Zindirectr� c C st t jddgdd�}t j| j| |dd�}|j|ks6t�|�|�}t j| j| |dd�}|j|ksdt�t�||� d S )NrL rM �xyz)r r� )r& r ) rC rp rD r� r r! r� r9 r: ) r1 r� r� r� r� r& rH r� rI r r r � test_replace_series_datetime_tzm s z9TestReplaceSeriesCoercion.test_replace_series_datetime_tzc C st t jddgdd�}t j| j| |dd�}|j|ks6t�|�|�}t j| j| |dd�}|j|ksdt�t�||� d S )NrL rM r� )r r� )r& r ) rC rp rD r� r r! r� r9 r: ) r1 r� r� r� r� r& rH r� rI r r r �%test_replace_series_datetime_datetime� s z?TestReplaceSeriesCoercion.test_replace_series_datetime_datetimezTest not implemented)rN c C s t �d S )N)r0 )r1 r r r �test_replace_series_period� s z4TestReplaceSeriesCoercion.test_replace_series_periodN)r" r2 r3 r r r� �__annotations__rC rb rd �keyrf rP �fixturer� r� r� r� r� rQ r| r� r� rR r� r r r r r� � sz r� )� __future__r r� r r �numpyrO rP Z pandas.compatr r ZpandasrC Zpandas._testingZ_testingr9 r� r$ r% r5 r} r� r� r� r r r r �<module> s( # < 8 "
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка