Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/numpy/lib/tests/__pycache__/test_arraysetops.cpython-37.pyc
Назад
B �5�g o � @ 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 d d dge j d�de jdfe j d d dge j d�e j ddd ge jd�ddfe j dddge jd�e je jdfg�dd� �Zej � de j d d dge jd�dd e j d d d d ge jd�fe j d d dge jd�e j dge jd�de j dd d ge jd�fe j d d dge jd�d d e j d d d d ge jd�fe j d d dge j d�dd e j dd d d ge 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 �ddddg�}t �dddddg�}t �dddg�}t||dd�}t||� t �dddddg�}t �dddddddg�}t �dddg�}t||�}t||� tg tg g �� d S ) N� � � � � � T)� assume_unique)�np�arrayr r )�self�a�b�ec�cZed� r �V/home/digitalm-up/venv/lib/python3.7/site-packages/numpy/lib/tests/test_arraysetops.py�test_intersect1d s zTestSetOps.test_intersect1dc C sR G dd� d�}|� }t ||�}t||� t dddgdddg�}t|dddg� 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 �ddddg�}t �ddddg�}t||ddd�\}}}t �dddg�}t||� t|| |� t|| |� t �dddddddg�}t �ddddddddg�}t||dd �\}}}t �ddddg�}t||� t|| |� t|| |� t �ddd dgd ddd gg�}t �ddd dgd dddgg�}t||ddd�\}}}t �||j�}t �||j�} t �ddd dg�} t| || � t| || � t �ddd ddgdd dd dgg�}t �ddd d gd ddd gg�}t||dd �\}}}t �||j�}t �||j�} t �dd dg�} t| || � t| || � d S )Nr r r r � T)r �return_indices� )r) r r � � � � )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 �ddddg�}t �dddddg�}t �dddg�}t||�}t||� t �dddg�}t �dddg�}t �ddddddg�}t||�}t||� t �ddddg�}t �ddddg�}t �ddddddg�}t||�}t||� tg tg g �� d S ) Nr r r r r r r( r* )r r r r )r r r r r r r r � test_setxor1dU s zTestSetOps.test_setxor1dc C s8 t �g �}t �dg�}t �ddg�}tg t|�� tdgt|dd�� tdgt|dd�� tddgt|ddd�� tg t|�� tdgt|�� tddd gt|dd d�� td d ddd gt|d d gdd gd�� tdd gt|d d�� tddd gt|dd gd�� tddgt|dd�� td d dgt|d d gd�� d S ) Nr r r )�to_begin)�to_end���)r5 r6 r r. 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 r r )�dtypeNr6 r r r5 g �?g @g "@c C s2 d� |�}tt|�� t|||d� W d Q R X d S )Nz dtype of `{}` must be compatible)�aryr6 r5 )�formatr � TypeErrorr )r r: �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 i����c C s. t j|||d�}t||� |j|jks*t�d S )N)r: r6 r5 )r r r r9 �AssertionError)r r: r= r> r? �actualr r r � test_ediff1d_scalar_handling� s z'TestSetOps.test_ediff1d_scalar_handlingc s� dd� }t j|tgdhd�� � fdd�}t �d��dd d g�}t �d d d gddd gdddgg�}|||� ddg}d}|||� t �d �}|||� |||� |||� |d|� ||d� |dd� g }|||� |||� |||� d S )Nc S s t �|��� �� }| |kS )N)r �asarray�flatten�tolist)r r r r r � _isin_slow� s z(TestSetOps.test_isin.<locals>._isin_slowr )Zotypes�excludedc s" t | |�}� | |�}t||� d S )N)r r )r r �x�y)� isin_slowr r �assert_isin_equal� s z/TestSetOps.test_isin.<locals>.assert_isin_equal� r r r r, � � r � � �! )r. r* )r r( )r. r r r( )r Z vectorize�boolr �reshaper ) r rH rM r r r �d�frJ r )rL r � test_isin� s( " zTestSetOps.test_isinc C s� �xXdD �]N}ddddg}dddddg| }t �dd ddg�}t||dd �}t||� d |d < t �d d ddg�}t||dd �}t||� d \|d < |d< t �dd dd g�}t||dd �}t||� t �ddddddddddddddg�}dddg| }d dd ddddddd dd d d g}t||�}t||� |dddg| }dddddddddddd ddg}t||�}t||� t �ddddg�}t �dddddg| �}t �dd ddg�}t||�}t||� t �dddddg�}t �ddddddg| �}t �dd dddg�}t||�}t||� t �ddg�}t �ddg| �}t �d d g�}t||�}t||� qW t �dg�}t �dg�}t �d g�}t||�}t||� ttg g �g � d S )N)r r, r r r r r r TF)r r* r )r r* )r r r r )r �multr r r r r r r � test_in1d� s\ & zTestSetOps.test_in1dc C sZ t �ddddddddg�}t �ddg�}t �ddddddddg�}t||�}t||� d S )Nr r r rV �eTF)r r r r )r r r r r r r r �test_in1d_char_array+ s z TestSetOps.test_in1d_char_arrayc C sf x`dD ]X}t �ddddddddddddddg�}dddg| }tt �t||��t||dd�� qW d S ) z Test in1d's invert parameter)r r, r r r r r T)�invertN)r r r r] r )r rY 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 �ddddddg�}tt||dd �|� tt||dd �|� tt||dd �|� tt||dd �|� d S ) Nr( r r r. �? rP FT)r )r r rU 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 r, F)r r r r )r �ar1�ar2r? �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 )Nr r, F)r r r r )r ra rb r? rc 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 )Nr, T)r r r r )r ra rb r? rc 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 N)r9 r, T)r r9 �int�objectr r r )r �dtra rb r? rc r r r �+test_in1d_both_arrays_have_structured_dtype_ s z6TestSetOps.test_in1d_both_arrays_have_structured_dtypec C sf t jddgtd�}t jddgtd�}t �ddg�}t �||�}t||� t j||dd�}t|t �|�� t jdddgtd�}|d d� }t jdddgtd�}|d d� }t �ddg�}t �||�}t||� t j||dd�}t|t �|�� t jdd dgtd�}|d d� }t jddgtd�}t �dd g�}t �||�}t||� t j||dd�}t|t �|�� d S ) N)r r )r9 T)r] )r r r r7 )r r F)r r rh r r r] )r ra rb r? rc r r r �'test_in1d_with_arrays_containing_tuplesi s0 z2TestSetOps.test_in1d_with_arrays_containing_tuplesc C s� t �dddddg�}t �dddddddg�}t �ddddddg�}t||�}t||� t �dddgdddgg�}t �dddddg�}t �ddddddg�}t||�}t||� tg tg g �� d S )Nr r r r r r r )r r r r ) r r r r r rJ rK Zez�zr r r � test_union1d� s zTestSetOps.test_union1dc C s� t �ddddddddg�}t �dddddddg�}t �ddg�}t||�}t||� t �d�}t �d �}t �d d g�}t||�}t||� tg tg g �� t �d t j�}tt|g �jt j� d S ) Nr( r r r r r r � � rO r )r r r r r Zuint32r r9 )r r r r r r r r �test_setdiff1d� s zTestSetOps.test_setdiff1dc C sJ t �dddg�}t �dddg�}t �ddg�}t||dd�}t||� d S )Nr r r r r T)r )r r r r )r r r r? rC r r r �test_setdiff1d_unique� s z TestSetOps.test_setdiff1d_uniquec C s< t �dddg�}t �dddg�}tt||�t �dg�� d S )Nr r r �s)r r r r )r r r r r r �test_setdiff1d_char_array� s z$TestSetOps.test_setdiff1d_char_arrayc C sb t �dddddg�}t �dddddddg�}t||�}t||�}t||�}t||�}t||� d S ) Nr r r r r* r. r r )r r r r r r r )r r r �c1Zaux1Zaux2�c2r r r � test_manyways� s zTestSetOps.test_manyways)%r" r# r$ r r' r3 r4 r8 �pytest�mark� parametrizer r Zint64�nanZfloat32�int8rA Zint16Zfloat64Zint32rD rX rZ r\ r^ r` rd re rf rj rk rm rp rq rs rv r r r r r sf '&)< !r c @ sf e Zd Zdd� Zdd� Zdd� Zdd� Zej� d d d g�d d � �Z dd� Z dd� Z dd� Z dd� ZdS )� TestUniquec C s� dd� }dddddddgd }ddddg}ddd dg}ddd dd ddgd }t �ddddgd�}g }|�t jd � |�t jd � |�d � |�d � x6|D ].}t �||�} t �||�} || | ||||� q�W d}t �t|�|�} || d d �< t �t|�|�} || d d �< || | ||||� ddg}t �tt ||��|�} t �tt ||��|�} || | ||||� dddg} t t � | �ddg� dddg}dddg} d dd dddg} 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� dt jdt jg}ddt jg}dd dg}ddd dg}dddg}tt � |�|� tt j |dd�||f� tt j |dd�||f� tt j |dd �||f� d t jd td t j�tdt j�g}d d td t j�g}dd dg}ddd ddg}dddg}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}dd dg}ddd dg}dddg}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}dd dg}ddd dg}dddg}tt � |�|� tt j |dd�||f� tt j |dd�||f� tt j |dd �||f� t jgd% }t jg}d g}d d d d g}d%g}tt � |�|� tt j |dd�||f� tt j |dd�||f� tt j |dd �||f� d S )&Nc 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)r; r r ) r r r0 r1 r ri Zbase_msgr@ �v�jZj1Zj2Zj3r r r � check_all� sL z,TestUnique.test_unique_1d.<locals>.check_allr r r r r, r r � AllInteger�AllFloatz datetime64[D]ztimedelta64[D]�O)� �iy �? y �? �)r r )r r T)r r� .)r~ )r~ r g @g �?)r� y @ �y �? �?g �natz 2020-12-26z 2020-12-24�D�hr )r �multiply�extend� typecodesr> r �empty�len�list�zipr r Z chararrayZzerosr r9 Zintprz �complexZ datetime64Z timedelta64)r r� r r r0 r1 r �typesri ZaaZbbZunq�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 r, r )�axis���) r r< �_run_axis_testsrh rg r Z AxisErrorr r )r r r r �test_unique_axis_errorsp s z"TestUnique.test_unique_axis_errorsc C s^ d}dddgdddgg}t �|�}tt|dd�t|dd�|� tt|dd�t|dd�|� d S )Nz Unique failed on list of listsr r )r� )r rE r r )r r@ ZinpZinp_arrr r r �test_unique_axis_listx s z TestUnique.test_unique_axis_listc C s g }|� tjd � |� tjd � |�d� |�d� |�dtfdtfg� |�dtfdtfg� x|D ]}| �|� qjW d}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� z datetime64[D]ztimedelta64[D]r r z&Non-bitwise-equal booleans test failedr, )r9 r7 r FTr )r� z"Negative zero equality test failedg �g )r� r r� r> rg �floatr� r �uint8rU �viewrT r r r )r r� r9 r@ �datarc r r r �test_unique_axis s" "zTestUnique.test_unique_axisr� r r7 c C s< t �ddddddddg�}t||d�}t|ddddg� d S )Nr r r r )r� )r r r r )r r� rJ �uniqr r r �test_unique_1d_with_axis� s z#TestUnique.test_unique_1d_with_axisc C sZ 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�}xXtt |��D ]H}t |�} || dk�r.d| |< nd | |< tt||d �t j| d�� �q W d S ) N)r r )r/ r9 r T)r� r~ r r� )r r )r/ r r )r r r r r r r )r� ) r r� r{ r r r9 r r �ranger� r� ) r Z single_zeror� �idxr� �cntr/ Zmultiple_zerosr� Zexpected_shaper r r �test_unique_axis_zeros� s0 z!TestUnique.test_unique_axis_zerosc C s� t jdddddddddddddddgdd�}t j�|d�}t �|�}t j|d d d �\}}}d }t|j|j|� t|j|j|� d S ) N�@ r r r r r_ r� )r9 T)r~ r� z6Unique returned different results when asked for index)r r �maZ masked_equalr r r� �mask)r rJ rK r� Zv2r� r r@ r r r �test_unique_masked� s $ z TestUnique.test_unique_maskedc C sH d}x>dD ]6}t �dgdgg|�}t j|dd�}t|||| � q W d S )Nz*sort order incorrect for integer type '%s'Zbhilqr7 r )r� )r r r r )r �fmtri r r r r r � test_unique_sort_order_with_axis� s z+TestUnique.test_unique_sort_order_with_axisc C s t �ddddgddddgddddgddddgg��|�}d}t �ddg�}tt|�|�|�|� d}t �ddddgddddgg�}tt|dd�|�|�|� d}t �dddgdddgdddgdddgg�}tt|dd�|�|�|� d}t �ddgddggddgddggg��|�}t j|ddgdd�}tt|dd�||� t|dd d d d �\}}}} d }t|| ||� d }t|| |� d }t| t �ddg�|� t|dd d d d �\}}}} d}t|d d �|f |� d}t|d d �|f |� d}t| t �dddg�|� d S )Nr r z&Unique with 1d array and axis=0 failedz&Unique with 2d array and axis=0 failed)r� z&Unique with 2d array and axis=1 failedz&Unique with 3d array and axis=2 failedr T)r� r~ r r� 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 Zastyper r Ztake) r r9 r� r@ rc Zdata3dr� r� r� r� r r r r� � sH * zTestUnique._run_axis_testsN)r" r# r$ r� r� r� r� rw rx ry r� r� r� r� r� r r r r r| � s *$ r| )�__doc__�numpyr Z numpy.testingr r r r Znumpy.lib.arraysetopsr r r r r r r r rw r r| r r r r �<module> s ( :
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка