Edit File: test_twodim_base.cpython-37.pyc
B ��Fd�B � @ s| d Z ddlmZmZmZ ddlmZmZmZm Z m Z m Z m Z ddl mZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m Z m Z ddl Z dd� Z G dd � d e�Z!G d d � d e�Z"G d d � d e�Z#G dd� de�Z$G dd� de�Z%G dd� de�Z&dd� Z'dd� Z(dd� Z)dd� Z*d d � Z+d d � Z,G d d!� d!e-�Z.G d"d#� d#e-�Z/G d$d%� d%e-�Z0G d&d'� d'e-�Z1e2d(k�rxe� dS ))z"Test functions for matrix module � )�division�absolute_import�print_function)�TestCase�run_module_suite� assert_equal�assert_array_equal�assert_array_max_ulp�assert_array_almost_equal� assert_raises)�arange�add�fliplr�flipud�zeros�ones�eye�array�diag� histogram2d�tri� mask_indices� triu_indices�triu_indices_from� tril_indices�tril_indices_from�vanderNc C s t | �}t�||�}|S )N)r r �outer)�n�data� 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_twodim_base.py�get_mat s r" c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )�TestEyec C s� t td�tddddgddddgddddgddddgg�� t tddd�tddddgddddgddddgddddggd�� t td�dktdtd�� d S )N� � r �f)�dtype� )r r r �bool)�selfr r r! � test_basic s zTestEye.test_basicc C s� t tddd�tddddgddddgddddgddddgg�� t tddd�tddddgddddgddddgddddgg�� d S )Nr$ r% )�kr ���)r r r )r* r r r! � test_diag+ s zTestEye.test_diagc C sj t tdd�tdddgdddgdddgdddgg�� t tdd�tddddgddddgddddgg�� d S )Nr$ r( r% r )r r r )r* r r r! �test_2d8 s zTestEye.test_2dc C sr t tdddd�tddddgddddgddddgg�� t tdddd�tdddgdddgdddgdddgg�� d S )Nr( r$ � )r, r r% ���)r r r )r* r r r! � test_diag2dD s zTestEye.test_diag2dc C s0 t tddd�ddgddgg� t tddd�ddgddgg� t tddd�ddgddgg� t tddd�ddgddgg� t tddd�ddgddgddgg� t tddd�ddgddgddgg� t tddd�ddgddgddgg� t tddd�ddgddgddgg� t tddd�ddgddgddgg� d S )Nr0 r% r r- r1 r( ���)r r )r* r r r! �test_eye_boundsP s $$$$zTestEye.test_eye_boundsc C s$ t tdddd�ddgddgg� d S )Nr0 �S3)r' � 1� )r r )r* r r r! � test_strings[ s zTestEye.test_stringsc C s$ t tddtd�ddgddgg� d S )Nr0 )r' TF)r r r) )r* r r r! � test_bool_ s zTestEye.test_boolN) �__name__� __module__� __qualname__r+ r. r/ r2 r4 r8 r9 r r r r! r# s r# c @ s6 e Zd Zdd� Zd dd�Zdd� Zdd � Zd d � ZdS ) �TestDiagc C s� dt d� �d�}td�}x td�D ]}|| |||f< q$W tt|�|� td�}|�� }x8td�D ],}|| |||d f< || ||d |f< qdW tt|dd�|� tt|dd�|� d S ) N�d � �l)r? r? )� rA r0 )r, r1 )r �astyper �ranger r �copy)r* �vals�br, �cr r r! � test_vectord s zTestDiag.test_vectorNc C s� |d kr dt d� d �d�}td�}x td�D ]}|||f ||< q0W tt|�|� |d }x$td�D ]}|||d f ||< qhW tt|d�|d d� � x$td�D ]}||d |f ||< q�W tt|d �|d d� � d S ) Nr> r? r% r@ )r? r r( r0 r1 )r" rB r rC r r )r* rE rF r, r r r! � test_matrixr s zTestDiag.test_matrixc C s( t dtd� d ddd�}| �|� d S )Nr> r? r% �Fr@ )�orderr' )r r" rI )r* rE r r r! �test_fortran_order� s zTestDiag.test_fortran_orderc C s� ddgddgddgg}t t|dd�g � t t|dd�dg� t t|dd�ddg� t t|d d�ddg� t t|d d�dg� t t|d d�g � d S ) Nr% r0 r( r$ r? � )r, r r- r1 r3 )r r )r* �Ar r r! �test_diag_bounds� s zTestDiag.test_diag_boundsc C s | � ttdggg� d S )Nr% )� assertRaises� ValueErrorr )r* r r r! � test_failure� s zTestDiag.test_failure)N)r: r; r< rH rI rL rO rR r r r r! r= c s r= c @ s e Zd Zdd� ZdS )� TestFliplrc C sx | � tttd�� td�}|d d �d d d�f }tt|�|� dddgdddgg}dddgdddgg}tt|�|� d S )Nr$ r- r r% r0 r( r? )rP rQ r r r" r )r* �arF r r r! r+ � s zTestFliplr.test_basicN)r: r; r< r+ r r r r! rS � s rS c @ s e Zd Zdd� ZdS )� TestFlipudc C sf t d�}|d d d�d d �f }tt|�|� dddgdddgg}dddgdddgg}tt|�|� d S )Nr$ r- r r% r0 r( r? )r" r r )r* rT rF r r r! r+ � s zTestFlipud.test_basicN)r: r; r< r+ r r r r! rU � s rU c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�TestHistogram2dc C s� t dddddg�}t dddd d g�}t�d d d �}t�d d d �}t||||f�d }t d d d d d d d d d g d d d d d d d d d g d d d d d d d d d g d d d d d d d d d g d d d d d d d d d g d d d d d d d d d g d d d d d d d d d g d d d d d d d d d g d d d d d d d d d g g �}t|j|� t|||�d }t|j|� tttd ��ttd ���\}}}t|td d �� t|t�d dd�� t|t�d dd�� d S )Ng̙� }��?g%���� �?g�$F�� ?g�;c�jY�?gI�+����?g�p5x���?g�?�_��?g�p��� �?g��b�d�?g�g<��=�?r r% � � � ) r �np�linspacer r �T�listrC r )r* �x�y�xedges�yedges�H�answerr r r! � test_simple� s0 zTestHistogram2d.test_simplec C s� t ddddddddg�}t ddddddddg�}t||dddgddggd d �\}}}t d ddddgdddddgdddddgdddddgdddddgdddddgg�}t||d d� t|t�ddd �� t|t�ddd�� d S )Nr% r0 r( r$ r? r )rM r? rM T)rC �normedg g @rA )r r r r rZ r[ )r* r^ r_ rb �xed�yedrc r r r! � test_asym� s " zTestHistogram2d.test_asymc C s� t dddddddddg �}t dddddddddg �}t||ddddgddddggdd�\}}}t dddgdddgdddgg�d }t||d� d S ) Nr% r0 r( r? T)re g �?g �?g "@)r r r )r* r^ r_ rb rf rg rc r r r! � test_norm� s ( zTestHistogram2d.test_normc C sD t j�d�d d }t||dddgddgfd�\}}}t|d� d S )Nr> g �?g ��.A)r$ r? r r% )rC )rZ �random�randr r )r* �rrb rf rg r r r! �test_all_outliers� s "z!TestHistogram2d.test_all_outliersc C sZ t g g ddgddgfd�\}}}t|tdgg�� t g g dd�\}}}t|t�d�� d S )Nr r% )�binsg r$ )r$ r$ )r r r rZ r )r* rT Zedge1Zedge2r r r! � test_empty� s zTestHistogram2d.test_emptyc C sp t dddddddg�}t ddd d d d dg�}d }t|||df�\}}}t ddddgddddgddddgddddgddddgddddgddddgddddgddddgddddgg �}t||� t|t dddddg�� t||d|f�\}}}t ddddddddddg ddddddddddg ddddddddddg ddddddddddg g�}t||� t|t dddddg�� d S )Nr g<D��瑷?g�� ��?g!,��3{�?g�l�yT�?g�f��.�?r% g( @TT\�?g���IZ�?gY�N���?gK?~�?g&W-��U�?) r g�������?g�������?g333333�?g�������?g �?g333333�?gffffff�?g�������?g�������?r% r$ g @g g �?g �?g �?g �?)r r r )r* r^ r_ �edgesrb Zxe�yerc r r r! � test_binparameter_combination� s: z-TestHistogram2d.test_binparameter_combinationN) r: r; r< rd rh ri rm ro rr r r r r! rV � s rV c @ s e Zd Zdd� ZdS )�TestTric C sJ t dddgdddgdddgg�}ttd�|� ttdtd�|�t�� d S )Nr% r r( )r' )r r r r) rB )r* �outr r r! � test_dtype s zTestTri.test_dtypeN)r: r; r< ru r r r r! rs s rs c c s� x�t jd t jd D ]l} t jd| d�}t �|�}t �|�}t|ddgddggfV t||jfV t|j|jfV t|j|jfV qW d S )N�AllFloat� AllInteger)r0 r0 )r' r% r ) rZ � typecodesr �tril�triur r\ r r' )r' rT rF rG r r r! �test_tril_triu_ndim2 s r{ c c s �xt jd t jd D ]�} t jddgddggddgddggddgddggg| d�}t jddgddggddgddggddgddggg| d�}t jddgddggddgddggddgddggg| d�}t �|�}t �|�}t||fV t||fV t|j|jfV t|j|jfV qW d S )Nrv rw r% r )r' )rZ rx r rz ry r r r' )r' rT Za_tril_desiredZa_triu_desiredZa_triu_observedZa_tril_observedr r r! �test_tril_triu_ndim3! s, r| c C st t �ddt jgdddgt jddgg�} t �dddgdddgt jddgg�}|j}tt �| �|� tt �| �|� d S )Nr% r )rZ r �infr\ r rz ry )�arrZout_trilZout_triur r r! �test_tril_triu_with_inf9 s r c C s� xRt jd D ]D} | dkrq t jd| d�}tt �|�j|j� tt �|�j|j� q W t jddgddggd d�}tt �|�j|j� tt �|�j|j� t jdd d�}tt �|�j|j� tt �|�j|j� d S ) N�All�V)r( r( )r' z2001-01-01T12:00z2002-02-03T13:56z2004-01-01T12:00z2003-01-03T13:45� datetime64zf4,f4)rZ rx r r rz r' ry r )rG r~ r r r! �test_tril_triu_dtypeF s r� c c sj t dtj�} t�d��dd�}t|| tddddddg�fV t dtjd�}t|| tdddg�fV d S ) Nr( rX r r% r0 r$ r? � )r rZ rz r �reshaper r )�iurT �iu1r r r! �test_mask_indices\ s r� c c s� t d�} t ddd�}t ddd�}t dddd�}t�ddddgdd d d gd d ddgddddgg�}t�dd��dd�}t|| tddd d d dddddg �fV t|| tdd d dddddddg �fV d|| < t|tddddgddd d gddddgddddgg�fV d||< t|tdddddgddd d d gdddddgdddddgg�fV d||< t|tddddgddddgddddgddddgg�fV d||< t|tdddddgddddd gdddddgdddddgg�fV d S )Nr$ r0 )r, r? )�m)r, r� r% r( rM rA r� rX rW rY � � � � � � � � � r- � i����)r rZ r r r� r )Zil1Zil2Zil3Zil4rT rF r r r! �test_tril_indicesf sJ r� c @ s e Zd Zdd� ZdS )�TestTriuIndicesc c s� t d�}t ddd�}t ddd�}t dddd�}t�ddddgdd d d gd d ddgddddgg�}t�dd��dd�}t|| tddddd d d dddg �fV t|| tdddddd d d d dddddg�fV d||< t|tddddgddddgd d ddgddddgg�fV d||< t|tdddddgd ddddgdddddgdddddgg�fV d||< t|tddddgddddgd d ddgddddgg�fV d||< t|tdddddgd ddddgdddddgdddddgg�fV d S )Nr$ r0 )r, r? )r� )r, r� r% r( rM rA r� rX rW rY r� r� r� r� r� r� r� r� r- r� r� i����)r rZ r r r� r )r* r� Ziu2Ziu3Ziu4rT rF r r r! �test_triu_indices� sJ ( z!TestTriuIndices.test_triu_indicesN)r: r; r< r� r r r r! r� � s r� c @ s e Zd Zdd� ZdS )�TestTrilIndicesFromc C s( t ttt�d�� t ttt�d�� d S )N)r0 )r0 r0 r0 )r rQ r rZ r )r* r r r! �test_exceptions� s z#TestTrilIndicesFrom.test_exceptionsN)r: r; r< r� r r r r! r� � s r� c @ s e Zd Zdd� ZdS )�TestTriuIndicesFromc C s( t ttt�d�� t ttt�d�� d S )N)r0 )r0 r0 r0 )r rQ r rZ r )r* r r r! r� � s z#TestTriuIndicesFrom.test_exceptionsN)r: r; r< r� r r r r! r� � s r� c @ s e Zd Zdd� Zdd� ZdS )� TestVanderc c s� t �ddddg�}t|�}t �dddddgdddddgdddddgdd d ddgg�}t||d d �dd �f fV |jd }x<td �D ]0}t||d �}t||d d �|| |�f fV q�W d S ) Nr r% r1 r( r� i����r$ �Q � rX rM )�N)rZ r r r �shaperC )r* rG �v�powersr� r r r r! r+ � s zTestVander.test_basicc c s� t dddgtjd�}t|�}t� dddgdddgdddgg�}t||fV t d d g�}t|d d �}t� d d dgdd dgg�}t||fV d S )NrY i�r� )r' �y r% � � y �? �?y �? �r( )r� y @y � �)r rZ �int8r r )r* rG r� �expectedr r r! � test_dtypes� s zTestVander.test_dtypesN)r: r; r< r+ r� r r r r! r� � s r� �__main__)3�__doc__� __future__r r r Z numpy.testingr r r r r r r �numpyr r r r r r r r r r r r r r r r r rZ r" r# r= rS rU rV rs r{ r| r r� r� r� �objectr� r� r� r� r: r r r r! �<module> s. $LI/ a /1"