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sJ ||}||}t |d j| ksJ t |d j| ksJ t||| t||| d S )N   )r   r   r      )
n_clustersZpooling_funcgUUUUUU?g        r   )nparrayZreshaper   meanZmedianfitsizeuniqueZlabels_shape	transformZinverse_transformr   )r   XZ
agglo_meanZagglo_medianZXt_meanZ	Xt_medianZX_full_meanZX_full_median r   j/home/kim/smarthome/.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   )Z
n_featuresZrandom_stater   )r   c                 S   s   g | ]}d | qS )Zfeatureagglomerationr   ).0ir   r   r   
<listcomp>6   s    z@test_feature_agglomeration_feature_names_out.<locals>.<listcomp>N)r   r   r   Zn_clusters_Zget_feature_names_outr   range)r   _Zagglor   Z	names_outr   r   r   ,test_feature_agglomeration_feature_names_out-   s   

r   )__doc__numpyr	   Znumpy.testingr   Zsklearn.clusterr   Zsklearn.datasetsr   Zsklearn.utils._testingr   r   r   r   r   r   r   <module>   s     