?????????? ????????? - ??????????????? - /usr/lib64/python3.9/site-packages/numpy/lib/tests/__pycache__/test_arraysetops.cpython-39.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||� |j|jks*J �d S )NrC )r r r rA )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 ( :
| ver. 1.6 |
Github
|
.
| PHP 8.2.30 | ??????????? ?????????: 0 |
proxy
|
phpinfo
|
???????????