Edit File: test_arraysetops.cpython-37.pyc
B ��Fd�= � @ s� d Z ddlmZmZmZ ddlZddlmZm Z m Z m Z m Z ddl mZmZmZmZmZmZmZmZ G dd� de �ZG dd � d e �Zed kr�e� dS ) z-Test functions for 1D array set operations. � )�division�absolute_import�print_functionN)�run_module_suite�TestCase�assert_array_equal� assert_equal� assert_raises)�ediff1d� intersect1d�setxor1d�union1d� setdiff1d�unique�in1d�isinc @ sl e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� ZdS )� 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�c�ed� r" ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/site-packages/numpy/lib/tests/test_arraysetops.py�test_intersect1d s zTestSetOps.test_intersect1dc 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# � test_setxor1d$ s zTestSetOps.test_setxor1dc C st 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�� ttt �d��t j��sPt�ttt �d�dd�t j��spt�d S ) Nr r r )�to_begin)�to_end���)r( r) r � r r% r& )r r r r � isinstance�matrix�AssertionError)r Z zero_elemZone_elemZtwo_elemr" r" r# � test_ediff1d< s" & zTestSetOps.test_ediff1dc 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_slowT s z(TestSetOps.test_isin.<locals>._isin_slowr )�otypes�excludedc s" t | |�}� | |�}t||� d S )N)r r )r r �x�y)� isin_slowr" r# �assert_isin_equalX s z/TestSetOps.test_isin.<locals>.assert_isin_equal� r r r � � � r � � �! )r+ r& )r r% )r+ r r r% )r � vectorize�bool�arange�reshaper ) r r3 r9 r r r �d�fr6 r" )r8 r# � test_isinP 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_in1dx s\ & zTestSetOps.test_in1dc C sZ t �ddddddddg�}t �ddg�}t �ddddddddg�}t||�}t||� d S )Nr r r rE �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 rL r )r rH r r r" r" r# �test_in1d_invert� 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+ �? r= FT)r )r rC rD r r r )r r r Zlong_br r" r" r# �test_in1d_ravel� s zTestSetOps.test_in1d_ravelc C sj t �dddddg�}t �dddddddg�}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# � 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 � � r< r" )r r r r rC �uint32r �dtype)r r r r r r" r" r# �test_setdiff1d� s zTestSetOps.test_setdiff1dc 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_manywaysN)�__name__� __module__� __qualname__r$ r' r/ rG rI rK rM rO rP rU rW rZ r" r" r" r# r s (< r c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )� 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�� 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_indexr r �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)�formatr r ) r r �i1�i2r �dtZbase_msg�msg�v�j�j1�j2�j3r" 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)ra rq .)r �multiply�extend� typecodes�appendr �empty�len�list�zipr r � chararray�zeros)r rm r r rd re r �typesrf �aa�bbZunq�inv�a1�a2Za2_invr" r" r# �test_unique_1d sN 0 zTestUnique.test_unique_1dc C sX t t| jt� t t| jdtfdtfg� t ttt�d�dd� t ttt�d�dd� d S )Nr r r; r )�axis���) r � TypeError�_run_axis_tests�object�int� ValueErrorr r rC )r r" r" r# �test_unique_axis_errorsf 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 r0 r r )r rg �inpZinp_arrr" r" r# �test_unique_axis_listn 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 )Nrn ro z datetime64[D]ztimedelta64[D]r r z&Non-bitwise-equal booleans test failedr; )rT r* r FTr )r� z"Negative zero equality test failedg �g )rt r ru rv r� �floatr� rC �uint8rD �viewrB r r r )r r} rT rg �data�resultr" r" r# �test_unique_axisu s" "zTestUnique.test_unique_axisc 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 rN r� )rT T)r` rb z6Unique returned different results when asked for index)r r �ma� masked_equalr r r� �mask)r r6 r7 rh �v2rr r rg r" r" r# �test_unique_masked� s , z TestUnique.test_unique_maskedc C s� t �ddddgddddgddddgddddgg��|�}d}t �ddg�}tt|�|�|�|� d}t �ddddgddddgg�}tt|dd�|�|�|� d}t �dddgdddgdddgdddgg�}tt|dd�|�|�|� d}t �|gd �}|d d d�f }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 �d ddg�|� 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 .r T)r� r` ra rb 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 �astyper r �dstack) r rT r� rg r� Zdata3d�uniq�idxr� �cntr" r" r# r� � sB *zTestUnique._run_axis_testsN) r[ r\ r] r� r� r� r� r� r� r" r" r" r# r^ � s e r^ �__main__)�__doc__� __future__r r r �numpyr Z numpy.testingr r r r r Znumpy.lib.arraysetopsr r r r r r r r r r^ r[ r" r" r" r# �<module> s ( q C