o
    h                     @   s.   d dl Zd dlmZmZmZ G dd dZdS )    N)assert_assert_equalassert_raisesc                   @   s,   e Zd Zdd Zdd Zdd Zdd Zd	S )
TestRegressionc                 C   s(   t d}ttt ||t| d S )Nz
[1 0; 1 0])npmatrixr   typekronselfx r   y/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/numpy/matrixlib/tests/test_regression.pytest_kron_matrix   s   
zTestRegression.test_kron_matrixc                 C   st   t jdgtd}tt|jt ju  tt|jt ju  t 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dr   r   r   test_matrix_properties   s   z%TestRegression.test_matrix_propertiesc                 C   s   dd }t t| d S )Nc                   S   s   t t dt d  d S )N   )r   asmatrixeyeonesr   r   r   r   mul   s   z=TestRegression.test_matrix_multiply_by_1d_vector.<locals>.mul)r   
ValueError)r   r   r   r   r   !test_matrix_multiply_by_1d_vector   s   z0TestRegression.test_matrix_multiply_by_1d_vectorc                 C   s:   t t jddd}t| jd t| jd d S )Nr      )   r"   r   )r   r   randomuniformr   stdshapeargmaxr
   r   r   r   test_matrix_std_argmax   s   z%TestRegression.test_matrix_std_argmaxN)__name__
__module____qualname__r   r   r    r(   r   r   r   r   r      s
    	r   )numpyr   numpy.testingr   r   r   r   r   r   r   r   <module>   s    