Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/__pycache__/test_align.cpython-37.pyc
Назад
B �5�g:, � @ sV d dl Zd dlZd dlZd dlZd dlmZmZmZm Z d dl m Z G dd� d�Z dS )� N)� DataFrame�Index�Series� date_rangec @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Ze j � d dde j gddegddegdde j gg�dd� �Zdd� Zdd� Zd)dd�Ze j � dddg�e j � d d d dg�e j � d d d g�e j � d d!d"d#d$g�d%d&� ����Zd'd(� ZdS )*�TestDataFrameAlignc C s: t ddddd�}t ddddd�}ttj�t|�d�|�}ttj�t|�d�|�}|�|�\}}|jj|jjksrt �|jj|jjks�t �|� d�}|�|�\}}|jjt j ks�t �|jjt j ks�t �|j|d d d �\}}|jjt j ks�t �|jjt j ks�t �|d j|d d � |jjt j k�s"t �|jjt j k�s6t �d S ) NZ2001� �Hz US/Eastern)Zperiods�freq�tzZ2H� z US/Centralr )�axis) r r �np�randomZrandn�len�align�indexr �AssertionErrorZ tz_convert�pytz�UTC)�selfZidx1Zidx2�df1�df2Znew1Znew2Z df1_central� r �[/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/test_align.py�test_frame_align_aware s" z)TestDataFrameAlign.test_frame_align_awarec s� |� |�\}}|j|jk s t�|j |dd�\}}|j|jks@t�|jd d�d d�f }|j |ddd�\}}t�|j|j� |j�|j�}|j� |�}|j� |�}|� |�j }|� |�j } |dk� � s�t�|j |ddd �\}}t�|j|j� t�|j|j� t�|j|j� |jd d�d d�f � � }|j |d d �\}}t�|j|j� t�|j|j� |j�|j�}|j� |�}|j� |�}|� |�j }|� |�j } |dk� � �s�t�|j |d d d �\}}t�|j|j� |j |d d d d�\}}t�|j|j� |j |jd d �df d d d d d�\}}t�|jtg �� |j |jd d �df d d d dd�\}}t�|jtg �� d} tjt| d��$ |j |jdd d�f d dd � W d Q R X |j} ttt| ��| d�� |j � dd �\} } t�| j|j� t�| j|j� t| t��s�t�|j � d d�\} } t�| j|j� � fdd�|jD �}t||j|jd�}t�| |� td ddgdddgd��}||d dk }tddggd gd d gd�}t�||� |�|d dkd�}tdddgdddgd��}t�||� d S ) NF)�copy���r r ���)r � fill_value�right)�joinr � )r �inner�pad)r r �method)r r r$ r z)No axis named 2 for object type DataFrame)�match� )r )Zbroadcast_axisc s i | ] }� |�qS r r )�.0�c)�sr r � <dictcomp>u s z7TestDataFrameAlign.test_align_float.<locals>.<dictcomp>)r �columns� r � )�a�br. r/ )r Z_mgrr �iloc�tm�assert_index_equalr+ r r � difference�reindex�values�allr r �pytestZraises� ValueErrorr �ranger � isinstancer �assert_frame_equal�where)r � float_frame�af�bf�otherZjoin_idxZdiff_aZdiff_bZ diff_a_valsZ diff_b_vals�msg�idx�leftr �expected�df�resultr )r) r �test_align_float* sv ""( z#TestDataFrameAlign.test_align_floatc C s@ t td�dddgd�}|j|dddd �\}}t�|j|j� d S ) Nr �A�B�C)r r+ r" r! r# )r r r$ )r r9 r r1 r2 r+ )r Z int_framer@ r> r? r r r �test_align_int� s z!TestDataFrameAlign.test_align_intc C s* |j |dddd�\}}t�|j|j� d S )Nr" r! r# )r r r$ )r r1 r2 r+ )r Zfloat_string_framer> r? r r r �test_align_mixed_type� s z(TestDataFrameAlign.test_align_mixed_typec C sR t td�dddgd�}|j|jd d �df ddd dd �\}}t�|jtg �� d S ) Nr rH rI rJ )r r+ r r" r! )r r r$ r )r r9 r r0 r1 r2 r r )r Zmixed_float_framer@ r> r? r r r �test_align_mixed_float� s "z)TestDataFrameAlign.test_align_mixed_floatc C sR t td�dddgd�}|j|jd d �df ddd dd �\}}t�|jtg �� d S ) Nr rH rI rJ )r r+ r r" r! )r r r$ r )r r9 r r0 r1 r2 r r )r Zmixed_int_framer@ r> r? r r r �test_align_mixed_int� s "z'TestDataFrameAlign.test_align_mixed_intz l_ordered,r_ordered,expectedTFc C s� t tjddd�ttd���tjtd�|d��d���d�}t tjd dd�ttd ���tjtd�|d��d���d�}|� |�\}}t |j |�s�t �t |j |�s�t �t �|j |j � d S ) Nr- �int64)�dtypeZaabbcaZcab)Zordered)rH rI rI r Zbabca)r r �aranger �listZastype�pdZCategoricalDtypeZ set_indexr r: r r r1 r2 )r Z l_orderedZ r_orderedrD Zdf_1Zdf_2Z aligned_1Z aligned_2r r r �test_align_categorical� s z)TestDataFrameAlign.test_align_categoricalc C s� t jjtd�td�td�gdd�}ttd�dd�}ttjddd �|d �}ttjddd �|d �}|j|d d �\}}|j|d d �\}}|} t � | |� t � | |� tddddtj tj gd |d �} t � | |� t � | |� |j|d d �\}}|j|d d �\}}t jjtd�td�td�gdd�} tddddddddg| d �} t � | |� t � | |� tddddgd | d �} t � | |� t � | |� d S )Nr& r )r. r/ r( )�namesr/ )�name� rO )rP )r rC )r r r r! r- � � � ) rS Z MultiIndexZ from_productr9 r r r rQ r r1 r; �nan) r ZmidxrB r r Zres1lZres1rZres2lZres2rZexpl�exprZexp_idxr r r �test_align_multiindex� s. z(TestDataFrameAlign.test_align_multiindexc C s� t dddgdddgd�td�d�}tdddgtd �d d �}|j|d d �\}}t dtjdtjdgdtjdtjdgd�td�d�}tddtjdtjgtd�d d �}t�||� t�||� |�|�\}}t�||� t�||� d S )Nr! r r )r. r/ ZACE)r r& r, ZABD�x)r rV r )r ZABCDE) r rR r r r r[ r1 r; Zassert_series_equal)r rE r) Zres1Zres2Zexp1Zexp2r r r � test_align_series_combinations� s "$ z1TestDataFrameAlign.test_align_series_combinationsNc C s� |j ||||||d�\}} d\} } || } } |d ks<|dkrf|jj|j|d�} | j| d�} | j| d�} |d ksv|dkr�|jj|j|d�} | j| d�} | j| d�} | j|||d�} | j|||d�} t�|| � t�| | � d S ) N)r r r$ �limit� fill_axis)NNr )�how)r r! )r+ )r r$ r` )r r r r4 r+ Zfillnar1 r; )r r. r/ r ra rb r$ r` Zaa�abZ join_indexZ join_columnsZeaZebr r r � _check_align� s z TestDataFrameAlign._check_align�methr# Zbfill�axr r! �faxrb r" �outerrC r c C s |}| � |||||� d S )N)�_check_align_fill)r rb re rf rg r= rE r r r �test_align_fill_method s z)TestDataFrameAlign.test_align_fill_methodc C s� |j dd�d d�f }|j dd �dd �f }|j d d�d d�f }| j||||||d� | j||||||dd� | j||||||d� | j||||||dd� | j||||||d� | j||||||dd� | j||||||d� | j||||||dd� d S ) Nr r, � r& r- )r ra rb r$ r! )r ra rb r$ r` )r0 rd ) r �frame�kindre rf rg rC r �emptyr r r ri s z$TestDataFrameAlign._check_align_fill)N)�__name__� __module__� __qualname__r rG rK rL rM rN r7 �markZ parametrizerS ZCategoricalIndexr rT r] r_ rd rj ri r r r r r s( [ # $r )�numpyr r7 r ZpandasrS r r r r Zpandas._testingZ_testingr1 r r r r r �<module> s
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка