Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexes/multi/__pycache__/test_setops.cpython-37.pyc
Назад
B �5�g�A � @ s d dl Zd dlZd dlZd dlmZmZmZmZm Z d dl m Z ej �dddg�ej �dddd d g�d d � ��Zej �d eeje eg�dd� �Zej jej �d eeje eg�dd� ��Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zej jd d �d d!� �Z d"d#� Z d$d%� Z d&d'� Z d(d)� Z ej �dddd d g�d*d+� �Z!d,d-� Z"d.d/� Z#ej jd d �d0d1� �Z$ej �d2e%d�e%d �g�d3d4� �Z&ej jd d �d5d6� �Z'd7d8� Z(ej jd d �d9d:� �Z)d;d<� Z*d=d>� Z+d?d@� Z,ej �dddd d g�dAdB� �Z-ej �dCdDgdDgfdDdDgdDgfdEdDgdEdDgfg�dFdG� �Z.ej �dHdIdddgfdIdJgddgfdIdKgddgfdLdMdNgddgfdLdKdJgddgfdOdJdKdMgddgfdLdJdMgdJdgfdLdMdKgddKgfdLdJdKgdJdKgfdLddKgddKgfg �dPdQ� �Z/dRdS� Z0dTdU� Z1dVdW� Z2dXdY� Z3dZd[� Z4dS )\� N)�CategoricalIndex�Index� IntervalIndex� MultiIndex�Series�caseg �?Zxxx�method� intersection�union� difference�symmetric_differencec C s4 d}t jt|d�� t| |�||d� W d Q R X d S )Nz!Input must be Index or array-like)�match)�sort)�pytest�raises� TypeError�getattr)�idxr r r �msg� r �\/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexes/multi/test_setops.py�test_set_ops_error_cases s r �klassc C s� | dd d� }| d d� }|t k r,||j�}|j||d�}|d krL|�� }n|}t�||� d}tjt|d�� |jdddg|d� W d Q R X d S ) N� ���� )r z.other must be a MultiIndex or a list of tuples)r � � ) r �valuesr � sort_values�tm�assert_index_equalr r r )r r r �first�secondZ intersect�expectedr r r r �test_intersection_base s r% c C s� | d d d� }| d d� }|t k r,||j�}|j||d�}|d krL|�� }n|}t�||� d}tjt|d�� |jdddg|d� W d Q R X d S ) Nr r )r z.other must be a MultiIndex or a list of tuples)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_union_base. s r& c s� | dd � � | d d� }| j � |d�}|d kr6|�� }|�|�sDt�t�||� � fdd�tjtt gD �}x&|D ] }| j ||d�}t�||� qpW d}t j t |d�� | j ddd g|d� W d Q R X d S ) N� )r c s g | ]}|� j ��qS r )r )�.0r )r# r r � <listcomp>O s z(test_difference_base.<locals>.<listcomp>z.other must be a MultiIndex or a list of tuples)r r r r ) r r �equals�AssertionErrorr r! �np�arrayr �listr r r )r r �answer�result�casesr r r )r# r �test_difference_baseC s r2 c s� | dd � }| d d� � | ddg }|j � |d�}|d krB|�� }t�||� � fdd�tjttgD �}x&|D ] }|j ||d�}t�||� qnW d}tj t |d�� |j dd d g|d� W d Q R X d S ) Nr r r )r c s g | ]}|� j ��qS r )r )r( r )r# r r r) e s z-test_symmetric_difference.<locals>.<listcomp>z.other must be a MultiIndex or a list of tuples)r r r ) r r r r! r, r- r r. r r r )r r r" r/ r0 r1 r r r )r# r �test_symmetric_differenceY s r3 c C s� t jddgddggddgd�} t�t�� | | A }W d Q R X |j| jksLt�| �� �ddg�}t�t�� | |A }W d Q R X |jd d gks�t�d S )N�a�b�A�B)�names) r � from_productr �assert_produces_warning� FutureWarningr8 r+ �copy�rename)r r0 �idx2r r r �$test_multiindex_symmetric_differenceo s r? c C s | j r t�| d d� j s t�d S )Nr )�emptyr+ )r r r r � test_empty| s rA c C sl | }|j | dd � |d�}| d d� j}|d kr8t|�}tj|d| jd�}t|t�sXt�|�|�sft�|j| jksvt�t � ||� | j | |d�}| d d� }|�|�s�t�|j| jks�t�| dd � j | |d�}| d d� }|�|�s�t�|j| jks�t�| d d� j | |d�}| d d� }|�|��s,t�|j| jk�s>t�| dd � }ddg|_|j ||d�}|jdk�srt�| j | � d�d |d�}t |�dk�s�t�|j |j|d�}|�|d d� ��s�t�|j g |d�}|�|��s�t�|j|jk�s�t�|j d g|d�}t�d d d d dg�}|j|_|j|jk�s2t�d}t jt|d�� |j dddddg|d� W d Q R X d S )N���)r r )Z sortorderr8 �foo�baz)NNr )rC �one)�barrE )rD �two)rC rG )�quxrE )rH rG z.other must be a MultiIndex or a list of tuples)r r r r' r )r r �sortedr � from_tuplesr8 � isinstancer+ r* r r! Z sortlevel�lenr r r )r r r" r0 �valsr$ Zchunkletr r r r �test_difference� sR rN c C s0 t �ddgddgg�} | �g �}t�|| � d S )Nr r r4 r5 )r r9 r r r! )r r0 r r r � test_difference_sort_special� s rO zNot implemented.)�reasonc C sJ t �ddgddgg�} | jg dd�}t �ddgddgg�}t�||� d S )Nr r r4 r5 T)r )r r9 r r r! )r r0 r$ r r r �!test_difference_sort_special_true� s rQ c C s� t �dt�d�dgddgg�} t �dt�d�dgdd gg�}d }tjt|d �� | �|�}W d Q R X t�|| � | j|d d �}t�|| � d S )Nr �2000r r4 r5 r r' �c�dz0sort order is undefined for incomparable objects)r F)r ) r r9 �pd� Timestampr r: �RuntimeWarningr r! )r �otherr r0 r r r �!test_difference_sort_incomparable� s rY c C sl t �dt�d�dgddgg�} t �dt�d�dgdd gg�}d }tjt|d �� | j|d d � W d Q R X d S )Nr rR r r4 r5 r r' rS rT zKThe 'sort' keyword only takes the values of None or False; True was passed.)r T)r )r r9 rU rV r r � ValueErrorr )r rX r r r r �&test_difference_sort_incomparable_true� s r[ c C s� | d d� d d d� }| dd � }|j ||d�}|d krHt�|| �� � t�|| �sXt�| j | |d�}t�|| � | j | d d� |d�}t�|| � d S )Nr r r )r r )r r r! r � equalContentsr+ )r r �piece1�piece2Z the_unionr r r � test_union� s r_ c C s� | d d� d d d� }| dd � }|j ||d�}|d krLt�|| dd� � t�|| dd� �sdt�| j | |d�}t�|| � | d d� j | dd � |d�}| d d� }|�|�s�t�d S )Nr r r )r r r )r r r! r\ r+ r* )r r r] r^ Zthe_intr@ r$ r r r �test_intersection s r` c C s� | � � �d�}d g| j }t| |�||d�}t| |�| |d��|�}t�||� t| |�|d d� |d�}t| |�| d d� |d��|�}t�||� d S )N�category)r r )Z to_flat_indexZastype�nlevelsr r= r r! )r r r rX Z res_namesr0 r$ r r r �test_setop_with_categorical, s rc c C s� t td�dd�}| j||d�}t| jg g| j d d�}tj||dd� | jt� |�d d� |d�}t| jg g| j | j d�}tj||dd� d }t j t |d �� | jt� |�|d� W d Q R X d S ) Nr rC )�name)r )�levels�codesr8 T)�exactr z.other must be a MultiIndex or a list of tuples)r )r �ranger r re rb r r! r, Zasarrayr8 r r r )r r rX r0 r$ r r r r � test_intersection_non_object< s ri c C sF t �ddgddgg�} t�| j| dd�| � t�| j| d d�| � d S )Nr r r4 r5 F)r )r r9 r r! r )r r r r �test_intersect_equal_sortN s rj c C sF t �ddgddgg�} t �ddgddgg�}t�| j| dd�|� d S )Nr r r4 r5 T)r )r r9 r r! r )r Zsorted_r r r � test_intersect_equal_sort_trueU s rk �slice_c C sJ t �ddgddgg�}|| }t�|�|�|� t�|j|dd�|� d S )Nr r r4 r5 F)r )r r9 r r! r )rl r rX r r r �test_union_sort_other_empty] s rm c C sV t �ddgddgg�}|d d� }|j|dd�}t �ddgddgg�}t�||� d S )Nr r r4 r5 T)r )r r9 r r r! )rl r rX r0 r$ r r r � test_union_sort_other_empty_sortl s rn c C sv t �dt�d�gddgg�} t�t�� | �| d d� �}W d Q R X t�|| � | j| d d� dd�}t�|| � d S )Nr rR r4 r5 F)r ) r r9 rU rV r r: rW r r! )r r0 r r r �"test_union_sort_other_incomparablew s ro c C sP t �dt�d�gddgg�} tjtdd�� | j| d d� dd� W d Q R X d S ) Nr rR r4 r5 zCannot compare)r T)r )r r9 rU rV r r r r )r r r r �'test_union_sort_other_incomparable_sort� s rp c C sL t �ddgddgg�} | jd }d}tjt|d�� | �|� W d Q R X d S )Nr4 r5 r r z<Can only union MultiIndex with MultiIndex or Index of tuples)r )r r9 re r r �NotImplementedErrorr )�mir r r r r �"test_union_non_object_dtype_raises� s rs c C sZ t �g g�} t jddgddggddgd�}| �|�}t �ddgddgg�}t�||� d S )Nr r r r' r4 r5 )r8 )r � from_arraysr r r! )rr �mi2r0 r$ r r r �%test_union_empty_self_different_names� s rv c C s` t jddgddggddgd�} t�d�}| �|�}tj| |d d � |�| �}tj| |d d � d S ) Nr r r r' r4 r5 )r8 r F)Z check_names)r rt rU Z RangeIndexr r r! )rr �riZ result_leftZ result_rightr r r �&test_union_multiindex_empty_rangeindex� s rx c C s\ t �ddgddgg�}t �ddgddgg�}tjtdd�� t|| �|dd � W d Q R X d S ) Nr4 r5 r r rS z The 'sort' keyword only takes)r T)r )r r9 r r rZ r )r Zidx1r> r r r �test_setops_disallow_true� s ry )�tuples� exp_tuples)�val1�test1)�val2�test2c C sV t j| ddgd�}t jdddgddgd�}|�|�}t j|ddgd�}t�||� d S )Nr" r# )r8 )r| r} )r~ r )r rJ r r r! )rz r{ �left�rightr0 r$ r r r � test_intersect_with_duplicates� s r� zdata, names, expected)r r4 r5 )r r rS rT )r r r c C s< t jg ddgd�}t j| g|d�}|�|�}||ks8t�d S )Nr4 r5 )r8 )r rJ Z_maybe_match_namesr+ )�datar8 r$ rr ru r0 r r r �test_maybe_match_names� s r� c C st t jddgddggddgd�} t jddgddggddgd�}| �|�}t jddgddggd dgd�}t�||� d S ) Nr r r r' rS r5 )r8 r4 )r rt r r r! )�mi1ru r0 r$ r r r �'test_intersection_equal_different_names� s r� c C sF t jdgdggddgd�} t �dgdgg�}| �|�}t�||� d S )Nr r rS r5 )r8 )r rt r r r! )rr ru r0 r r r �!test_intersection_different_names� s r� c C sh t �d| d| gddddgg�}t �d| dgdddgg�}|�|�}t �d| gddgg�}t�||� d S )Nr r' r r g @)r rt r r r! )Z nulls_fixturer� ru r0 r$ r r r �3test_intersection_with_missing_values_on_both_sides� s r� c C sN t �dtjgddgg�} t �dtjdgdddgg�}| �|�}t�||� d S )Ng �?r r g @r' )r rt r, �nanr r r! )r� ru r0 r r r � test_union_nan_got_duplicated s r� c C s� | j st| ttf�rd S | �� j�� }t�|dgt |� g�}t�|d g| dgt |�d g�}|� |�}t � ||� � � |� |�}t � ||� � � d S )Nr r )r@ rK r r �uniquer �tolistr rt rL r r r! r )�indexr r� ru r0 r r r �test_union_duplicates s & r� )5�numpyr, r ZpandasrU r r r r r Zpandas._testingZ_testingr �markZ parametrizer r- r. r% Zarm_slowr& r2 r3 r? rA rN rO ZxfailrQ rY r[ r_ r` rc ri rj rk �slicerm rn ro rp rs rv rx ry r� r� r� r� r� r� r� r r r r �<module> st " B $"
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка