Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/__pycache__/test_set_index.cpython-37.pyc
Назад
B �5�g{e � @ s� d Z ddlmZmZ ddlZddlZddlmZmZm Z m Z m Z m Z m Z mZmZ ddlmZ G dd� d�ZG dd� d�ZG d d � d �ZdS ) z/ See also: test_reindex.py:TestReindexSetIndex � )�datetime� timedeltaN) � Categorical� DataFrame� DatetimeIndex�Index� MultiIndex�Series� date_range� period_range� to_datetimec @ s~ e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z e j�dddddgdg�e j�ddd g�e j�d dd g�d d � ���Ze j�dddddgdg�e j�d dd g�d d!� ��Ze j�dddddgdg�e j�d dd g�d"d#� ��Zd$d%� Ze j�d&eeejed'd(� d)d(� g�e j�d*d+d,d-d.g�e j�d dd g�d/d0� ���Ze j�d&eeejed1d(� g�e j�d*d+d2d,d-d.g�e j�d dd g�d3d4� ���Ze j�d5eeejeed6d(� d7d(� g�e j�d8eeejeed9d(� d:d(� g�e j�d*d+d2d-d.g�e j�d dd g�d;d<� ����Z e j�d=dd g�e j�d dd g�d>d?� ��Z d@dA� Z dBdC� Z dDdE� Z dFdG� Z!dHS )I� TestSetIndexc C sL dddgdddgd�}t |�}dd d g}||d < t�|d �}|�|� d S ) N� g @� � � � )�t1�t2)r � )r r )r r �tuples)r r � from_tuples� set_index)�self�d�dfr �index� r �_/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/test_set_index.py�test_set_index_multiindex s z&TestSetIndex.test_set_index_multiindexc C s� t ddd�ddd�ddd d �d d d d �gddddgd�}|�ddg�}|ddg }tj|d |d gddgd�|_t�||� d S )Nr r )�a�pr � )r �mr � � )r r# r! r � � r r# r! �x)�columns)�names)r r r � from_arraysr �tm�assert_frame_equal)r r �result�expectedr r r �test_set_index_empty_column' s z(TestSetIndex.test_set_index_empty_columnc C sT t tdd�tdd�g d��}|�ddg�}|j�� j}|ddg j}t�||� d S )Nzdatetime64[ns])ZdtypeZint64)r �b�cr r1 )r r r r Zto_frameZdtypesr, Zassert_series_equal)r Zdf1�df2r. r/ r r r � test_set_index_empty_dataframe9 s z+TestSetIndex.test_set_index_empty_dataframec C s� t �dddg�}ttj�dd�|d�}|�|jd �}|jd d �dd �f }|jd d �df j |_ |jd g|j _ t � ||� d S )N)�foor )r5 r )�barr r )r) r r )r r r �np�random�randnr r) Ziloc�valuesr r* r, r- )r r) r r. r/ r r r � test_set_index_multiindexcolumnsD s z-TestSetIndex.test_set_index_multiindexcolumnsc C sv t dgdd��d�}td|i�}|�|�jd jdks:t�t t|j��d jdksVt�|�|j�jd jdksrt�d S )Nz2014-01-01 10:10:10�UTC)�tzz Europe/Rome�Ar r$ ) r � tz_convertr r r �hour�AssertionErrorr r> )r �idxr r r r �test_set_index_timezoneO s z$TestSetIndex.test_set_index_timezonec C sB t dd� td�D �tj�d�d��}|�d�}t|jt�s>t �d S )Nc S s g | ]}t d dd�t|� �qS )i� r )r r )�.0�ir r r � <listcomp>[ s zBTestSetIndex.test_set_index_cast_datetimeindex.<locals>.<listcomp>i� )r> �Br> ) r �ranger7 r8 r9 r � isinstancer r rA )r r �idfr r r �!test_set_index_cast_datetimeindexX s z.TestSetIndex.test_set_index_cast_datetimeindexc C s� t ddddd�}tdddgdd d gd �|d ��� }|�d �}tdddgdd d gd �t|d d�d �}|j�d �|_t�||� |�d dg�}t j |dddggd dgd�}tddd d gi|d�}t�||� d S )Nz2006-10-29 00:00:00r �Hz US/Pacific)�periods�freqr= r r r r r )r r1 )�datar r )�namer )r* r1 )r ) r r � reset_indexr r r � _with_freqr, r- r r+ )r Zdir �res�expZ exp_indexr r r �test_set_index_dstc s " z TestSetIndex.test_set_index_dstc C sj |}t t�t|��d d d� �}|�|�}t�|j|� tj t dd�� |�|d d d� � W d Q R X d S )N���zLength mismatch)�matchr ) r r7 Zarange�lenr r, �assert_index_equalr �pytest�raises� ValueError)r Zfloat_string_framer rB r r r �test_set_indexv s zTestSetIndex.test_set_indexc C s t �� }d|j_|�|j�jjdgks*t�tj|ddg j j ddgd�}tj|ddddg j j ddddgd�}|�ddg�}|�|j�jjddgks�t�t |�|j�jt�s�t�t � |�|j�j|� |j� ddg�}t |�|j|g�jt�s�t�t � |�|j|g�j|� d S )NrP r> rG )r* �C�D)r, Z makeDataFramer rP r r* rA r r+ �Tr: rI rY �rename)r r �miZmi2�idx2r r r �test_set_index_names s " z!TestSetIndex.test_set_index_namesc C sH t dddgdddgd�dd d gd �}|�|j�tj��}t�||� d S ) Ng�������?g������@gffffff @g @gffffff@g������ @)r> rG i� i� i� )r )r r r Zastyper7 Zint32r, r- )r r r3 r r r �test_set_index_cast� s "z TestSetIndex.test_set_index_cast�keysr> r^ rG )�tuple�as�label�inplaceTF�dropc s� |� t |t�r,tj� fdd�|D �|d�}nt� | |d�}|rN� j|dd�n� }||_|r�� �� }|j||dd�}|d ks�t �n� j||d �}t � ||� d S ) Nc s g | ] }� | �qS r r )rD r( )r r r rF � s z<TestSetIndex.test_set_index_drop_inplace.<locals>.<listcomp>)r* )rP r )�axisT)rk rj )rk ) rI �listr r+ r rk r �copyr rA r, r- ) r �frame_of_index_colsrk rj rf rB r/ r. Z return_valuer )r r �test_set_index_drop_inplace� s z(TestSetIndex.test_set_index_drop_inplacec s� |� t |t�r|n|g}tj� jg� fdd�|D � d g| d�}|rT� j|dd�n� �� }||_� j||dd�}t� ||� d S )Nc s g | ] }� | �qS r r )rD r( )r r r rF � s z6TestSetIndex.test_set_index_append.<locals>.<listcomp>)r* r )rl T)rk �append) rI rm r r+ r rk rn r r, r- )r ro rk rf rB r/ r. r )r r �test_set_index_append� s &z"TestSetIndex.test_set_index_appendc C s\ |j dg|dd�}t|t�r |n|g}|j dg| |dd�}|j ||dd�}t�||� d S )Nr_ T)rk rq )r rI rm r, r- )r ro rk rf r r/ r. r r r �#test_set_index_append_to_multiindex� s z0TestSetIndex.test_set_index_append_to_multiindexc C sh t dddgdddgd��}t dddgitddgd d ��}|j|j�d d � � }|�d �}t�||� d S ) Nr r r r r1 r2 )�val�keyrt ru )rP c S s | dkS )Nr r )Zindxr r r �<lambda>� � z<TestSetIndex.test_set_index_after_mutation.<locals>.<lambda>)r r �locr �mapr r, r- )r r r/ r3 r. r r r � test_set_index_after_mutation� s z*TestSetIndex.test_set_index_after_mutation�boxc C s t | �gS )N)rm )r( r r r rv � rw zTestSetIndex.<lambda>c C s t �| g�S )N)r r+ )r( r r r rv � rw zappend, index_name)TN)TrG )T�test)FNc C s� |}||j _||d �}|tkrPd}tjt|d�� |j|||d� W d Q R X njt|dd �} | d krrt|dd �gn| } |j|||d�} |jdgd|d�} |r�|g| n| | j _t � | | � d S )NrG z%['one', 'two', 'three', 'one', 'two'])rW )rk rq r* rP F) r rP rm rZ r[ �KeyErrorr �getattrr* r, r- ) r ro rk rq � index_namer{ r ru �msgZname_mirP r. r/ r r r � test_set_index_pass_single_array� s z-TestSetIndex.test_set_index_pass_single_arrayc C s t �| g�S )N)r r+ )r( r r r rv rw )Tr> c C s� |}||j _d||d �g}d|tjtttgkr4d ndg}|j|||d�} |jddgd|d�} |rp| jddd�n| } |r�|g| n|| j _ t � | | � d S )Nr> rG )rk rq Fr )rl ) r rP r7 �arrayrm rg �iterr rk r* r, r- ) r ro rk rq r r{ r rf r* r. r/ r r r �test_set_index_pass_arrays s z'TestSetIndex.test_set_index_pass_arrays�box2c C s t �| g�S )N)r r+ )r( r r r rv 9 rw c C s | j S )N)rP )r( r r r rv : rw �box1c C s t �| g�S )N)r r+ )r( r r r rv E rw c C s | j S )N)rP )r( r r r rv F rw c C s� |}||j _||d �||d �g}|j|||d�} ||d �||d �g}t|d t�r�|d dkr�t|d t�r�|d dkr�dn|} |j|d g| |d�} | j|d g|dd�} t�| | � d S )Nr> )rk rq r r FT)r rP r rI �strr, r- ) r ro rk rq r r� r� r rf r. Z first_dropr/ r r r �$test_set_index_pass_arrays_duplicate1 s z1TestSetIndex.test_set_index_pass_arrays_duplicaterq c C sV |}t j|d |d gddgd�}|j|||d�}|jddgd|d�}t�||� d S )Nr> rG )r* )rk rq F)r r+ r r, r- )r ro rk rq r rf r. r/ r r r � test_set_index_pass_multiindexm s z+TestSetIndex.test_set_index_pass_multiindexc C s� t �d�}d|_ttj�d�|jd��}|�d�}t � |j |� ttj�d�|d��}|�d�}t � |j |� |� � �d�}t � |j |� d S )Nr" rG )r> rG ) r, ZmakeCategoricalIndexrP r r7 r8 r9 r: r rY r rQ )r �cir rJ r r r �(test_construction_with_categorical_indexz s z5TestSetIndex.test_construction_with_categorical_indexc C s� t dddddgdddddgttd�td �d d �ttd�td �d d �d ��}xJddddgddgddggD ],}|�|��� }|j|jd�}t�||� qfW d S )Nr r r" � � � �" ZabaabZbacF)� categoriesZorderedT)r> rG �C1�C2r� r� r> )r) ) r r rm r rQ Zreindexr) r, r- )r r �colsr. r r r �)test_set_index_preserve_categorical_dtype� s z6TestSetIndex.test_set_index_preserve_categorical_dtypec C s8 t ddddddgddddddgtd�d��}t|�d�d d �|_|j�d �|_tdddgdd �}|�d ��d �}|jdd d�}t � |jj d |� t � |jj d t ddgdd �� |jj ddgks�t�|�dd�}t � |jj d t ddgdd �� t � |jj d |� |jj ddgk�s t�t tj�d��}tddddddgdd�}tddddddgdd�}tdddd�}|�d �}|�|�}|j|d d�}|j|d d�}tdddgdd�}tddgdd�}t � |jj d |� t � |jj d |� t � |jj d |� t � |j�d�|� t � |j�d�|� t � |j�d�|� d S )Nr r1 z2011-07-19 07:00:00z2011-07-19 08:00:00z2011-07-19 09:00:00r )ri r �valuer T)�utcz US/Pacific)rP r< ri )rq r r z US/Eastern)r= z2012-04-01 09:00z2012-04-02 09:00z2011-01-01 09:00z Asia/Tokyo)rM r= r )r rH r �popr r? r Z tz_localizer r, rY �levelsr r* rA Z swaplevelr7 r8 r rR �get_level_values)r r r/ �idx1rc �idx3� expected1� expected2r r r �test_set_index_datetime� sp z$TestSetIndex.test_set_index_datetimec C s t tj�d��}tdddd�}|�|�}tdddd�}|�|��|�}td dd d�}|�|�}|j|d d �}|j|d d �}tdddd�}tdddd�}t�|jj d |� t�|jj d |� t�|jj d |� t�|j� d �|� t�|j� d�|� t�|j� d�|� d S )Nr z 2011-01-01r �M)rM rN z2013-01-01 09:00r rL Z2005r> T)rq r r ) r r7 r8 r rq r r, rY r r� r� )r r r� rc r� r� r� r r r �test_set_index_period� s" z"TestSetIndex.test_set_index_periodN)"�__name__� __module__� __qualname__r r0 r4 r; rC rK rU r] rd re rZ �mark� parametrizerp rr rs rz r r r7 r� rm r� r� r� r� r� r� r� r� r� r r r r r s| ! Rr c @ s� e Zd Zdd� Zej�dddg�ej�dddg�dd� ��Zej�dddg�ej�dddg�ejjd egd gd �d d � ���Z ejjd e e e j edd� gdddddgd �ejjdddgddgd �ej�dddg�ej�dddg�dd� ����Zd S ) �TestSetIndexInvalidc C sd |}t jtdd�� |jddd� W d Q R X t jtdd�� |j|d |d gdd� W d Q R X d S )NzIndex has duplicate keys)rW r> T)Zverify_integrity)rZ r[ r\ r )r ro r r r r � test_set_index_verify_integrity s z3TestSetIndexInvalid.test_set_index_verify_integrityrq TFrk c C s� |}t jtdd�� |jdddg||d� W d Q R X t jtdd��$ |j|d |d dg||d� W d Q R X d }t jt|d�� |jt|d �||d� W d Q R X t jt|d��( |jd|d t|d �g||d� W d Q R X d S ) Nz['foo', 'bar', 'baz'])rW r5 r6 Zbaz)rk rq �Xr> rG z%[('foo', 'foo', 'foo', 'bar', 'bar')])rZ r[ r} r rg )r ro rk rq r r� r r r �test_set_index_raise_keys s ("z-TestSetIndexInvalid.test_set_index_raise_keysr{ �set)Zidsc C sz |}d}t jt|d�� |j||d �||d� W d Q R X t jt|d��( |jd|d ||d �g||d� W d Q R X d S )Nz,The parameter "keys" may be a column key, .*)rW r> )rk rq )rZ r[ � TypeErrorr )r ro r{ rk rq r r� r r r � test_set_index_raise_on_type& s "z0TestSetIndexInvalid.test_set_index_raise_on_typec C s t �| g�S )N)r r+ )r( r r r rv 8 rw z TestSetIndexInvalid.<lambda>r r znp.arrayr� r �lengthr r Z too_shortZtoo_longc C s� |}t j�dd|f�}d}tjt|d�� |j||�||d� W d Q R X tjt|d��" |jd|j||�g||d� W d Q R X d S )Nr r" z<Length mismatch: Expected 5 rows, received array of length.*)rW )rk rq r> )r7 r8 �randintrZ r[ r\ r r> ) r ro r{ r� rk rq r r: r� r r r �test_set_index_raise_on_len6 s z/TestSetIndexInvalid.test_set_index_raise_on_lenN)r� r� r� r� rZ r� r� r� r� r� r r r7 r� r� r� r r r r r� s "r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �TestSetIndexCustomLabelTypec C s� G dd� d�}|dd�}|dd�}t |ddg|d d gi�}t |ddgitd d g|d �d �}|�|�}t�||� |�|g�}t�||� |d d�}d}tjt|d�� |�|� W d Q R X tjt|d�� |�|g� W d Q R X d S )Nc @ s& e Zd Zdd� Zed�dd�ZeZdS )zKTestSetIndexCustomLabelType.test_set_index_custom_label_type.<locals>.Thingc S s || _ || _d S )N)rP �color)r rP r� r r r �__init__V s zTTestSetIndexCustomLabelType.test_set_index_custom_label_type.<locals>.Thing.__init__)�returnc S s dt | j�� d�S )Nz<Thing �>)�reprrP )r r r r �__str__Z s zSTestSetIndexCustomLabelType.test_set_index_custom_label_type.<locals>.Thing.__str__N)r� r� r� r� r� r� �__repr__r r r r �ThingU s r� �One�red�Two�bluer r r r )rP )r �Three�pinkz<Thing 'Three'>)rW )r r r r, r- rZ r[ r} ) r r� �thing1�thing2r r/ r. �thing3r� r r r � test_set_index_custom_label_typeR s z<TestSetIndexCustomLabelType.test_set_index_custom_label_typec C s� G dd� dt �}|ddg�}|ddg�}t|ddg|d d gi�}t|ddgitd d g|d �d �}|�|�}t�||� |�|g�}t�||� |d dg�}d}tjt|d�� |�|� W d Q R X tjt|d�� |�|g� W d Q R X d S )Nc @ s e Zd Zed�dd�ZdS )zXTestSetIndexCustomLabelType.test_set_index_custom_label_hashable_iterable.<locals>.Thing)r� c S s$ t | �}d�tt|��}d|� d�S )Nz, z frozenset({z}))�sorted�joinry r� )r �tmpZ joined_reprsr r r r� � s zaTestSetIndexCustomLabelType.test_set_index_custom_label_hashable_iterable.<locals>.Thing.__repr__N)r� r� r� r� r� r r r r r� s r� r� r� r� r� r r r r )rP )r r� r� z frozenset\(\{'Three', 'pink'\}\))rW ) � frozensetr r r r, r- rZ r[ r} ) r r� r� r� r r/ r. r� r� r r r �-test_set_index_custom_label_hashable_iterablex s zITestSetIndexCustomLabelType.test_set_index_custom_label_hashable_iterablec C s� G dd� dt �}|dd�}|dd�}tddgd d gg||gd �}d }tjt|d �� |�|� W d Q R X tjt|d �� |�|g� W d Q R X d S )Nc @ s" e Zd Zdd� Zed�dd�ZdS )zRTestSetIndexCustomLabelType.test_set_index_custom_label_type_raises.<locals>.Thingc S s || _ || _d S )N)rP r� )r rP r� r r r r� � s z[TestSetIndexCustomLabelType.test_set_index_custom_label_type_raises.<locals>.Thing.__init__)r� c S s dt | j�� d�S )Nz<Thing r� )r� rP )r r r r r� � s zZTestSetIndexCustomLabelType.test_set_index_custom_label_type_raises.<locals>.Thing.__str__N)r� r� r� r� r� r� r r r r r� � s r� r� r� r� r� r r r r )r) z,The parameter "keys" may be a column key, .*)rW )r� r rZ r[ r� r )r r� r� r� r r� r r r �'test_set_index_custom_label_type_raises� s zCTestSetIndexCustomLabelType.test_set_index_custom_label_type_raisesc C s` t tj�d��}tdddd�}tdddd�}|�|�}t�|j|� |�|�}t�|j|� d S )Nr z 2011/01/01r� )rM rN Z2013r> )r r7 r8 r r r, rY r )r r r� rc r r r �test_set_index_periodindex� s z6TestSetIndexCustomLabelType.test_set_index_periodindexc C sd t ddddgi�}d}tjt|d�� |�dd�}W d Q R X t tdddgdd�d �}t�||� d S ) Nr r r r zwIn a future version of pandas all arguments of DataFrame\.set_index except for the argument 'keys' will be keyword-only)rW T)rP )r )r r, Zassert_produces_warning� FutureWarningr r r- )r r r� r. r/ r r r � test_drop_pos_args_deprecation� s z:TestSetIndexCustomLabelType.test_drop_pos_args_deprecationN)r� r� r� r� r� r� r� r� r r r r r� Q s &' r� )�__doc__r r �numpyr7 rZ Zpandasr r r r r r r r r Zpandas._testingZ_testingr, r r� r� r r r r �<module> s , nK
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка