Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexes/multi/__pycache__/test_constructors.cpython-37.pyc
Назад
B �5�g�d � @ s� d dl mZm Z d dlZd dlZd dlZd dlmZ d dlZ d dlm Z m Z m Z m Z mZ d dlmZ dd� Zdd� Zd d � Zd d � Zd d� Zdd� Zdd� Zdd� Zdd� Zej� de j dddd�e j d d dd�fed ddd d �ed d dd!d �fe j d"ddd�e j d#d dd�fg�d$d%� �Z d&d'� Z d(d)� Z!d*d+� Z"ej� d,d-d-gd-d.gd-gd.gd-d.ggd/d/gd/d0gd/gd0gd1d2d-gd.fd-d.gfd/d3d4d/gd0fd1d.gd-d5gd3d0gd6d7d8g�d9d:� �Z#ej� d;d-d.dgd/d0gfg d/d0gfd-d.dgg fg�d<d=� �Z$d>d?� Z%d@dA� Z&dBdC� Z'dDdE� Z(dFdG� Z)dHdI� Z*dJdK� Z+dLdM� Z,dNdO� Z-ej� dPg g fdQdRdSgg fg d/d0dTgfg�dUdV� �Z.ej� dWe/e0dX���dYdZ� �Z1ej� d[d-d-gd-d.gd-gd.gd/d/gd/d0gd/gd0gg�d\d]� �Z2d^d_� Z3d`da� Z4ej� dbdcddg�ej� dedfdg� dhdg� didg� g�djdk� ��Z5dldm� Z6dndo� Z7ej� dpe d-d.dgdQdq�e d/d0gdRdq�dQdRgfe d-d.dgdQdq�d/d0gdQdgfd-d.dgd/d0gdfg�drds� �Z8dtdu� Z9dvdw� Z:dxdy� Z;dzd{� Z<ej� d|e d-d.ddXg�d-d.ddXgd-d.gddXgd}d~gge d-d.ddXg�e�=d-d.gddXgd}d~gg�dg�d�d�� �Z>d�d�� Z?ej� d�dd�d�gfd�d�gd�d�gfg�d�d�� �Z@ej� d�d�d/d0dTgd�fg�d�d�� �ZAd�d�� ZBd�d�� ZCd�d�� ZDd�d�� ZEd�d�� ZFdS )�� )�date�datetimeN)�'construct_1d_object_array_from_listlike)�Index� MultiIndex�Series� Timestamp� date_rangec C sl t ddddggddddggd gd �} t| t �s2t�tddddgd d �}t�| jd |� | jd gksht�d S ) N�foo�bar�baz�quxr � � � �first)�levels�codes�names)�name)r � isinstance�AssertionErrorr �tm�assert_index_equalr r )�result�expected� r �b/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexes/multi/test_constructors.py� test_constructor_single_level s "r c C sz d} t jt| d�� tg g d� W d Q R X d} t jt| d�� tg d� W d Q R X t jt| d�� tg d� W d Q R X d S )Nz non-zero number of levels/codes)�match)r r z Must pass both levels and codes)r )r )�pytest�raises� ValueErrorr � TypeError)�msgr r r �test_constructor_no_levels! s r% c C s� ddgddgg} ddddgddddgg}dgdgf}d}t jt|d �� t| ||d � W d Q R X tddgddggddddgddddggd d �}d gd gg}t jt|d �� |j|d� W d Q R X t jt|d �� |j|d� W d Q R X d S )Nr r �one�twor r r z(MultiIndex\.name must be a hashable type)r )r r r )r r ZfoorZbarr)r )r r! r# r �renameZ set_names)r r r r$ �miZrenamedr r r �"test_constructor_nonhashable_names- s r* c C s� t �dg�t �dg�t �dg�g}dg}d}tjt|d�� t||d� W d Q R X d}d }d }tjt|d��. tdgd ggd dddgd dd dggd� W d Q R X tjt|d��* tdgd ggd d d d gd d ggd� W d Q R X tjt|d�� | �� �dgd gg� W d Q R X tjt|d��$ | �� �d d d d gd d gg� W d Q R X | �� jd d d d gd d ggdd� tjt|d��& tdgd ggd dgd d ggd� W d Q R X d S )Nr r r �az+Length of levels and codes must be the same)r )r r zbOn level 0, code max \(3\) >= length of level \(1\)\. NOTE: this index is in an inconsistent statez Unequal code lengths: \[4, 2\]z"On level 0, code value \(-2\) < -1�br � F)r Zverify_integrity���) �np�arrayr r! r" r �copy� set_levels� set_codes)�idxr r r$ Z length_errorZ label_errorZcode_value_errorr r r �(test_constructor_mismatched_codes_levelsE s& "2. ("r5 c C sV t tjd tjddggddddddggd�} t tjd tjddggddddddggd�}t�| |� t tjd tjdd ggddddddggd�} t tjd tjdd ggddddddggd�}t�| |� t ddddd ggddddddggd��tjd tjdd gg�} t�| |� t tjd tjdd ggddddddggd��ddddddgg�} t�| |� d S ) N� r r ���r r r- )r r �s� ) r r/ �nan�pd�NaTr r r2 r3 )r r r r r �test_na_levelsk s$ (( (( $ (r= c C s� t �dddg�} t �dddddddg�}|d }t| | g||gdd�}|jd d |ks\t�d |d< |jd d |kszt�| d }d | d< |jd d |ks�t�d S ) Nr+ r, �cr r r T)r r r1 � ZPANDA)r/ r0 r r r r )r r �valr) r r r �test_copy_in_constructor� s rA c C s� dd� t | j| j�D �}tj|| jd�}t�|| � t�tj t d�gddgg�}|jd � t t d�g��snt �|jd � t ddg��s�t �d S ) Nc S s g | ]\}}t �|��|��qS r )r/ �asarray�take)�.0�lev� level_codesr r r � <listcomp>� s z$test_from_arrays.<locals>.<listcomp>)r Z20130101r+ r, r r )�zipr r r � from_arraysr r r r; r< r �equalsr r )r4 �arraysr r r r �test_from_arrays� s rL c C sd dd� t | j| j�D �}tjt|�| jd�}t�|| � d}t j t |d�� t�d� W d Q R X d S )Nc S s g | ]\}}t �|��|��qS r )r/ rB rC )rD rE rF r r r rG � s z-test_from_arrays_iterator.<locals>.<listcomp>)r z/Input must be a list / sequence of array-likes.)r r ) rH r r r rI �iterr r r r r! r# )r4 rK r r$ r r r �test_from_arrays_iterator� s rN c C s< t dd� t| j| j�D ��}tj|| jd�}t�|| � d S )Nc s s& | ] \}}t t�|��|��V qd S )N)�tupler/ rB rC )rD rE rF r r r � <genexpr>� s z*test_from_arrays_tuples.<locals>.<genexpr>)r ) rO rH r r r rI r r r )r4 rK r r r r �test_from_arrays_tuples� s rQ )�idx1�idx2z 2011-01-01�Dr )�freq�periodsz 2015-01-01�Hz2015-01-01 10:00z US/Eastern)rU rV �tzz Asia/Tokyoz1 daysz2 hoursc C s| t �| |g�}t�|�d�| � t�|�d�|� t �t| �t|�g�}t�|�d�| � t�|�d�|� t�||� d S )Nr r )r rI r r �get_level_valuesr )rR rS r �result2r r r �=test_from_arrays_index_series_period_datetimetz_and_timedelta� s r[ c C s t ddddd�} t dddd�}tjdddd�}tjd ddd�}t�| |||g�}t�|�d �| � t�|�d �|� t�|�d �|� t�|�d�|� t�t | �t |�t |�t |�g�}t�|�d �| � t�|�d �|� t�|�d �|� t�|�d�|� t�||� d S ) Nz2015-01-01 10:00rT r z US/Eastern)rU rV rX rW )rU rV z1 daysz 2011-01-01r r r ) r r; �timedelta_range� period_ranger rI r r rY r )rR rS Zidx3Zidx4r rZ r r r �)test_from_arrays_index_datetimelike_mixed� s r^ c C s� t jtd�td�dd�} t jtd�td�dd�}t�| |g�}t�|�d�| � t�|�d�|� t�t| �t|�g�}t�|�d�| � t�|�d�|� t�| j |j g�}t�|�d�| � t�|�d�|� d S )N�abcaab�bacF)� categories�orderedTr r ) r; �CategoricalIndex�listr rI r r rY r �values)rR rS r rZ Zresult3r r r �)test_from_arrays_index_series_categorical� s rf c C s� d} t jt| d�� tjg d� W d Q R X tjg gdgd�}t|t�sJt�tg dd�}t� |j d |� |j dgksxt�xZdD ]R}g g| }t d �d |� }tj||d�}tg g| g g| |d �}t� ||� q~W d S ) Nz)Must pass non-zero number of levels/codes)r )rK �A)rK r )r r )r r �ABC)r r r ) r r! r" r rI r r r r r r r rd )r$ r r �NrK r r r r �test_from_arrays_empty s rj �invalid_sequence_of_arraysr r r+ r, )r )r r )r+ )r+ r, )r ))r r )r )r ))r+ r, c C s. d}t jt|d�� tj| d� W d Q R X d S )Nz.Input must be a list / sequence of array-likes)r )rK )r r! r# r rI )rk r$ r r r � test_from_arrays_invalid_input# s rl z idx1, idx2c C s0 d}t jt|d�� t�| |g� W d Q R X d S )Nz ^all arrays must be same length$)r )r r! r" r rI )rR rS r$ r r r �"test_from_arrays_different_lengthsE s rm c C st t dddgdd�} t dddgd d�}tj| |gd d �}tdddgdddggd ddgd ddggd d �}t�||� d S ) Nr r r r )r r+ r, r> r )r r )r r r )r r rI r r )r+ r, r r r r r �$test_from_arrays_respects_none_namesO s ,rn c C st d} t jt| d�� t�g � W d Q R X tddgddggddgddggdd gd �}tjd dd gd �}t�||� d S ) Nz-Cannot infer number of levels from empty list)r r r r r- r r+ r, )r r r ))r r )r r- )r )r r! r# r � from_tuplesr r )r$ r r r r r �test_from_tuples_ s (rp c C s� t ddgddggddgddggddgd�} t jtddgddg�ddgd �}t�|| � d }tjt|d �� t �d� W d Q R X d S ) Nr r r r- r r+ r, )r r r )r z/Input must be a list / sequence of tuple-likes.)r )r ro rH r r r r! r# )r r r$ r r r �test_from_tuples_iteratorm s ( rq c C s8 t jg ddgd�} t jg g gddgd�}t�| |� d S )Nr+ r, )r )rK r )r ro rI r r )r r r r r �test_from_tuples_empty} s rr c C s" t �| �}|j| jk�� s t�d S )N)r ro re �allr )r4 r r r r � test_from_tuples_index_values� s rt c C s^ dddg} d}t jt|d�� t| dd� W d Q R X t jt|d�� t| dd� W d Q R X d S ) N)r r r )r r r )r r r z*Names should be list-like for a MultiIndex)r �abc)r r+ )r r! r" r )Zlir$ r r r � test_tuples_with_name_string� s rv c C sf t jdddgdddggdddgd ��ddg�} tjd d gd d �}t jddgdg|d�}t�| |� d S )Nr r r- )r r r r+ r, r> )�columns)r r )r- )r r )r+ r, )r )rw �index)r; � DataFrameZ set_indexr ro r Zassert_frame_equal)r r4 r r r r �!test_from_tuples_with_tuple_label� s rz c C s, d} t jt| d�� t�g � W d Q R X d S )Nz)Must pass non-zero number of levels/codes)r )r r! r" r � from_product)r$ r r r �#test_from_product_empty_zero_levels� s r| c C sD t jg gdgd�} tg dd�}t�| jd |� | jdgks@t�d S )Nrg )r )r r )r r{ r r r r r r )r r r r r �!test_from_product_empty_one_level� s r} z first, secondr r r r> c C s@ ddg}t j| |g|d�}t | |gg g g|d�}t�||� d S )Nrg �B)r )r r r )r r{ r r )r �secondr r r r r r �"test_from_product_empty_two_levels� s r� ri r- c C sT dddg}t t| ��}tjg |g g|d�}tg |g gg g g g|d�}t�||� d S )Nrg r~ �C)r )r r r )rd �ranger r{ r r )ri r Zlvl2r r r r r �$test_from_product_empty_three_levels� s r� � invalid_inputc C s. d}t jt|d�� tj| d� W d Q R X d S )NzDInput must be a list / sequence of iterables|Input must be list-like)r )� iterables)r r! r# r r{ )r� r$ r r r � test_from_product_invalid_input� s r� c C s` t ddd�} t�ddg| g�}tdtd�fdtd�fdtd�fdtd�fg�}t�|j|� d S )Nz 2000-01-01r )rV r z 2000-01-02)r r r{ r r r Zassert_numpy_array_equalre )Zdt_indexr) Zetalonr r r � test_from_product_datetimeindex� s r� c C s< t td��} ddg}t�| |g�}tj|jd | dd� d S )Nr9 r+ r, r T)�exact)r r� r r{ r r Z_levels)�rng�otherr) r r r � test_from_product_rangeindex� s r� rb FT�fc C s | S )Nr )�xr r r �<lambda>� � r� c C s t | �S )N)r )r� r r r r� � r� c C s | j S )N)re )r� r r r r� � r� c C sh ddg}t jtd�td�| d�}t jtd�td� td�| d�}t�|||�g�}t�|�d�|� d S )Nr r r_ r` )ra rb r )r; rc rd r r{ r r rY )rb r� r r4 r r r r r �*test_from_product_index_series_categorical� s r� c C sb dddg} dddg}ddg}t j| |g|d �}d d d d dddddg }t j||d �}t�||� d S )Nr r �buzr+ r, r> r r )r )r r+ )r r, )r r> )r r+ )r r, )r r> )r� r+ )r� r, )r� r> )r r{ ro r r )r r r r �tuplesr r r r �test_from_product� s r� c C s� dddg} dddg}ddg}d d d d d ddddg }t j||d�}t jt| |g�|d�}t�||� d}tjt|d�� t �d� W d Q R X d S )Nr r r� r+ r, r> r r )r r+ )r r, )r r> )r r+ )r r, )r r> )r� r+ )r� r, )r� r> )r z-Input must be a list / sequence of iterables.)r r ) r ro r{ rM r r r r! r# )r r r r� r r r$ r r r �test_from_product_iterator s$ r� za, b, expected_names)r c C sV t �| |g�}t dddgddggddddddgddddddgg|d�}t�||� d S )Nr r r r+ r, r )r r r )r r{ r r )r+ r, Zexpected_namesr r r r r � test_from_product_infer_names$ s r� c C s| t dddgdd�} t ddgdd�}tj| |gd d �}tdddgddggd d ddddgd dd dd dggd d �}t�||� d S ) Nr r r r )r r+ r, r )r r )r r r )r r r{ r r )r+ r, r r r r r �%test_from_product_respects_none_names; s r� c C sN t �td��} ddg}t�| |g�}| jdd� t�| |g�}t�||� d S )Nr r+ r, F)�write)r/ r0 r� r r{ Zsetflagsr r )r+ r, r r r r r �test_from_product_readonlyI s r� c C sz | }ddg|_ t|�}ttddddddgd d ��}t�||� t|d d �}ttddddddgd d �d d �}t�||� d S ) Nr r )r r& )r r' )r r& )r r' )r r& )r r' �object)�dtyperg )r )r r r r )r4 rx r r r r r � test_create_index_existing_nameT s2 r� c C s^ t jddgddgddgddggddgd�} tjdddd gddgd �}t�| �}t�||� d S ) Nr+ r, �L1�L2)rw )r+ r+ )r+ r, )r, r+ )r, r, )r )r; ry r ro � from_framer r )�dfr r r r r �test_from_frame s & r� � non_framer9 � � c C s( t jtdd�� t�| � W d Q R X d S )NzInput must be a DataFrame)r )r r! r# r r� )r� r r r �test_from_frame_error� s r� c s� t �tdddd�ddddddgt jddddd d gd d �d d d dd d gd��} | j�� }tjtdddd�ddddddgt jddddd d gd d �d d d dd d ggdddd gd�}t�| �� � fdd�t � j �D �}t � |� � ||ks�t �d S )NZ19910905r� z US/Eastern)rV rX r r r+ r, r> T)rb r� �y�z)�datesr+ r, r> r� )r c s i | ]\}}� j | j|�qS r )r r� )rD �ir )r) r r � <dictcomp>� s z2test_from_frame_dtype_fidelity.<locals>.<dictcomp>)r; ry r Z CategoricalZdtypes�to_dictr rI r� � enumerater r r r )r� Zoriginal_dtypesZ expected_miZ mi_dtypesr )r) r � test_from_frame_dtype_fidelity� s r� znames_in,names_out)r� r� )r� r� r� r� c C sP t jddgddgddgddggt�ddg�d�}tj|| d�}|j|ksLt�d S )Nr+ r, )r� r� )r� r� )rw )r )r; ry r ro r� r r )Znames_inZ names_outr� r) r r r �test_from_frame_valid_names� s r� znames,expected_error_msg)Z bad_inputz*Names should be list-like for a MultiIndexz9Length of names must match number of levels in MultiIndexc C s\ t jddgddgddgddggt�ddg�d�}tjt|d�� tj|| d� W d Q R X d S )Nr+ r, )r� r� )r� r� )rw )r )r )r; ry r ro r r! r" r� )r Zexpected_error_msgr� r r r � test_from_frame_invalid_names� s r� c C s@ t g g gg g gddgd�} t jg g gddgd�}t�| |� d S )Nr+ r, )r r r )rK r )r rI r r )r+ r, r r r � test_index_equal_empty_iterable� s r� c C s� ddgdddgg} t | ddddddgddddddggdd� tjtdd��0 t | ddddddgddddddggdd� W d Q R X tjtdd��0 t | ddddddgddddddggdd� W d Q R X d S )Nr r r )r r Z sortorderz%.* sortorder 2 with lexsort_depth 1.*)r z%.* sortorder 1 with lexsort_depth 0.*)r r r! r" )r r r r � test_raise_invalid_sortorder� s (2r� c C s: t jdddgd dd�} tddd d d �}t�| |g�}t jdddgdd�}t�|jd |� t�|jd |� t� d�}t � � }t � � }xbt �|||g|||g�D ]F\}}t�|g|gg�} t| jd t j�s�t�t| jd t j�s�t�q�W t� � } t�| g|gg�} t| jd t j��r t�t| jd t j��s6t�d S )Nz2013-04-01 9:00z2013-04-02 9:00z2013-04-03 9:00r z Asia/Tokyo)rX z 2010/01/01r� �Mz US/Eastern)rV rU rX r r �today)r; � DatetimeIndexr r rI r r r r/ Z datetime64r r� r � itertools�productr{ r r r ) rR rS r4 Z expected1Zdate1Zdate2Zdate3Zd1Zd2rx Zdate4r r r �test_datetimeindex� s& "r� c C s� t jddgddd�} t jddgdd d�}t�| |g�}|jddgksHt�t�|jd | � t�|jd |� t�t | �t |�g�}|jddgks�t�t�|jd | � t�|jd |� d S ) Nz2013/01/01 09:00z2013/01/02 09:00Zdt1z US/Pacific)r rX z2014/01/01 09:00z2014/01/02 09:00Zdt2z Asia/Tokyor r ) r; r� r rI r r r r r r )rx rw r r r r �test_constructor_with_tz s r� c C s� t �� } | | g}t|�}|jtks&t�t�|g�}|jd }|jtksJt�t� |g�}|jd }|jtksnt�t� dd� |D ��}|jd }|jtks�t�d S )Nr c S s g | ] }|f�qS r r )rD r� r r r rG ; s z9test_multiindex_inference_consistency.<locals>.<listcomp>) r r� r r� r� r r rI r r{ ro )�vZarrr4 r) rE r r r �%test_multiindex_inference_consistency) s r� )Gr r r� �numpyr/ r Zpandas.core.dtypes.castr Zpandasr; r r r r r Zpandas._testingZ_testingr r r% r* r5 r= rA rL rN rQ �markZ parametrizer] r\ r[ r^ rf rj rl rm rn rp rq rr rt rv rz r| r} r� rd r� r� r� r� r� r� r� r� r� r� r� r� r� r0 r� r� r� r� r� r� r� r� r� r r r r �<module> s� & 6 . 8* + (
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка