Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexes/multi/__pycache__/test_get_set.cpython-37.pyc
Назад
B �5�gP@ � @ s& d dl Zd dlZd dlmZ d dlZd dlmZmZ d dl m Z d.dd�Z dd� Z d d � Zd d � Zd d� Zdd� Zdd� Zdd� Zdd� Zdd� Zej�dddg�dd � �Zd d � Zej�d ddg�d d!� �Zd"d#� Zd$d%� Z ej�dddg�d&d'� �Z ej�dddg�d(d)� �Z d*d+� Z d,d-� Z dS )/� N)�DatetimeTZDtype)�CategoricalIndex� MultiIndexFc C sT t | �t |�kst�x:t| |�D ],\}}t�|�}t�|�}tj|||d� q W d S )N)� check_dtype)�len�AssertionError�zip�npZasarray�tmZassert_numpy_array_equal)�actual�expectedr Zact�exp� r �]/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexes/multi/test_get_set.py�assert_matching s r c C s~ ddg| _ | �d�dks t�| �d�dks.t�d}tjt|d�� | �d� W d Q R X tjtdd�� | �d� W d Q R X d S )N� r z/Too many levels: Index has only 2 levels, not 3)�match� zLevel fourth not foundZfourth)�names�_get_level_numberr �pytest�raises� IndexError�KeyError)�idx�msgr r r � test_get_level_number_integer s r c C sh t jdddgdddgtjdddd �gd d d gd �} t�t�d�t�d�tdd�d��}t� || j � d S )Nr r � �a�b�c�20200101�UTC)�periods�tz�int�string�dt)r �int64�O�utc)r$ )r% r& r' ) r � from_product�pd� date_range�Seriesr �dtyper r �assert_series_equal�dtypes)� idx_multityper r r r �test_get_dtypes# s r3 c C s^ t �dddgdddgtjdddd �g�} t�t�d �t�d �td d �d��}t� || j � d S )Nr r r r r r r! r" )r# r$ r( r) r* )r$ )Zlevel_0Zlevel_1Zlevel_2) r r+ r, r- r. r r/ r r r0 r1 )r2 r r r r � test_get_dtypes_no_level_name3 s r4 c C sT | }t jtdd�� |j�d� W d Q R X t jtdd�� |j�d� W d Q R X d S )NzToo many levels)r r znot a valid level number���)r r r �indexr )Z multiindex_dataframe_random_data�framer r r �#test_get_level_number_out_of_boundsF s r8 c C s� | j | jkst�dd� |D �}| �|�}| j|ks6t�|j|ksDt�d}tjt|d�� |�|| � W d Q R X dd� |D �}|j|dd�}|d ks�t�|j|ks�t�| j|d dd �}| j|ks�t�|j|d |d gks�t�|j|d ddd �}|d k�st�|j|d |d gk�s t�| j|dd gd �}| j|k�s@t�|j|k�sPt�|j|dd gdd �}|d k�srt�|j|k�s�t�d S ) Nc S s g | ] }|d �qS )ZSUFFIXr )�.0�namer r r � <listcomp>R s z)test_set_name_methods.<locals>.<listcomp>z9Length of names must match number of levels in MultiIndex)r c S s g | ] }|d �qS )ZSUFFIX2r )r9 r: r r r r; Y s T)�inplacer )�levelr )r= r< )�rename� set_namesr r r r � ValueError)r Z index_namesZ new_names�indr Z new_names2�resr r r �test_set_name_methodsO s0 rC c C s� | j }dd� |D �}| j}|\}}dd� |D �}dd� |D �}||g}d}tjt|d�� || _ W d Q R X tjt|d�� || _W d Q R X d S )Nc S s g | ]}d d� |D ��qS )c S s g | ] }|d �qS )r r )r9 �levr r r r; u s z=test_set_levels_codes_directly.<locals>.<listcomp>.<listcomp>r )r9 r= r r r r; u s z2test_set_levels_codes_directly.<locals>.<listcomp>c S s g | ]}|d d �qS )r r r )r9 �xr r r r; y s c S s g | ]}|d d �qS )r r )r9 rE r r r r; z s z[Cc]an't set attribute)r )�levels�codesr r �AttributeError)r rF � new_levelsrG � major_codes� minor_codes� new_codesr r r r � test_set_levels_codes_directlyq s rM c C s� | j }dd� |D �}| �|�}t|j |� t| j |� | �� }t�t�� |j|dd�}W d Q R X |d ksnt�t|j |� | j|d dd�}t|j |d |d g� t| j |� | j|d dd�}t|j |d |d g� t| j |� | j|ddgd�}t|j |� t| j |� | �� }t�t�� |j|d ddd�}W d Q R X |d k�sPt�t|j |d |d g� t| j |� | �� }t�t�� |j|d ddd�}W d Q R X |d k�s�t�t|j |d |d g� t| j |� | �� }t�t�� |j|ddgdd�}W d Q R X |d k�st�t|j |� t| j |� | �� }�xfd D �]\}tj t d d ��. t�t�� | jd gd|d� W d Q R X W d Q R X t| j |j dd � tj t d d ��8 t�t��" | j ddddddgd|d� W d Q R X W d Q R X t| j |j dd � tj t dd ��, t�t�� | jd d|d� W d Q R X W d Q R X t| j |j dd � tj t dd ��, t�t�� | j dd|d� W d Q R X W d Q R X t| j |j dd � �q@W d S )Nc S s g | ]}d d� |D ��qS )c S s g | ] }|d �qS )r r )r9 rD r r r r; � s z.test_set_levels.<locals>.<listcomp>.<listcomp>r )r9 r= r r r r; � s z#test_set_levels.<locals>.<listcomp>T)r< r )r= r )r= r< )TFz^On)r r )r r r � � z^Levelsz^Codes)rF � set_levelsr �copyr �assert_produces_warning� FutureWarningr r r r@ � set_codesrG � TypeError)r rF rI �ind2�inplace_returnZoriginal_indexr< r r r �test_set_levels� sn & 0 $ $rX c C s� | j }|\}}dd� |D �}dd� |D �}||g}| �|�}t|j |� t| j |� | �� }t�t�� |j|dd�}W d Q R X |d ks�t�t|j |� | j|d dd�}t|j |d |d g� t| j |� | j|d dd�}t|j |d |d g� t| j |� | j|ddgd�}t|j |� t| j |� | �� }t�t�� |j|d ddd �}W d Q R X |d k�snt�t|j |d |d g� t| j |� | �� }t�t�� |j|d ddd �}W d Q R X |d k�s�t�t|j |d |d g� t| j |� | �� }t�t�� |j|ddgdd �}W d Q R X |d k�s6t�t|j |� t| j |� t� d d� t d �D ��}t d d d �}t� dd� |D ��}|j|dd�} | � |��s�t�|�� } t�t�� | j|ddd� W d Q R X | � |��s�t�d S )Nc S s g | ]}|d d �qS )r r r )r9 rE r r r r; � s z"test_set_codes.<locals>.<listcomp>c S s g | ]}|d d �qS )r r )r9 rE r r r r; � s T)r< r )r= r )r= r< c S s g | ] }d |f�qS )r r )r9 �ir r r r; s � � ���c S s g | ] }d |f�qS )r r )r9 rY r r r r; s )rG r= )rG r= r< ) rG rT r rQ r rR rS r r � from_tuples�range�equals) r rG rJ rK rL rV rW rA r �resultr r r �test_set_codes� sb ra c C s | j | j }}| j}tjtdd�� | �|d g� W d Q R X tjtdd�� | �|d g� W d Q R X tjtdd�� | �|d g� W d Q R X tjt dd�� | �|d � W d Q R X tjt dd�� | �|d � W d Q R X tjt dd�� | �|d � W d Q R X tjt dd�� | j|d ddgd � W d Q R X tjt dd�� | j|dd � W d Q R X tjt dd�� | j|d ddgd � W d Q R X tjt dd�� | j|dd � W d Q R X tjtdd�� | j|d ddgd � W d Q R X tjt d d�� | j|dd � W d Q R X d S ) NzLength of levels)r r zLength of codeszLength of nameszlist of lists-likez list-liker )r= zNames must be a) rF rG r r r r@ rP rT r? rU )r rF rG r r r r �%test_set_levels_codes_names_bad_input s4 rb r< Tc C sT t ddggddggdgd�}t �ddgg�}|jdd| d�}| rD|}t�||� d S )Nr r �first)rF rG r )r= r< )r r+ r? r �assert_index_equal)r< r �mr` r r r � test_set_names_with_nlevel_1M s rf c C s� t �ddgddgg�} d}tjt|d�� | �ddgd �}W d Q R X t ddgddggd d d d gd d d d ggddgd �}t�||� d S ) N�pythonZcobrai� i� zxIn a future version of pandas all arguments of MultiIndex.set_names except for the argument 'names' will be keyword-only)r �kind�yearr r )rF rG r )r r+ r rR rS r? rd )r r r` r r r r �)test_multi_set_names_pos_args_deprecation\ s rj �orderedc C s� t �td�ddddgg�}ttd�| d�}|j|dd�}t |ddddgg|jd �}t�||� |�d�}ttd �|j |j d �}t�||� d S ) NZxyzxr r r r Zbac)rk )r= )rF rG Zbacb)� categoriesrk ) r � from_arrays�listr rP rG r rd Zget_level_valuesrl rk )rk r6 Zcidxr` r Z result_lvlZ expected_lvlr r r �test_set_levels_categoricalm s ro c C s� ddddddg} dddgd }t j| |gddgd �}tjtj�d d �d d ddg|d�}|�� }|jd kslt �|j j dks|t �d|j d< |jd ks�t �|j j dks�t �d S )NZhans�grethe�1�2�3r �Name�Number)r � rN �one�two�three�four)�columnsr6 )rt ru g�(\���X@))rp �4rw ) r rm r, Z DataFramer �randomZrandnZ sort_indexZ_is_copyr r6 r �at)Zlev1Zlev2r Zdfr r r �test_set_value_keeps_names~ s r c C st dddg} dgd }t j| |gddgd�}|jttdd d g�dd �}dddg}t j||gddgd�}t�||� d S ) Nr r r �black�size�color)r rs rr rq )r= )r rm rP �mapr% r rd )�sizes�colorsr6 r` Zexpected_sizesr r r r � test_set_levels_with_iterable� s r� c C s> | j d d d d� }t�t�� | j|d|d� W d Q R X d S )Nr r\ )rG r= r< )rG r rR rS rT )r r< rL r r r �!test_set_codes_inplace_deprecated� s r� c C s8 | j d �� }t�t�� | j|d|d� W d Q R X d S )Nr )rF r= r< )rF rQ r rR rS rP )r r< Z new_levelr r r �"test_set_levels_inplace_deprecated� s r� c C sp t jdddgddgd�} d}tjt|d�� | �d d d gd �}W d Q R X t jd ddgddgd�}t�||� d S )N)r rw )r rw )r rw �foo�bar)r zzIn a future version of pandas all arguments of MultiIndex.set_levels except for the argument 'levels' will be keyword-only)r r r r r )r rw )r rw )r rw )r r] r rR rS rP rd )r r r` r r r r �$test_set_levels_pos_args_deprecation� s r� c C sx d}t jt|d��2 | �ddddddgddddddggddg�}W d Q R X tjddd d d d gd dgd�}t �||� d S )NzxIn a future version of pandas all arguments of MultiIndex.set_codes except for the argument 'codes' will be keyword-only)r r r r r )r� rw )r� rx )r� rw )Zbazrx )�quxrw )r� rx rc �second)r )r rR rS rT r r] rd )r r r` r r r r �$test_set_codes_pos_args_depreciation� s 6 r� )F)!�numpyr r Zpandas.core.dtypes.dtypesr Zpandasr, r r Zpandas._testingZ_testingr r r r3 r4 r8 rC rM rX ra rb �markZ parametrizerf rj ro r r� r� r� r� r� r r r r �<module> s0 "PJ/
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка