Current File : //lib64/python3.9/site-packages/numpy/lib/tests/__pycache__/test_arraysetops.cpython-39.opt-1.pyc
a
z[ycKp � @ st d Z ddlZddlmZmZmZmZ ddlm Z m
Z
m Z m Z m
Z
mZmZmZ ddlZG dd� d�ZG dd� d�ZdS ) z-Test functions for 1D array set operations.
� N)�assert_array_equal� assert_equal�
assert_raises�assert_raises_regex)�ediff1d� intersect1d�setxor1d�union1d� setdiff1d�unique�in1d�isinc
@ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zej �
d e j g d �e j
d
�de jdfe j g d �e j
d
�e j g d�e jd
�ddfe j g d�e jd
�e je jdfg�dd� �Zej �
de j g d �e jd
�dde j g d�e jd
�fe j g d �e jd
�e j dge jd
�de j g d�e jd
�fe j g d �e jd
�dde j g d �e jd
�fe j g d �e j
d
�d d e j g d �e j
d
�fg�d d!� �Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Z d0d1� Z d2d3� Z d4d5� Z d6d7� Z d8d9� Z!d:d;� Z"d<d=� Z#d>d?� Z$dS )@�
TestSetOpsc C s� t �g d��}t �g d��}t �g d��}t||dd�}t||� t �g d��}t �g d��}t �g d��}t||�}t||� tg tg g �� d S )N�� � � � �r � � r r )r r r T��
assume_unique)r r r r r )r r r r r r r )�np�arrayr r )�self�a�b�ec�cZed� r �F/usr/lib64/python3.9/site-packages/numpy/lib/tests/test_arraysetops.py�test_intersect1d s
zTestSetOps.test_intersect1dc C sL G dd� d�}|� }t ||�}t||� t g d�g d��}t|g d�� d S )Nc @ s e Zd Zdd� ZdS )z4TestSetOps.test_intersect1d_array_like.<locals>.Testc S s
t �d�S )Nr )r �arange�r r r r! � __array__% s z>TestSetOps.test_intersect1d_array_like.<locals>.Test.__array__N)�__name__�
__module__� __qualname__r% r r r r! �Test$ s r) �r r r )r r )r r) r �resr r r! �test_intersect1d_array_like" s
z&TestSetOps.test_intersect1d_array_likec C s� t �g d��}t �g d��}t||ddd�\}}}t �g d��}t||� t|| |� t|| |� t �g d��}t �g d��}t||dd�\}}}t �g d��}t||� t|| |� t|| |� t �g d �g d
�g�}t �g d �g d �g�}t||ddd�\}}}t �||j�}t �||j�} t �g d
��}
t|
|| � t|
|| � t �g d�g d�g�}t �g d�g d�g�}t||dd�\}}}t �||j�}t �||j�} t �g d��}
t|
|| � t|
|| � d S )N�r r r r )r r r � T)r �return_indices)r r r )r r r r r r r )r � r r r r r r )r/ )r r r r. )r r0 r � )r r r r. )�
� r0 � )r r. r r0 )r r r r. r. )r r r0 r r )r r r r )r2 r3 r0 r )r r r0 )r r r r Z
unravel_index�shape) r r r r �i1�i2�eeZefZui1Zui2Zear r r! �test_intersect1d_indices. s<
z#TestSetOps.test_intersect1d_indicesc C s� t �g d��}t �g d��}t �g d��}t||�}t||� t �g d��}t �g d��}t �g d��}t||�}t||� t �g d��}t �g d��}t �g d��}t||�}t||� tg tg g �� d S ) Nr r )r r r r* )r. r r )r r r r r r. )r r0 r r )r. r r r0 )r r r r �r r r r r r r r! �
test_setxor1dU s
zTestSetOps.test_setxor1dc C s, t �g �}t �dg�}t �ddg�}tg t|�� tdgt|dd�� tdgt|dd�� tddgt|ddd�� tg t|�� tdgt|�� tg d�t|d d
d�� tg d �t|d d
gd dgd�� tdd
gt|d
d�� tg d�t|d dgd�� td dgt|d d�� tg d�t|d d
gd�� d S )Nr r r )�to_begin)�to_end���)r< r= )r r r4 r r4 )r r. r r r0 r r. r0 )r r r0 )r r. r )r r r r )r Z zero_elemZone_elemZtwo_elemr r r! � test_ediff1dm s"
�zTestSetOps.test_ediff1dz ary, prepend, append, expectedr* ��dtypeNr= )r r r r< )� �?g @g "@c C sF d� |�}tt|�� t|||d� W d � n1 s80 Y d S )Nz dtype of `{}` must be compatible��aryr= r< )�formatr � TypeErrorr )r rD �prepend�append�expected�msgr r r! �!test_ediff1d_forbidden_type_casts� s
�z,TestSetOps.test_ediff1d_forbidden_type_castszary,prepend,append,expectedi i )r r r r r )r r r r )r r r r r ���)r r r rL c C s t j|||d�}t||� d S )NrC )r r r )r rD rG rH rI �actualr r r! � test_ediff1d_scalar_handling� s �
z'TestSetOps.test_ediff1d_scalar_handlingc s� dd� }t j|tgdhd�� � fdd�}t �d��g d��}t �g d �g d
�g d �g�}|||� d d
g}d}|||� t �d�}|||� |||� |||� |d|� ||d� |dd� g }|||� |||� |||� d S )Nc S s t �|��� �� }| |v S �N)r �asarrayZflatten�tolist)r r r r r! �
_isin_slow� s z(TestSetOps.test_isin.<locals>._isin_slowr )ZotypesZexcludedc s" t | |�}� | |�}t||� d S rO )r
r )r r �x�y�Z isin_slowr r! �assert_isin_equal� s
z/TestSetOps.test_isin.<locals>.assert_isin_equal� �r r r )r2 � � )r r r )� � �! )r4 r0 )r r. )r4 r r r r. )r Z vectorize�boolr# �reshaper ) r rR rV r r r �d�frS r rU r! � test_isin� s(
zTestSetOps.test_isinc C s dD �]�}g d�}g d�| }t �g d��}t||dd�}t||� d|d< t �g d ��}t||dd�}t||� d
\|d< |d < t �g d ��}t||dd�}t||� t �g d
��}g d�| }g d�}t||�}t||� |g d�| }g d�}t||�}t||� t �g d��}t �g d�| �}t �g d��}t||�}t||� t �g d��}t �g d�| �}t �g d��}t||�}t||� t �ddg�}t �ddg| �}t �ddg�}t||�}t||� qt �dg�}t �dg�}t �dg�}t||�}t||� ttg g �g � d S )N�r r2 r r )TFTTTr r0 r )FFTT)r r0 r )TFTF�r r r r r r r r r r r r r r rX )FTFTTTTTTFTFFF)r r r )TTTTTTTTTTTFTT)r r r r r )r r r r r r )TFTTTr r F�r r r r )r �multr r r r r r r! � test_in1d� sX
zTestSetOps.test_in1dc C sB t �g d��}t �ddg�}t �g d��}t||�}t||� d S )N)r r r r` �er rh r r r )TFTFFTFFre r: r r r! �test_in1d_char_array+ s
z TestSetOps.test_in1d_char_arrayc C sH dD ]>}t �g d��}g d�| }tt �t||��t||dd�� qdS )z Test in1d's invert parameterrc rd rX T��invertN)r r r rk r )r rf r r r r r! �test_in1d_invert4 s zTestSetOps.test_in1d_invertc C s� t �d��dd�}t �dd��dd�}t �dd��dd�}t �g d��}tt||dd �|� tt||d
d �|� tt||dd �|� tt||d
d �|� d S ) Nr. r r r4 �? rZ )FFFTTTTr F)r r# r_ r r r )r r r Zlong_br r r r! �test_in1d_ravel= s zTestSetOps.test_in1d_ravelc C s< d g}t �dgd �}t �dg�}t �||�}t||� d S �Nr r2 Fre �r �ar1�ar2rI �resultr r r! � test_in1d_first_array_is_objectJ s
z*TestSetOps.test_in1d_first_array_is_objectc C s: d}t �d gd �}t �dg�}t �||�}t||� d S ro re rp r r r! � test_in1d_second_array_is_objectQ s
z+TestSetOps.test_in1d_second_array_is_objectc C s< d g}t �d gd �}t �dg�}t �||�}t||� d S )Nr2 Tre rp r r r! � test_in1d_both_arrays_are_objectX s
z+TestSetOps.test_in1d_both_arrays_are_objectc C s` t �dtfdtfg�}t jdg|d�}t jdgd |d�}t �dg�}t �||�}t||� d S )NZfield1Zfield2)r Nr@ r2 T)r rA �int�objectr r r )r �dtrq rr rI rs r r r! �+test_in1d_both_arrays_have_structured_dtype_ s z6TestSetOps.test_in1d_both_arrays_have_structured_dtypec C s` t jddgtd�}t jddgtd�}t �ddg�}t �||�}t||� t j||dd�}t|t �|�� t jg d�td�}|d d� }t jg d�td�}|d d� }t �ddg�}t �||�}t||� t j||dd�}t|t �|�� t jg d�td�}|d d� }t jddgtd�}t �dd g�}t �||�}t||� t j||dd�}t|t �|�� d S )
N�r r r@ Trj )r{ )r r r r> )r{ �r r r F)r r rx r r rk rp r r r! �'test_in1d_with_arrays_containing_tuplesi s0
z2TestSetOps.test_in1d_with_arrays_containing_tuplesc C s� t �g d��}t �g d��}t �g d��}t||�}t||� t �g d�g d�g�}t �g d��}t �g d��}t||�}t||� tg tg g �� d S )N)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 r r r )r r r r r r )r r r r ) r r r r r rS rT Zez�zr r r! � test_union1d� s
zTestSetOps.test_union1dc C s� t �g d��}t �g d��}t �ddg�}t||�}t||� t �d�}t �d�}t �ddg�}t||�}t||� tg tg g �� t �dt j�}tt|g �jt j� d S ) N)r. r r r r r r r r~ r. r � � rY r )r r r
r r# Zuint32r rA r: r r r! �test_setdiff1d� s
zTestSetOps.test_setdiff1dc C sF t �g d��}t �g d��}t �ddg�}t||dd�}t||� d S )N)r r r )r r r r r Tr )r r r
r )r r r rI rM r r r! �test_setdiff1d_unique� s
z TestSetOps.test_setdiff1d_uniquec C s8 t �g d��}t �g d��}tt||�t �dg�� d S )N)r r r )r r �sr )r r r r
)r r r r r r! �test_setdiff1d_char_array� s z$TestSetOps.test_setdiff1d_char_arrayc C sR t �g d��}t �g d��}t||�}t||�}t||�}t||�}t||� d S )N)r r r r r0 )r4 r0 r r r r r )r r r r r r
r )r r r Zc1Zaux1Zaux2Zc2r r r! �
test_manyways� s
zTestSetOps.test_manyways)%r& r' r( r" r, r9 r; r? �pytest�mark� parametrizer r Zint64�nanZfloat32�int8rK Zint16Zfloat64Zint32rN rb rg ri rl rn rt ru rv rz r} r� r� r� r� r� r r r r! r s~ '����
������
)<
!r c @ sn e Zd Zdd� Zdd� Zdd� Zdd� Zej� d d
d g�d d
� �Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS )�
TestUniquec C sd dd� }g d�d }g d�}g d�}g d�d }t �g d�d�}g }|�t jd � |�t jd
� |�d � |�d � |D ].}t �||�} t �||�}
|| |
||||� q|d
}t �t|�|�} || d d �< t �t|�|�}
||
d d �< || |
||||� ddg}t �tt ||��|�} t �tt ||��|�}
|| |
||||� g d�} t
t � | �ddg� g d�}g d�} g d�} t |�}
t
|
| � t |dd�\}}t
|| � t
|| � t � d�}d|d< t j |dd�\}}t
|t �
d�� g }t j |dd�d }t j |dd�d }t j |ddd �dd � \}}t|jt j� t|jt j� t|jt j� t|jt j� d t jd t jg}d d t jg}g d �}g d �}g d!�}tt � |�|� tt j |dd�||f� tt j |dd�||f� tt j |dd"�||f� d#t jd$td%t j�td t j�g}d$d#td%t j�g}g d&�}g d'�}g d(�}tt � |�|� tt j |dd�||f� tt j |dd�||f� tt j |dd"�||f� t �d)�}t �d*�|t �d+�|g}t �d+�t �d*�|g}g d �}g d �}g d!�}tt � |�|� tt j |dd�||f� tt j |dd�||f� tt j |dd"�||f� t �d)�}t �dd,�|t �dd-�|g}t �dd-�t �dd,�|g}g d �}g d �}g d!�}tt � |�|� tt j |dd�||f� tt j |dd�||f� tt j |dd"�||f� t jgd. }t jg}d/g}g d0�}d.g}tt � |�|� tt j |dd�||f� tt j |dd�||f� tt j |dd"�||f� d S )1Nc
S s� d}|� d|�}t| �}t|||� |� d|�}t| ddd�\}} t|||� t| ||� |� d|�}t| ddd�\}} t|||� t| ||� |� d|�}t| ddd�\}} t|||� t| ||� |� d|�}t| ddd�\}}
} t|||� t|
||� t| ||� |� d |�}t| ddd�\}}
} t|||� t|
||� t| ||� |� d
|�}t| ddd�\}}
} t|||� t|
||� t| ||� |� d |�}t| ddd�\}}
} } t|||� t|
||� t| ||� t| ||� d S ) Nz check {0} failed for type {1}�values� return_indexTF�return_inverse�
return_countsz return_index and return_inversez return_index and return_countsz return_inverse and return_countsz.return_index, return_inverse and return_counts)rE r r )
r r r6 r7 r ry Zbase_msgrJ �v�jZj1Zj2Zj3r r r! � check_all� sN � z,TestUnique.test_unique_1d.<locals>.check_all)r r r r r r r r2 )r r r r )r r r r )r r r r r r r )r r r r �
AllInteger�AllFloat�
datetime64[D]�timedelta64[D]�O)� �i)� �? � �? �r r� r� )�r r r� r| r* )r r r r r r T)r� r r� .)r� r )r� r� g @rB )r r r )r r r r )r r r )r� y @ �y �? �?� )r r r )r r r r r )r r r �natz
2020-12-26z
2020-12-24�D�hr r )r r r r )r Zmultiply�extend� typecodesrH r �empty�len�list�zipr r Z chararrayZzerosr rA Zintpr� �complexZ
datetime64Z timedelta64)r r� r r r6 r7 r �typesry ZaaZbb�unq�invZa1Za2Za2_invZa1_idxZa3_idxZa3_invZuaZua_idxZua_invZua_cntr� Zall_nansr r r! �test_unique_1d� s� 0
��
zTestUnique.test_unique_1dc C s\ t t| jt� t t| jdtfdtfg� t tjtt�d�dd� t tjtt�d�dd� d S )Nr r r2 r ��axis���) r rF �_run_axis_testsrx rw r Z AxisErrorr r# r$ r r r! �test_unique_axis_errorsp s �z"TestUnique.test_unique_axis_errorsc C sZ d}g d�g d�g}t �|�}tt|dd�t|dd�|� tt|dd�t|dd�|� d S )Nz Unique failed on list of lists�r r r r r� r )r rP r r )r rJ �inpZinp_arrr r r! �test_unique_axis_listx s
z TestUnique.test_unique_axis_listc C s g }|� tjd � |� tjd � |�d� |�d� |�dtfdtfg� |�dtfdtfg� |D ]}| �|� qhd}tjdtjd �� d
d ��
t �}tj d d
gd
d
ggt d �}t
t|dd�||� d}t� ddgddgddgddgg�}t� ddgg�}t
t|dd�||� d S )Nr� r� r� r� r r z&Non-bitwise-equal booleans test failedr2 r@ r> r FTr r� z"Negative zero equality test failedg �r� )r� r r� rH rw �floatr� r# �uint8r_ �viewr^ r r r )r r� rA rJ �datars r r r! �test_unique_axis s"
"zTestUnique.test_unique_axisr� r r> c C s, t �g d��}t||d�}t|g d�� d S )N)r r r r r r r r r� r- �r r r r )r r� rS �uniqr r r! �test_unique_1d_with_axis� s z#TestUnique.test_unique_1d_with_axisc
C sV t jdt jd�}t|ddddd�\}}}}t|j|j� t|t jdd�� t|t �dg�� t|t �ddg�� t|t �dg�� t|d dddd�\}}}}t|j|j� t|t jdd�� t|t �g �� t|t �g �� t|t �g �� d
}t j|d�}tt |��D ]H}t
|�} || dk�r,d| |< nd | |< tt||d �t j| d�� �qd S ) N)r r )r5 rA r T�r� r� r� r� )r r )r5 r r )r r r r r r r r� ) r r� r� r r rA r r �ranger� r� )
r Z single_zeror� �idxr� �cntr5 Zmultiple_zerosr� Zexpected_shaper r r! �test_unique_axis_zeros� s6 ��
�z!TestUnique.test_unique_axis_zerosc C sh t jg d�dd�}t j�|d�}t �|�}t j|ddd�\}}}d}t|j|j|� t|j|j|� d S )N)�@ r r r r rm rm r r r r r r rm r r� r@ r T)r� r� z6Unique returned different results when asked for index)r r ZmaZ masked_equalr r r� �mask)r rS rT r� Zv2r� r rJ r r r! �test_unique_masked� s
�
z TestUnique.test_unique_maskedc C sD d}dD ]6}t �dgdgg|�}t j|dd�}t|||| � qd S )Nz*sort order incorrect for integer type '%s'Zbhilqr> r r� r� )r �fmtry r r r r r! � test_unique_sort_order_with_axis� s
z+TestUnique.test_unique_sort_order_with_axisc
C s� t �g d�g d�g d�g d�g��|�}d}t �ddg�}tt|�|�|�|� d}t �g d�g d�g�}tt|dd�|�|�|� d}t �g d �g d
�g d �g d
�g�}tt|dd�|�|�|� d }t �ddgddggddgddggg��|�}t j|ddgd d�}tt|d d�||� t|dd
d
d
d�\}}}} d}t|| ||� d}t|| |� d}t| t �d d g�|� t|dd
d
d
d�\}}}} d}t|d d �|f |� d}t|d d �|f |� d}t| t �g d��|� d S )N)r r r r )r r r r z&Unique with 1d array and axis=0 failedr r z&Unique with 2d array and axis=0 failedr� z&Unique with 2d array and axis=1 failed)r r r r� z&Unique with 3d array and axis=2 failedr Tr� z-Unique's return_index=True failed with axis=0z/Unique's return_inverse=True failed with axis=0z.Unique's return_counts=True failed with axis=0z-Unique's return_index=True failed with axis=1z/Unique's return_inverse=True failed with axis=1z.Unique's return_counts=True failed with axis=1)r r r )r r Zastyper r Ztake)
r rA r� rJ rs Zdata3dr� r� r� r� r r r! r� � s\
��"
������zTestUnique._run_axis_testsc C sj t �ddt jt jt jg�}t �|�}t j|dd�}t|t �dt jg�� t|t �dt jt jt jg�� d S )Nr F)Z equal_nan)r r r� r r )r r r� Znot_unqr r r! �test_unique_nanequals s
z TestUnique.test_unique_nanequalsN)r& r' r( r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r! r� � s *
$
,r� )�__doc__Znumpyr Z
numpy.testingr r r r Znumpy.lib.arraysetopsr r r r r
r r r
r� r r� r r r r! �<module> s ( :
Mini Shell