
    Bwg                     4    d dl Zd dlmZmZmZ  G d d      Zy)    N)assert_assert_equalassert_raisesc                   $    e Zd Zd Zd Zd Zd Zy)TestRegressionc                     t        j                  d      }t        t        t        j                  ||            t        |             y )Nz
[1 0; 1 0])npmatrixr   typekronselfxs     \/var/www/horilla/myenv/lib/python3.12/site-packages/numpy/matrixlib/tests/test_regression.pytest_kron_matrixzTestRegression.test_kron_matrix   s-    IIl#T"''!Q-($q'2    c                    t        j                  dgt              }t        t	        |j
                        t         j                  u        t        t	        |j                        t         j                  u        t        j                  dg      j                         \  }}t        t	        |      t         j                  u        t        t	        |      t         j                  u        y )Ng      ?)dtypeg        )	r	   r
   floatr   r   realimagnonzerondarray)r   acds       r   test_matrix_propertiesz%TestRegression.test_matrix_properties   s    IIse5)QVV		)*QVV		)*yy#'')1Q2::%&Q2::%&r   c                 *    d }t        t        |       y )Nc                      t        j                  t        j                  d            t        j                  d      z   y )N   )r	   asmatrixeyeones r   r   mulz=TestRegression.test_matrix_multiply_by_1d_vector.<locals>.mul   s#    KKq	"2771:-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                  j                  ddd            }t	        |j                         j                  d       t	        |j                         j                  d       y )Nr      )   r*   r$   )r	   r!   randomuniformr   stdshapeargmaxr   s     r   test_matrix_std_argmaxz%TestRegression.test_matrix_std_argmax   sM    KK		))!Q78QUUW]]B'QXXZ%%r*r   N)__name__
__module____qualname__r   r   r'   r0   r$   r   r   r   r      s    3
''+r   r   )numpyr	   numpy.testingr   r   r   r   r$   r   r   <module>r6      s     > >+ +r   