o
    Ó\i­  ã                   @   sP   d 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„ Zd	d
„ ZdS )z2
Tests for sklearn.cluster._feature_agglomeration
é    N)Úassert_array_equal)ÚFeatureAgglomeration)Ú
make_blobs)Úassert_array_almost_equalc                  C   sz  d} t  g d¢¡ dd¡}t| t jd}t| t jd}| |¡ | |¡ t  t  |j	¡¡| ks2J ‚t  t  |j	¡¡| ks?J ‚t  |j	¡|j
d ksLJ ‚t  |j	¡|j
d ksYJ ‚| |¡}| |¡}|j
d | kslJ ‚|j
d | ksuJ ‚|t  dg¡ksJ ‚|t  dg¡ks‰J ‚| |¡}| |¡}t  |d ¡j| ksŸJ ‚t  |d ¡j| ks«J ‚t| |¡|ƒ t| |¡|ƒ d S )Né   )r   r   r   é   )Ú
n_clustersÚpooling_funcgUUUUUUÕ?g        r   )ÚnpÚarrayÚreshaper   ÚmeanÚmedianÚfitÚsizeÚuniqueÚlabels_ÚshapeÚ	transformÚinverse_transformr   )r   ÚXÚ
agglo_meanÚagglo_medianÚXt_meanÚ	Xt_medianÚX_full_meanÚX_full_median© r   ú‘/var/www/www-root/data/www/176.119.141.140/sports-predictor/venv/lib/python3.10/site-packages/sklearn/cluster/tests/test_feature_agglomeration.pyÚtest_feature_agglomeration   s,   





r   c                  C   sN   t ddd\} }tdd}| | ¡ |j}| ¡ }tdd„ t|ƒD ƒ|ƒ dS )	z9Check `get_feature_names_out` for `FeatureAgglomeration`.é   r   )Ú
n_featuresÚrandom_stater   )r   c                 S   s   g | ]}d |› ‘qS )Úfeatureagglomerationr   )Ú.0Úir   r   r   Ú
<listcomp>6   s    z@test_feature_agglomeration_feature_names_out.<locals>.<listcomp>N)r   r   r   Ún_clusters_Úget_feature_names_outr   Úrange)r   Ú_Úagglor   Ú	names_outr   r   r   Ú,test_feature_agglomeration_feature_names_out-   s   

ÿr-   )Ú__doc__Únumpyr
   Únumpy.testingr   Úsklearn.clusterr   Úsklearn.datasetsr   Úsklearn.utils._testingr   r   r-   r   r   r   r   Ú<module>   s     