Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/reshape/concat/__pycache__/test_concat.cpython-37.pyc
Назад
B �5�gZX � @ s� d dl mZmZ d dlmZ d dlmZ d dlZd dl Z d dl m m Z d dlZd dlmZmZmZmZmZmZmZ d dlmZ d dlmZ d dl m Z d dl m Z G d d � d �Z e j!�"d eeg�e j!�"d ej#d �dd� ��Z$e j!�"d e eg�e j!�"d ej#d �dd� ��Z%dd� Z&dd� Z'dd� Z(e j!�"ddddgdddgg�d d � �Z)e j!�"d e�*d e�+d d �i�ej,e�+d d �d d!�g�d"d#� �Z-d$d%� Z.d&d'� Z/e j!�"d(d)gd* d+fd)gd* d,fd)d-d)gd,fd)d-d)gd+fg�d.d/� �Z0d0d1� Z1d2d3� Z2d4d5� Z3dS )6� )�abc�deque)�Decimal)�catch_warningsN)� DataFrame�Index� MultiIndex� PeriodIndex�Series�concat� date_range)� SparseArray)�!create_series_with_explicit_dtype)� to_decimalc @ s� e Zd Zdd� Zejdd� �Zdd� Zdd� Ze j � d d d g�d d � �Z d d� Z dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Z d-d.� Z d/S )0�TestConcatenatec C s� t dddd�}t dddd�}ttj�d�|�}ttj�d�|�}|�� }|�� }t||g�}t|jt �slt �|jd |jd ks�t �d S ) Nz 12/31/1990z 12/31/1999zA-DEC)�freqz 12/31/2000z 12/31/2009� r ) r r �np�random�randnZ to_periodr � isinstance�indexr �AssertionError)�selfZd1Zd2�s1�s2�result� r �]/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/reshape/concat/test_concat.py�test_append_concat s z"TestConcatenate.test_append_concatc C s� t tj�dd��}t tjjdddd��dd��}t dditd�d �}t|||gdd d �}x |jj D ]}|j j d ksbt �qbW t|||gdd d �}xz|jj D ]n}|j jd kr�|j j |jj d j j ks�t �q�|j jdkr�|j j |jj d j j ks�t �q�|jr�|j j d k s�t �q�W t tj�dd��}t||||gdd d �}x||jj D ]p}|j jd k�rb|j j d k�s�t �nH|j jdk�r�|j j |jj d j j k�s�t �n|j�r<|j j d k �s<t ��q<W d S )N� � r r )�size� � �foo)r T)�axis�copyF�f)�i�u)r r r r �randint�reshape�ranger Z_mgr�blocks�values�baser �dtype�kindZ is_object)r �df�df2Zdf3r �bZdf4r r r �test_concat_copy/ s. z TestConcatenate.test_concat_copyc C s� t tj�dd��}t tj�dd��}t tj�dd��}t tj�dd��}t||gddgd�}t�dddddddgdddddddgg�}t tj|j|jf |d�}t � ||� t||gddgd�}t�ddddddgddddddgg�}t tj|j|jf |d�}t � ||� t tj�dd��}t tj�dd��}t||gddgdd�}t tj |j|jf |d �}t � ||� t||gddgdd�}t tj |j|jf |d �}t � ||� d S ) Nr r! r r# )�keys� )r )r7 r& )�columns) r r r r r r Z from_arraysZr_r/ �tm�assert_frame_equalZc_)r r3 r4 r � exp_index�expectedZ exp_index2r r r �test_concat_with_group_keysQ s* & & z+TestConcatenate.test_concat_with_group_keysc C s� t tj�dd��}|jd d �ddgf |jd d �dgf |jd d �dgf g}ddd d g}t|dd ddg|gd gd �}t�|jj d t |d d �� t�|jj d t ddddg�� |jj d d gks�t �d S )Nr r r r# r8 r! �three�two�one�zeroZ group_key)r& r7 �levels�names)�name) r r r r �ilocr r: �assert_index_equalr9 rC r rD r )r r3 �pieces�levelr r r r � test_concat_keys_specific_levelsq s < z0TestConcatenate.test_concat_keys_specific_levels�mapping�dictc s |dkr t n|}|ttj�dd��ttj�dd��ttj�dd��ttj�dd��d��� t� �� �}t� �}t� fdd�|D �|d�}t� ||� t� dd �}t� fd d�|D �|dd �}t� ||� d d dg}t� |d�}t� fdd�|D �|d�}t� ||� d S )NrL r r! )r% �bar�bazZquxc s g | ] }� | �qS r r )�.0�k)�framesr r � <listcomp>� s z7TestConcatenate.test_concat_mapping.<locals>.<listcomp>)r7 r# )r& c s g | ] }� | �qS r r )rO rP )rQ r r rR � s )r7 r& rN r% rM c s g | ] }� | �qS r r )rO rP )rQ r r rR � s ) rL r r r r �listr7 r r: r; )r rK Znon_dict_mapping_subclass� constructorZ sorted_keysr r= r7 r )rQ r �test_concat_mapping� s" z#TestConcatenate.test_concat_mappingc C s> t tj�dd��}t tj�dd��}ddgddgg}dd g}t||||gd d d d g||d�}t||||g�}t|dgg ddddgddddgddddgg|d g d�}||_t�||� t||||gd d d d g|d�}|jj dks�t �t||||gd d d d gdd gd�}|jj dk�st �t� |jj d t ddgdd�� d S )Nr# r! r r% rN rA r@ �first�second)r% rA )r% r@ )rN rA )rN r@ )r7 rC rD r )rC �codesrD )r7 rC )NNN)r7 rD )rV rW N)rE )r r r r r r r r: r; rD r rG rC r )r r3 r4 rC rD r r= r< r r r �test_concat_keys_and_levels� s8 z+TestConcatenate.test_concat_keys_and_levelsc C s� t tj�dd�dgd�}t tj�dd�dgd�}d}tjt|d��$ t||gd d gd d d ggd� W d Q R X d}tjt|d��$ t||gd d gd d d ggd� W d Q R X d S )Nr# r! �a)r r r5 z Values not found in passed level)�matchrA r@ r% rM rN )r7 rC zKey one not in level)r r r r �pytest�raises� ValueErrorr )r r3 r4 �msgr r r �"test_concat_keys_levels_no_overlap� s (z2TestConcatenate.test_concat_keys_levels_no_overlapc C s` ddddg}t tjddddgd d �tjddddgd d �tjddddgd d �tjddddgd d �d �|d �}t tjddddgd d �tjddddgd d �tjddddgd d �tjddddgd d �d �|d �}|j|dd�}t tj|j|jgdd�|d �}t�||� t tj� dd�dgd�}t tj� dd�dgd�}t ||gddgddgd�}|j j dk�s\t �d S )N�A�B�C�Dr# r8 r! r Zf8)r1 �i8)ra rb rc rd )r9 T)� ignore_indexr )r& rZ )r r5 rA r@ rV rW )r7 rD )rV rW )r r �array�appendZ concatenater/ r: r; r r r r rD r )r r9 �df1r4 Zappendedr= r3 r r r r � test_crossed_dtypes_weird_corner� s* z0TestConcatenate.test_crossed_dtypes_weird_cornerc C s@ t ddd�td�d�}t ddd�td�d�}t||g|d� d S )Nr% rM )ra )rb r# r8 )r )rb )rb r# )�sort)r r- r )r rk ri r4 r r r �test_with_mixed_tuples� s z&TestConcatenate.test_with_mixed_tuplesc C s� t dddd�}tjddd�}t||d�}t||d�}t|�dd �|d�}tt�|d ��dd �|d d gd �}t||gd d �}t� ||� tt�|d ��dd �|d d gd �}t||gd d �}t� ||� tt�|d��dd�|d d d gd �}t|||gd d �}t� ||� tt�|d��dd�|d d d d dgd �}t|||||gd d �}t� ||� d|_ tt�|d��dd�|dd d gd �}t|||gd d �}t� ||� d|_ tt�|d��dd�|dd dgd �}t|||gd d �}t� ||� tt�|d��dd�|d d d gd �}t|||gd dd�}t� ||� tt� |d��dd �|� � d d gd �}t|||g�}t� ||� tt� |d��dd �d gd�}t|||gdd�}t� ||� d S )Nz 01-Jan-2013r �H)Zperiodsr �int64)r1 )r ���r# r8 r )r r9 )r& r! r$ r% rM T)r& rf )r9 )rf ) r r �aranger r r, �repeatr r: r; rE Ztile�tolist)r r Zarrr r r3 r= r r r r �test_concat_mixed_objs� sT " & " " " & z&TestConcatenate.test_concat_mixed_objsc C s� t dt�d��d�tjgi�}t|jdg |jdg g�}t�|j |j � dd l }t d|� ddd�|� ddd�gi�}t|jdg |jdg g�}t�|j |j � t d d gd gd i�}t|jdg |jdg g�}t�|j |j � d S ) N�dateZ20130101�UTCr r# i� i� r8 �textz some words� ) r �pd� TimestampZ tz_localizeZNaTr rF r: �assert_series_equalZdtypes�datetime)r r3 r r{ r r r �test_dtype_coerceionB s " z$TestConcatenate.test_dtype_coerceionc C sN t tj�dd��}t|gdgd�}t||gddgd�}t�||d d� � d S )Nr r r% )r7 rM )r r r r r r: r; )r r3 r r= r r r �test_concat_single_with_keyX s z+TestConcatenate.test_concat_single_with_keyc C s& t jtdd�� tg � W d Q R X d S )NzNo objects to concatenate)r[ )r\ r] r^ r )r r r r �test_concat_no_items_raises_ s z+TestConcatenate.test_concat_no_items_raisesc C sl t tj�dd��}|d d� d d |dd � g}t|�}t�||� tjt dd�� td d g� W d Q R X d S )Nr r r$ z All objects passed were None)r[ ) r r r r r r: r; r\ r] r^ )r r3 rH r r r r �test_concat_exclude_nonec s z(TestConcatenate.test_concat_exclude_nonec C s� t dddgdddgdddgg�}td ||d d� |d d� |d��}t||d d� |d d� |d��}t�||� td ||d d� |d d� |gdd d d d gd �}t||d d� |d d� |gd d d d gd �}t�||� d S )Nr � � r8 r# )rZ r5 �c�d�e)r5 r� r� r� rZ r5 r� r� r� )r7 )r r r: r; )r Zdf0r r= r r r �test_concat_keys_with_nonel s $" .,z*TestConcatenate.test_concat_keys_with_nonec C sV t �� }t �� d d d� }t||gddd�}t||gddd�}t|�t|�ksRt�d S )Nr8 �outerr# )�joinr& )r: ZmakeTimeSeriesr �lenr )r �ts1Zts2�left�rightr r r �test_concat_bug_1719z s z$TestConcatenate.test_concat_bug_1719c C s` t t�d��}t t�d��}d |_|_t||gdd�}t||d��}ddg|_t� ||� d S )Nr$ z same namer# )r& )r r# ) r r ZzerosZonesrE r r r9 r: r; )r Zts0r� r r= r r r �test_concat_bug_2972� s z$TestConcatenate.test_concat_bug_2972c C s� t ddddgddddgddddgd��}t ddddgd d d d gddddgd ��}t ddddd dgddddd dgddddd dgddddd dgg�}ddddddg|_t||gd d�}t�||� d S )Nr � Zrrr)�firmNo�prc� stringvarrw r � � r# r8 r! r )rc �miscr� r� r� r� rc r� )r& )r r9 r r: r; )r ri r4 r= r r r r �test_concat_bug_3602� s &z$TestConcatenate.test_concat_bug_3602c s� t dddg�� t dddg��t ddddddg�}t�t� �fdd�|� t�t� �gdd�|� t�td d � � �fD �dd�|� t�tt� �f�dd�|� G � �fd d �d �}t�t|� dd�|� G � �fd d�dtj�}t�t|� dd�|� d S )Nr# r8 r! r r$ r� T)rf c s s | ] }|V qd S )Nr )rO r3 r r r � <genexpr>� s z8TestConcatenate.test_concat_iterables.<locals>.<genexpr>c s( e Zd Zed�dd�Z� �fdd�ZdS )z>TestConcatenate.test_concat_iterables.<locals>.CustomIterator1)�returnc S s dS )Nr8 r )r r r r �__len__� s zFTestConcatenate.test_concat_iterables.<locals>.CustomIterator1.__len__c s< y� �d�| S t k r6 } z t|�W d d }~X Y nX d S )N)r r# )�KeyError� IndexError)r r �err)ri r4 r r � __getitem__� s zJTestConcatenate.test_concat_iterables.<locals>.CustomIterator1.__getitem__N)�__name__� __module__� __qualname__�intr� r� r )ri r4 r r �CustomIterator1� s r� c s e Zd Z� �fdd�ZdS )z>TestConcatenate.test_concat_iterables.<locals>.CustomIterator2c 3 s � V �V d S )Nr )r )ri r4 r r �__iter__� s zGTestConcatenate.test_concat_iterables.<locals>.CustomIterator2.__iter__N)r� r� r� r� r )ri r4 r r �CustomIterator2� s r� )r r: r; r r r �Iterable)r r= r� r� r )ri r4 r �test_concat_iterables� s z%TestConcatenate.test_concat_iterablesc C sV t td�ddd gd�g}|dd� td�D �7 }t|dd �j}|d j}t�||� d S ) Nr! rZ r# )r r9 c S s" g | ]}t td �dddgd��qS )r! Nr# rZ )r r9 )r r- )rO r) r r r rR � s z5TestConcatenate.test_concat_order.<locals>.<listcomp>�d T)rk r )r r- r r9 r: rG )r �dfsr r= r r r �test_concat_order� s z!TestConcatenate.test_concat_orderc C sb t tjddgdd��}t tddg��}t||gdd�}t ddtd�td�gtd�}t�||� d S )Nr# r8 �Int64)r1 T)rf ) r rx rg r r r �objectr: rz )r rZ r5 r r= r r r �.test_concat_different_extension_dtypes_upcasts� s z>TestConcatenate.test_concat_different_extension_dtypes_upcastsc C sR t ttd��ttd��gddgd�}t ttd��ttd��d��}t�||� d S )Nr! r �First�Another)r7 )r� r� )r r r- r: rz )r r= r r r r �test_concat_ordered_dict� s " z(TestConcatenate.test_concat_ordered_dictN) r� r� r� r �tdZ skip_array_manager_invalid_testr6 r> rJ r\ �mark� parametrizerU rY r` rj rl rs r| r} r~ r r� r� r� r� r� r� r� r� r r r r r s, " ) ! E r �pdt�dt�floatc C sj |t d�j}|tjdg| |d��|tjtjg| |d��|tjdg| |d��g}t|�}|jj| ksft�d S )N)r1 r# )r1 �ndminr$ ) r� �ndimr rg �nanr r/ r1 r )r� r� �dimsr� �xr r r �!test_concat_no_unnecessary_upcast� s r� r� c C sz t dd��f |� j}|tjdg| |d��|tjtjg|d��|tjdg| |d��g}t|�}|jjdkslt�W d Q R X d S )NT)�recordr# )r1 r� )r� r$ �float64) r r� r rg r� r r/ r1 r )r� r� r� r� r� r r r �test_concat_will_upcast� s r� c C sD t ddgi�} t dg i�}t ddgi�}t| |g�}t�||� d S )Nr% r# g �?)r r r: r; )ri r4 r= r r r r �0test_concat_empty_and_non_empty_frame_regression s r� c C s^ t tdddg��} tddgddgddggd��t�tjd��}t| | gdd�}t � ||� d S )Nr r# r8 )�data)r& ) r r r �astyperx � SparseDtyper rn r r: r; )rZ r= r r r r �test_concat_sparse s r� c C sn t tj�dd g�td�} t dgtd�}t dd dgdddgd��t�tjd ��}t | |gdd�}t � ||� d S )Nr# )r1 r )r� r )r& ) r rx Zarraysr r� r� r� r r� r r: rz )rZ r5 r= r r r r �test_concat_dense_sparse s r� r7 r� r( c C s� t dddgdddgd��}tdd d gd d �}td ddgdd �}t|||gd| d�}dddd gddd dgddd dgg}t�| d df| d df| d d f| d dfg�}t ||d�}t�||� d S )Nr# r8 r! r r$ r� )rZ r5 � � rw r� )rE r r� r� r� )r& r7 r rZ r5 )r9 )r r r r � from_tuplesr: r; )r7 r3 r r r Zexpected_valuesZexpected_columnsr= r r r �test_duplicate_keys s ". r� �objra r )rE c C s" t | | g�}t|t| ��s t�d S )N)r r �typer )r� r r r r � test_concat_preserves_subclass- s r� c C s� t dtjdddgdd�i�} t dt�ddd g�i�}t| |gd d �}t ddddddd gidd�}t�||� t|| gd d �}t dddd dddgidd�}t�||� d S ) NrZ r# r8 r! r� )r1 r r$ r� T)rf )r rx rg r r r: r; )ri r4 r r= r r r �(test_concat_frame_axis0_extension_dtypes; s r� c C s^ t ddgidd�} t ddgidd�}t| |gdd�}t dd gd dgd �dd�}t�||� d S ) NrZ ro r� )r1 r5 r# T)rf )rZ r5 )r r r: r; )Zdf_aZdf_br r= r r r �+test_concat_preserves_extension_int64_dtypeI s r� )r7 � integrity�redr! TF�bluec C sl t ddi�t ddi�t ddi�g}t|| |d�}tt| dddg��}t dddgt�|�d�}t�||� d S ) NrZ r# r5 r8 r� r! )r7 Zverify_integrity)r )r r rS �zipr r� r: rz )r7 r� Z series_listr Ztuplesr= r r r �test_concat_repeated_keysR s "r� c C s� t jdgddd�} tddgi| d�}td gdd d�}td gtjgd �|d�}t||gd d �}td | d gtd�}td d gtjtjgtjdgd�|d�}t� ||� d S )Nz2021-04-08 21:21:14+00:00zdatetime64[ns, UTC]z Time (UTC))r1 rE rc g�!��u��?)r� r r� zMaybe Time (UTC))ra rb r9 )r& r )r1 )ra rb rc )r ) rx Z DatetimeIndexr r r r� r r� r: r; )Zdtir� �idxr� r r< r= r r r � test_concat_null_object_with_dtid s r� c C sl t �ddg�} tddgg| d�}tdgt�d�d�}t||g�}tddgtjtjgg| d�}t� ||� d S )N)rb r# )rc r# r# r8 )r9 r )r r9 ) r r� r rx Z RangeIndexr r r� r: r; )�miri r4 r r= r r r �,test_concat_multiindex_with_empty_rangeindexx s r� c C s� t dddggdgd�} t dddggd gd�}d }tjt|d �� t| |gd �}W d Q R X t dddgdddggdd gd�}t�||� d S ) Nr# r8 r! rZ )r r r$ r� r5 ziIn a future version of pandas all arguments of concat except for the argument 'objs' will be keyword-only)r[ r )r r: Zassert_produces_warning� FutureWarningr r; )r3 r4 r_ r r= r r r � test_concat_posargs_deprecation� s r� )4� collectionsr r �decimalr �warningsr �numpyr r\ Z pandas.util._test_decorators�utilZ_test_decoratorsr� Zpandasrx r r r r r r r Zpandas._testingZ_testingr: Zpandas.core.arraysr Zpandas.core.constructionr Z pandas.tests.extension.decimalr r r� r� Zsctypesr� r� r� r� r� r� ZSubclassedDataFramerp ZSubclassedSeriesr� r� r� r� r� r� r� r r r r �<module> sL $ I &"
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка