Файловый менеджер - Редактировать - /opt/imh-python/lib/python3.9/site-packages/numpy/lib/tests/__pycache__/test_arraysetops.cpython-39.pyc
Ðазад
a e�hH� � @ 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!� �Zej � d"g d#��d$d%� �Zej � d"g d#��d&d'� �Zd(d)� Zej � d"g d#��d*d+� �Zej � d"g d#��d,d-� �Zd.d/� Zej � d"g d#��d0d1� �Z ej � d"dd2g�d3d4� �Z d5d6� Z ej � d7e je jfe je jfe j e j fe j e j fe j e jfe je j fg�ej � d"g d#��d8d9� ��Z!ej � d"g d#��d:d;� �Z"d<d=� Z#d>d?� Z$d@dA� Z%dBdC� Z&dDdE� Z'dFdG� Z(dHdI� Z)dJdK� Z*dLdM� Z+dNdO� Z,dPdQ� Z-dS )R� 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 ��/root/rpmbuild/BUILDROOT/imh-python39-modules-3.9.7-92.el8.x86_64/opt/imh-python/lib/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_handling�kind)N�sort�tablec 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� �dkr�g } || |� ||| � || | � tt jt jfD ]�} �dk�r| t jk�rq�| t jt jhv �r:t jg d �| d�} n| thv �rTt �g d��} t jg | d�} || | � || | � || | � q�d S )Nc S s t �|��� �� }| |v S )N)r �asarray�flatten�tolist)r r r r r! � _isin_slow� s z(TestSetOps.test_isin.<locals>._isin_slowr )Zotypes�excludedc s& t | |�d�}� | |�}t||� d S )N�rO )r r )r r �x�y�Z isin_slowrO r 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. rQ r@ �TFF)r Z vectorize�boolr# �reshaper �int64�float64) r rO rU r[ r r r �d�frX rA �arZ empty_arrayr rZ r! � test_isin� s@ zTestSetOps.test_isinc C sJ dD �]�}g d�}g d�| }t �g d��}t||d|d�}t||� d|d< t �g d ��}t||d|d�}t||� d \|d< |d < t �g d ��}t||d|d�}t||� t �g d ��}g d�| }g d�}t|||d�}t||� |g d�| }g d�}t|||d�}t||� t �g d��}t �g d�| �}t �g d��}t|||d�}t||� t �g d��}t �g d�| �}t �g d��}t|||d�}t||� t �ddg�}t �ddg| �}t �ddg�}t|||d�}t||� qt �dg�}t �dg�}t �dg�}t|||d�}t||� |dv �rFttg g |d�g � d S )N�r r2 r r �TFTTT�r rO r0 r )FFTT)r r0 r )TFTF�r r r r r r r r r r r r r r r] )FTFTTTTTTFTFFFrW )r r r )TTTTTTTTTTTFTT)r r r r r )r r r r r r )TFTTTr r F> NrP �r r r r )r rO �multr r r r r r r! � test_in1d sZ zTestSetOps.test_in1dc C sB t �g d��}t �ddg�}t �g d��}t||�}t||� d S )N)r r r rh �er rs r r r )TFTFFTFFrp r: r r r! �test_in1d_char_arrayA s z TestSetOps.test_in1d_char_arrayc C s� dD ]D}t �g d��}g d�| }tt �t|||d��t||d|d�� q|dv r�dD ]Z}t jg d�t jd�}g d�| }t j|t jd�}tt �t|||d��t||d|d�� qVd S ) z Test in1d's invert parameterrl ro r] rW T��invertrO > NrP r@ N)r r r rv r �float32)r rO rq r r r r r! �test_in1d_invertJ s � � �zTestSetOps.test_in1d_invertc C s� t �d��dd�}t �dd��dd�}t �dd��dd�}t �g d��}tt||d|d �|� tt||d |d �|� tt||d|d �|� tt||d |d �|� d S ) Nr. r r r4 �? r_ )FFFTTTTrn F)r r# re r r r )r rO r r Zlong_br r r r! �test_in1d_ravel_ s, � �� �� ��zTestSetOps.test_in1d_ravelc C s� t jg d�t jd�}t jg d�t jd�}t jg d�td�}t|t||�� tt �|�t||dd�� t jg d�t jd�}t jg d�t jd�}t �g d ��}t||dd �}t||� d S ) z(Hit the standard isin code with integers)r r r r r r � e��Ar@ )r r r r{ )r r r r r r r T�rv r )r r r r r r{ rm r N)r r rf rd r r rv )r r r rI r r r r r! �!test_in1d_hit_alternate_algorithmt s z,TestSetOps.test_in1d_hit_alternate_algorithmc C s^ t �ddg�}t �g d��}t �ddg�}t|t|||d�� tt �|�t||d|d�� dS )z&Test that in1d works for boolean inputTF)FFFrW ru N)r r r r rv )r rO r r rI r r r! �test_in1d_boolean� s � �z TestSetOps.test_in1d_booleanrP c C sb t j�d�}|jdddd�}|jdddd�}t||�}|�d�}|�d�}t|t|||d�� dS )z(Test that in1d works for timedelta inputr �d r2 )�size�timedelta64[s]rW N)r �randomZ RandomState�randintr �astyper )r rO Zrstater r �truthZ a_timedeltaZ b_timedeltar r r! �test_in1d_timedelta� s z TestSetOps.test_in1d_timedeltac C sR t jg d�dd�}|}t�t�� t||dd� W d � n1 sD0 Y d S )N�r r r r� r@ rQ rW )r r �pytest�raises� ValueErrorr �r r r r r r! � test_in1d_table_timedelta_fails� s z*TestSetOps.test_in1d_table_timedelta_failsz dtype1,dtype2c C s� t �|t j�}t jg d�|d�}|r8t jg d�|d�}nt jg d�|d�}t �g d��}|dko�t|t jkot|t jk|t jko�|t jkf�}|r�tjt dd�� t |||d � W d � q�1 s�0 Y nt t |||d �|� d S ) z7Test that in1d works as expected for mixed dtype input.)r r r r r@ )i����r � )r� r � )TTFFrQ zexceed the maximum)�matchrW N) r Z issubdtypeZ signedintegerr �any�int8�int16r� r� � RuntimeErrorr r ) r Zdtype1Zdtype2rO Zis_dtype2_signed�ar1�ar2rI Zexpect_failurer r r! �test_in1d_mixed_dtype� s �.z TestSetOps.test_in1d_mixed_dtypec C s� t jd D ]~}t jg d�td�}t jg d�|d�}t jg d�td�}tt|||d�|� || }}t jg d�td�}tt|||d�|� q dS ) z4Test that in1d works as expected for bool/int input.� AllIntegerrc r@ �r r r r )FTTrW )TTTTN)r � typecodesr rd r r )r rO rA r r rI r r r! �test_in1d_mixed_boolean� s z"TestSetOps.test_in1d_mixed_booleanc C s< d g}t �dgd �}t �dg�}t �||�}t||� d S �Nr r2 Frp �r r� r� rI �resultr r r! � test_in1d_first_array_is_object� s z*TestSetOps.test_in1d_first_array_is_objectc C s: d}t �d gd �}t �dg�}t �||�}t||� d S r� rp r� r r r! � test_in1d_second_array_is_object� s z+TestSetOps.test_in1d_second_array_is_objectc C s< d g}t �d gd �}t �dg�}t �||�}t||� d S )Nr2 Trp r� r r r! � test_in1d_both_arrays_are_object� 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 �dtr� r� rI r� 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@ Tr| )r� )r r r r> )r� �r r r F)r r r� r r rv r� r r r! �'test_in1d_with_arrays_containing_tuples� s0 z2TestSetOps.test_in1d_with_arrays_containing_tuplesc C s t �g d��}t �g d��}ttt||dd� t jg d�td�}t jg d�td�}ttt||dd� t jt jfD ]�}t jg d�|d�}t jd t �|�j g|d�}tt t||dd� t j||d d�}t |d gd gd � t j||dd�}t |d gd gd � qpd S )z&Test that in1d raises expected errors.)r r r r r )r r r. r0 r2 Z quicksortrW )r r r r r r@ rQ )r> r r r r r> NTFr rP ) r r r r� r r� �int32rf Ziinfo�maxr� r )r r� r� Zobj_ar1Zobj_ar2rA Z overflow_ar2r� r r r! �test_in1d_errors s" �zTestSetOps.test_in1d_errorsc 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 rX rY Zez�zr r r! � test_union1d3 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 � � r^ r )r r r r r# Zuint32r rA r: r r r! �test_setdiff1dE 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_uniqueW 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! �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 �c1Zaux1Zaux2�c2r r r! � test_manywaysc s zTestSetOps.test_manyways).r& r' r( r" r, r9 r; r? r� �mark� parametrizer r rf �nanrw r� rK r� rg r� rN rk rr rt rx rz r} r~ r� r� �uint8Zuint16r� r� r� r� r� r� r� 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 r� Zbase_msgrJ �v�jZj1Zj2Zj3r r r! � check_allr 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 r� �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 �multiply�extendr� rH r �empty�len�list�zipr r Z chararrayZzerosr rA Zintpr� �complexZ datetime64Z timedelta64)r r� r r r6 r7 r �typesr� ZaaZbb�unq�invZa1Za2Za2_invZa1_idxZa3_idxZa3_invZuaZua_idxZua_invZua_cntr� Zall_nansr r r! �test_unique_1dp 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_testsr� r� r Z AxisErrorr r# r$ r r r! �test_unique_axis_errors 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 rR r r )r rJ �inpZinp_arrr r r! �test_unique_axis_list! 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 r� �floatr� r# r� re �viewrd r r r )r r� rA rJ �datar� 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� rX �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_zerosD 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 ry ry r r r r r r ry r r� r@ r T)r� r� z6Unique returned different results when asked for index)r r �maZ masked_equalr r r� �mask)r rX rY r� Zv2r� r rJ r r r! �test_unique_maskedh 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 �fmtr� r r r r r! � test_unique_sort_order_with_axisu 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 r� r r Ztake) r rA r� rJ r� 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� n s * $ ,r� )�__doc__�numpyr 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 ( d
| ver. 1.1 | |
.
| PHP 8.3.30 | Ð“ÐµÐ½ÐµÑ€Ð°Ñ†Ð¸Ñ Ñтраницы: 0 |
proxy
|
phpinfo
|
ÐаÑтройка