OwlCyberSecurity - MANAGER
Edit File: test_interaction.cpython-38.pyc
U �i]�/ � @ s d Z ddlmZmZmZ ddlZddlZddlZddlZ ddl m Z m Z m Z mZmZmZmZ dd� Zdd� Zd d � Zd d � Zd d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Z dd � Z d d � Z d d � Z d!d"� Z d#d$� Z!d%d&� Z"d'd(� Z#d)d*� Z$G d+d,� d,e%�Z&d-d.� Z'd/d0� Z(dS )1z�Tests of interaction of matrix with other parts of numpy. Note that tests with MaskedArray and linalg are done in separate files. � )�division�absolute_import�print_functionN)�assert_� assert_equal� assert_raises�assert_raises_regex�assert_array_equal�assert_almost_equal�assert_array_almost_equalc C s~ t �ddgddgg�} tt| dddgd d �f t j�� t �t �d��dd��}t|d d�t �d �f |d d�d f j � d S ) N� � � � r �2 � � ���) �np�matrixr � isinstance�asmatrix�arange�reshaper �array�T)�m�x� r �H/usr/lib/python3/dist-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 sR t �dddgg�} t j| d d�}t �dddgg�}t||� tt|�t jk� d S �Nr r r ��axis)r r �sortr r �type��a�actual�expectedr r r �test_sort_matrix_none+ s r, c C sT t �dddgg�} t j| dd d�}t �dddgg�}t||� tt|�t jk� d S r# )r r � partitionr r r' r( r r r �test_partition_matrix_none4 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 r7 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 r0 r r r r r1 � � )r Z typecodesr r r �inner)ZdtZscar5 r6 r r r � test_inner_scalar_and_matrixG s r< c C sN t jddgtd�} t jddggtd�}tt �| d�|� tt �d| �|� d S r/ )r r r2 r r; r4 r r r �'test_inner_scalar_and_matrix_of_objectsQ s r= c C s2 t �ddgddgg�} t �d��dd�j}t �| |d gg dgdgddgg�}tt|jd �t jk� tt|jd �t j k � t |jd j d� t �d��ddd�}t t t j| |d gg dgdgddgg� t �| |d gg dgdgddd gg�}tt|jd �t j k� tt|jd �t jk � t |jd j d � d S ) Nr r r r �readonlyZ writeonly�allocate)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_subtypeZ s( ����rE c C sh t �ddgddgg�} t jt jt jfD ]<}|| �}tt|�t jk� || dd�}tt|�t jk � q&d S )Nr r r r F)�subok)r r Z zeros_likeZ ones_likeZ empty_liker r' )r) � like_functionrC �cr r r rG u s rG c C s� t jdddgdddggdd�} | jdd d d �}t| |k� | jd d d �}t| |� tt|�t j� | jdd d d �}t| |� t| |k � tt|�t jk � d S )Nr r r r r r Zf4r0 TF)rF �copyZi4)rI )r r Zastyper r r' �r) rC r r r �test_array_astype� s rK 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 �rN c C sN t jddgtd�} t jddggtd�}tt �| d�|� tt �d| �|� d S r/ )r r r2 r Zmultiplyr4 r r r �test_object_scalar_multiply� s rO c C s 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 �]X}t j dd��R}t � d � || dd�}tt|t j�� tt � t �|�� � tt|�dk� W 5 Q R X t j dd���}t � d � || dd�}tt|t j�� tt �|d ��ort �|d � �ort �|d � � tt|�dkd � tt|d jt�� W 5 Q R X t j dd��D}t � d � || �}tt �|�� t|t j k� tt|�dk� W 5 Q R X 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 rA �isscalar�nan�warnings�catch_warnings� simplefilter�anyZisnan�len� issubclass�category�RuntimeWarning)�mat�fr! �wr r r �test_nanfunctions_matrices� sB $� rb 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$ rP r rQ )r r )r r9 )r r rT Z nanargminZ nanargmaxZnansumZnanprodZnanmeanZnanvarZnanstdr r rA rU Z nancumsumZ nancumprod)r_ r` r! r r r �"test_nanfunctions_matrices_general� s0 � rc 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) ra �rr r r �test_average_matrix� s rf c C sN t �dd�} | | }t �|| �}t �| �}t �|�}t �||�}t||� d S )Nr r )r ZlinspaceZtrapzr r )r rd re ZmxZmyZmrr r r �test_trapz_matrix� s rg c C s@ t t�t�d��tj�s t�t tjt�d�dd�tj�s<t�d S )Nr )Zto_begin)r r Zediff1dr �AssertionErrorr r r r �test_ediff1d_matrix s ri 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 r1 )r r Zapply_along_axisr r r )rl r r+ �resultr r r � test_apply_along_axis_matrix s rn c C s| t �ddg�} t �| �}ttt �| | ��t j� ttt �||��t j� ttt �| |��t j� ttt �|| ��t j� d S rj )r Zonesr r r' Zkronr@ r )r) r r r r �test_kron_matrix s ro 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 �|�ddddgg� tt �|�dgdgdgdgg� tt� �fdd�� d S ) Nr r r r re rH c s t jd� �f S )NZrc)r �r_r rJ r r �<lambda>4 � z4TestConcatenatorMatrix.test_matrix.<locals>.<lambda>)r rq r r' r r r rL )�selfZab_rZab_cr rJ r � test_matrix' s z"TestConcatenatorMatrix.test_matrixc C s@ t jdddgdf }tt|�t j� tt �|�dddgg� d S )Nre r r r )r rq r r' r r )rt re r r r �test_matrix_scalar6 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) rC rH �dr* r+ r r r �test_matrix_builder; s z*TestConcatenatorMatrix.test_matrix_builderN)�__name__� __module__� __qualname__ru rv rx r r r r rp % s rp c C s� z"t t�ddg�t�ddg�� W nn tk r� } zPt| �}|�dd�}t�d�}zt ||� W n tk r~ t ||� Y nX W 5 d } ~ X Y n X td��d S )Nr r zshapes (2L,), (1L, 2L)zshapes (2,), (1, 2)z| Arrays are not equal (shapes (2,), (1, 2) mismatch) x: array([1, 2]) y: matrix([[1, 2]])z Did not raise) r r r r rh �str�replace�textwrap�dedent)�e�msgZmsg2Z msg_referencer r r �%test_array_equal_error_message_matrixG 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 rV �infr r r )Zm1Zm2Zm3Zm4Zm5Z assert_funcr r) r r r � test_array_almost_equal_matrix] s r� ))�__doc__Z __future__r r r Zpytestr~ rW Znumpyr Z numpy.testingr r r r r r r r r" r, r. r7 r< r= rE rG rK rN rO rb rc rf rg ri rn ro r2 rp r� r� r r r r �<module> s8 $ ) "