Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/series/indexing/__pycache__/test_getitem.cpython-37.pyc
Назад
B �5�g�T � @ s� d Z ddlmZmZmZ ddlZddlZddlmZm Z ddl m Z ddl Z ddl mZmZmZmZmZmZmZmZmZ ddlmZ ddlmZ ddl m Z G d d � d �Z G d d � d �Z G d d� d�Z G dd� d�Z!G dd� d�Z"dd� Z#ej$�%deddg�eeddd��eedddd��g�d d � �Z&d d � Z'd d!� Z(d"d#� Z)d$d%� Z*d&d'� Z+d(d)� Z,d*d+� Z-d,d-� Z.d.d/� Z/d0d1� Z0d2d3� Z1d4d5� Z2d6d7� Z3d8d9� Z4ej$�%d:d;d<g�d=d>� �Z5dS )?zJ Series.__getitem__ test classes are organized by the type of key passed. � )�date�datetime�timeN)� conversion� timezones)� is_scalar) � Categorical� DataFrame� DatetimeIndex�Index�Series� Timestamp� date_range� period_range�timedelta_range)� IndexingError)�BDayc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z e j � dddg�dd� �Ze j � dddg�dd� �Zdd � Zd d � Zd d � Zd!S )"�TestSeriesGetitemScalarsc C sv t dddgdddgdd�}|d }|dks.t�t ddgddgdd�}t dddgdddgdd�}|d }t�||� d S ) N)� r )� r )� r g g�������?g�������?Zfoo)�index�name)r �AssertionError�tm�assert_series_equal)�self�ser�result�expected� r �_/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/series/indexing/test_getitem.py�$test_getitem_float_keys_tuple_values' s z=TestSeriesGetitemScalars.test_getitem_float_keys_tuple_valuesc C sB t ddgt�d�t�d�gd�}|jd }|| }|dks>t�d S )Nr r �O�i8)r )r �np�dtyper r )r r �keyr r r r! � test_getitem_unrecognized_scalar6 s z9TestSeriesGetitemScalars.test_getitem_unrecognized_scalarc C sF t t�dd�t�dd�d�}d}tjt|d�� |d W d Q R X d S )N� � )r z2index -11 is out of bounds for axis 0 with size 10)�matchi����)r r Z rands_array�pytest�raises� IndexError)r r �msgr r r! �#test_getitem_negative_out_of_boundsA s z<TestSeriesGetitemScalars.test_getitem_negative_out_of_boundsc C s. d}t jt|d�� |t|� W d Q R X d S )Nz3index \d+ is out of bounds for axis 0 with size \d+)r+ )r, r- r. �len)r �datetime_seriesr/ r r r! �%test_getitem_out_of_bounds_indexerrorH s z>TestSeriesGetitemScalars.test_getitem_out_of_bounds_indexerrorc C s2 t g td�}tjtdd�� |d W d Q R X d S )N)r& z-1)r+ ���)r �objectr, r- �KeyError)r r r r r! �4test_getitem_out_of_bounds_empty_rangeindex_keyerrorN s zMTestSeriesGetitemScalars.test_getitem_out_of_bounds_empty_rangeindex_keyerrorc C s� t tj�d�ddddddgd�}tjtdd�� |d W d Q R X tjtd d�� |d W d Q R X t tj�d�ddddddgd�}tjtdd�� |d W d Q R X tjtd d�� |d W d Q R X d S ) N� r r r )r z^5$)r+ r) z^'c'$�c)r r% �random�randnr, r- r6 )r r r r r! �%test_getitem_keyerror_with_int64indexU s z>TestSeriesGetitemScalars.test_getitem_keyerror_with_int64indexc C s" t �d�}|| |d ks t�d S )Nr) )r% �int64r )r r2 �idxr r r! �test_getitem_int64g s z+TestSeriesGetitemScalars.test_getitem_int64c C s8 t td�ttd��d�}|ttd�� }t�||� d S )Nr) )r )r �range�listr r )r r r r r r! �test_getitem_regressionl s z0TestSeriesGetitemScalars.test_getitem_regression�tzstrz Europe/Berlinzdateutil/Europe/Berlinc C sh t �|�}tddd|d�}t||jd�}td|d�}tdd d d d �}t�||�}|| || ksdt �d S ) Nz2012-12-24 16:00z2012-12-24 18:00�H)�start�end�freq�tz)r �dataz2012-12-24 17:00)rH i� � � � r ) r Z maybe_get_tzr r �hourr r r Zlocalize_pydatetimer )r rC rH r �tsZ time_pandas�dtZ time_datetimer r r! �test_getitem_pydatetime_tzt s z3TestSeriesGetitemScalars.test_getitem_pydatetime_tzrH z US/Easternzdateutil/US/Easternc C sB t dd|d�}ttj�t|��|d�}|d }t�||d � d S )Nz1/1/2000r* )�periodsrH )r z1/3/2000r )r r r% r: r; r1 r �assert_almost_equal)r rH �rngr r r r r! � test_string_index_alias_tz_aware� s z9TestSeriesGetitemScalars.test_string_index_alias_tz_awarec C sn t dddd�}ttj�t|��|d�}|jdk|jdk@ }|tdd� }|| }|j � d �|_ t � ||� d S )Nz1/1/2000z1/5/2000Z5min)rG )r � � ) r r r% r: r; r1 rM �minuter r � _with_freqr r )r rS rN �maskr r r r r! �test_getitem_time_object� s z1TestSeriesGetitemScalars.test_getitem_time_objectc C sJ t td�td�g�}tddg|d�}|jd }||d }||ksFt�d S )Nz 12-31-1999z 12-31-2000r r )r r )r r r �ilocr )r Zcatsr r r r r r! �%test_getitem_scalar_categorical_index� s z>TestSeriesGetitemScalars.test_getitem_scalar_categorical_indexc C s� t dddd�}tt�t|��|d�}d}|�|�}|dks>t�|| }||jd ksXt�d}tj t |d �� |�d � W d Q R X tj t |d �� |d W d Q R X d S ) Nz1 day 10:11:12�hi� )rG rQ )r z6 days, 23:11:12� z!^Timedelta\('50 days 00:00:00'\)$)r+ z50 days) r r r% �aranger1 �get_locr r[ r, r- r6 )r rS r r' �indexerr r/ r r r! �$test_getitem_str_with_timedeltaindex� s z=TestSeriesGetitemScalars.test_getitem_str_with_timedeltaindexN)�__name__� __module__� __qualname__r" r( r0 r3 r7 r< r? rB r, �mark� parametrizerP rT rZ r\ rb r r r r! r & s r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zej � d �d d � �Z ej � de eddd�d�dddgge eddd�d�dgge eddd�d�g ge deddd��dddgge deddd��dggg�dd� �Zdd� Zdd� Zd d � ZdS ) �TestSeriesGetitemSlicesc C sB t dd�}|�� }|d }t ddd�}t||d�}t�||� d S )Nz1/1/2008z1/1/2009Z2008z 2008-01-01z 2008-12-31)rE rF )r )r Z to_seriesr r r )r Zarrr r rS r r r r! �1test_getitem_partial_str_slice_with_datetimeindex� s zITestSeriesGetitemSlices.test_getitem_partial_str_slice_with_datetimeindexc C sv t dddddg�}ttj�t|��|d�}|dd � }|dd � }t�||� |dd� }|dd� }t�||� d S )Nz1/1/2000z1/2/2000z1/3/2000z1/4/2000)r r � )r r r% r: r; r1 r r )r r> rN r r r r r! �-test_getitem_slice_strings_with_datetimeindex� s zETestSeriesGetitemSlices.test_getitem_slice_strings_with_datetimeindexc C s� t dddd�}tt�t|��|d�}|dd� }|jdd � }t�||� |dd � }|jdd � }t�||� |d d� }|jd d � }t�||� d S ) Nz1 day 10:11:12r] i� )rG rQ )r z5 dayz6 day�V � )r r r% r_ r1 r[ r r )r rS r r r r r r! �2test_getitem_partial_str_slice_with_timedeltaindex� s zJTestSeriesGetitemSlices.test_getitem_partial_str_slice_with_timedeltaindexc C s� t dddd�}tt�t|��|d�}|dd � }|jdd � }t�||� |dd � }|jdd � }t�||� |d }||jd ks�t�d S ) Nz1 day 10:11:12�usi� )rG rQ )r r z1 day 10:11:12.001i� z1 days, 10:11:12.001001i� ) r r r% r_ r1 r[ r r r )r rS r r r r r r! �<test_getitem_partial_str_slice_high_reso_with_timedeltaindex� s zTTestSeriesGetitemSlices.test_getitem_partial_str_slice_high_reso_with_timedeltaindexc C sR t jt| d�� |d d �tjf }W d Q R X |jd d �tjf }t �||� d S )N)�check_stacklevel)r �assert_produces_warning� FutureWarningr% Znewaxis�valuesrR )r r2 �using_array_managerr r r r r! �test_getitem_slice_2d� s z-TestSeriesGetitemSlices.test_getitem_slice_2dz&ignore:Using a non-tuple:FutureWarningc C sj t dddd�}ttj�d�|d�}tddd �g}t�t�� || }W d Q R X ||d }t� ||� d S ) NZ20090415Z20090519Z2B)rG � )r r8 � r ) r r r% r: r; �slicer rr rs r )r r �sra r r r r r! � test_getitem_median_slice_bug� s z5TestSeriesGetitemSlices.test_getitem_median_slice_bugzslc, positionsi� r Nr r i� i� c C s< t dddgtdddg��}|| }|�|�}t�||� d S )Nr r r z 2019-01-01z2019-01-01T06:00:00z 2019-01-02)r r Ztaker r )r ZslcZ positionsr r r r r r! �test_getitem_slice_date s z/TestSeriesGetitemSlices.test_getitem_slice_datec C sd d}t jt|jdd�d�� |dd� W d Q R X t jt|jdd�d�� |dd� W d Q R X d S ) NzUcannot do slice indexing on DatetimeIndex with these indexers \[{key}\] of type floatz4\.0)r' )r+ g @g $@z4\.5g @)r, r- � TypeError�format)r r2 r/ r r r! � test_getitem_slice_float_raises s z7TestSeriesGetitemSlices.test_getitem_slice_float_raisesc C st t td�ttd��d�}|dd � }t�||� |dd � }t�||dd � � |d d� }t�||d d� � d S )Nr* )r i�i����r r )r r@ rA r r )r r r r r r! �test_getitem_slice_bug& s z.TestSeriesGetitemSlices.test_getitem_slice_bugc C s^ t tj�d�ddddddddgd �}|d d� }t |jd d� ddddgd �}t�||� d S ) N� r rj r8 r* rJ � � )r )r r% r: r; rt r r )r r r r r r r! �test_getitem_slice_integers1 s $ z3TestSeriesGetitemSlices.test_getitem_slice_integers)rc rd re ri rk rn rp rv r, rf �filterwarningsr{ rg ry r r| r r� r� r r r r! rh � s $ rh c @ s� e Zd Zej�deeje e g�dd� �Z dd� Z ej�deeje g�dd� �Z ej�deeje g�ej�dejejejg�d d � ��Zd d � Zd S )�TestSeriesGetitemListLike�boxc C sL t ddg�}t dgtd�}||�}d}tjt|d�� || W d Q R X d S )N�A�B�C)r& zANone of \[Index\(\['C'\], dtype='object'\)\] are in the \[index\])r+ )r r5 r, r- r6 )r r� r r' r/ r r r! �test_getitem_no_matches: s z1TestSeriesGetitemListLike.test_getitem_no_matchesc C sf t tdddd��}|ddg }t tjddd�tjd dd�gddgd d �}t�||� |jd ksbt�d S ) Nz 2000-01-01r* �D)rQ rG r rj z 2000-01-03)rG z 2000-01-05z Period[D])r r& )r r �pdZPeriodr r r& r )r r r �expr r r! �*test_getitem_intlist_intindex_periodvaluesF s zDTestSeriesGetitemListLike.test_getitem_intlist_intindex_periodvaluesc C sb t ddd��d �}tj�|�}ttt|��|d�}|jd d� }|dg�}|| }t � ||� d S )Nz 2000-01-03r )rQ )r r r ) r rX r� Z IntervalIndexZ from_breaksr r@ r1 r[ r r )r r� �dti�iir r r' r r r r! �*test_getitem_intlist_intervalindex_non_intR s zDTestSeriesGetitemListLike.test_getitem_intlist_intervalindex_non_intr& c C s| t td���|�}tddd�}tj�||g�}ttt|��d d d� |d�}|dg�}t j t dd �� || W d Q R X d S ) Nrj z 2000-01-03r )rQ r4 )r r) �5)r+ ) r r@ �astyper r� Z MultiIndexZ from_productr r1 r, r- r6 )r r& r� r> r� �mir r' r r r! �-test_getitem_intlist_multiindex_numeric_level^ s zGTestSeriesGetitemListLike.test_getitem_intlist_multiindex_numeric_levelc C sh t dddg�}tjdg|d�}tjtdd�� || W d Q R X tjtdd�� |j| W d Q R X d S )Nr r r rj )r& �4)r+ )r r% �arrayr, r- r6 �loc)r Z uint_dtyper r' r r r! �test_getitem_uint_array_keyk s z5TestSeriesGetitemListLike.test_getitem_uint_array_keyN)rc rd re r, rf rg rA r% r� r r r� r� r� r= �float64Zuint64r� r� r r r r! r� 9 s $ r� c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )�TestGetitemBooleanMaskc C sH |}||� � k}|t|� }|| }t�||� t�|j|j| � d S )N)�medianrA r r �assert_index_equalr )r � string_seriesr rY r r r r r! �test_getitem_booleanw s z+TestGetitemBooleanMask.test_getitem_booleanc C s� t g tjd�}d|j_||�� }|jjdks2t�|jtjksBt�t ddg�}t tt g dd�d�}|t g td� }t � ||� d}t j t|d�� |t g td� W d Q R X t j t|d�� |t d gtd� W d Q R X d S ) N)r& Z index_namer� r� r= )r& r zsUnalignable boolean Series provided as indexer \(index of the boolean Series and of the indexed object do not match)r+ T)r r% r= r r Zisnar r& r5 r r r r, r- r �bool)r r r r r/ r r r! �test_getitem_boolean_empty� s z1TestGetitemBooleanMask.test_getitem_boolean_emptyc C s� |}||� � k}|�t�}|| }|| }t�||� |�� }|�� }d||< d||< t�||� tj|dd�< d} tj t | d�� || W d Q R X tj t | d�� d||< W d Q R X d S )Nr) r* z=Cannot mask with non-boolean array containing NA / NaN values)r+ ) r� r� r5 r r �copyr% �nanr, r- � ValueError) r r� r rY Zomaskr r �s2�copr/ r r r! �test_getitem_boolean_object� s" z2TestGetitemBooleanMask.test_getitem_boolean_objectc C sl t dddd�}t�ddddg�}t|j�}|| }|jjjd ksDt�ttd��}|| }|jjd ksht�d S )Nz 2016-01-01rj z US/Pacific)rQ rH TF) r r% r� r �_dataZ_values�baser r@ )r r� r' r �resZser2Zres2r r r! � test_getitem_boolean_dt64_copies� s z7TestGetitemBooleanMask.test_getitem_boolean_dt64_copiesc C sj |}|j dt� d�|�� k}d}tjt|d�� || W d Q R X tjt|d�� |j| W d Q R X d S )Nr )rG zsUnalignable boolean Series provided as indexer \(index of the boolean Series and of the indexed object do not match)r+ )�shiftr r� r, r- r r� )r r2 rN Z mask_shiftedr/ r r r! �test_getitem_boolean_corner� s z2TestGetitemBooleanMask.test_getitem_boolean_cornerc C s0 |� � }||dk }||dk }t�||� d S )Nr )Z sort_valuesr r )r r� Zordered�selr� r r r! �$test_getitem_boolean_different_order� s z;TestGetitemBooleanMask.test_getitem_boolean_different_orderc C s~ t dddd�}tjt|�td�}d|dd�< || }|dd� }|j|jksPt�t�||� d|d < || }|jd kszt�d S ) Nz1/1/2000z3/1/2000r� )rG )r& Tr* � � ) r r% Zzerosr1 r� rG r r r� )r rS rY Zmaskedr r r r! �-test_getitem_boolean_contiguous_preserve_freq� s zDTestGetitemBooleanMask.test_getitem_boolean_contiguous_preserve_freqN) rc rd re r� r� r� r� r� r� r� r r r r! r� v s r� c @ s e Zd Zdd� ZdS )�TestGetitemCallablec C s� t dtd�d�}|dd� }||jd ks.t�|dd� }|jddg }t�||� |d d� }|jd d d g }t�||� d S ) Nrj ZABCD)r c S s dS )Nr� r )�xr r r! �<lambda>� � z;TestGetitemCallable.test_getitem_callable.<locals>.<lambda>r� c S s ddgS )Nr� r� r )r� r r r! r� � r� r� c S s ddddgS )NTFr )r� r r r! r� � r� r r r )r rA r� r r r r[ )r r r r r r r! �test_getitem_callable� s z)TestGetitemCallable.test_getitem_callableN)rc rd re r� r r r r! r� � s r� c C sN dd� | D �}| | }| t | dk� }| | dk }t�||� t�||� d S )Nc s s | ]}|d kV qdS )r Nr )�.0r� r r r! � <genexpr>� s z)test_getitem_generator.<locals>.<genexpr>r )�iterr r )r� �genr Zresult2r r r r! �test_getitem_generator� s r� �seriesr z 2012-01-01r )rQ ZCET)rQ rH c C sT t jtd| d�� | d d �d f }W d Q R X t�| �d d �d f }t �||� d S )Nz&Support for multi-dimensional indexing)r+ rq )r rr rs r% ZasarrayZassert_numpy_array_equal)r� ru r r r r r! � test_getitem_ndim_deprecated s r� c C s> | }|d }d}t jt|d�� |ddd�f W d Q R X d S )Nr� z \(2000, slice\(3, 4, None\)\))r+ i� r rj )r, r- r} )Z/multiindex_year_month_day_dataframe_random_data�dfr r/ r r r! �4test_getitem_multilevel_scalar_slice_not_implemented s r� c C sR t td��} td| d�}t| | d�}d}tjt|d�� ||dk W d Q R X d S )Nr* )r zWIndexing a Series with DataFrame is not supported, use the appropriate DataFrame column)r+ r) )rA r@ r r r, r- r} )rS r r� r/ r r r! � test_getitem_dataframe_raises# s r� c C sZ t td��} t�dddg�}t dddg�| |< t dddd dd d d d dg �}t�| |� d S )Nr* r rj rU � rJ r r r r) r8 rx r� )r r@ r% r� r r )r r> r r r r! �'test_getitem_assignment_series_aligment/ s r� c C s� t ddddgddddgd�} tjtd d �� | d W d Q R X tjtd d �� | j�d � W d Q R X tjtd d �� | jj�d � W d Q R X d S ) Nr r) r8 r� g @g @g @)r �None)r+ )r r, r- r6 r r` Z_engine)r r r r! �9test_getitem_duplicate_index_mistyped_key_raises_keyerror9 s r� c C s: t td��} td�f}| | }| |d }t�||� d S )Nr) r r )r r@ ry r r )r r' r r r r r! �,test_getitem_1tuple_slice_without_multiindexF s r� c C sZ | | dk }|j | j ks t�| dddg }|j | j ks:t�| dd� }|j | j ksVt�d S )Nr r rj r) r* )r r )r2 r r r r! �test_getitem_preserve_nameO s r� c C s� t tj�d�ttddd��d�} dddddg}t�dddddg�}tjt d d �� | | W d Q R X tjt d d �� | | W d Q R X d S ) Nr* r r� r )r r) rx r� z not in index)r+ ) r r% r: r; rA r@ r� r, r- r6 )r ZindsZarr_indsr r r! � test_getitem_with_integer_labelsZ s r� c C s: | j d t� }d}tjt|d�� | | W d Q R X d S )Nr z,Timestamp\('1999-12-31 00:00:00', freq='B'\))r+ )r r r, r- r6 )r2 �dr/ r r r! �test_getitem_missingf s r� c C sx | dddg }|dddg }| j d |j d ks4t�|j d |j d ksLt�| d |d ks`t�|d |d kstt�d S )Nr r r )r r )r� Z object_seriesZslice1Zslice2r r r! �test_getitem_fancyn s r� c C s | d }t |tj�st�d S )Nr) )� isinstancer% r� r )r2 �valuer r r! �test_getitem_box_float64w s r� c C s> t td�dddddgd�} t| d �s*t�| d dks:t�d S )Nr) r9 �a�b)r r )r r@ r r )�objr r r! �test_getitem_unordered_dup| s r� c C sN t td�dddddgtjd�} t ddgddgtjd�}| d }t�||� d S )Nr) r� r� r� )r r& r rj )r r@ r% r= r r )r r r r r r! �test_getitem_dups� s r� c C st t td�tdddddg�d�} | d }| jddg }t�||� t�t�� | j� | d�}W d Q R X t�||� d S )Nr) r� r� r9 )r r r ) r r@ r r[ r r rr rs r � get_value)r r r r r r! � test_getitem_categorical_str� s r� c C s( t ddddddgd�} | d d d� d S )Nr r� r� r9 )r r4 )r )r r r r! �+test_slice_can_reorder_not_uniquely_indexed� s r� � index_valsZaabcdZaadcbc C s, t td�t| �d�}|d }|dks(t�d S )Nr) )r r )r r@ rA r )r� rz r r r r! �0test_duplicated_index_getitem_positional_indexer� s r� )6�__doc__r r r �numpyr% r, Zpandas._libs.tslibsr r Zpandas.core.dtypes.commonr Zpandasr� r r r r r r r r r Zpandas._testingZ_testingr Zpandas.core.indexingr Zpandas.tseries.offsetsr r rh r� r� r� r� rf rg r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r! �<module> sL , =t
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка