
    G/Ph                     <    d dl Zd dlmZmZmZ  G d d          ZdS )    N)assert_assert_equalassert_raisesc                   &    e Zd Zd Zd Zd Zd ZdS )TestRegressionc                     t          j        d          }t          t          t          j        ||                    t          |                     d S )Nz
[1 0; 1 0])npmatrixr   typekronselfxs     e/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/numpy/matrixlib/tests/test_regression.pytest_kron_matrixzTestRegression.test_kron_matrix   s?    Il##T"'!Q--(($q''22222    c                    t          j        dgt                    }t          t	          |j                  t           j        u            t          t	          |j                  t           j        u            t          j        dg                                          \  }}t          t	          |          t           j        u            t          t	          |          t           j        u            d S )Ng      ?)dtypeg        )	r	   r
   floatr   r   realimagnonzerondarray)r   acds       r   test_matrix_propertiesz%TestRegression.test_matrix_properties   s    Ise5)))QV	)***QV	)***y#''))1Q2:%&&&Q2:%&&&&&r   c                 6    d }t          t          |           d S )Nc                  |    t          j        t          j        d                    t          j        d          z   d S )N   )r	   asmatrixeyeones r   r   mulz=TestRegression.test_matrix_multiply_by_1d_vector.<locals>.mul   s-    Kq		""271::----r   )r   
ValueError)r   r%   s     r   !test_matrix_multiply_by_1d_vectorz0TestRegression.test_matrix_multiply_by_1d_vector   s)    	. 	. 	. 	j#&&&&&r   c                    t          j        t           j                            ddd                    }t	          |                                j        d           t	          |                                j        d           d S )Nr      )   r*   r$   )r	   r!   randomuniformr   stdshapeargmaxr   s     r   test_matrix_std_argmaxz%TestRegression.test_matrix_std_argmax   s`    K	))!Q7788QUUWW]B'''QXXZZ%r*****r   N)__name__
__module____qualname__r   r   r'   r0   r$   r   r   r   r      sP        3 3 3
' ' '' ' '+ + + + +r   r   )numpyr	   numpy.testingr   r   r   r   r$   r   r   <module>r6      sa        > > > > > > > > > >+ + + + + + + + + +r   