ó
‡ˆ\c           @   sO   d  Z  d d l Z d d l m Z d d l m Z d d l m Z d „  Z d S(   s2   
Tests for sklearn.cluster._feature_agglomeration
iÿÿÿÿN(   t   FeatureAgglomeration(   t   assert_no_warnings(   t   assert_array_almost_equalc          C   s4  d }  t  j d d d g ƒ j d d ƒ } t d |  d t  j ƒ } t d |  d t  j ƒ } t | j | ƒ t | j | ƒ t  j t  j	 | j
 ƒ ƒ |  k s¡ t ‚ t  j t  j	 | j
 ƒ ƒ |  k sÈ t ‚ t  j | j
 ƒ | j d k sí t ‚ t  j | j
 ƒ | j d k st ‚ | j | ƒ } | j | ƒ } | j d |  k sIt ‚ | j d |  k sbt ‚ | t  j d d g ƒ k s„t ‚ | t  j d g ƒ k s¢t ‚ | j | ƒ } | j | ƒ } t  j	 | d ƒ j |  k sât ‚ t  j	 | d ƒ j |  k st ‚ t | j | ƒ | ƒ t | j | ƒ | ƒ d  S(   Ni   i    i   t
   n_clusterst   pooling_funcg      @g        (   t   npt   arrayt   reshapeR    t   meant   medianR   t   fitt   sizet   uniquet   labels_t   AssertionErrort   shapet	   transformt   inverse_transformR   (   R   t   Xt
   agglo_meant   agglo_mediant   Xt_meant	   Xt_mediant   X_full_meant   X_full_median(    (    sO   lib/python2.7/site-packages/sklearn/cluster/tests/test_feature_agglomeration.pyt   test_feature_agglomeration   s4    $''%%"""(	   t   __doc__t   numpyR   t   sklearn.clusterR    t   sklearn.utils.testingR   R   R   (    (    (    sO   lib/python2.7/site-packages/sklearn/cluster/tests/test_feature_agglomeration.pyt   <module>   s
   