Edit File: test_defmatrix.cpython-37.pyc
B ��Fdy: � @ s d dl mZmZmZ d dlZd dlZd dlmZmZm Z d dl m Z m Z m Z mZmZmZmZmZ d dlmZ d dlmZ G dd� de �ZG d d � d e �ZG d d � d e �ZG d d� de �ZG dd� de �ZG dd� de �Z G dd� de �Z G dd� de �Z G dd� de �Z e dk�re � dS )� )�division�absolute_import�print_functionN)�matrix�asmatrix�bmat)�TestCase�run_module_suite�assert_� assert_equal�assert_almost_equal�assert_array_equal�assert_array_almost_equal� assert_raises)� matrix_power)�matc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestCtorc C s t �ddgddgg�}t|�}tt �|j|k�� td�}t||g||gg�}t �ddddgddddgddddgddddgg�}tt �|j|k�� tt �|j|k�� t �ddgdd gg�}tddddgdddd gg�}tt �t||g�|k�� t �d�}t|�} t| jd k� d S ) N� � � � zA,A;A,A� � � � )r r ) �np�arrayr r �all�Ar �arange�shape) �selfr �mA�B�C�D�EZAEresult�vec�mvec� r) ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/site-packages/numpy/matrixlib/tests/test_defmatrix.py� test_basic s" zTestCtor.test_basicc C s t ttd� d S )N�invalid)r � ValueErrorr )r! r) r) r* �test_exceptions% s zTestCtor.test_exceptionsc C s, t �ddgddgg�}t �ddgddgg�}t �ddddgddddgddddgddddgg�}t �ddddgddddgddddgddddgg�}tt �td �|k�� tt �td d |id �|k�� tttd d |id � tt �td d |id |id �|k�� td||d�||d�d �}tt �||k�� d S )Nr r r r r r r r zA,A;A,Ar )�ldict)�gdict)r/ r0 zA,B;C,D)r r# )r$ r% )r r r r r r � TypeError)r! r r# ZAresultZ mixresult�b2r) r) r* �test_bmat_nondefault_str) s" "z!TestCtor.test_bmat_nondefault_strN)�__name__� __module__� __qualname__r+ r. r3 r) r) r) r* r s r c @ st e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� ZdS ) �TestPropertiesc C s� t ddddgddddgddddgddddgg�}t ddddg�}t dd dd g�j}d }t||jdd �� t||jdd �� t||�� � t|tj|dd �� t|tj|dd �� t|t�|�� d S )zeTest whether matrix.sum(axis=1) preserves orientation. Fails in NumPy <= 0.9.6.2127. r r r r r r � r r � � )�axisN)r �Tr �sumr r )r! �MZsum0Zsum1Zsumallr) r) r* �test_sum> s zTestProperties.test_sumc C s� t dddgdddgg�}t|�� d� t|�d�t dd d gg�� t|�d�t dgd gg�� tt�|�d� ttj|dd �t dd d gg�� ttj|dd �t dgd gg�� t dddg�}t|�� dk� d S ) Nr r r r r r i� r � � �x )r; )r r �prodr r )r! �x�yr) r) r* � test_prodQ s zTestProperties.test_prodc C s� t dddgdddgg�}t|�� d� t|�d�t dddgg�� t|�d�t dgdgg�� tt�|�d� ttj|dd�t dddgg�� ttj|dd�t dgdgg�� d S ) Nr r r r r r r )r; )r r �maxr )r! rD r) r) r* �test_max^ s zTestProperties.test_maxc C s� t dddgdddgg�}t|�� d� t|�d�t dddgg�� t|�d�t dgdgg�� tt�|�d� ttj|dd�t dddgg�� ttj|dd�t dgdgg�� d S ) Nr r r r r r r )r; )r r �minr )r! rD r) r) r* �test_minh s zTestProperties.test_minc C sh t �d��d�}t|�� dk� tt �|�d�t �ddg�k�� tt �|�d�t �ddg�k�� d S )Nr )r r r r r r )r r �reshaper �ptpr r )r! rD r) r) r* �test_ptpr s "zTestProperties.test_ptpc C sT t �d��d�}|�t j�}t|jdd�|jdd�� t|jdd�|jdd�� d S )N� )r r r )�ddofr )r r rK �viewr r �var)r! rD �mxr) r) r* �test_varx s zTestProperties.test_varc C s dd l m} t�ddgddgg�}t|�}tt�|�|�|j�� tt� t�t� |�|j k��� tt� t�t� |�|j k��� tt� ||j k�� |d| }t|�}tt�|�|�|j�� tt� t�t� |�|j k��� tt� t�t� |��� |j k��� d S )Nr g �?g @g @g @y @)� numpy.linalg�linalgr r r r �allclose�inv�Ir � transposer< �Hr �conj)r! rU r r" r# �mBr) r) r* r+ ~ s zTestProperties.test_basicc C s@ t t�d��dd��}t ddgddgdd gg�}t|j|� d S ) Nr r r g��:���gn�)t ��?g#��q��g#��q�?gn�)t ��?g�6�B�q��)r r r rK r rX )r! rD Zxpinvr) r) r* � test_pinv� s zTestProperties.test_pinvc C sr t �d��dd�}t|�}t|�d }tt �||d k�� tt �|t|d �k�� tt �|t|d �k� � tt �||k �� tt �||k�� tt �||k�� tt �||k � � tt �||k � � tt �||k�� tt �||k�� tt �||k� � tt �||k�� tt �||k� � tt �||k�� tt �t|�dk� � tt �t|dk��� d S )N�d r@ g�������?r )r r rK r r r �any�abs)r! r r" r\ r) r) r* �test_comparisons� s& z TestProperties.test_comparisonsc C s: t �d��dd�}t|�}d|d< t|d |d k� d S )Nr^ r@ i����)r r )r r rK r r )r! r r" r) r) r* � test_asmatrix� s z TestProperties.test_asmatrixc C s@ t ddgddgg�}t|�� t d�k� t|�� t d�k� d S )Nr r r g �?)r r r= �mean)r! r r) r) r* � test_noaxis� s zTestProperties.test_noaxisc C s( t ddgddgg�}tt|�dk� d S )Nr r z matrix([[1, 0], [0, 1]]))r r �repr)r! r r) r) r* � test_repr� s zTestProperties.test_reprc C s* t d�}t dgdgdgg�}t||� d S )NzTrue; True; FalseTF)r r )r! r r# r) r) r* � test_make_bool_matrix_from_str� s z-TestProperties.test_make_bool_matrix_from_strN)r4 r5 r6 r? rF rH rJ rM rS r+ r] ra rb rd rf rg r) r) r) r* r7 = s r7 c @ s e Zd Zdd� ZdS )� TestCastingc C s� t �d��dd�}t|�}|�� }t �dt j�d }|| }t|jj t jk� tt � ||k�� tt � ||d k�� |�� }t �dt j �}|| }t|jj t j k� tt � ||k�� d S )Nr^ r@ )r@ r@ g�������?) r r rK r �copy�ones�float64r �dtype�typer � complex128)r! r r" r\ �OZmCr) r) r* r+ � s zTestCasting.test_basicN)r4 r5 r6 r+ r) r) r) r* rh � s rh c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TestAlgebrac C s. dd l m} t�ddgddgg�}t|�}t�d�}x2td�D ]&}tt�|| j |�� t� ||�}q>W |� |�}t�d�}x4td�D ](}tt�|| j |�� t� ||�}q�W tt�|| j t� ||��� tt�|| j || �� tt�d| j d| �� t|�}|d9 }tt�|j d| �� d S ) Nr g �?g @g @g @r r r ) rT rU r r r �identity�ranger rV r �dotrW )r! rU r r" r# �iZAinvZmA2r) r) r* r+ � s$ zTestAlgebra.test_basicc C st t d�}|�� }|dC }|�� }|dC }|�� }|dC }t||d � t|t�||�� tt�||�t�d�� dS )z<Test raising a matrix to an integer power works as expected.z 1. 2.; 3. 4.r ���N)r ri r r rs �eye)r! �m�m2�mi�m4r) r) r* �test_pow� s zTestAlgebra.test_powc C sr t ddgddgg�}y d| W n tk r4 Y n X | �d� y|t� W n tk rb Y n X | �d� dS )z:Check that 'not implemented' operations produce a failure.g �?g @g @g @z)matrix.__rpow__ doesn't raise a TypeErrorz?matrix.__mul__ with non-numeric object doesn't raisea TypeErrorN)r r1 �fail�object)r! r r) r) r* �test_notimplemented� s z TestAlgebra.test_notimplementedN)r4 r5 r6 r+ r{ r~ r) r) r) r* rp � s rp c @ s e Zd Zdd� ZdS )�TestMatrixReturnc & C s` t dgdd�}dddgfdddt�dg�d �}d d d d ddddddddddddddd d d d d d!d"d#d$d%d d&d'd(d)d*d+d,d-d.g&}x�t|�D ]v}|�d/�s�||kr�q�t||�}t|tj�r�|� d� |� d� ||kr�|| }nd0}||� }t t |�t kd1| � q�W t t |j �t k� t t |j�t k� t d2g��� \}} t t |�tjk� t t | �tjk� d S )3Ng �?�f8)rl )�intc)g g �?r )r )r r )�astype�clip�compress�repeatrK �swapaxesrs �argmin�choose�dump�dumps�fill�getfield�getA�getA1�item�nonzero�put�putmask�resize� searchsorted�setflags�setfield�sort� partition� argpartition�take�tofile�tolist�tostring�tobytesr r_ r= �argmaxrI rG rc rQ rL rC �std�ctypes�itemset�_r) z%sg )r r r �dir� startswith�getattr� isinstance� collections�Callabler� r� r rm �real�imagr� �ndarray) r! �aZ methodargsZexcluded_methods�attrib�f�args�b�c�dr) r) r* �test_instance_methods s@ z&TestMatrixReturn.test_instance_methodsN)r4 r5 r6 r� r) r) r) r* r s r c @ s e Zd Zdd� ZdS )� TestIndexingc C sf t t�dt��}t�dt�}dddg|d d �df< |dk|d d �df< t|ddgddgddgg� d S ) N)r r )r r g�������?g�������?g333333�?r g �?r )r r �zeros�floatr )r! rD rE r) r) r* r+ = s zTestIndexing.test_basicN)r4 r5 r6 r+ r) r) r) r* r� < s r� c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dS )�TestNewScalarIndexingc C s t ddgddgg�| _d S )Nr r r r )r r� )r! r) r) r* �setUpF s zTestNewScalarIndexing.setUpc C s | j }|d }t|jd� d S )Nr r )r� r �ndim)r! r� rD r) r) r* �test_dimesionsI s z$TestNewScalarIndexing.test_dimesionsc C s* | j }t�||g�}t|jdddg� d S )Nr )r� r r r r )r! r� rD r) r) r* �test_array_from_matrix_listN s z1TestNewScalarIndexing.test_array_from_matrix_listc C s$ | j }t|�� ddgddgg� d S )Nr r r r )r� r r� )r! r� r) r) r* �test_array_to_listS s z(TestNewScalarIndexing.test_array_to_listc C s� | j }|ddddgf }tt|t�� t|tdddgg�� |ddg }tt|t�� t|tddgddgg�� |dgdggddgddggf }tt|t�� t|tddgddgg�� d S )Nr r r r r )r� r r� r r )r! r� rD r) r) r* �test_fancy_indexingW s z)TestNewScalarIndexing.test_fancy_indexingc C s� t dddgdddgg�}t|d d t dddgg�� t|d d jd� t|d jd� t|d d �df jd � t d�}t|d d� t|d d� t|d d �df j|j� d S ) Nr r r r r r r )r r )r r )r r )r r r )r! rD r) r) r* �test_matrix_elementc s z)TestNewScalarIndexing.test_matrix_elementc C s* t t�dt��}t|d |d d � d S )N)r r )r r r )r r r� r� r )r! rD r) r) r* �test_scalar_indexingo s z*TestNewScalarIndexing.test_scalar_indexingc C s� t t�d��}t|dd d �f ddgg� t|dd d �f ddgg� t|d d �df dgdgg� t|d d �df dgdgg� d S )Nr r r )r r rv r )r! rD r) r) r* �test_row_column_indexings s z.TestNewScalarIndexing.test_row_column_indexingc C sv t �d�}d|_t|�}t|d d �t �ddg�f |d d �df � t|t �dddg�d d �f |dd d �f � d S )Nr )r r TFr )r r r r r r )r! r rD r) r) r* �test_boolean_indexingz s ,z+TestNewScalarIndexing.test_boolean_indexingc C sv t �d�}d|_t|�}t|d d �ddgf |d d �d d d�f � t|dddgd d �f |d d d�d d �f � d S )Nr )r r r r ru r )r r r r r )r! r rD r) r) r* �test_list_indexing� s ,z(TestNewScalarIndexing.test_list_indexingN) r4 r5 r6 r� r� r� r� r� r� r� r� r� r� r) r) r) r* r� E s r� c @ s e Zd Zdd� Zdd� ZdS )� TestPowerc C sP t �ddgddgg�}ttt|d��t jk� t|�}ttt|d��tk� d S )Nr r r )r r r rm r r� r r )r! r� r) r) r* �test_returntype� s zTestPower.test_returntypec C s, t tddgddggd�ddgddgg� d S )Nr r r )r r )r! r) r) r* � test_list� s zTestPower.test_listN)r4 r5 r6 r� r� r) r) r) r* r� � s r� c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )� TestShapec C s* t �dgdgg�| _tdgdgg�| _d S )Nr r )r r r� r rw )r! r) r) r* r� � s zTestShape.setUpc C s t | jjd� t | jjd� d S )N)r r )r r� r rw )r! r) r) r* � test_shape� s zTestShape.test_shapec C s, t t�| j�jd� t t�| j�jd� d S )N)r )r r �ravelr� r rw )r! r) r) r* �test_numpy_ravel� s zTestShape.test_numpy_ravelc C s( t | j�� jd� t | j�� jd� d S )N)r )r r )r r� r� r rw )r! r) r) r* �test_member_ravel� s zTestShape.test_member_ravelc C s( t | j�� jd� t | j�� jd� d S )N)r )r r )r r� �flattenr rw )r! r) r) r* �test_member_flatten� s z TestShape.test_member_flattenc C s. t �dddgdddgg�}tt �|�ddddddg� tt j|dd�ddddddg� tt �|j�ddddddg� tt j|jd d�ddddddg� tdddgdddgg�}tt �|�ddddddg� tt j|dd�ddddddg� tt �|j�ddddddg� tt j|jd d�ddddddg� d S ) Nr r r r r r �F)�orderr )r r r r� r< r )r! rD r) r) r* �test_numpy_ravel_order� s " z TestShape.test_numpy_ravel_orderc C s� t dddgdddgg�}t|�� ddddddgg� t|jdd�ddddddgg� t|j�� ddddddgg� t|jjd d�ddddddgg� d S ) Nr r r r r r r� )r� r )r r r� r< )r! rD r) r) r* �test_matrix_ravel_order� s z!TestShape.test_matrix_ravel_orderc C s6 t t�| j| j�� �� t t�| j| j�� � � d S )N)r r �may_share_memoryr� r� r� )r! r) r) r* �test_array_memory_sharing� s z#TestShape.test_array_memory_sharingc C s6 t t�| j| j�� �� t t�| j| j�� � � d S )N)r r r� rw r� r� )r! r) r) r* �test_matrix_memory_sharing� s z$TestShape.test_matrix_memory_sharingN) r4 r5 r6 r� r� r� r� r� r� r� r� r� r) r) r) r* r� � s r� �__main__)!� __future__r r r r� �numpyr r r r Z numpy.testingr r r r r r r r Znumpy.matrixlib.defmatrixr �numpy.matrixlibr r r7 rh rp r r� r� r� r� r4 r) r) r) r* �<module> s$ ( / =* D 1