File: //lib/python3/dist-packages/numpy/linalg/tests/__pycache__/test_regression.cpython-310.pyc
o
6��a� � @ 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 �eig�sortr )�self�rva�a�va�ve� r �D/usr/lib/python3/dist-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 �eighr )r �rvals�cov�vals�vecsr 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 �np�zeros)r r �m�n�u�s�vh�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 �choleskyr )r r r4 �ap�bpr r r �test_lapack_endian@ s ��
z!TestRegression.test_lapack_endianc C s t �dd�}t j�|� d S )Ni� �B )r- �eyer r, )r �xr r r �test_large_svd_32bitJ s z#TestRegression.test_large_svd_32bitc C s~ dD ]:}t tfD ]3}tj||d�}tj|dd�}t�t�|�dk�}t|d� ttj� |�d� t
dtjj |dd�� qqd S ) N))� � )rC rC )rC rB r F)�
compute_uvg �?� r+ ��ord) �float�complexr- �onesr r, �
count_nonzero�absoluter
� matrix_rankr
r8 )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 � � � t �
d t� tttftj|dd� W d � n1 s�w 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 rE r r rF r+ r6 �nuc�error������)r r r rB )r- r �objectr r8 r r r r r7 �inf�warnings�catch_warnings� simplefilter�DeprecationWarning�AttributeError� TypeError)r �
testvectorr8 �
testmatrixr r r �test_norm_object_arrayZ sT
� ���
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 �?)�rcond)r- �random�randnr r �lstsqr )
r �size�n_rhs�Gr1 r4 �u_lstsq�res�rank�svr r r � test_lstsq_complex_larger_rhs� s
z,TestRegression.test_lstsq_complex_larger_rhsN) �__name__�
__module__� __qualname__r r' r5 r9 r= rA rQ r` rn r r r r r
s
0r )�__doc__rX �numpyr- r r r r r r �
numpy.testingr r r
r r r
r r r r r �<module> s