Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexes/numeric/__pycache__/test_indexing.cpython-37.pyc
Назад
B �5�g?Q � @ s� d dl Zd dlZd dlmZmZmZmZmZm Z m Z d dl m Z ejdd� �ZG dd� d�ZG dd� d�ZG d d � d �ZG d d � d �ZG d d� d�ZG dd� d�ZG dd� d�ZdS )� N)� Float64Index�Index� Int64Index� RangeIndex�Series� Timestamp� UInt64Indexc C s dddddg} t | �S )Nl l l l l )r )Zlarge� r �`/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexes/numeric/test_indexing.py� index_large s r c @ s� e Zd Zej�dddddg�dd� �Zej�dddddg�dd � �Zej�d d d d g�dd� �Zej�ddddg�dd� �Z dd� Z dd� Z dd� Z dd� Z dd� Zd d � Zej�d d gd ged!�gd"gg�ej�ddddg�d#d$� ��ZdS )%� TestGetLoc�methodN�pad�backfill�nearestc C sD t dddg�}|jd|d�dks$t�|r@|jd|dd�dks@t�d S )Nr � � )r )r � tolerance)r �get_loc�AssertionError)�selfr �indexr r r � test_get_loc s zTestGetLoc.test_get_locc C sL t dddg�}|rd}nd}tjt|d�� |jddg|d� W d Q R X d S )Nr r r znot supported betweenz invalid key)�match)r )r �pytest�raises� TypeErrorr )r r r �msgr r r � test_get_loc_raises_bad_label s z(TestGetLoc.test_get_loc_raises_bad_labelz method,loc)r r )r r )r r c C s> t dddg�}|�d|�|ks"t�|jd|dd�|ks:t�d S )Nr r r g�������?)r )r r r )r r �locr r r r �test_get_loc_tolerance+ s z!TestGetLoc.test_get_loc_tolerancec C s< t dddg�}tjtdd�� |jd|dd� W d Q R X d S ) Nr r r z1.1)r g�������?g�������?)r )r r r �KeyErrorr )r r r r r r �%test_get_loc_outside_tolerance_raises3 s z0TestGetLoc.test_get_loc_outside_tolerance_raisesc C s< t dddg�}tjtdd�� |jdddd � W d Q R X d S ) Nr r r zmust be numeric)r g�������?r �invalid)r )r r r � ValueErrorr )r r r r r �!test_get_loc_bad_tolerance_raises9 s z,TestGetLoc.test_get_loc_bad_tolerance_raisesc C s: t dddg�}tjtdd�� |jddd� W d Q R X d S )Nr r r ztolerance .* valid if)r g�������?)r )r r r r$ r )r r r r r �'test_get_loc_tolerance_no_method_raises> s z2TestGetLoc.test_get_loc_tolerance_no_method_raisesc C s@ t dddg�}tjtdd�� |jddddgd� W d Q R X d S ) Nr r r ztolerance size must match)r g�������?r )r )r r r r$ r )r r r r r �&test_get_loc_raises_missized_toleranceC s z1TestGetLoc.test_get_loc_raises_missized_tolerancec C s� t dddg�}x@dD ]8}|�d|�dks,t�|d k r|jd|dd�dkst�qW x<dD ]4\}}|�d |�|ksrt�|jd |d d�|ksVt�qVW tjtd d �� |�d � W d Q R X tjtdd �� |�d� W d Q R X tjtdd �� |jdddd� W d Q R X tjtdd �� |�d� W d Q R X tjtdd �� |�d� W d Q R X tjtdd �� |jddd d� W d Q R X tjtdd �� |jddt�d g�d� W d Q R X tjtdd �� |jddt�dd g�d� W d Q R X d S ) Ng g �?g @)Nr r r r r )r ))r r )r r )r r g�������?g�������?z^'foo'$)r �fooz^1\.5$g �?r g�������?)r r z^True$Tz^False$Fzmust be numericgffffff�?r z must contain numeric elementsz+tolerance size must match target index sizer ) r r r r r r! r$ �np�array)r �idxr r r r r �test_get_loc_float64H s2 " z TestGetLoc.test_get_loc_float64c C s� t tjddg�}|�d�dks"t�|�tj�dks6t�t tjdtjg�}|�d�dksZt�d}tjt|d�� |�tj� W d Q R X t tjdtjtjg�}|�d�dks�t�d}tjt|d�� |�tj� W d Q R X d S )Nr r r z7'Cannot get left slice bound for non-unique label: nan')r z6'Cannot get left slice bound for non-unique label: nan) r r) �nanr r r r r! � slice_locs)r r+ r r r r �test_get_loc_nai s zTestGetLoc.test_get_loc_nac C s� t ddg�}|�d�dks t�tjtdd�� |�d� W d Q R X tjtdd�� |�tj� W d Q R X tjtdd�� |�tjg� W d Q R X d S ) Nr r r z^3$)r � z^nan$z'\[nan\]' is an invalid key) r r r r r r! r) r- r )r r+ r r r �test_get_loc_missing_nan| s z#TestGetLoc.test_get_loc_missing_nan�valsr g �?z 2019-12-31�testc C s6 t |�}tjtdd�� |jtj|d� W d Q R X d S )Nr- )r )r )r r r r! r r) r- )r r2 r r+ r r r �(test_get_loc_float_index_nan_with_method� s z3TestGetLoc.test_get_loc_float_index_nan_with_method)�__name__� __module__� __qualname__r �mark� parametrizer r r r" r% r& r' r, r/ r1 r r4 r r r r r s ! r c @ s2 e Zd Zdd� Zej�dddg�ej�dejdddd d gej d �d fejdddd d gej d �d fejddd d d gej d �dfejddd d d gej d �dfg�dd� ��Z dd� Z ej�dd ddddgdddgfdddddgdddgfdddddgdddgfd ddddgdddgfdddddgdddgfdddddgdddgfd ddddgdd d gfdddddgd d dgfdddddgdd dgfd d dddgdd d gfdd dddgd d dgfdd dddgdd dgfd ddddgdddgfdddddgdd dgfdddddgdd dgfg�d d � �Z ej�d e eeejg�ej�d e ed!d!d"gdd"d"gd"d#d#ggdd dgdddgdd dgg���d$d%� ��Zd&d'� Zej�d(d d d)dgfddd d gfddd)dgfg�d*d+� �Zej�d,eeeeg�ej�d-d.d/g�d0d1� ��Zej�d-d ddg�d2d3� �Zd4d5� Zd6d7� Zd8d9� Z d:d;� Z d<d=� Z dS )>�TestGetIndexerc C sP t dddddg�}t dddg�}|�|�}tjdddgtjd�}t�||� d S ) Nr r r0 � � � ���)�dtype)r � get_indexerr) r* �intp�tm�assert_almost_equal)r �index1�index2�r1�e1r r r �test_get_indexer� s z TestGetIndexer.test_get_indexer�reverseTFzexpected,methodr> r r )r? r Zffillr r Zbfillc C s^ t dddddg�}t dddg�}|r@|d d d� }|d d d� }|j||d�}t�||� d S ) Nr r r0 r; r< r= r> )r )r r@ rB rC )r rI �expectedr rD rE �resultr r r �test_get_indexer_methods� s z'TestGetIndexer.test_get_indexer_methodsc C sj t t�d��}tjtdd�� |jddgdd� W d Q R X tjtdd�� |jddgdd� W d Q R X d S ) N� ztolerance argument)r r r )r zlimit argument)�limit)r r) �aranger r r$ r@ )r r r r r �test_get_indexer_invalid� s z'TestGetIndexer.test_get_indexer_invalidz$method, tolerance, indexer, expectedNr< � r g�������?g������?g !@� c C s: t t�d��}|j|||d�}t�|tj|tjd�� d S )NrM )r r )r? )r r) rO r@ rB �assert_numpy_array_equalr* rA )r r r �indexerrJ r �actualr r r �test_get_indexer_nearest� s z'TestGetIndexer.test_get_indexer_nearest�listtypeztolerance, expectedg333333�?g�������?g �?c C sD t t�d��}|jdddgd||�d�}t�|tj|tjd�� d S )NrM g�������?g������?g !@r )r r )r? )r r) rO r@ rB rS r* rA )r r rJ rW r rU r r r �+test_get_indexer_nearest_listlike_tolerance� s z:TestGetIndexer.test_get_indexer_nearest_listlike_tolerancec C st t t�d��}tjtdd�� |jddgddd� W d Q R X tjtdd�� |jddgddd d gd � W d Q R X d S ) NrM zlimit argument)r r r r )r rN ztolerance size must matchr r0 )r r )r r) rO r r r$ r@ )r r r r r � test_get_indexer_nearest_error� s z-TestGetIndexer.test_get_indexer_nearest_errorzmethod,expected� c C sz t t�d��d d d� }|jdddg|d�}t�|tjdddgtjd�� |jd d d g|d�}t�|tj|tjd�� d S ) NrM r> r r< rQ )r r; )r? g�������?g������?g !@)r r) rO r@ rB rS r* rA )r r rJ r rU r r r �#test_get_indexer_nearest_decreasing� s z2TestGetIndexer.test_get_indexer_nearest_decreasing� idx_classr r@ �get_indexer_non_uniquec C s� |t d��}tdddg�}t||�|�}tjdddgtjd�}|dkrTt�||� n0tjdtjd�}t�|d |� t�|d |� d S ) Nr; TFr> )r? r@ r0 r r ) r r �getattrr) r* rA rB rS rO )r r r\ Z numeric_index�otherrK rJ �missingr r r �-test_get_indexer_numeric_index_boolean_target� s z<TestGetIndexer.test_get_indexer_numeric_index_boolean_targetc C sn t dddg�}t ddg�}tjtdd�� |j||d� W d Q R X tjtdd�� |j||d� W d Q R X d S ) Nr r r0 TFzCannot compare)r )r )r r r r r@ )r r �left�rightr r r �,test_get_indexer_with_method_numeric_vs_bool s z;TestGetIndexer.test_get_indexer_with_method_numeric_vs_boolc C s� t dddg�}t ddg�}|�|�}dtjt|�tjd� }t�||� |�|�}dtjt|�tjd� }t�||� |�|�d }dtjt|�tjd� }t�||� |�|�d }dtjt|�tjd� }t�||� d S ) Nr r r0 TFr> )r? r ) r r@ r) Zones�lenrA rB rS r] )r rb rc �resrJ r r r � test_get_indexer_numeric_vs_bool s z/TestGetIndexer.test_get_indexer_numeric_vs_boolc C s� t dddg�}t�|�|�tjdddgtjd�� dd d g}t�|�|d �tjd ddgtjd�� t�|�|d �tjdddgtjd�� t�|�|d�tjdddgtjd�� d S )Ng g �?g @r r r )r? g��������g �?g�������?r r> r r )r rB rS r@ r) r* rA )r r+ �targetr r r �test_get_indexer_float64. s ""z'TestGetIndexer.test_get_indexer_float64c C s< t ddtjg��tjg�}tjdgtjd�}t�||� d S )Nr r )r? )r r) r- r@ r* rA rB rS )r rK rJ r r r �test_get_indexer_nan? s z#TestGetIndexer.test_get_indexer_nanc C s� t tddd��}t t�d��}|�|�}tjddddddddddg tjd �}t�||� t t�d��}|j|d d �}tjddddddddddg tjd �}t�||� t t�d��}|j|d d �}tjdddddddddd g tjd �}t�||� d S )Nr � r rM r> r r0 r; )r? r )r r r< ) r �ranger) rO r@ r* rA rB rS )r r rh rT rJ r r r �test_get_indexer_int64E s $ $ $z%TestGetIndexer.test_get_indexer_int64c C s t t�d��d�d d �}|�|�}tjddddd d ddddg tjd �}t�||� t t�d��d�d d �}|j|d d �}tjddddd d d d d d g tjd �}t�||� t t�d��d�d d �}|j|dd �}tjddddd d ddddg tjd �}t�||� d S )NrM �uint64r< l r r> r r r0 r; )r? r )r r ) r r) rO Zastyper@ r* rA rB rS )r r rh rT rJ r r r �test_get_indexer_uint64V s $ $ $z&TestGetIndexer.test_get_indexer_uint64) r5 r6 r7 rH r r8 r9 r) r* rA rL rP rV �list�tupler �ziprX rY r[ r r r r ra rd rg ri rj rm ro r r r r r: � s\ , $ ,2 r: c @ sp e Zd Zej�deejddd��e e ddd��e ejddd��g�ej�d e e ejeg�d d � ��Zd d � ZdS )� TestWherer r< Zfloat64)r? r rk r rn �klassc C sr dgt |� }|}|�||��}dgdgt |�d }t|jg|dd � �� �}|�||��}t�||� d S )NTFr )re �wherer Z _na_value�tolistrB �assert_index_equal)r rt r ZcondrJ rK r r r � test_whereh s zTestWhere.test_wherec C st t dddg�}t�dddg�}tjdgtjd�}t dddg�}|�||�}t�||� |�| |�}t�||� d S )Nr r= r FTr )r? )r r) r* Zint64ru rB rw Zputmask)r r+ �maskr_ rJ rK r r r �test_where_uin64{ s zTestWhere.test_where_uin64N)r5 r6 r7 r r8 r9 r r) rO r rl r rp rq r* r rx rz r r r r rs g s " rs c @ sJ e Zd Zej�deeeg�dd� �Z dd� Z ej�deeg�dd� �Z dS ) �TestTakert c C s8 |ddddgdd�}|� dddg�}|j|jks4t�d S )Nr r r0 r; r( )�namer )�taker| r )r rt r Ztakenr r r �test_take_preserve_name� s z TestTake.test_take_preserve_namec C s\ t dddgdd�}|�t�dddg��}t dddgdd�}t�||� |jt�dddg�d d �}t ddtjgdd�}t�||� |jt�dddg�d d d �}t dddgdd�}t�||� d }tjt |d�� |jt�dddg�d d � W d Q R X tjt |d�� |jt�dddg�d d � W d Q R X d}tjt |d�� |�t�ddg�� W d Q R X d S )Ng �?g @g @�xxx)r| r r r> T)� fill_valueF)� allow_fillr� zJWhen allow_fill=True and fill_value is not None, all indices must be >= -1)r ������z3index -5 is out of bounds for (axis 0 with )?size 3) r r} r) r* rB rw r- r r r$ � IndexError)r r+ rK rJ r r r r � test_take_fill_value_float64� s$ $$z%TestTake.test_take_fill_value_float64c C sd |dddgdd�}|� t�dddg��}|dddgdd�}t�||� |j}d|� d �}tjt|d �� |j t�dddg�d d � W d Q R X |j t�dddg�d d d�}|dddgdd�}t�||� tjt|d �� |j t�dddg�d d � W d Q R X tjt|d �� |j t�dddg�d d � W d Q R X d}tjt |d �� |� t�ddg�� W d Q R X d S )Nr r r0 r )r| r r> z Unable to fill values because z cannot contain NA)r T)r� F)r� r� r� r� z3index -5 is out of bounds for (axis 0 with )?size 3) r} r) r* rB rw r5 r r r$ r� )r rt r+ rK rJ r| r r r r �test_take_fill_value_ints� s$ $ $$z"TestTake.test_take_fill_value_intsN) r5 r6 r7 r r8 r9 r r r r~ r� r� r r r r r{ � s r{ c @ s8 e Zd Zej�deeeg�dd� �Z dd� Z dd� Z dS ) � TestContainsrt c C s" |dddddg�}d |ks t �d S )Nr r r r0 r; )r )r rt r r r r �test_contains_none� s z TestContains.test_contains_nonec C s" t ddtjg�}tj|ks t�d S )Ng �?g @)r r) r- r )r r r r r �test_contains_float64_nans� s z'TestContains.test_contains_float64_nansc C s t ddtjg�}d|ks t�d S )Ng �?g @)r r) r- r )r r r r r � test_contains_float64_not_nans� s z+TestContains.test_contains_float64_not_nansN) r5 r6 r7 r r8 r9 r r r r� r� r� r r r r r� � s r� c @ sj e Zd Zej�deeg�dd� �Zej�deeg�dd� �Z ej�deeg�dd� �Z dd � Z d d � Z d S ) � TestSliceLocsr? c C s� t tjddddddddg|d ��}t|�}|jdd �d|fksBt�|jd d �d |fksZt�|�d d �d ksnt�|�dd�d |fks�t�|jd d�dks�t�|jdd�dks�t�|d d d� }|�d d�dks�t�|�dd �dks�t�d S )Nr r r r< r= rZ rQ rM )r? )�startr0 rR )r0 r= )�end)r r= )r rZ r> )r r= )r r< )r r) r* re r. r )r r? r �nrE r r r �test_slice_locs� s "z TestSliceLocs.test_slice_locsc C s� t tjddddddddg|d ��}t|�}|�d d �d |fksBt�|�d d�dksVt�|d d d� }|�dd�dksxt�|�dd�d|fks�t�d S )Nr r r r< r= rZ rQ rM )r? g @g $@r0 g @g %@)r0 rR r> g !@g �?)r r= )r r) r* re r. r )r r? r r� rE r r r �test_slice_locs_float_locs� s "z(TestSliceLocs.test_slice_locs_float_locsc C s| t tjddddg|d��}|�dd�dks.t�|�dd�dksBt�|d d d� }|�dd�dksdt�|�dd�dksxt�d S ) NrM � � )r? )r r0 � � r> )r r) r* r. r )r r? r rE r r r �test_slice_locs_dup_numeric� s z)TestSliceLocs.test_slice_locs_dup_numericc C sd t tjddg�}|�d�dks"t�|�tj�dks6t�t dtjtjddg�}|�tj�dks`t�d S )Nr r )r r0 )r r0 r )r r< )r r) r- r. r )r r r r r �test_slice_locs_na s z TestSliceLocs.test_slice_locs_nac C s` t tjddg�}tjtdd�� |jdd� W d Q R X tjtdd�� |jdd� W d Q R X d S )Nr r � )r g �?)r� )r� )r r) r- r r r! r. )r r r r r �test_slice_locs_na_raises s z'TestSliceLocs.test_slice_locs_na_raisesN) r5 r6 r7 r r8 r9 �int�floatr� r� r� r� r� r r r r r� � s r� c @ sz e Zd Zej�ddddg�ej�dddg�dd � ��Zej�ddddg�ej�d d d g�ej�d ddg�dd� ���ZdS )�TestGetSliceBounds�kind�getitemr Nzside, expected)rb r; )rc r< c C s, t td��}|jd||d�}||ks(t�d S )Nr= r; )r� �side)r rl �get_slice_boundr )r r� r� rJ r rK r r r � test_get_slice_bounds_within s z/TestGetSliceBounds.test_get_slice_bounds_withinr� rb rc zbound, expected)r> r )rM r= c C s, t td��}|j|||d�}||ks(t�d S )Nr= )r� r� )r rl r� r )r r� r� rJ �boundr rK r r r � test_get_slice_bounds_outside s z0TestGetSliceBounds.test_get_slice_bounds_outside)r5 r6 r7 r r8 r9 r� r� r r r r r� s r� )�numpyr) r Zpandasr r r r r r r Zpandas._testingZ_testingrB Zfixturer r r: rs r{ r� r� r� r r r r �<module> s $ z W"C8
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка