Current File : //lib64/python3.9/site-packages/numpy/matrixlib/tests/__pycache__/test_interaction.cpython-39.pyc
a
z[ycc. � @ s d Z ddlZddlZddlZddlZddlmZmZm Z m
Z
m Z m Z m
Z
dd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Z d"d#� Z d$d%� Z d&d'� Z d(d)� Z G d*d+� d+�Z!d,d-� Z"d.d/� Z#dS )0z�Tests of interaction of matrix with other parts of numpy.
Note that tests with MaskedArray and linalg are done in separate files.
� N)�assert_� assert_equal�
assert_raises�assert_raises_regex�assert_array_equal�assert_almost_equal�assert_array_almost_equalc C s| t �ddgddgg�} tt| g d�d d �f t j�� t �t �d��dd��}t|d d�t �d �f |d d�d f j � d S )
N� � � � )r r r �2 � �
���)
�np�matrixr �
isinstance�asmatrix�arange�reshaper �array�T)�m�x� r �L/usr/lib64/python3.9/site-packages/numpy/matrixlib/tests/test_interaction.py�test_fancy_indexing s r c C sD ddg} ddg}t �| | g�}t jj�|| |�}tt|t j�� d S )Nr r r r )r r Z
polynomialZ polyutilsZ mapdomainr r )Zdom1Zdom2r �resr r r �test_polynomial_mapdomain s
r c C sN t �g d�g�} t j| d d�}t �g d�g�}t||� tt|�t ju � d S )N�r
r r ��axis�r r r
)r r �sortr r �type��a�actual�expectedr r r �test_sort_matrix_none) s
r* c C sP t �g d�g�} t j| dd d�}t �g d�g�}t||� tt|�t ju � d S )Nr r r! r# )r r � partitionr r r% r& r r r �test_partition_matrix_none2 s
r, c C sN t jddgtd�} t jddggtd�}tt �| d�|� tt �d| �|� d S �Nr r
�Zdtyper � )r r �objectr �dot��arr�desiredr r r �%test_dot_scalar_and_matrix_of_objects< s r5 c C s� t jd t jd d D ]n} t jd| d�d }t jddgdd gg| d�}t jdd
gd d gg| d�}tt �||�|� tt �||�|� qd S )
NZ
AllIntegerZAllFloat�?r r. r r r
r r/ � � )r Z typecodesr r r �inner)ZdtZscar3 r4 r r r � test_inner_scalar_and_matrixE s r: c C sN t jddgtd�} t jddggtd�}tt �| d�|� tt �d| �|� d S r- )r r r0 r r9 r2 r r r �'test_inner_scalar_and_matrix_of_objectsO s r; c C s0 t �ddgddgg�} t �d��dd�j}t �| |d gg dgdgddgg�}tt|jd �t ju � tt|jd �t j u� t
|jd j d� t �d��ddd�}t t
t j| |d gg dgdgddgg� t �| |d gg dgdgg d �g�}tt|jd �t j u � tt|jd �t ju� t
|jd j d
� d S ) Nr r
r r �readonly� writeonly�allocate)r
r
)r= r> Z
no_subtype)r r
r
)r r r r r Znditerr r% Zoperands�ndarrayr �shaper � RuntimeError)r' �b�ir r r �!test_iter_allocate_output_subtypeX s( ����rD c C sh t �ddgddgg�} t jt jt jfD ]<}|| �}tt|�t ju � || dd�}tt|�t ju� q&d S )Nr r
r r F)�subok)r r Z
zeros_likeZ ones_likeZ
empty_liker r% )r' �
like_functionrB �cr r r rF s s rF c C s� t jg d�g d�gdd�} | jdddd�}t| |u � | jddd �}t| |� tt|�t j� | jdddd�}t| |� t| |u� tt|�t ju� d S )
Nr# )r r r Zf4r. TF)rE �copyZi4)rH )r r Zastyper r r% �r' rB r r r �test_array_astype~ s
rJ c C s. t �ddgddgg�} ttdt j| | g� d S )Nr r
r r z shape too large to be a matrix)r r r �
ValueError�stack)r r r r �
test_stack� s
�rM c C sN t jddgtd�} t jddggtd�}tt �| d�|� tt �d| �|� d S r- )r r r0 r Zmultiplyr2 r r r �test_object_scalar_multiply� s rN c C sD t �t �d��} t jt jfD ]n}|| dd�}tt|t j�� t|jdk� || dd�}tt|t j�� t|jdk� || �}tt �|�� q t j | d< t jt jfD �]�}t
j dd��\}t
� d � || dd�}tt|t j�� tt �
t �|�� � tt|�dk� W d � n1 �s0 Y t
j dd���}t
� d � || dd�}tt|t j�� tt �|d
��o�t �|d � �o�t �|d � � tt|�dkd
� tt|d jt�� W d � n1 �s�0 Y t
j dd��N}t
� d � || �}tt �|�� t|t j k� tt|�dk� W d � q�1 �s40 Y q�d S )Nr r r! �r r r �r r T)�record�always)r r )r r )r
r zno warning raised)r r �eyeZnanminZnanmaxr r r@ �isscalar�nan�warnings�catch_warnings� simplefilter�anyZisnan�len�
issubclass�category�RuntimeWarning)�mat�fr �wr r r �test_nanfunctions_matrices� sB
0
$�4
ra c C s0 t �t �d��} t jt jt jt jt jt jt j fD ]n}|| dd�}t
t |t j�� t
|j dk� || dd�}t
t |t j�� t
|j dk� || �}t
t �
|�� q0t jt jfD ]~}|| dd�}t
t |t j�� t
|j dk� || dd�}t
t |t j�� t
|j dk� || �}t
t |t j�� t
|j dk� q�d S ) Nr r r! rO r rP )r r )r r7 )r r rS Z nanargminZ nanargmaxZnansumZnanprodZnanmeanZnanvarZnanstdr r r@ rT Z nancumsumZ
nancumprod)r^ r_ r r r r �"test_nanfunctions_matrices_general� s, � rb c C s� t �t j�dd��} t| �d�t �| d�� t �ddgddgg�}t �ddgddgg�}t j|d|d�}tt|�t j� t|dd gg� d S )
Nr r r r
r r )r" Zweightsg @g������
@) r r ZrandomZrandr ZmeanZaverager r% )�yr' r` �rr r r �test_average_matrix� s re c C sN t �dd�} | | }t �|| �}t �| �}t �|�}t �||�}t||� d S )Nr r )r ZlinspaceZtrapzr r )r rc rd ZmxZmy�mrr r r �test_trapz_matrix� s
rg c C s@ t t�t�d��tj�s J �t tjt�d�dd�tj�s<J �d S )Nr )Zto_begin)r r Zediff1dr r r r r �test_ediff1d_matrix s rh c C s� dd� } t �ddgddgg�}t �ddgddgg�}t �| d|�}tt|t j�� t||� t �| d|�}tt|t j�� t||� d S ) Nc S s | d S �Nr
r )�rowr r r �double
s z,test_apply_along_axis_matrix.<locals>.doubler r r
r r r/ )r r Zapply_along_axisr r r )rk r r) �resultr r r � test_apply_along_axis_matrix s
rm c C s| t �ddg�} t �| �}ttt �| | ��t j� ttt �||��t j� ttt �| |��t j� ttt �|| ��t j� d S ri )r Zonesr r r% Zkronr? r )r' r r r r �test_kron_matrix s
rn c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestConcatenatorMatrixc s� ddg� ddg�t jd� �f }t jd� �f }tt|�t j� tt|�t j� tt �|�g d�g� tt �|�dgdgdgdgg� tt� �fdd �� d S )
Nr r
r r rd rG )r r
r r c s t jd� �f S )N�rc)r �r_r rI r r �<lambda>2 � z4TestConcatenatorMatrix.test_matrix.<locals>.<lambda>)r rq r r% r r r rK )�selfZab_rZab_cr rI r � test_matrix% s z"TestConcatenatorMatrix.test_matrixc C s> t jdddgdf }tt|�t j� tt �|�g d�g� d S )Nrd r r
r )r r
r )r rq r r% r r )rt rd r r r �test_matrix_scalar4 s z)TestConcatenatorMatrix.test_matrix_scalarc C sp t �dg�}t �dg�}t �dg�}t �dg�}t jd }t �||g||gg�}t||� tt|�t|�� d S )Nr r
r r z
a, b; c, d)r r rq Zbmatr r% )rt r' rB rG �dr( r) r r r �test_matrix_builder9 s
z*TestConcatenatorMatrix.test_matrix_builderN)�__name__�
__module__� __qualname__ru rv rx r r r r ro # s ro c C sj t �t��.} tt�ddg�t�ddg�� W d � n1 s>0 Y t| j�}t �
d�}t||� d S )Nr r
zl
Arrays are not equal
(shapes (2,), (1, 2) mismatch)
x: array([1, 2])
y: matrix([[1, 2]])) �pytestZraises�AssertionErrorr r r r �str�value�textwrap�dedent)�exc_info�msgZ
msg_referencer r r �%test_array_equal_error_message_matrixE s
<
r� c C s� t �ddgg�} t �dt jgg�}t �dt j gg�}t �t jt jgg�}t �ddgt jt jgg�}ttfD ]@}| ||||fD ],}|||� t �|�}|||� |||� q~qld S )Ng �?g @)r r rU �infr r r )Zm1Zm2Zm3Zm4Zm5Z assert_funcr r' r r r � test_array_almost_equal_matrixT s
r� )$�__doc__r| r� rV Znumpyr Z
numpy.testingr r r r r r r r r r* r, r5 r: r; rD rF rJ rM rN ra rb re rg rh rm rn ro r� r� r r r r �<module> s6 $
)
"
Mini Shell