\c@sUddlZddljZddlZddlmZddlmZddl m Z ddl m Z ddl m Z ddl m Z ddl mZdd l mZdd lmZdd lmZdd lmZdd lmZddlmZdZdZdZdZdZdZdZdZdZdZ dS(iN(t csr_matrix(tdatasets(t assert_false(tassert_array_equal(t assert_equal(tassert_raises_regexp(tassert_raise_message(tassert_greater(tsilhouette_score(tsilhouette_samples(tpairwise_distances(tcalinski_harabaz_score(tdavies_bouldin_scorec Cstj}|j}t|}tj|}tj|}|j}xK||||gD]7}t|dd}t ||dd}t |dt ||dd} t j || ||kr|} nt j | | t ||dddt |jdddd}t ||dddt |jdddd} t |dt | dt j | |||kr|} q[t j | | q[WdS(Ntmetrict euclideant precomputedit sample_sizeit random_state(Rt load_iristdataRtspt dok_matrixt lil_matrixttargetR RRtpytesttapproxtinttshape( tdatasettX_densetX_csrtX_doktX_liltytXtDtscore_precomputedtscore_euclideantscore_dense_without_samplingtscore_dense_with_sampling((sNlib/python2.7/site-packages/sklearn/metrics/cluster/tests/test_unsupervised.pyttest_silhouettes8              cCsdgdgdgdgdgdgg}tjddddddg}t||}ttj|t||}t|ddddddgdS( Ngg?g@g@iiig?(tnptarrayRRtisnanR R(R"tlabelst silhouettetss((sNlib/python2.7/site-packages/sklearn/metrics/cluster/tests/test_unsupervised.pyttest_cluster_size_1>s *! c BCsdddddddddd d d d d dddddddddddddddddddddddd dd dddd dd!d"dd#d d$dd d d%d&d'dd ddddd(dd gB}tjdT}||tjd)d*<||j7}d+d,d-d.d/d0d1d2d3d4d5d6g }d7d7d8d8d7d7d8d7d7d8d8d7g }i d9d36d:d+6d;d06d<d26d=d,6d>d/6d?d66d@d.6dAd46dBd-6dCd56dDd16}dE}d7d8dFdFd7d7d8d7d7dFdFd8g }i dGd36dHd06dId+6dJd26dKd/6dEd66dLd,6dMd16dNd.6dHd46dOd56dOd-6}dB}x|||f|||fgD]\} } } g|D]} | | ^qq} tj| t|tj| dPdQdRdStj| t|tj| dPdQdRdSq[WdS(UNgRQ@g@g@gRQ@gp= ף@gRQ@gRQ@gףp= W @gRQ@g\(\@g@gGz@gGz@gGz@gGz@g@g@gGz@g\(\ @g@g\(\@gGz@g@g@gRQ@g@g@gRQ@g@gRQ@g@g\(\@g@gRQ@g@gGz@g\(\ @g@g@gGz@i itBELtBRAtCHItCUBtEGYtFRAtINDtISRtUSAtUSStYUGtZAIiigQ?g(\?gffffff?g333333?g)\(?g?gRQ?g?g(\?gQ?gp= ף?g{GzgQ?igGz?g)\(?gzG?gGz?g{Gz?g?g(\?gQ?gףp= ?R Rtabsg{Gz?(i i ( R)tzerost tril_indicestTRRR R*R( tlowerR#tnamestlabels1t expected1tscore1tlabels2t expected2tscore2R,texpectedtscoretname((sNlib/python2.7/site-packages/sklearn/metrics/cluster/tests/test_unsupervised.pyttest_silhouette_paper_exampleUsJ  ' *&#*&#    cCstj}|j}tj|jd}ttdttj |t ||tj |jd}ttdttj |t ||dS(NisJNumber of labels is %d\. Valid values are 2 to n_samples - 1 \(inclusive\)( RRRR)tarangeRRt ValueErrortlentuniqueRR=(RR"R!((sNlib/python2.7/site-packages/sklearn/metrics/cluster/tests/test_unsupervised.pyttest_correct_labelsizes   cCsptj}|j}|j}tt||ddt||tt||ddt||dS(Nii (RRRRRRRR (RR"R,((sNlib/python2.7/site-packages/sklearn/metrics/cluster/tests/test_unsupervised.pyttest_non_encoded_labelss   $cCsMtj}|j}|j}ttt|t|t||dS(N(RRRRRRtlist(RR"R!((sNlib/python2.7/site-packages/sklearn/metrics/cluster/tests/test_unsupervised.pyttest_non_numpy_labelss    cCsDtjjdd}ttd||jddtjddS(s+Assert message when there is only one labeltseedisNumber of labels isi iN(R)trandomt RandomStateRRMtrandR=(tfunctrng((sNlib/python2.7/site-packages/sklearn/metrics/cluster/tests/test_unsupervised.pytassert_raises_on_only_one_labels cCsDtjjdd}ttd||jddtjddS(s7Assert message when all point are in different clustersRTisNumber of labels isi iN(R)RURVRRMRWRL(RXRY((sNlib/python2.7/site-packages/sklearn/metrics/cluster/tests/test_unsupervised.pyt(assert_raises_on_all_points_same_clusters cCsDtttttdttjd dgddgdtdtddgddggddgddgdddgddggdd d gd d ggddd gdd ggdd dgd dggd}dgddgddgdd gd}tjt||d ddddS(Ng?i iiiigiiii-i((i ii$i(RZR R[RR)tonesRR(R"R,((sNlib/python2.7/site-packages/sklearn/metrics/cluster/tests/test_unsupervised.pyttest_calinski_harabaz_scores  "j.cCsttttttjd dgddgdtjdksSttddgddggddgddgdtjdkstddgddggdddgd d ggddd gddggdddgd dggd}dgddgddgddgd}tjt||dtjd dddgddgddgddgg}ddddg}tjt||d d ddS( Ni iiiigiiig?g@(i i( RZR R[R)R\RRtAssertionErrortsqrt(R"R,((sNlib/python2.7/site-packages/sklearn/metrics/cluster/tests/test_unsupervised.pyttest_davies_bouldin_scores  00j.**(!tnumpyR)t scipy.sparsetsparseRRRtsklearnRtsklearn.utils.testingRRRRRRtsklearn.metrics.clusterRR tsklearn.metricsR R R R(R/RKRPRQRSRZR[R]R`(((sNlib/python2.7/site-packages/sklearn/metrics/cluster/tests/test_unsupervised.pyts2   *  1