Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/__pycache__/test_combine_first.cpython-37.pyc
Назад
B �5�gEC � @ s� d dl m Z d dlZd dlZd dlmZmZ d dlZd dlm Z m Z m Z m Z d dl mZ G dd� d�Zej�de dd d �e dd d �fe�d d �e�d d �fe�d�e�d�fejd d d�ejd ddd�fg�dd� �Zdd� Zdd� Zdd� ZdS ) � )�datetimeN)�find_common_type�is_dtype_equal)� DataFrame�Index� MultiIndex�Seriesc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Ze j � d e ddd�e ddd�e ddd�ge je je jge ddd�e ddd�e ddd�gfe je je jge ddd�e ddd�e ddd�ge ddd�e ddd�e ddd�gfe ddd�e je jge ddd�e ddd�e ddd�ge ddd�e ddd�e ddd�gfe ddd�e ddd�e ddd�ge ddd�e je jge ddd�e ddd�e ddd�gff�dd� �Zdd� Zdd� Zdd� Zdd� Zd d � Ze j � d dd g�d d!� �Zd"d#� Zd$S )%�TestDataFrameCombineFirstc C s� t ddgtd�d�}t td�td�d�}t||d��}t ddgtdd�d�}t td�tdd�d�}t||d��}ttd�d d d d gd�d d dd gd�}|�|�}t�||� d S ) N�a�b� )�index)�A�B� � Zababr � � )r �ranger �list� combine_first�tm�assert_frame_equal)�selfr r �f�g�exp�combined� r �c/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/test_combine_first.py�test_combine_first_mixed s & z2TestDataFrameCombineFirst.test_combine_first_mixedc C sZ |d d� |dd � }}|� |�}|�|j�}t�||� t�|j|j�sPt�t�|d |d � |� � }d|d< |d= |� � }d|d< |d= |� |�}|d dk� � s�t�t�|d |d � t�|d |d � t�|d |d � |d d� � � | }}d|d< |� |�}|d d d� dk� � �s0t�d|d d d�< |� |�}|d d d� dk� � �sht�|d d� }|dd � } |� | �}t�|d �|j�|d � t�|d �| j�| d � |� t � �} t�| |� t � � |�} t�| |� |� t d d gd ��} d | jk�st�t d dgit d dd�gd �} t dgd�} | � | �} d| k�sVt�d S )Nr r r �Cr r �D� Zfaz�boo)r r i� r )�columns) r Zreindexr r r Z equalContentsr% �AssertionError�assert_series_equal�copy�allr r )r Z float_frame�head�tailr Zreordered_frameZfcopyZfcopy2r r Zcomb�df�df2�resultr r r �test_combine_first# sN z,TestDataFrameCombineFirst.test_combine_firstc C s� t ddddg�}tddddg|d �}tddddg|d �}td d d d g|d �}t|||d��}t ddddg�}tddddg|d �}tddddg|d �}td d d d g|d �}t|||d��}|�|�}t|j�d ks�t�d S )Nr r �c�eg @g "�g @g Y@)r � � r �a )Zcol0�col2Zcol3r )Zcol1r5 Zcol5)r r r r �lenr% r& )r �idxZser1Zser2Zser3Zframe1Zframe2r r r r � test_combine_first_mixed_bug` s z6TestDataFrameCombineFirst.test_combine_first_mixed_bugc C s� t ddddgddddggddd d gd �}t d d ggd gddgd�}|�|�}t�||� tj|jd< |�|�}d |jd< t�||� d S )Ng �?g @FTg @g @r r Zbool1Zbool2)r% �- r )r r% )r r )r r r r �np�nan�loc)r r, �otherr. r r r �$test_combine_first_same_as_in_updater s z>TestDataFrameCombineFirst.test_combine_first_same_as_in_updatec C s� t dtjddtjgtjddtjdgd��}t dddtjddgtjtjdddd gd��}|�|�}t d d d d d dgtjd d dddgd��}t�||� d S )Ng �?g @g @g @g @)r r g @g @g @r r � r r3 r � )r r: r; r r r )r �df1r- r. �expectedr r r � test_combine_first_doc_example� s ( (z8TestDataFrameCombineFirst.test_combine_first_doc_examplec C s� t tjddgdtjdgtjddgg�}t dtjdgdddggd d gd �}tdddgd td �}|�|�d }t�||� |�|�d }t�||� d S ) Ng @Tgffffff�g @Fg�����LE�g �g�������?r r )r )�name�dtype)r r: r; r �boolr r r' )r rA r- rB � result_12� result_21r r r �-test_combine_first_return_obj_type_with_bools� s $" zGTestDataFrameCombineFirst.test_combine_first_return_obj_type_with_boolszdata1, data2, data_expectedi� r r r? c C s@ t d|i�t d|i� }}|�|�}t d|i�}t�||� d S )Nr )r r r r )r �data1�data2Z data_expectedrA r- r. rB r r r �-test_combine_first_convert_datatime_correctly� s zGTestDataFrameCombineFirst.test_combine_first_convert_datatime_correctlyc C s$ t t�d�dggddgd�}t dgdggdgd�}|d jdksDt�|d jd ksVt�|�|�}t t�d�tjgddgd �ddgd�}t�||� |d jdks�t�|d jd ks�t�|j d d � �|�}t t j t j gddgd �ddgd�}t�||� |d jd k�s t�|d jd k�s t�d S ) Nz 2011-01-01r r r )r% r3 r zdatetime64[ns]�int64)r r r Zfloat64) r �pd� TimestamprE r& r �NaTr r Zilocr: r; )r ZdfaZdfb�resr r r r � test_combine_first_align_nan� s " z6TestDataFrameCombineFirst.test_combine_first_align_nanc C s� t �d��d�}tddg|t jddd�d�}t �d ��d�}tdd g|t jd dd�d�}|dg �|�}tt jd dd �t jddd �gt jddd �t jgd�ddgt jdddd�d�}|d jdks�t �|d jdks�t �t � ||� t jdddd �}td|i�}t jdddd �}td|i�}|�|�}t � ||� |d jdk�sFt �t j ddd d gd d �}td|idd d d!gd"�}t j d#d$d%gd d �}td|idd&d gd"�}|�|�}t j dd#dd$d d gd d �} td| iddd d&d d!gd"�}t � ||� t jddd d �}td|i�}t �dd�}td|i�}|�|�}t � ||� |d jd'k�sHt �t jdd(d d �}td|i�}t �dd�}td|i�}|�|�}t jdd d �t jd(d d �t �d�g} td| i�}t � ||� |d jd)k�s�t �d S )*Nz20100101 01:01�UTC� UTCdatetime�abcZ20140627r )�periods)r% �datar z20121212 12:12ZxyzZ20140628z2010-01-01 01:01)�tzz2012-12-12 12:12z2010-01-01 01:01:00)rT rU r r" )rV �freq)r% r zdatetime64[ns, UTC]z 2015-01-01z 2015-01-05ZDATEz 2015-01-03z 2011-01-01rP z 2011-01-03z 2011-01-04z US/Easternr? r r )r z 2012-01-01z 2012-01-02z 2012-01-03r3 zdatetime64[ns, US/Eastern]z 2015-01-02�object) rN � to_datetimeZ tz_localizer Z date_ranger rO rP rE r& r r Z DatetimeIndex) r rJ rA rK r- rQ r Zdts1�dts2�exp_dtsr r r �test_combine_first_timezone� s| z5TestDataFrameCombineFirst.test_combine_first_timezonec C s� t �ddddg�}td|idddd gd �}t �d d d g�}td|idddgd �}|�|�}t �dd dd ddg�}td|idddddd gd �}t�||� |d jdks�t�d S )Nz1 dayrP z3 dayZ4dayZTDr r? r r )r z10 dayz11 dayz12 dayr r3 z4 dayztimedelta64[ns])rN ZTimedeltaIndexr r r r rE r& )r rJ rA rK r- rQ r] r r r r � test_combine_first_timedelta: s z6TestDataFrameCombineFirst.test_combine_first_timedeltac C sp t jddddgdd�}td|idd d d gd �}t jd ddgdd�}td|iddd gd �}|�|�}t jddddddgdd�}td|iddd dd d gd �}t�||� |d j|jks�t�t jd ddgdd�}td|iddd gd �}|�|�}t jddd�t jd dd�t j t jddd�t jddd�t jddd�g}td|iddd dd d gd �}t�||� |d jdk�slt�d S )Nz2011-01rP z2011-03z2011-04�M)rY �Pr r? r r )r z 2012-01-01z2012-02z2012-03r r3 z2012-01z 2012-01-02z 2012-01-03r" rZ ) rN Z PeriodIndexr r r r rE r& �PeriodrP ) r rJ rA rK r- rQ r] r r\ r r r �test_combine_first_periodH s, z3TestDataFrameCombineFirst.test_combine_first_periodc C s� t dddddgidd�}t dddgidd�}|�|�}t dddddgi�}t�||� |�|�}t dddddgi�}t�||� d S ) Nr r r r? r rM )rE r3 )r r r r )r rA r- rG Z expected_12rH Z expected_21r r r �test_combine_first_intg s z0TestDataFrameCombineFirst.test_combine_first_int�valg �?c C sH t d|gi�}t ddgi�}|�|�}t dg|gd��}t�||� d S )N�isNum�isBoolT)rg rf )r r r r )r re rA r- rQ r r r r �(test_combine_first_with_asymmetric_otheru s zBTestDataFrameCombineFirst.test_combine_first_with_asymmetric_otherc C s� t ddgtjgd d�|d�}t dgtjgd�|d�}|jddgdd � |jddgdd � |�|�}t ddgtjgd d�|d��ddg�}t�||� d S ) NZ962Z85r )r r )rE r r T)Zinplace)r rN ZNAZ set_indexr r r )r Znullable_string_dtyper, r- r. rB r r r �'test_combine_first_string_dtype_only_na� s zATestDataFrameCombineFirst.test_combine_first_string_dtype_only_naN)�__name__� __module__� __qualname__r r/ r8 r> rC rI �pytest�mark� parametrizer rN rP rL rR r^ r_ rc rd rh ri r r r r r s6 = " " " 0[ r zscalar1, scalar2i� r r z 2020-01-01r" z 2020-01-02z89 daysz60 min)�left�rightr? rp )rp rq �closedc C s� |}t ||ggddgd�}t | |ggddgd�}t|jd |jd g�}t|d�sf|jd |jd krl| }n|}|�|�}t |||ggdddgd�} | d �|�| d< t�|| � d S )Nr r )r% r0 rZ )r r Zdtypesr r Zastyper r ) Zscalar1Zscalar2Z nulls_fixtureZna_value�framer= Z common_dtypere r. rB r r r � test_combine_first_timestamp_bug� s rt c C s� t tjtjggddgd�} t tddd�tddd�ggddgd�}| �|�}t tjtddd�tddd�ggdddgd�}t�||� d S )Nr r )r% i� r r r0 )r rN rP r r r r )rs r= r. rB r r r �$test_combine_first_timestamp_bug_NaT� s $ *ru c C sH t jdddddtjgdddddd ggddgd �} tdddddddgi| d �}t jdddddd gddddddggddgd �}tdddddd g|d �}|�td |i��}t jdddddddddd tjg ddddddddddd g gddgd �}ttjtjdddddtjdtjdg d dtjddtjtjdtjdtjg d�|d �}t�||� d S )Nr r0 r r r r? r3 r r )�names)r �dg �?g @g @g @g @g @)r0 rw ) r Z from_arraysr: r; r r r r r )Zmi1r, Zmi2�srQ Z mi_expectedrB r r r �&test_combine_first_with_nan_multiindex� s , * &ry c C s� t ddgtd�d�} t td�td�d�}t| |d��}t ddgtdd�d�}t tdd �tdd�d�}t||d ��}tddtjtjgd d dd gtjtjddgd �d d dd gd�}|�|�}t�||� d S )Nr r r )r )r r r r ���r )r r! r )r r r! r )r r r r: r; r r r )Za_columnZb_columnrA Zc_columnr- rB r r r r � test_combine_preserve_dtypes� s r{ )r �numpyr: rm Zpandas.core.dtypes.castr r ZpandasrN r r r r Zpandas._testingZ_testingr r rn ro rb Z TimedeltaZIntervalrt ru ry r{ r r r r �<module> s$ |*
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка