
    -i                     L    S r SSKrSSKJr  SSKJr  SSKJr  SSK	J
r
  S rS rg)	z2
Tests for sklearn.cluster._feature_agglomeration
    N)assert_array_equal)FeatureAgglomeration)
make_blobs)assert_array_almost_equalc                     Sn [         R                  " / SQ5      R                  SS5      n[        U [         R                  S9n[        U [         R
                  S9nUR                  U5        UR                  U5        [         R                  " [         R                  " UR                  5      5      U :X  d   e[         R                  " [         R                  " UR                  5      5      U :X  d   e[         R                  " UR                  5      UR                  S   :X  d   e[         R                  " UR                  5      UR                  S   :X  d   eUR                  U5      nUR                  U5      nUR                  S   U :X  d   eUR                  S   U :X  d   eU[         R                  " S/5      :X  d   eU[         R                  " S/5      :X  d   eUR                  U5      nUR                  U5      n[         R                  " US   5      R                  U :X  d   e[         R                  " US   5      R                  U :X  d   e[        UR                  U5      U5        [        UR                  U5      U5        g )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           c/var/www/html/venv/lib/python3.13/site-packages/sklearn/cluster/tests/test_feature_agglomeration.pytest_feature_agglomerationr       s   J
##Aq)A%"''RJ':BIIVLNN1Q77299Z//01Z???77299\1123zAAA77:%%&!''!*44477<''(AGGAJ666 ""1%G&&q)I==z)))??1+++bhhy))))#''' ..w7K 229=M99[^$))Z77799]1%&++z999j22;?Il44]CYO    c                      [        SSS9u  p[        SS9nUR                  U 5        UR                  nUR	                  5       n[        [        U5       Vs/ s H  nSU 3PM
     snU5        gs  snf )z9Check `get_feature_names_out` for `FeatureAgglomeration`.   r   )
n_featuresrandom_stater	   )r
   featureagglomerationN)r   r   r   n_clusters_get_feature_names_outr   range)r   _agglor
   	names_outis         r   ,test_feature_agglomeration_feature_names_outr.   -   sl    3DA A.E	IIaL""J++-I-2:->?->s	#->??s   A-)__doc__numpyr   numpy.testingr   sklearn.clusterr   sklearn.datasetsr   sklearn.utils._testingr   r    r.    r!   r   <module>r6      s'     , 0 ' <P@
r!   