o
    ×tBhC  ã                   @   sd   d Z ddlZddlZddlmZ ddlmZ ddlmZ ddlm	Z	 ddl
mZ dd	„ Zd
d„ ZdS )zQ
Testing for Elliptic Envelope algorithm (sklearn.covariance.elliptic_envelope).
é    N)ÚEllipticEnvelope)Úassert_almost_equal)Úassert_array_almost_equal)Úassert_array_equal)ÚNotFittedErrorc                  C   s   t j d¡} |  dd¡}tdd}t t¡ | |¡ W d   ƒ n1 s&w   Y  t t¡ | 	|¡ W d   ƒ n1 s@w   Y  | 
|¡ | |¡}| |¡}| 	|¡}t|| |¡ ƒ t| |¡|jƒ t| |t  d¡¡d||dk j d ƒ t|dkƒt|dk ƒksŽJ ‚d S )Nr   éd   é
   gš™™™™™¹?©Úcontaminationéÿÿÿÿg      Y@)ÚnpÚrandomÚRandomStateÚrandnr   ÚpytestÚraisesr   ÚpredictÚdecision_functionÚfitÚscore_samplesr   ÚmahalanobisÚdist_r   ÚscoreÚonesÚsizeÚsum)ÚrndÚXÚclfÚy_predÚscoresÚ	decisions© r"   ú†/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/sklearn/covariance/tests/test_elliptic_envelope.pyÚtest_elliptic_envelope   s&   
ÿÿ



$ÿ r$   c                  C   s¨   ddgddgddgg} t dd | ¡}t ƒ  | ¡}t| ddgg¡| ddgg¡|j ƒ t| ddgg¡| ddgg¡|j ƒ t| ddgg¡| ddgg¡ƒ d S )Né   é   gš™™™™™É?r	   g       @)r   r   r   r   r   Úoffset_)ÚX_trainÚclf1Úclf2r"   r"   r#   Útest_score_samples$   s   þþÿr+   )Ú__doc__Únumpyr   r   Úsklearn.covariancer   Úsklearn.utils._testingr   r   r   Úsklearn.exceptionsr   r$   r+   r"   r"   r"   r#   Ú<module>   s    