Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexes/__pycache__/test_setops.cpython-37.pyc
Назад
B �5�g�l � @ sH d Z ddlmZ ddlZddlZddlZddlmZ ddlm Z m Z m Z m Z m Z mZmZmZmZmZ ddlmZ ddlmZmZmZ ejejfejej fgZ dd� Z d d � Z ej �!d ej"ej#fej$ej"fej$ej#fej$ej%fg�d d � �Z&ej �!ddddddddddddddd d g�ej �!d d d g�d!d"� ��Z'd#d$� Z(ej �!d%d&d'd'd(gd(d(gg�d)d*� �Z)G d+d,� d,�Z*ej �!d-d.d/d0d1g�d2d3� �Z+d4d5� Z,ej �!d6e e e e d7d8� ed9d8� eg�d:d;� �Z-ej �!d6e e e e ed<d8� g�d=d>� �Z.d?d@� Z/dAdB� Z0ej �!dCd&ej1g�dDdE� �Z2ej �!d6e e e edFd8� g�dGdH� �Z3G dIdJ� dJ�Z4dS )KzX The tests in this package are to ensure the proper resultant dtypes of set operations. � )�datetimeN)�find_common_type) �CategoricalIndex� DatetimeIndex� Float64Index�Index� Int64Index� MultiIndex�Series�TimedeltaIndex� Timestamp� UInt64Index)�is_datetime64tz_dtype�is_signed_integer_dtype� pandas_dtypec C s* | � � }| � � }|�|�j|jks&t�d S )N)� sort_values�union�dtype�AssertionError)�index�idx1�idx2� r �V/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexes/test_setops.py�test_union_same_types& s r c C s� | }|}t |j|jg�}|jtjkp.|jtjk}t|j�}t|j�}|�� }|�� }|�|�}|�|�} |r�|st|r�|jt�d�ks�t�| jt�d�ks�t�n |j|ks�t�| j|ks�t�d S )N�O)r r �np�uint64r r r r ) � index_flatZ index_flat2r r Z common_dtypeZ any_uint64Z idx1_signedZ idx2_signed�res1�res2r r r �test_union_different_types. s r! zidx_fact1,idx_fact2c C sT | d�}|d�}|� |�}|� |�}|j|j|jfks:t�|j|j|jfksPt�d S )N� � )r r r )Z idx_fact1Z idx_fact2r r r r r r r �"test_compatible_inconsistent_pairsJ s r$ zleft, right, expected)�int64r% r% )r% r �object)r% �float64r' )r r' r' )r r r )r' r' r' )zdatetime64[ns]r% r& )zdatetime64[ns]r r& )zdatetime64[ns]r' r& )zdatetime64[ns, CET]r% r& )zdatetime64[ns, CET]r r& )zdatetime64[ns, CET]r' r& )z Period[D]r% r& )z Period[D]r r& )z Period[D]r' r& �names)�foor) r) )r) �barNc C s~ t | �} t |�}tg | |d d�}tg ||d d�}|�|�}|j|ksLt�|j|d ks^t�|�|�}|j|d kszt�d S )Nr )r �name� � )r r r r r r+ � intersection)�left�right�expectedr( �a�b�resultr r r �test_union_dtypes_ s r5 c C s^ t �t�� | | O } W d Q R X t �t�� | | M } W d Q R X t �t�� | | N } W d Q R X d S )N)�tm�assert_produces_warning� FutureWarning)r r r r �%test_dunder_inplace_setops_deprecated� s r9 �valuesr, r- � c C s8 t | �}t ddg�}|�|�}t dg�}t�||� d S )Nr; )r r. r6 �assert_index_equal)r: r2 r3 r4 r1 r r r � test_intersection_duplicates� s r= c @ s� e Zd Zej�dddg�ej�dddddg�d d � ��Zd d � Zd d� Zdd� Z dd� Z ej�ddddddg�dd� �Z ej�ddddddg�dd � �Z ej�ddddddg�d d � �Z ej�ddddddg�d d � �Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)S )*� TestSetOps�caseg �?Zxxx�methodr. r � difference�symmetric_differencec C s0 d}t jt|d�� t||�|� W d Q R X d S )Nz!Input must be Index or array-like)�match)�pytest�raises� TypeError�getattr)�selfr? r@ r �msgr r r �test_set_ops_error_cases� s z#TestSetOps.test_set_ops_error_casesc s� t |t�rd S |d d� }|d d� � |�� �}t�|� �s@t�t|j�rNd S � fdd�tj t t gD �}x&|D ] }|�|�}t�|� �snt�qnW t |t �r�d}t jt|d�� |�dddg� W d Q R X d S ) N� r; c s g | ]}|� j ��qS r )r: )�.0�klass)�secondr r � <listcomp>� s z5TestSetOps.test_intersection_base.<locals>.<listcomp>z.other must be a MultiIndex or a list of tuples)rC r, r- )� isinstancer r. r6 � equalContentsr r r r �arrayr �listr rD rE rF )rH r �first� intersect�casesr? r4 rI r )rN r �test_intersection_base� s z!TestSetOps.test_intersection_basec s� |dd � }|d d� � |}|� � �}t�||�s6t�t|j�rDd S � fdd�tjtt gD �}x>|D ]6}t |t �sd|� |�}t�||�sdt||t |�f��qdW t |t �r�d}tjt|d�� |� dddg� W d Q R X d S ) Nr; rK c s g | ]}|� j ��qS r )r: )rL rM )rN r r rO � s z.TestSetOps.test_union_base.<locals>.<listcomp>z.other must be a MultiIndex or a list of tuples)rC r, r- )r r6 rQ r r r r rR r rS rP r �typer rD rE rF ) rH r rT � everythingr rV r? r4 rI r )rN r �test_union_base� s& zTestSetOps.test_union_basec s |dd � }|d d� � t |t�s*|�� r0g }n |dd � }|�� |�}t�||�sXt�� fdd�tjt t gD �}xd|D ]\}t |t t f�r�t |�t |�ks�t�t�|�� j|�� j� qx|�||�}t�||�sxt�qxW t |t��rd}tjt|d�� |�dddg|� W d Q R X d S ) Nr- � c s g | ]}|� j ��qS r )r: )rL rM )rN r r rO � s z3TestSetOps.test_difference_base.<locals>.<listcomp>z.other must be a MultiIndex or a list of tuples)rC r, r; )rP r Z is_booleanrA r6 rQ r r rR r rS r r rX Zassert_numpy_array_equalr Zasi8r rD rE rF ) rH �sortr rT �answerr4 rV r? rI r )rN r �test_difference_base� s& z TestSetOps.test_difference_basec s( t |t�rd S t|�dk r d S |d |dd � ksF|d |d d� krJd S |dd � }|d d� � |ddg }|�� �}t�||�s�t�� fdd�tjt t gD �}xF|D ]>}|�|�}t |�r�|� |�}t� ||� q�t�||�s�t�q�W t |t��r$d}tjt|d�� |�ddd g� W d Q R X d S ) Nr- r r, ���c s g | ]}|� j ��qS r )r: )rL rM )rN r r rO s z8TestSetOps.test_symmetric_difference.<locals>.<listcomp>z.other must be a MultiIndex or a list of tuples)rC r; )rP r �lenrB r6 rQ r r rR r rS r r r< r rD rE rF ) rH r rT r] r4 rV r? r1 rI r )rN r �test_symmetric_difference� s. ( z$TestSetOps.test_symmetric_differencezfname, sname, expected_name)�Arb rb )rb �BN)rb NN)Nrc N)NNNc C s" |}|j st�d� |�� �|�}|�� �|�}|�|�}|�� �|�} t�|| � |�� �|�}|�|��|�}|�|�}|�� �|�} t�|| � |�|��|�}|�� �|�}|�|�}|�� �|�} t�|| � |�|��|�}|�|��|�}|�|�}|�|��|�} t�|| � d S )Nz&Not for MultiIndex or repeated indices) � is_uniquerD �skip�copy� set_namesr r6 r< �drop) rH r �fname�sname� expected_namer rT rN r r1 r r r �test_corner_union s. z TestSetOps.test_corner_unionc C s` |}|j st�d� |�� �|�}|dd � �|�}|�|��� }|�|��� } t�|| � d S )Nz&Not for MultiIndex or repeated indicesr, ) rd rD re rf rg r r r6 r< ) rH r ri rj rk r rT rN r r1 r r r �test_union_unequalF s z TestSetOps.test_union_unequalc C s& |}|j st�d� |�� �|�}|�� �|�}|�|�}|�� �|�} t�|| � |�� �|�}|�|��|�}|�|�}|�|��|�} t�|| � |�|��|�}|�� �|�}|�|�}|�|��|�} t�|| � |�|��|�}|�|��|�}|�|�}|�|��|�} t�|| � d S )Nz&Not for MultiIndex or repeated indices) rd rD re rf rg r. r6 r< rh ) rH r ri rj rk r rT rN rU r1 r r r �test_corner_intersect\ s. z TestSetOps.test_corner_intersectc C sh |}|j st�d� |�� �|�}|dd � �|�}|�|��� }|dd � �|��� } t�|| � d S )Nz&Not for MultiIndex or repeated indicesr, ) rd rD re rf rg r. r r6 r< ) rH r ri rj rk r rT rN rU r1 r r r �test_intersect_unequal� s z!TestSetOps.test_intersect_unequalc C s� t |t�r |�tt|j���}n |�d�}t�|�}|�|�}|j |j ksNt �|�|d d� �}|j |j kspt �|d d� �|�}|j |j ks�t �d S )Nr) r ) rP r �renamerS �range�nlevelsr Zasarrayr. r+ r )rH r �otherr4 r r r �.test_intersection_name_retention_with_nameless� s z9TestSetOps.test_intersection_name_retention_with_namelessc C s8 |j s d S |j||d�}|d d� }tj||dd� d S )N)r\ r T)�exact)rd rA r6 r< )rH r r\ r4 r1 r r r �$test_difference_preserves_type_empty� s z/TestSetOps.test_difference_preserves_type_emptyc sv t � t�r � fdd�|D �}� �|d �� � �|d �}� �|�sFt�� �|�}� d d� �|d �}t�||� d S )Nc s g | ]}|g� j �qS r )rr )rL �x)r r r rO � s zDTestSetOps.test_difference_name_retention_equals.<locals>.<listcomp>r r, r- )rP r rp �equalsr rA r6 r< )rH r r\ r( rs r4 r1 r )r r �%test_difference_name_retention_equals� s z0TestSetOps.test_difference_name_retention_equalsc C s> |j s d S |�|d d� �}|j||d�}tj||dd� d S )Nr )r\ T)ru )rd r. rA r6 r< )rH r r\ �inter�diffr r r �(test_intersection_difference_match_empty� s z3TestSetOps.test_intersection_difference_match_emptyN)�__name__� __module__� __qualname__rD �mark� parametrizerJ rW rZ r^ ra rl rm rn ro rt rv ry r| r r r r r> � sN "'& r> r@ r. r rA rB c C s� t | t�rd S | �d�}t| |�||d�}t| |�| |d�}t�||� t| |�|d d� |d�}t| |�| d d� |d�}t�||� d S )N�category)r\ rK )rP r ZastyperG r6 r< )r r\ r@ rs r4 r1 r r r �test_setop_with_categorical� s r� c C sD | j r d S dd� }| }|ddddg }|||� |�|�js@t�d S )Nc S s | � |��|� | ��st�d S )N)r. rx r )r/ r0 r r r � check_intersection_commutative� s zPtest_intersection_duplicates_all_indexes.<locals>.check_intersection_commutativer r, r- )�emptyr. rd r )r r� �idxZidx_non_uniquer r r �(test_intersection_duplicates_all_indexes� s r� �clsc C s t | t| �d�S )N)� categories)r �set)rw r r r �<lambda>� � r� c C s t | td�S )N)r )r r& )rw r r r r� r� c C s� | ddddg�}| dddg�}| ddddddg�}t |t�rPtddddddg�}|�|�}t�||� |j|dd�}t�||� d S )Nr, r- r; r[ F)r\ )rP r r r r6 r< )r� r2 r3 r1 r4 r r r �0test_union_duplicate_index_subsets_of_each_other� s r� c C s t | td�S )N)r )r r& )rw r r r r� r� c C sX | dddg�}| ddg�}| dddg�}|� |�}t�||� |� |�}t�||� d S )Nr, r )r r6 r< )r� r2 r3 r1 r4 r r r �1test_union_with_duplicate_index_and_non_monotonic s r� c C sR t ddddg�} t dddg�}t dddddddg�}| j|dd�}t�||� d S )Nr, r- r; �1�0F)r\ )r r r6 r< )r2 r3 r1 r4 r r r �+test_union_duplicate_index_different_dtypes) s r� c C sH t dddg�} t ddddg�}| �|�}t ddddg�}t�||� d S )Nr r, r- )r r r6 r< )r2 r3 r4 r1 r r r �(test_union_same_value_duplicated_in_both2 s r� �dupc C sV t tjdddg�}t tj| ddg�}|j|dd�}t tj| dddg�}t�||� d S )Nr, r- F)r\ g �?g @)r r �nanr r6 r< )r� r2 r3 r4 r1 r r r �test_union_nan_in_both; s r� c C s t | td�S )N)r )r r& )rw r r r r� L r� c C s\ | dddg�}| dddg�}| ddddg�}|� |�}t�||� |� |�}t�||� d S )Nr, r r- )r r6 r< )r� r2 r3 r1 r4 r r r �<test_union_with_duplicate_index_not_subset_and_non_monotonicE s r� c @ s: e Zd Zdd� Zejjddgdd�dd� �Zej�d ed d d d dgdd�dfed d d d dgdd�dfed d d d dg�dfg�dd� �Z ejjddgdd�ej�ddddg�dd� ��Z dd� Z ejjddgdd�d d � �Z ej�d e jeeg�ejjddgdd�d d � ��Zejjddgdd�d!d"� �Zejjddgdd�ej�d#d$d%g�d&d'� ��Zd(d)� Zejjddgdd�d*d+� �Zejjddgdd�d,d-� �Zej�d.d/d0g�d1d2� �Zejjd3d4�ej�d.d/d0g�d5d6� ��Zd7d8� Zej�d9ed:d;e jg�ed<d=d>g�fed:d;g�ee jd<d=d>g�fg�d?d@� �Z dAdB� Z dCS )D�TestSetOpsUnsortedc C sX t ddd�t ddd�g}t|td�}tdgtd�}|�|�}tg td�}t�||� d S )Ni� r- � � )r Zaa)r r r& r. r6 r< )rH Zdt_dates�index1�index2r4 r1 r r r �test_intersect_str_dates_ s z+TestSetOpsUnsorted.test_intersect_str_datesr �stringT)Zindirectc C sl |d d� }|d d� }|j ||d�}|d kr>t�||�� � t�||�sNt�|j ||d�}||ksht�d S )Nr# r" )r\ )r. r6 r< r rQ r )rH r r\ rT rN rU rz r r r �test_intersectioni s z$TestSetOpsUnsorted.test_intersectionzindex2,keeps_namer; r[ rK � � )r+ rs Fc C sZ t dddddgdd�}t dddg�}|�||�}|r:d|_|j|jksJt�t�||� d S )Nr, r- r; r[ rK r )r+ )r r. r+ r r6 r< )rH r� Z keeps_namer\ r� r1 r4 r r r �#test_intersection_name_preservationv s z6TestSetOpsUnsorted.test_intersection_name_preservationz$first_name,second_name,expected_name)rb rb rb )rb rc N)Nrc Nc C sD |dd� }|d d� }||_ ||_ |j||d�}|j |ks@t�d S )NrK r# r" )r\ )r+ r. r ) rH r Z first_name� second_namerk r\ rT rN rU r r r �$test_intersection_name_preservation2� s z7TestSetOpsUnsorted.test_intersection_name_preservation2c C s� t ddgdd�}t ddgdd�}t dd gd d�}|j|j||d �|d �}|j||d �j||d �}t�||� t ddgd d�}t g d d�}t g dd�} |j|j| |d �|d �}|j||d �j| |d �}t�||� d S )Nr, r- �i1)r+ rK r� �i2r; r[ �i3)r\ �j1�j2�j3)r r r6 r< ) rH r\ r� r� r� r r1 r� r� r� r r r �test_chained_union� s z%TestSetOpsUnsorted.test_chained_unionc C s^ |dd� }|d d� }|d d� }|j ||d�}|d krJt�||�� � t�||�sZt�d S )NrK r# r" )r\ )r r6 r< r rQ r )rH r r\ rT rN rY r r r r � test_union� s z TestSetOpsUnsorted.test_unionrM c C sh |dd� }|d d� }|d d� }||j �}|j||d�}|d krTt�||�� � t�||�sdt�d S )NrK r# r" )r\ )r: r r6 r< r rQ r ) rH r rM r\ rT rN rY r? r4 r r r �test_union_from_iterables� s z,TestSetOpsUnsorted.test_union_from_iterablesc C st |dd� }|j ||d�}||k| ks,t�|j g |d�}||k| ksLt�tg �j ||d�}||k| kspt�d S )NrK r# )r\ )r r r )rH r r\ rT r r r r �test_union_identity� s z&TestSetOpsUnsorted.test_union_identityzsecond_name,expected)NN)r+ r+ c C sx |dd� }|d d� }|dd� }d|_ ||_ |j||d�}t�||�sNt�|d krf|j d kstt�n|j |kstt�d S )NrK r# r" r+ )r\ )r+ rA r6 rQ r ) rH r r� r1 r\ rT rN r] r4 r r r �!test_difference_name_preservation� s z4TestSetOpsUnsorted.test_difference_name_preservationc C s. |dd� }d|_ |�g |�}t�||� d S )NrK r# r+ )r+ rA r6 r< )rH r r\ rT r4 r r r �test_difference_empty_arg� s z,TestSetOpsUnsorted.test_difference_empty_argc C sB |dd� }d|_ |�||�}t|�dks.t�|j |j ks>t�d S )NrK r# r+ r )r+ rA r` r )rH r r\ rT r4 r r r �test_difference_identity� s z+TestSetOpsUnsorted.test_difference_identityc C sP |dd� }|d d� }|� ||�}|dd� }|d kr@|�� }t�||� d S )NrK r# r" )rA r r6 r< )rH r r\ rT rN r4 r1 r r r �test_difference_sort� s z'TestSetOpsUnsorted.test_difference_sort�opnamerA rB c C s� t dtd�dg�}t dtd�dg�}t�||�}t�t�� ||�}W d Q R X t dtd�dtd�g�}|dkrz|d d� }t�||� tj||dd�}||�}t�||� d S ) Nr; �2000r, r- �1999rA F)r\ )r r �operator� methodcallerr6 r7 �RuntimeWarningr< )rH r� r2 r3 �opr4 r1 r r r � test_difference_incomparable s z/TestSetOpsUnsorted.test_difference_incomparablezNot implemented)�reasonc C sZ t dtd�dg�}t dtd�dg�}tj||dd�}tjtdd �� ||� W d Q R X d S ) Nr; r� r, r- r� T)r\ zCannot compare)rC )r r r� r� rD rE rF )rH r� r2 r3 r� r r r �!test_difference_incomparable_true s z4TestSetOpsUnsorted.test_difference_incomparable_truec C sx t �tdddgdddg��}t �ddg�}|j||d �}t �d d dg�}|d krX|�� }t�||� t�||�stt�d S ) Nr) r* �bazr, r- r; )r) r, )r* r; )r\ )r* r- )r� r; ) r � from_tuples�ziprB r r6 r< rQ r )rH r\ r� r� r4 r1 r r r � test_symmetric_difference_mi s z/TestSetOpsUnsorted.test_symmetric_difference_mizindex2,expectedr r, g @g @g c C s@ t dtjddg�}|j||d�}|d kr0|�� }t�||� d S )Nr, r- r; )r\ )r r r� rB r r6 r< )rH r� r1 r\ r� r4 r r r �!test_symmetric_difference_missing) s z4TestSetOpsUnsorted.test_symmetric_difference_missingc C s� t ddddgdd�}t�ddddg�}t ddg�}|j||d�}t�||�sPt�|jdks^t�|j|d |d �}t�||�s~t�|jd ks�t�d S ) Nr, r- r; r[ r� )r+ rK )r\ �new_name)Z result_namer\ )r r rR rB r6 rQ r r+ )rH r\ r� r� r1 r4 r r r �#test_symmetric_difference_non_index: s z6TestSetOpsUnsorted.test_symmetric_difference_non_indexN) r} r~ r r� rD r� r� r� r r� r� r� r� r rR r rS r� r� r� r� r� r� r� Zxfailr� r� r� r� r� r r r r r� \ s< " * r� )5�__doc__r r� �numpyr rD Zpandas.core.dtypes.castr Zpandasr r r r r r r r r r Zpandas._testingZ_testingr6 Zpandas.api.typesr r r r' r% r Z COMPATIBLE_INCONSISTENT_PAIRSr r! r� r� Z makeIntIndexZmakeRangeIndexZmakeFloatIndexZ makeUIntIndexr$ r5 r9 r= r> r� r� r� r� r� r� r� r� r� r� r r r r �<module> s� 0 & <
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка