Current File : //lib64/python3.9/site-packages/numpy/linalg/tests/__pycache__/test_regression.cpython-39.opt-1.pyc
a
z[yc� � @ sf d Z ddlZddlZddlmZmZmZmZmZm Z ddl
m Z m Z m
Z
mZmZmZ G dd� d�ZdS )z" Test functions for linalg module
� N)�linalg�arange�float64�array�dot� transpose)�assert_�
assert_raises� assert_equal�assert_array_equal�assert_array_almost_equal�assert_array_lessc @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�TestRegressionc C sR t g d��}tdtd�}d|_|d }t�|�\}}|�� |�� t||� d S )N)
y��ӝ'�Y@ yF.�<2/3� y�ʹLT�W��9>�/@y�ʹLT�W��9>�/�y=J���&� yt��V) 忷B����$@yt��V) 忷B����$�yZynFr62@ y�(
��3%@ y� �: @ yC r
~� y�Y�qs`�< yt&���Aۼ � ��dtype)�
r � )r r r �shaper Zeig�sortr )�selfZrva�a�va�ve� r �H/usr/lib64/python3.9/site-packages/numpy/linalg/tests/test_regression.py�test_eig_build s
z TestRegression.test_eig_buildc C s> g d�}t g d�g d�g d�g�}t�|�\}}t||� d S )N)g"���&Q@gg��dV@g��7���Z@)g'�U��lS@� 4h�� @�%p��J/@)r g�+��>V@研�qi0�)r r gJ%~���X@)r r Zeighr )r ZrvalsZcovZvalsZvecsr r r �test_eigh_build& s �z TestRegression.test_eigh_buildc C sl t ddgddgddgddgg�}|j\}}t�|�\}}}tt|d d �|d �f �|�}t|t�d�� d S )Ng � �?� @� @)� r$ ) r r r �svdr r r �npZzeros)r r �m�n�u�sZvh�br r r �test_svd_build1 s
z TestRegression.test_svd_buildc C s t ttjtg d��d� d S )N)r! r" r# �fro)r �
ValueErrorr �normr )r r r r �test_norm_vector_badarg; s z&TestRegression.test_norm_vector_badargc C sF t ddgddggdd�}t |dd�}t�|�}t�|�}t||� d S )Ng�C$� 3@g�&��u�g.�ܷ#@z>f8r z<f8)r r Zcholeskyr )r r r+ ZapZbpr r r �test_lapack_endian@ s ��
z!TestRegression.test_lapack_endianc C s t �dd�}t j�|� d S )Ni� �B )r& Zeyer r% )r �xr r r �test_large_svd_32bitJ s z#TestRegression.test_large_svd_32bitc C s~ dD ]t}t tfD ]f}tj||d�}tj|dd�}t�t�|�dk�}t|d� ttj� |�d� t
dtjj |dd�� qqd S ) N))� � )r6 r6 )r6 r5 r F)Z
compute_uvg �?� r$ ��ord) �float�complexr& Zonesr r% Z
count_nonzero�absoluter
Z matrix_rankr
r/ )r r �tr �w�cr r r �test_svd_no_uvO s
z TestRegression.test_svd_no_uvc C s� t jt �ddg�ddgtd�}t�|�}t|ddg� t|jt �d�k� tj|dd�}t|ddg� t|jt �d�k� tj|dd�}t|ddg� t|jt �d�k� tt tj|dd� tt tj|dd� tt tj|t j
d� tt tj|t j
d� t � � �2 t �
d t� tttftj|dd� W d � n1 �s:0 Y tt tj|d
d� tt tj|d d� t jt �ddg�ddgg d �gtd�}t�|�}t|ddg� t|jt �d�k� tj|dd�}t|ddg� t|jt �d�k� tttj|dd� tt tj|t j
d� tt tj|t j
d� tt tj|dd� tt tj|dd� tt tj|d
d� tttj|dd� tttj|d d� tt tj|d
d� d S )Nr r7 r r r8 r$ r- Znuc�error������)r r r r5 )r& r �objectr r/ r r r r r. �inf�warnings�catch_warnings� simplefilter�DeprecationWarning�AttributeError� TypeError)r Z
testvectorr/ Z
testmatrixr r r �test_norm_object_arrayZ sR
�&��
z%TestRegression.test_norm_object_arrayc
C sx d}d}t j�||�dt j�||� }t j�||�dt j�||� }|�|�}tj||d d�\}}}} t||� d S )N� �F y �?)Zrcond)r& ZrandomZrandnr r Zlstsqr )
r �sizeZn_rhs�Gr) r+ Zu_lstsq�resZrank�svr r r � test_lstsq_complex_larger_rhs� s
z,TestRegression.test_lstsq_complex_larger_rhsN) �__name__�
__module__� __qualname__r r r, r0 r1 r4 r@ rL rS r r r r r
s
0r )�__doc__rF Znumpyr& r r r r r r Z
numpy.testingr r r
r r r
r r r r r �<module> s
Mini Shell