
    Pi                     N    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 Zd ZdS )	z2
Tests for sklearn.cluster._feature_agglomeration
    N)assert_array_equal)FeatureAgglomeration)
make_blobs)assert_array_almost_equalc                  
   d} t          j        g d                              dd          }t          | t           j                  }t          | t           j                  }|                    |           |                    |           t          j        t          j        |j	                            | k    sJ t          j        t          j        |j	                            | k    sJ t          j        |j	                  |j
        d         k    sJ t          j        |j	                  |j
        d         k    sJ |                    |          }|                    |          }|j
        d         | k    sJ |j
        d         | k    sJ |t          j        dg          k    sJ |t          j        dg          k    sJ |                    |          }|                    |          }t          j        |d                   j        | k    sJ t          j        |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_medians           /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/sklearn/cluster/tests/test_feature_agglomeration.pytest_feature_agglomerationr       sO   J
##Aq))A%"'RRRJ':BIVVVLNN1Q729Z/0011Z????729\12233zAAAA7:%&&!'!*44447<'((AGAJ6666 ""1%%G&&q))I=z))))?1++++bhy))))))#'''' ..w77K 229==M9[^$$)Z77779]1%&&+z9999j22;??IIIl44]CCYOOOOO    c                      t          dd          \  } }t          d          }|                    |            |j        }|                                }t          d t          |          D             |           dS )z9Check `get_feature_names_out` for `FeatureAgglomeration`.   r   )
n_featuresrandom_stater	   )r
   c                     g | ]}d | S )featureagglomeration ).0is     r   
<listcomp>z@test_feature_agglomeration_feature_names_out.<locals>.<listcomp>6   s!    ???	#	#	#???r!   N)r   r   r   n_clusters_get_feature_names_outr   range)r   _agglor
   	names_outs        r   ,test_feature_agglomeration_feature_names_outr2   -   s    333DAq A...E	IIaLLL"J++--I??U:->->???    r!   )__doc__numpyr   numpy.testingr   sklearn.clusterr   sklearn.datasetsr   sklearn.utils._testingr   r    r2   r(   r!   r   <module>r9      s         , , , , , , 0 0 0 0 0 0 ' ' ' ' ' ' < < < < < <P P P@
 
 
 
 
r!   