o
    ZZŽ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   ÚtypeZkron©ÚselfÚx© r   úT/var/www/auris/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      ð?)Zdtypeg        )	r   r   Úfloatr   r   ÚrealÚimagZnonzeroZ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ÚeyeZ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   ZstdÚshapeZ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   Znumpy.testingr   r   r   r   r   r   r   r   Ú<module>   s    