U
    ÉZjŸ  ã                   @   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   úh/var/www/html/TRUCKING_PROJECT/venv/lib/python3.8/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kƒ tt|jƒt jkƒ t  dg¡ ¡ \}}tt|ƒt jkƒ tt|ƒt jkƒ 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   ZmatÚ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   ZasmatrixÚ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   