
    Wh                     L    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y)	z2
Tests for sklearn.cluster._feature_agglomeration
    N)assert_array_equal)FeatureAgglomeration)
make_blobs)assert_array_almost_equalc                     d} t        j                  g d      j                  dd      }t        | t         j                        }t        | t         j
                        }|j                  |       |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                  |      }|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 |j                  |      }|j                  |      }t        j                  |d         j                  | k(  sJ t        j                  |d         j                  | k(  sJ t        |j                  |      |       t        |j                  |      |       y )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           j/var/www/html/jupyter_env/lib/python3.12/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                      t        dd      \  } }t        d      }|j                  |        |j                  }|j	                         }t        t        |      D cg c]  }d| 	 c}|       yc c}w )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.   -   sh    3DAq A.E	IIaL""J++-I-2:->?s	#??s   A1)__doc__numpyr   numpy.testingr   sklearn.clusterr   sklearn.datasetsr   sklearn.utils._testingr   r    r.    r!   r   <module>r6      s'     , 0 ' <P@
r!   