Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexes/multi/__pycache__/test_indexing.cpython-37.pyc
Назад
B �5�g�y � @ s� d dl mZ d dlZd dlZd dlmZmZ d dlZ d dlm Z m Z m Z m Z d dlmZ G dd� d�Zdd� ZG d d � d �Zd d � Zd d� Zej�ddgd e dgd �g�ej�ddddddge dddddg�g�dd� ��Zej�ddge dg�g�ej�ddge dg�g�dd� ��ZG dd� d�ZG dd� d�ZG d d � d �Z d d � Z ej�d ej d!d"gd#d$d%ggd ej d&fej d!d"gd#d$d%ggd'ej d#fd(fd!d"d#gd$ej d$ggd'd"ej fd&fg�d)d*� �Z ej�d+ej d'd,gd-d.dgge d d,d�ej d'fej d'd,gd-d.dgge d d-d�ej d/fd'd,d-gd.ej dgge d'd-d�d,ej fd-fd'd,d-gd.ej dgge d'd-d�d,ej fd0fg�d1d2� �Z!d3d4� Z"dS )5� )� timedeltaN)�InvalidIndexError�PerformanceWarning)� Categorical�Index� MultiIndex� date_rangec @ s e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zej � d e j d d gdddggde j dfe j d d gdddggde j d fe j d d gdddggde j dfd d dgde j dggdd e j fdfd d dgde j dggdd e j fdfd d dgde j dggdd e j fdfg�dd� �Z dS )� TestSliceLocsc C sz |� d�\}}|�dd�}|dks&t�|�d d�}|dks>t�|�dd �}|dt|�fks^t�|�dd�}|d ksvt�d S ) Nr )�foo�two)Zqux�one)� � )r r r �bar�baz)� � )� sortlevel� slice_locs�AssertionError�len)�self�idx� sorted_idx�_�result� r �^/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexes/multi/test_indexing.py�test_slice_locs_partial s z%TestSliceLocs.test_slice_locs_partialc C s� t �� }|�� }|j}t|�|jd |jd �� }|| }|dd� �� }t �|j|j� t|�|jd tdd� |jd tdd� �� }|| }|dd� �� }t �|j|j� d S )Nr � � � )�seconds� ) �tm�makeTimeDataFrame�stack�index�slicer �assert_almost_equal�valuesr )r �df�stackedr ZslobZsliced�expectedr r r �test_slice_locs% s z TestSliceLocs.test_slice_locsc C s� t �� }|�� }|j}tjtdd�� |�d� W d Q R X tjtdd��" |�|jd tdd� d� W d Q R X t � dd�}|�� }|j}tjtdd�� |�tdd�� W d Q R X tjtdd�� |�|jd d � W d Q R X d S ) Nz^Level type mismatch)�match)r � r r! )r" )r r r )r �a) r$ r% r& r' �pytest�raises� TypeErrorr r ZmakeCustomDataframe)r r+ r, r r r r �"test_slice_locs_with_type_mismatch9 s & z0TestSliceLocs.test_slice_locs_with_type_mismatchc C s� t tt�d��tt�d��tt�d��gt�ddddddddg�t�ddddddddg�t�ddddddddg�gd�}d}tjt|d�� |�d d � W d Q R X |� d�\}}|�d d � d S ) Nr r r r r0 )�levels�codesz4[Kk]ey length.*greater than MultiIndex lexsort depth)r/ )r r r )r r r ) r r �np�arange�arrayr2 r3 �KeyErrorr r )r r' �msgZ sorted_indexr r r r �test_slice_locs_not_sortedJ s & z(TestSliceLocs.test_slice_locs_not_sortedc C s� t ddddgdddggdddddddddg dddddddddg gd�}|�dd �}|d ks^t�|�dd �}|d ksvt�|�d d �}|d ks�t�|�dd �}|d ks�t�|�dd �}|dks�t�|�dd�}|dt|�fks�t�d S )Nr r r r# r r0 )r6 r7 )r r )r r )r0 r# r )r r )r# r0 )r0 � ���� )r r r r )r r' r r r r � test_slice_locs_not_contained\ s 0 z+TestSliceLocs.test_slice_locs_not_containedz$index_arr,expected,start_idx,end_idxr1 �b�c�d�e)r r0 N)rB rE )r r0 )rC rE c C s( t �|�}|j||d�}||ks$t�d S )N)�start�end)r � from_arraysr r )r � index_arrr- � start_idx�end_idxr r r r r �"test_slice_locs_with_missing_valuev s z0TestSliceLocs.test_slice_locs_with_missing_value)�__name__� __module__� __qualname__r r. r5 r= rA r2 �mark� parametrizer8 �nanrL r r r r r s ""r c C s� d}t jt|d��$ | �t�t| �d tj�d� W d Q R X t jt|d��$ | �t�t| �d tj�d� W d Q R X t jt|d�� | �dd� W d Q R X d S )Nz,putmask: mask and data must be the same size)r/ r r )r2 r3 � ValueErrorZputmaskr8 Zonesr Zbool_)r r< r r r � test_putmask_with_wrong_mask� s ((rT c @ s" e Zd Zdd� Zdd� Zdd� Zej�dde j d gd d d ggde j d ge j d d d ge j d�fde j d gd d d gge j d fge j dge j d�fdd d ge j d d ggde j fge j dge j d�fdd d ge j d d gge j d d ge j d d d ge j d�fg�dd� �Z dd� Zdd� Zdd� Zdd� ZdS )�TestGetIndexerc C sH t t�d��}t t�d��}tjdddddddgtjd�}tjdddddddgtjd�}t||g||gd�}|d d� }|dddg }|�|�}t�|tjddd gtjd�� |j|d d �}tjd ddddgtjd�} t�|| � |j|d d d � d d �} t�| | d d d � � |j|d d �} t�|| � |j|d d �}tjdddddgtjd�} t�|| � |j|d d d � d d �} t�| | d d d � � |j|dd �} t�|| � |�|j �}|�|�} t�|| � |�dddg�}|d d d gk� � �s�t �t t t d��t t d�� �}t t t d���}d}tjt|d�� |�|� W d Q R X d S )Nr r r r r0 )�dtype)r6 r7 r r? �pad)�method�ffill�backfill�bfillr@ � z8Reindexing only valid with uniquely valued Index objects)r/ )r r8 r9 r: �intpr � get_indexerr$ r) r* �allr �list�ranger2 r3 r )r Z major_axisZ minor_axisZ major_codesZ minor_codesr' Zidx1Zidx2�r1�e1�r2Zrffill1Zrbfill1Zrexp1r< r r r �test_get_indexer� sD z TestGetIndexer.test_get_indexerc C sp t �ddg�}d}tjt|d�� |jdgdd� W d Q R X d}tjt|d�� |jdgd d d � W d Q R X d S ) N)r1 r )rB r zJmethod='nearest' not implemented yet for MultiIndex; see GitHub issue 9365)r/ r1 �nearest)rX z,tolerance not implemented yet for MultiIndexrW r )rX � tolerance)r � from_tuplesr2 r3 �NotImplementedErrorr^ )r �midxr< r r r �test_get_indexer_nearest� s z'TestGetIndexer.test_get_indexer_nearestc C sL t �tdddg�ttdddd��g�}|�|�}t�|tjdtj d �� d S ) Nr1 rB rC z 2012-01-01r0 �H)�periods�freq� )rV ) r � from_productr r r^ r$ �assert_numpy_array_equalr8 r9 r] )r rj r r r r �!test_get_indexer_categorical_time� s z0TestGetIndexer.test_get_indexer_categorical_timezindex_arr,labels,expectedr r r0 r r r? )rV r c C s$ t �|�}|�|�}t�||� d S )N)r rH r^ r$ rq )r rI �labelsr- r r r r r �#test_get_indexer_with_missing_value� s z2TestGetIndexer.test_get_indexer_with_missing_valuec C s" t �dddgddddgg�}t �dgdddgg�}|�|�}tjdddg|jd �}t�||� |j|d d �}tjd ddg|jd �}t�||� |j|d d �}tjd ddg|jd �}t�||� |j|dd �}tjdddg|jd �}t�||� |j|dd �}tjdddg|jd �}t�||� d S )Nr? r r r r0 r r# � )rV rZ )rX r r[ rW rY )r rp r^ r8 r: rV r$ r) )r � mult_idx_1� mult_idx_2�indexerr- �backfill_indexer� pad_indexerr r r �test_get_indexer_methods� s" z'TestGetIndexer.test_get_indexer_methodsc C s� t �ddgdddgddgg�}t �dd d d d d dg�}|js@t�|jsJt�|jsTt�|js^t�|d |d k srt�|d |d k r�|d k s�n t�|d |d ks�t�|d |d k r�|d k s�n t�|d |d k r�|d k s�n t�|d |d k �r |d k �s"n t�|d |d k �s8t�|�|�}tjdddddddg|j d�}t � ||� |j|dd�}tjdddddddg|j d�}t � ||� |j|dd�}tjdddddddg|j d�}t � ||� |d |d k �r |d k �sn t�|d |d k �r*|d k �s0n t�|d |d k �rT|d k �sZn t�|d |d k �r~|d k �s�n t�|d |d k �r�|d k �s�n t�|d |d k�s�t�|d |d k �r�|d k �s�n t�|d |d k �r|d k �sn t�|d |d k �r<|d k �sBn t�|d |d k �rf|d k �sln t�|d |d k �r�|d k �s�n t�|d |d k �r�|d k �s�n t�|�|�}tjddddddddddddg |j d�}t � ||� |j|dd�}tjddddddddddddg |j d�}t � ||� |j|dd�} tjddddddddddddg | j d�}t � || � d S )Nr r0 r r r# r ru )r r r> )r r ro )r r# ru )r r r# )r ru ru )r ru r> )r0 r# r> r r? )rV rZ )rX rW � r> ro r@ r[ ) r rp rh Z is_monotonicr Z is_uniquer^ r8 r: rV r$ r) ) r rv rw Zindexer_no_fillr- Zindexer_backfilledZindexer_paddedrx ry rz r r r �%test_get_indexer_three_or_more_levels/ sj &&&* *********** $ $ $z4TestGetIndexer.test_get_indexer_three_or_more_levelsc C s� t �ddggd �}t �ddg�}|d |d k rB|d k sHn t�|d |d k s\t�|�|�}tjd d g|jd �}t� ||� |j|d d �}tjdd g|jd �}t� ||� |j|d d �}tjddg|jd �}t� ||� d S )Nr r r )r r0 r r )r r0 r r ru r r> r? )rV r[ )rX rY r ) r rp rh r r^ r8 r: rV r$ r) )r rv rw rx r- ry rz r r r � test_get_indexer_crossing_levels� s & z/TestGetIndexer.test_get_indexer_crossing_levelsc C s� t �td�ddgg�}d}tjt|d�� |j|d d� dd� W d Q R X d }tjt|d�� |j|d d� d d � W d Q R X d S ) Nr0 �A�Bz;limit argument only valid if doing pad, backfill or nearest)r/ r? r )�limitz?tolerance argument only valid if doing pad, backfill or nearestZpiano)rg )r rp ra r2 r3 rS r^ )r �mir< r r r �!test_get_indexer_kwarg_validation� s z0TestGetIndexer.test_get_indexer_kwarg_validationN)rM rN rO re rk rr r2 rP rQ r8 rR r: r] rt r{ r} r~ r� r r r r rU � s" 4 00 $ 4c.rU c C s� | d dkst �| dd� }| dddg }|�|�s8t �| ddddddg }| t�ddddddg� }| ddddg }|�|�s�t �|�|�s�t �d S ) Nr )r r r r0 r TFr )r �equalsr8 r: )r r r- Zresult2r r r � test_getitem� s r� c C sB | � d�\}}|�d�tdd�ks&t�|�d�tdd�ks>t�d S )Nr r r0 r r r )r �get_locr( r )r r r r r r �test_getitem_group_select� s r� �ind1Tr �ind2Fc C sF t �dddddg�}t�|| |� t �ddg�}t�|| |� d S )N)r@ r )r\ r )r! r0 )�( r )�2 r )r rh r$ �assert_index_equal)r� r� r r- r r r �test_getitem_bool_index_all� s r� c C s\ t �dg�}t�|| |� t tjg tjd�tjg tjd�gg g gd�}t�|| |� d S )N)r@ r )rV )r6 r7 )r rh r$ r� r8 r: �int64)r� r� r r- r r r � test_getitem_bool_index_single� s r� c @ s� e Zd Zdd� Zdd� Zdd� Zej�de e e e g�ej�de e e e g�d d � ��Z ej�d d d g�ej�de e ge e gg�dd� ��Zdd� Zej�d d d g�dd� �Zdd� Zdd� Zdd� Zdd � Zd d � Zd S ) � TestGetLocc C sR |� d�dkst�|� d�dks$t�tjtdd�� |� d� W d Q R X tjtdd�� |� d � W d Q R X d }tjt|d�� |j d d d � W d Q R X ttt� d��tt� d��tt� d��gt� ddddddddg�t� ddddddddg�t� ddddddddg�gd�}tjtdd�� |� d� W d Q R X |� d�t dd�k�sNt�d S )N)r r r )r r r0 z^10$)r/ )r r z^'quux'$ZquuxzEonly the default get_loc method is currently supported for MultiIndexr rf )rX r r r )r6 r7 z ^\(1, 1\)$)r r )r r r ) r� r r2 r3 r; ri r r r8 r9 r: r( )r r r< r' r r r � test_get_loc s" & zTestGetLoc.test_get_locc C s� t ddddg�}|�d�}tdd�}||ks0t�t dddddg�}|�d�}d}||ks\t�tjtdd�� |�d� W d Q R X d S ) Nr r r rC r1 rB �2)r/ )r r� r( r r2 r3 r; )r r' r r- �rsZxpr r r �test_get_loc_duplicates s z"TestGetLoc.test_get_loc_duplicatesc C s� t tt�d��tt�d��tt�d��gt�ddddddddg�t�ddddddddg�t�ddddddddg�gd�}|�d�\}}tdd�}|| �d��d�}||ks�t�|� |�s�t�|�d�\}}d}||ks�t�|d ks�t�t j t d d �� |�d � W d Q R X t j t d d �� |� d��d� W d Q R X t j t d d �� |j ddd �jddd � W d Q R X t dgttd��gt�ddddg�t�ddddg�gd�}|�dtd d �f�\}}td d �}||k�s�t�|� |�d���s�t�d S )Nr r r r r0 )r6 r7 )r r )r r r z ^\(2, 2\)$)r/ )r r z^2$)�leveli� )r r r8 r9 r: Z get_loc_levelr( Z droplevelr r� r2 r3 r; �dropr` ra )r r' �locZ new_indexr- Z exp_indexr r r r �test_get_loc_level( s6 & "( z TestGetLoc.test_get_loc_level�dtype1�dtype2c C sL t �ddg��|�t �ddg��|�g}t�|�}|�|d �dksHt�d S )Nr r r )r8 r: Zastyper rp r� r )r r� r� r6 r r r r � test_get_loc_multiple_dtypesN s ( z'TestGetLoc.test_get_loc_multiple_dtypesr� r r �dtypesc C sf ddgddgg}ddg}|\}}t jddg|d�||< |d�||< t�|�}|�t|��dksbt�d S ) Nr1 rB rC rD r r )rV r0 )r8 r: r rp r� �tupler )r r� r� r6 �keyZ lev_dtypeZ key_dtyper r r r �test_get_loc_implicit_castV s z%TestGetLoc.test_get_loc_implicit_castc C s� ddgt jddd�g}t�|�}|�d�dks2t�|�d�dksDt�tjtd d �� |�d � W d Q R X tjtd d �� |�d � W d Q R X d S )NFTr r� )rV )r r r )r r z^\(False, True\)$)r/ )FTz^\(True, False\)$)TF) r8 r9 r rp r� r r2 r3 r; )r r6 r r r r �test_get_loc_cast_boolb s z!TestGetLoc.test_get_loc_cast_boolc C s^ ddgddgg}ddg}t jd|gt|�d�||< |||< t�|�}|�t|��dksZt�d S )Nr1 rB rC rD r )rV r0 )r8 r: �typer rp r� r� r )r r� Z nulls_fixturer6 r� r r r r �test_get_loc_nano s zTestGetLoc.test_get_loc_nanc C s� t �ddgddgg�}t|�d�t�s*t�tjtdd�� |�d� W d Q R X tjtd d�� |�t j � W d Q R X tjt d d�� |�t j g� W d Q R X d S ) Ng �?g @g @g @r z^3$)r/ r0 z^nan$zunhashable type: 'list') r rH � isinstancer� r( r r2 r3 r; r8 rR r4 )r r r r r �test_get_loc_missing_nany s z#TestGetLoc.test_get_loc_missing_nanc C s� t �tjdggd �}tddd �}|�tj�|ks6t�t �tjddtjgg�}t�ddddg�}t � |�tj�|� t �tjdggd �}tddd �}|�tjdf�|ks�t�d S )Nr r r TFr0 r ) r rp r8 rR r( r� r rH r: r$ rq )r r r- r r r �1test_get_loc_with_values_including_missing_values� s z<TestGetLoc.test_get_loc_with_values_including_missing_valuesc C st t dddgdddddd d d ggdddd d d d d d d g d ddddd d dddg gddgd�}|�d�tdd�kspt�d S )N�Dr� �Cr � � �% �9 �C �K �R r r r0 r r# r ru �tag�day)r6 r7 �names)r r� r( r )r r' r r r �test_get_loc_duplicates2� s . z#TestGetLoc.test_get_loc_duplicates2c C sr t dgddgdggddgddgddggdddgdd�}d }t�t�� |�|�}W d Q R X |tddd �ksnt�d S ) Nr1 r ru r �x�y�z)r6 r7 r� Z sortorder)r1 ru )r r$ Zassert_produces_warningr r� r( r )r r r� r r r r � test_get_loc_past_lexsort_depth� s z*TestGetLoc.test_get_loc_past_lexsort_depthc C s: t �ddg�}d}tjt|d�� |�g � W d Q R X d S )N)r1 r )rB r zunhashable type)r/ )r rh r2 r3 r4 r� )r r r< r r r �#test_multiindex_get_loc_list_raises� s z.TestGetLoc.test_multiindex_get_loc_list_raisesN)rM rN rO r� r� r� r2 rP rQ �int�float�bool�strr� r� r� r� r� r� r� r� r� r r r r r� s & $ r� c @ s6 e Zd Zdd� Zej�deee j e j g�dd� �Z dS )� TestWherec C s: t �ddg�}d}tjt|d�� |�d� W d Q R X d S )N)r r )r r z2\.where is not supported for MultiIndex operations)r/ T)r rh r2 r3 ri �where)r �ir< r r r � test_where� s zTestWhere.test_where�klassc C sF t �ddg�}ddg}d}tjt|d�� |�||�� W d Q R X d S )N)r r )r r FTz2\.where is not supported for MultiIndex operations)r/ )r rh r2 r3 ri r� )r r� r� Zcondr< r r r �test_where_array_like� s z TestWhere.test_where_array_likeN)rM rN rO r� r2 rP rQ r` r� r8 r: �pd�Seriesr� r r r r r� � s r� c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Ze j j d d� �Z dS )� TestContainsc C s4 t �ddgddgg�}d|ks"t�d|jks0t�d S )Nr r� r r )r rp r Z_engine)r rj r r r �test_contains_top_level� s z$TestContains.test_contains_top_levelc C sr t dgtddd�gddddddgddddd d ggd d gd �}dt�d�f|ksRt�x|jD ]}||ksZt�qZW d S ) Nr� z 2012-01-01r )rm r r? r r r0 r r� )r6 r7 r� )r r r� � Timestampr r* )r r� �valr r r �test_contains_with_nat� s z#TestContains.test_contains_with_natc C s( d|ks t �d|kst �d |ks$t �d S )N)r r )r r )r )r r r r r � test_contains� s zTestContains.test_containsc C s^ t �dtjdgg�}tj|ks"t�t �ddgtjdgg�}tj|ksHt�dtjf|ksZt�d S )Nr r r0 )r rH r8 rR r )r r r r r � test_contains_with_missing_value� s z-TestContains.test_contains_with_missing_valuec C s� t �ddgddgg�}d|ks"t�|�d�}d|jd ks>t�d|ksJt�t �ddgdd gg�}d|kslt�|�d�}d|jd ks�t�d|ks�t�d S ) Nr r r0 r r r1 rB rC rD )r rp r r� r6 )r r r r r � test_multiindex_contains_dropped� s z-TestContains.test_multiindex_contains_droppedc C sT t jdddd�}t�|t�t|��g�}|d |ks8t�d|ksDt�d|ksPt�d S )Nz09:30:00z16:00:00z30 min)rn r Zelement_not_exitz0 day 09:30:00)r� Ztimedelta_ranger rH r8 r9 r r )r Ztxr r r r �test_contains_td64_level� s z%TestContains.test_contains_td64_levelc C s& t �td�td�g�}d|ks"t�d S )Ni@B )i@B r )r rH ra r )r r r r r �test_large_mi_contains s z#TestContains.test_large_mi_containsN) rM rN rO r� r� r� r� r� r� r2 rP Zslowr� r r r r r� � s r� c C s� t �tddddd�dgdgg�} t�dt�t| ��i| �}|jtj d d �dd d �f df }t �td d ddd �dgdgg�}tj t�d t|�d �|dd�}t � ||� d S )Nz2019-01-01T00:15:33�d rl �date)rm rn �namer� r0 r z2019-1-2z2019-01-02T00:15:33z2019-01-05T03:15:33)rF rG rn r� � )�datar' r� ) r rp r r� Z DataFramer8 r9 r r� Z IndexSlicer� r$ Zassert_series_equal)r r+ r ZqidxZ should_ber r r �!test_timestamp_multiindex_indexer s " r� z index_arr,expected,target,algor1 rB rC rD rE �leftr �rightc C s* t �| �}|j||dd�}||ks&t�d S )Nr� )Zside�kind)r rH Zget_slice_boundr )rI r- �target�algor r r r r �'test_get_slice_bound_with_missing_value# s r� z$index_arr,expected,start_idx,end_idxr r0 r )r r )r0 r c C s( t �| �}|j||d�}||ks$t�d S )N)rF rG )r rH Z slice_indexerr )rI r- rJ rK r r r r r �%test_slice_indexer_with_missing_value2 s r� c s\ d} dd� dgd | dgd | dgd | t jg| dgd | dg| dgd | t jg| dgd | dg| gD �� xltt� ��D ]\}t�� �}|�� | �|ks�t�t j|d t j d �}|� � fd d�|D ��}t � ||� q�W tt� ��}t j d gt|� t j d �}tddgd | �}|� |g� fd d�|D � �}t � ||� d S )Nr c S s g | ] }t |��qS r )r� )�.0Zarrr r r � <listcomp>H s z%test_pyint_engine.<locals>.<listcomp>r r@ r r ro r> )rV c s g | ] }� | �qS r r )r� r� )�keysr r r� ] s r? c s g | ] }� | �qS r r )r� r� )r� r r r� d s )r8 rR ra r r rh r� r r9 r] r^ r$ rq r: r` r� )�Nr r' r- r Zidces�missingr )r� r �test_pyint_engineB s& * r� )#�datetimer �numpyr8 r2 Z pandas.errorsr r Zpandasr� r r r r Zpandas._testingZ_testingr$ r rT rU r� r� rP rQ r� r� r� r� r� r� rR r� r( r� r� r r r r �<module> sH u 7 , $ 6B "0 &&*8
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка