Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/__pycache__/test_multilevel.cpython-37.pyc
Назад
B �5�g�9 � @ sr d dl Zd dlZd dlZd dlmZmZmZ d dlm Z ddddddd d d d d g Z G dd� d�Z G dd� d�Z dS )� N)� DataFrame� MultiIndex�Series�sum�prod�min�maxZmedian�meanZskewZmad�std�varZsemc @ s` e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z e j �de�e j �dddg�e j �dddg�e j �dddg�dd � ����Ze j �de�e j �dddg�e j �d ddg�e j �dddg�e j �dddg�d d � �����Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Ze j �d.d/d.g�d0d1� �Zd2d3� Zd4d5� Zd6S )7�TestMultiLevelc C s� |}t �t�� |jdd�}W d Q R X |j|jdd�}|jdd��tj�}t � ||� |d j|jdd�}|d jdd��tj�}t j ||dd� t �t�� |j jddd�}W d Q R X |j|jdd�}|jdd��tj�j }t � ||� d S ) N�month)�level� �AF)� check_names)�axisr )�columnsr ) �tm�assert_produces_warning� FutureWarningr �reindex�index�groupby� transform�np�assert_frame_equal�assert_series_equal�T)�self�/multiindex_year_month_day_dataframe_random_data�ymd� month_sums�result�expected� r% �R/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/test_multilevel.py�test_reindex_level s z!TestMultiLevel.test_reindex_levelc s4 |� � fdd�}|d� |d� |d� |d� d S )Nc s� t t| �}t�t�� � jdd�}W d Q R X |� |dd�}� jdd��tj�}|� |�}t� ||� t t | �}|� d |d dd�}� d jdd��tj�}|� d |�}d|_ t� ||� d S )Nr )r r ) �getattrr r r r r r r r r r �namer )�opname�opr" r# Z broadcastedr$ )r! r% r& � _check_op6 s z3TestMultiLevel.test_binops_level.<locals>._check_op�sub�add�mul�divr% )r r r, r% )r! r& �test_binops_level3 s z TestMultiLevel.test_binops_levelc C s0 |}|j ddg }|jddg }t�||� d S )Nr � )�foo�one)�barr4 )�iloc�locr r )r � multiindex_dataframe_random_data�framer$ Z reindexedr% r% r& � test_reindexM s zTestMultiLevel.test_reindexc C s� |}|j d d d� }|�|�}|j |ks,t�|j| }|j |ksDt�|j}|j|d�}|j|ksdt�|jd d �|f }|j|ks�t�d S )N� )r )r r �AssertionErrorr7 r r )r r r! Z new_index�chunkZymdTr% r% r& � test_reindex_preserve_levelsT s z+TestMultiLevel.test_reindex_preserve_levelsc C s^ |}|d }|j �d�}|�|�}|�dd� �}|�dd� �}|�|j �}tj||dd� d S )Nr r c S s | d S )N� r% )�xr% r% r& �<lambda>o � z7TestMultiLevel.test_groupby_transform.<locals>.<lambda>c S s | d S )Nr? r% )r@ r% r% r& rA p rB F)r )r �get_level_valuesr �applyr r r r ) r r8 r9 �sZgrouper�groupedZappliedr$ r# r% r% r& �test_groupby_transformg s z%TestMultiLevel.test_groupby_transformc C s\ t dgdgdggdgdgdggdddgd�}ttj�d �gd d d d g|d�}|jdd� d S )Nr3 r5 �bazr r4 �two�three)�levels�codes�names� �a�b�c�d)r r )r )r r r �randomZrandr )r �midx�dfr% r% r& �test_groupby_cornert s z"TestMultiLevel.test_groupby_cornerc C s� t �ddddddg�}tddd d d d gd dddddgg|d�}|jdd�|j�dd� � }|jddd�}|�� }|jddgk�� s�t �d S )N)�f1�s1)rW �s2)�f2rX )rZ rY )�f3rX )r[ rY r r? r2 rN � � � � � r; � � )r )r c S s | d dkS )Nr )rZ r[ r% )�ur% r% r& rA � rB z:TestMultiLevel.test_groupby_level_no_obs.<locals>.<lambda>r )r r rZ r[ ) r � from_tuplesr r7 r �mapr r �allr< )r rT rU Zdf1rF r# r% r% r& �test_groupby_level_no_obs~ s ( z(TestMultiLevel.test_groupby_level_no_obsc C sJ |}|d d� j }|d |d< t|jt�s.t�|d |d k�� sFt�d S )Nr\ )i� r r^ )i� r r; )r � isinstancer r r< rf )r r r! rU r% r% r& �.test_setitem_with_expansion_multiindex_columns� s z=TestMultiLevel.test_setitem_with_expansion_multiindex_columnsc C s� t dddgt�dddg�d�}t dd d gt�d d dg�d�}|| }|j�|j�}|�|�|�|� }t�||� |d d d � |d d d � }|j�|j�}|�|�|�|� }t�||� d S )Nr r? r2 )r r )r r? )�Br2 )�datar rN r\ r] )�Zr )rl r? ���)r r rd r �unionr r r )r r@ �y�resZ exp_index�expr% r% r& �test_alignment� s z TestMultiLevel.test_alignmentr+ r r r �skipnaTF�sortc sn |}|j ||d�}|�� �fdd��}t�t�� t|� �|�d�} W d Q R X |r^| j|d�} t�|| � d S )N)r rt c s t | � ��d�S )N)rs )r( )r@ )r+ rs r% r& rA � rB z:TestMultiLevel.test_series_group_min_max.<locals>.<lambda>)r rs )r )r �aggr r r r( � sort_indexr ) r r+ r rs rt Z series_with_multilevel_indexZserrF �leftside� rightsider% )r+ rs r& �test_series_group_min_max� s z(TestMultiLevel.test_series_group_min_maxr c s |}t j|jdddgf< t j|jdddgf< |jj| }� dkrF|}n|j}|j|� |d�} g �� ���fdd�} | �| �} t� t �� t |��|� �d�} W d Q R X |r�| j |� d �} |j |� d �}|� � �j| �|�} t�| � � �| � t�| � � �| � t�| | � d S ) Nr r? r^ r )r r rt c s �� | � t| ���� d�S )N)rs r )�appendr( )r@ )r r+ �piecesrs r% r& �aggf� s z1TestMultiLevel.test_frame_group_ops.<locals>.aggf)r r rs )r r )r �nanr6 r rM r r ru r r r r( rv Z _get_axisrK �rename�assert_index_equalr )r r+ r r rs rt r8 r9 � level_namerF r| rw rx Z level_indexr% )r r+ r{ rs r& �test_frame_group_ops� s( z#TestMultiLevel.test_frame_group_opsc s� t �t�d��d�t�t�d�d�g�}ttj�t |�d�|d�}x�dD ]��d� � �fdd�}t � t �� t |d ��d� d �}W d Q R X |d jdd ��|�}t �||� t � t �� t |��d� d �}W d Q R X |jdd ��|�}t �||� qHW d S ) Nr\ r; )r )r r rN c s t | ��� d�S )N)�ddof)r( )r@ )r� �methr% r& rA � rB z7TestMultiLevel.test_std_var_pass_ddof.<locals>.<lambda>r )r r� )r )r Z from_arraysr Zarange�repeatZtiler rS �randn�lenr r r r( r ru r r )r r rU Zaltr# r$ r% )r� r� r& �test_std_var_pass_ddof� s $ z%TestMultiLevel.test_std_var_pass_ddofc C s^ |}|t kr|d }t�t�� |jddgd�}W d Q R X |jddgd��� }t�||� d S )Nr �yearr )r )r r r r r r Z assert_equal)r r Zframe_or_seriesr! r# r$ r% r% r& �test_agg_multiple_levels s z'TestMultiLevel.test_agg_multiple_levelsc C s� |}|j ddgd��� }|j�d�}|j�d�}|� ||g��� }tj||dd� |jj|jjd d� ksnt�|j |jjd d� d��� }t�||� d S )Nr r )r F)r r? )r r r rC r r rM r< )r r r! r# Zk1Zk2r$ �result2r% r% r& �test_groupby_multilevel s z&TestMultiLevel.test_groupby_multilevelc C s d S )Nr% )r r% r% r& �&test_groupby_multilevel_with_transform s z5TestMultiLevel.test_groupby_multilevel_with_transformc C sD t �ddddg�}ttj�dd�||d�}|�d�|d< |�� }d S ) N)r3 r4 )r3 rI )r5 r4 )r5 rI rN )r r r )ZTotals� )r rd r r rS r� r Z _consolidate)r r rU r% r% r& �test_multilevel_consolidate" s z*TestMultiLevel.test_multilevel_consolidatec C s� t dddgddggddddddgddddddggd�}ttj�d�|d �}ttj�dd �|d �}|d }|jd }|d d� }|j�d�|_t � ||� t � ||� t j t d d �� |d W d Q R X |jd }|�d�}|d d� }|j�d�|_t �||� t �||� t dddgddggddddddgddddddggd�}ttj�d�|d �}ttj�dd �|d �}|d }|jd }|d d� }|j�d�|_t � ||� t � ||� |jd }|�d�}|d d� }|j�d�|_t �||� t �||� d S )N)r3 r5 r )r3 rH r )r3 �quxr r r r? )rK rL r] )r rN z ^\(\('foo', 'bar', 0\), 2\)$)�match))r3 r5 r r? )r3 r5 )r3 rH )r3 r� )r r r rS r� r r7 r Z droplevelr r �pytestZraises�KeyErrorZxsr )r r Zseriesr9 r# r� r$ r% r% r& �test_level_with_tuples* sH $ $ z%TestMultiLevel.test_level_with_tuplesc C s� |}|j ddgdd�}|jdddddd g }t�||� |jj ddgddd �}t�||j� |jddg }t�||� |d jddg }t�||d � |jjd d �ddgf }t�||j� d S ) Nr3 r� r )r r r? r^ r_ r` )r r r )r r6 r r r r7 r )r r8 r9 r# r$ r% r% r& �$test_reindex_level_partial_selectionZ s z3TestMultiLevel.test_reindex_level_partial_selectionrR rN c C sx dddg}t dgdgdg|gd��}||jdk �|�}|�� j}tdgdgdggg g g gdddgd�}t�||� d S ) NrO rP rQ r r? r2 )rO rP rQ rR )rK rL rM )r rO r �firstr r r r )r rR Z group_keysrU �gr# r$ r% r% r& �+test_empty_frame_groupby_dtypes_consistencym s $z:TestMultiLevel.test_empty_frame_groupby_dtypes_consistencyc C sZ ddddg}ddddg}t j|ddgd�}t||d�}|�|j��� }t|�d ksVt�d S ) N)�600809Z20061231)r� Z20070331)r� Z20070630ZdemoZSTK_IDZRPT_Date)rM )r r2 )r rd r r r r� r� r< )r Zidx_tp�dt�idxrE r# r% r% r& � test_duplicate_groupby_issues{ s z,TestMultiLevel.test_duplicate_groupby_issuesc C sP ddgg}ddt jfdg}t|t�|�d�}|jjj}|jjj}t� ||� d S )Nr@ r rO rP )rO rQ g )r ) r r} r r rd ZdtypesrO rP r r )r rk r rU r$ r# r% r% r& � test_subsets_multiindex_dtype� s z,TestMultiLevel.test_subsets_multiindex_dtypeN) �__name__� __module__� __qualname__r' r1 r: r> rG rV rg ri rr r� �markZ parametrize� AGG_FUNCTIONSry r� r� r� r� r� r� r� r� r� r� r� r% r% r% r& r s8 "( 0r c @ s e Zd ZdZdd� ZdS )� TestSortedz+everything you wanted to test about sortingc C s� t dddgdddggddd ddd gdd dddd gg�}td tt|��i|d d �}|jjd ks`t�|�� }|jjdksxt�td dddd git �ddddg�d d �}|j t j dd�dd�f d d �f }t � ||� d S )Nr rj �CrQ rP rO r r r? �colZint64)r ZdtypeFTrN r\ )rj rO )rj rQ )r� rO )r� rP )r r �ranger� r Z is_monotonicr< rv rd r7 �pdZ IndexSlicer r )r r� rU �sortedr$ r# r% r% r& �test_sort_non_lexsorted� s 4 $z"TestSorted.test_sort_non_lexsortedN)r� r� r� �__doc__r� r% r% r% r& r� � s r� )�numpyr r� Zpandasr� r r r Zpandas._testingZ_testingr r� r r� r% r% r% r& �<module> s& {
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка