Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/reshape/concat/__pycache__/test_append_common.cpython-37.pyc
Назад
B �5�g)n � @ sN d dl Zd dlZd dlZd dlmZmZmZmZ d dl m Z G dd� d�Z dS )� N)� Categorical� DataFrame�Index�Seriesc @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z e j � dddddg�dd� �Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1S )2�TestConcatAppendCommonzE Test common dtype coercion rules between concat and append. c C s� t �d�t �d�t �d�g}t jddd�t jddd�t jddd�g}t �d�t �d�t �d�g}t jd d d �t jd d d �t jd d d �g}dddgdddgdtjdgtdddg�dddg||||d� | _d S ) Nz 2011-01-01z 2011-01-02z 2011-01-03z US/Eastern)�tzz1 daysz2 daysz3 daysz2011-01�M)�freqz2011-02z2011-03TF� � � g�������?gffffff @�X�Y�Z�a�b�c) �bool�int64�float64�category�objectzdatetime64[ns]zdatetime64[ns, US/Eastern]ztimedelta64[ns]z period[M])�pd� Timestamp� Timedelta�Period�np�nanr �data)�self�methodZdt_dataZtz_dataZtd_dataZ period_data� r! �d/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/reshape/concat/test_append_common.py� setup_method s* z#TestConcatAppendCommon.setup_methodc C sn t |t�r2|dkr"|jdks0t�qj|j|ksjt�n8t |t�rf|�d�rV|jdksdt�qj|j|ksjt�nt�dS )zr Check whether obj has expected dtype depending on label considering not-supported dtypes r r Zperiodz Period[M]N)� isinstancer �dtype�AssertionErrorr � startswith� ValueError)r �obj�labelr! r! r" �_check_expected_dtype8 s z,TestConcatAppendCommon._check_expected_dtypec C s< x6| j �� D ](\}}| �t|�|� | �t|�|� q W d S )N)r �itemsr+ r r )r �typ�valsr! r! r" � test_dtypesJ s z"TestConcatAppendCommon.test_dtypesc C s� �x�| j �� D �]�\}}|}|}|dkrZtt|�t|� �}tt|�t|� t|� �}n|| }|| | }t|��t|��}t|�}t�||� t|��t|�t|�g�}t|�}t�||� t|dd�} t|dd�} | �| �}t|�}t�||� t|dd�} t|dd�} | �| �}t|dd�}t�||� tj t dd�� t|��|� W d Q R X tj t dd�� t|��t|�|g� W d Q R X t |�jt |�dd�}t |�}tj ||dd � t jt |�t |�gdd�}tj ||dd � t |�jt |�t |�gdd�}t |�}t� ||� t jt |�t |�t |�gdd�}t� ||� t |dd�} t |dd�} | j| dd�}t |�}tj ||dd � t j| | gdd�}tj ||dd � t |dd�} t |dd�} | j| dd�}t |dd�}tj ||dd � t j| | gdd�}tj ||dd � d } tj t | d�� t |��|� W d Q R X tj t | d�� t |��t |�|g� W d Q R X tj t | d�� t �t |�|g� W d Q R X tj t | d�� t �t |�t |�|g� W d Q R X qW d S ) Nr �x)�name�yzall inputs must be Index)�matchT)� ignore_index)�check_index_typezPcannot concatenate object of type '.+'; only Series and DataFrame objs are valid)r r, r �listr �append�tm�assert_index_equal�pytestZraises� TypeErrorr �assert_series_equalr �concat)r �typ1�vals1�vals2�vals3�exp_data� exp_data3�res�exp�i1�i2�s1�s2�msgr! r! r" �test_concatlike_same_dtypesP s� z2TestConcatAppendCommon.test_concatlike_same_dtypesc C s� �x�| j �� D �]�\}}�x�| j �� D �]�\}}|}d }d }||krHq&n|dks&|dkrZq&|dkrp|dkrp|}n>|dkr�|dkr�|}n(|dks�|dks�|dks�|dkr�t}t}|| }|| | } t|��t|��} t||d�} t�| | � t|��t|�t|�g�} t| |d�} t�| | � t|�jt|�dd�} t||d�} tj| | dd � t j t|�t|�gdd�} tj| | dd � t|�jt|�t|�gdd�} t| |d�} t�| | � t j t|�t|�t|�gdd�} t�| | � q&W qW d S ) Nr r )r r zdatetime64[ns, US/Eastern]ztimedelta64[ns])r% T)r4 )r5 ) r r, r r r7 r8 r9 r r< r r= ) r r> r? Ztyp2r@ rA Zexp_index_dtypeZexp_series_dtyperB rC rD rE r! r! r" � test_concatlike_dtypes_coercion� sP z6TestConcatAppendCommon.test_concatlike_dtypes_coercionc C sB t �ddg�}t �ddg�}tt �d�t �d�t �d�t �d�g�}|�|�}t�||� t |d t j�snt �t |d t j�s�t �t |�}t |�}|�|�}t� |t |ddddgd�� t |j d t j�s�t �t |j d t j�s�t �t �||g�}t� |t |ddddgd�� t |j d t j��s&t �t |j d t j��s>t �d S ) Nz 2011-01-01z 2011-01-02z1 daysz2 daysr ���r )�index)r � DatetimeIndex�TimedeltaIndexr r r r7 r8 r9 r$ r& r r< Zilocr= )r Zdti�tdirE rD �dts�tdsr! r! r" �.test_concatlike_common_coerce_to_pandas_object s* zETestConcatAppendCommon.test_concatlike_common_coerce_to_pandas_objectc C s� |}t jddg|d�}t jddg|d�}t jddddg|d�}|�|�}t�||� t|�}t|�}|�|�}t�|t|ddddgd�� t �||g�}t�|t|ddddgd�� d S ) Nz 2011-01-01z 2011-01-02)r z 2012-01-01z 2012-01-02r r )rN )r rO r7 r8 r9 r r< r= ) r �tz_aware_fixturer �dti1�dti2rE rD �dts1�dts2r! r! r" �test_concatlike_datetimetz% s z1TestConcatAppendCommon.test_concatlike_datetimetzr �UTCz US/Easternz Asia/TokyoZEST5EDTc C s� t jddd|d�}t jddg|d�}td|d d gd �}td|d d gd �}t jdd dddg|d�}td|d d gd �}t�|�|�|� t�t �||g�|� d S ) Nz 2014-07-15z 2014-07-17�D)�start�endr r z 2014-07-11z 2014-07-21)r r �A�B)rN �columnsz 2014-07-16)r Z date_rangerO r r8 �assert_frame_equalr7 r= )r r Zix1Zix2Zdf1Zdf2Zexp_idxrE r! r! r" � test_concatlike_datetimetz_short: s z7TestConcatAppendCommon.test_concatlike_datetimetz_shortc C s� |}t jddg|d�}t �ddg�}tt jd|d�t jd|d�t �d�t �d�gtd�}|�|�}t�||� t|�}t|�}|�|�}t� |t|ddddgd �� t � ||g�}t� |t|ddddgd �� t jddgd d�} tt jd|d�t jd|d�t jdd d�t jdd d�gtd�}|�| �}t|�}t| �} |�| �}t� |t|ddddgd �� t � || g�}t� |t|ddddgd �� d S ) Nz 2011-01-01z 2011-01-02)r z 2012-01-01z 2012-01-02)r% r r )rN z US/Pacific) r rO r r r r7 r8 r9 r r< r= ) r rU r rV rW rE rD rX rY Zdti3Zdts3r! r! r" �$test_concatlike_datetimetz_to_objectK s> z;TestConcatAppendCommon.test_concatlike_datetimetz_to_objectc C s� t jddgdd�}t jddgdd�}t jddddgdd�}|�|�}t�||� t|�}t|�}|�|�}t�|t|ddddgd �� t �||g�}t�|t|ddddgd �� d S ) Nz2011-01z2011-02r )r z2012-01z2012-02r r )rN )r � PeriodIndexr7 r8 r9 r r< r= )r �pi1�pi2rE rD �ps1�ps2r! r! r" � test_concatlike_common_period� s z4TestConcatAppendCommon.test_concatlike_common_periodc C s� t jddgdd�}t jddgdd�}tt jddd�t jddd�t jddd�t jddd�gtd�}|�|�}t�||� t|�}t|�}|�|�}t� |t|d d d d gd �� t � ||g�}t� |t|d d d d gd �� d S ) Nz2011-01z2011-02r )r z 2012-01-01z 2012-02-01r\ )r% r r )rN ) r re r r r r7 r8 r9 r r< r= )r rf rg rE rD rh ri r! r! r" �1test_concatlike_common_period_diff_freq_to_object� s zHTestConcatAppendCommon.test_concatlike_common_period_diff_freq_to_objectc C sx t jddgdd�}t �ddg�}tt jddd�t jddd�t �d�t �d�gtd�}|�|�}t� ||� t |�}t |�}|�|�}t� |t |dd dd gd �� t � ||g�}t� |t |dd dd gd �� tt �d�t �d�t jddd�t jddd�gtd�}|�|�}t� ||� t |�}t |�}|�|�}t� |t |dd dd gd �� t � ||g�}t� |t |dd dd gd �� d S ) Nz2011-01z2011-02r )r z1 daysz2 days)r% r r )rN ) r re rP r r r r r7 r8 r9 r r< r= )r rf rQ rE rD rh rS r! r! r" �0test_concatlike_common_period_mixed_dt_to_object� s< zGTestConcatAppendCommon.test_concatlike_common_period_mixed_dt_to_objectc C sL t ddtjgdd�}t dddgdd�}t ddtjdddgdd�}t�tj||gdd�|� t�|j|dd�|� t ddgdd�}t ddgdd�}t ddddg�}t�tj||gdd�|� t�|j|dd�|� t dd tjgdd�}t tjdddgdd�}t dd tjtjdddgd d�}t�tj||gdd�|� t�|j|dd�|� d S ) Nr r r )r% T)r4 r � � r )r r r r8 r< r r= r7 )r rH rI rE r! r! r" �test_concat_categorical� s z.TestConcatAppendCommon.test_concat_categoricalc C s| t tdddgdddgd��}t tdddgdddgd��}tj||gdd�}t tddddddgdddgd��}t�||� d S )Nr r r )� categoriesT)r4 )r r r r= r8 r< )r r r �result�expectedr! r! r" �6test_union_categorical_same_categories_different_order� s zMTestConcatAppendCommon.test_union_categorical_same_categories_different_orderc C sn t ddtjgdd�}t dddg�}t ddtjdddgdd�}t�tj||gdd�|� t�|j|dd�|� t dddddtjgdd�}t�tj||gdd�|� t�|j|dd�|� t ddgdd�}t ddg�}t ddddg�}t�tj||gdd�|� t�|j|dd�|� t ddddg�}t�tj||gdd�|� t�|j|dd�|� t d d tjgdd�}t dddg�}t d d tjdddgdd�}t�tj||gdd�|� t�|j|dd�|� t dddd d tjgdd�}t�tj||gdd�|� t�|j|dd�|� t d d tjgdd�}t d d d g�}t d d tjd d d g�}t�tj||gdd�|� t�|j|dd�|� t d d d d d tjg�}t�tj||gdd�|� t�|j|dd�|� t d d gdd�}t tjtjtjg�}t d d tjtjtjg�}t�tj||gdd�|� t�|j|dd�|� t tjtjtjd d g�}t�tj||gdd�|� t�|j|dd�|� d S )Nr r r )r% r T)r4 r rm rn r r r )r r r r8 r< r r= r7 )r rH rI rE r! r! r" � test_concat_categorical_coercion sP z7TestConcatAppendCommon.test_concat_categorical_coercionc C s� t ddtjgdd�}t dddgdd�}t ddddtjg�}t ddtjdddddddtjg dd�}t�tj|||gdd�|� t�|j||gdd�|� t ddddtjddtjdddg dd�}t�tj|||gdd�|� t�|j||gdd�|� t dd d gdd�}t ddd gdd�}t dd dg�}t dd d ddd dd dg �}t�tj|||gdd�|� t�|j||gdd�|� t dd ddd d ddd g �}t�tj|||gdd�|� t�|j||gdd�|� t dd d gdd�}t ddd gdd�}t d d dg�}t dd d ddd d d dg �}t�tj|||gdd�|� t�|j||gdd�|� t d d ddd d ddd g �}t�tj|||gdd�|� t�|j||gdd�|� d S )Nr r r )r% �floatT)r4 � � � r rm rn � )r r r r8 r< r r= r7 )r rH rI �s3rE r! r! r" �&test_concat_categorical_3elem_coercionA s6 & & z=TestConcatAppendCommon.test_concat_categorical_3elem_coercionc C s< t ddgdd�}t ddgdd�}t ddg�}t ddgdd�}t dtjg�}t dddgdd�}t dddddddddtjdddg �}tj||||||gdd�}t�||� |j|||||gdd�}t�||� t ddddtjddddddddg �}tj||||||gdd�}t�||� |j|||||gdd�}t�||� d S ) Nr r r )r% rv r T)r4 )r r r r r= r8 r< r7 ) r rH rI rz Zs4Zs5Zs6rE rD r! r! r" �&test_concat_categorical_multi_coercionk s $ $ z=TestConcatAppendCommon.test_concat_categorical_multi_coercionc C s� t tddtjgdd��}t tdddgdd��}t tddtjdddgdd��}t�tj||gdd�|� t�|j|dd�|� t tddtjdddddtjg dd��}t�tj|||gdd�|� t�|j||gdd�|� d S )Nr r T)Zordered)r4 ) r r r r r8 r< r r= r7 )r rH rI rE r! r! r" � test_concat_categorical_ordered� s & z6TestConcatAppendCommon.test_concat_categorical_orderedc C s� t tjtjtjgtjd�dd�}t tjdg�}t tjtjtjdg�}t�tj||gdd�|� t�|j |dd�|� t dtjgdd�}t tjtjg�}t dtjtjtjgdd�}t�tj||gdd�|� t�|j |dd�|� t tjtjgdd�}t tjtjg�}t tjtjtjtjg�}t�tj||gdd�|� t�|j |dd�|� t�tj||gdd�|� t�|j |dd�|� t tjtjgdd�}t tjtjgdd�}t tjtjtjtjgdd�}t�tj||gdd�|� t�|j |dd�|� d S )N)r% r r T)r4 ru ) r r �arrayr r r8 r< r r= r7 )r rH rI rE r! r! r" �$test_concat_categorical_coercion_nan� s, z;TestConcatAppendCommon.test_concat_categorical_coercion_nanc C s� t g dd�}t ddgdd�}t�tj||gdd�|� t�|j|dd�|� t�tj||gdd�|� t�|j|dd�|� t g dd�}t g dd�}t�tj||gdd�|� t�|j|dd�|� t g dd�}t g dd�}t�tj||gdd�|� t�|j|dd�|� t�tj||gdd�|� t�|j|dd�|� t g dd�}t tjtjg�}t tjtjg�}t�tj||gdd�|� t�|j|dd�|� t�tj||gdd�|� t�|j|dd�|� d S )Nr )r% r r T)r4 r )r r8 r< r r= r7 r r )r rH rI rE r! r! r" � test_concat_categorical_empty� s. z4TestConcatAppendCommon.test_concat_categorical_emptyc C s t ddgddgd�}ddg}t||d��}t ddddgddgd�}ddddg}t||d�tddddg�d�}t�t�||g�|� t�|�|�|� t ddgddd gd�}ddg}t||d��} tj|| gd d �} ttd �ddddgd��}t�| |� |j| d d �} t�| |� d S ) Nr r )rp r r )Zcatsr. r )rN r T)r4 Zabab) r r r r8 rb r r= r7 r6 ) r �catr. Zdf�cat2r@ rE Zcat3rA Zdf_different_categoriesrD r! r! r" � test_categorical_concat_append� s z5TestConcatAppendCommon.test_categorical_concat_appendN) �__name__� __module__� __qualname__�__doc__r# r+ r/ rK rL rT rZ r: �markZ parametrizerc rd rj rk rl ro rs rt r{ r| r} r r� r� r! r! r! r" r s, %lI 50 @*&&r ) �numpyr r: Zpandasr r r r r Zpandas._testingZ_testingr8 r r! r! r! r" �<module> s
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка