ó ‡ˆ\c@sdZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z m Z ddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZejd dƒZejd dƒZejjedƒZdddddddgZd„Zd„Z d„Z!d„Z"d„Z#d„Z$d„Z%dS(svThis test for the LFW require medium-size data downloading and processing If the data has not been already downloaded by running the examples, the tests won't run (skipped). If the test are run, the first execution will be long (typically a bit more than a couple of minutes) but as the dataset loader is leveraging joblib, successive runs will be fast (less than 200ms). iÿÿÿÿN(tpartial(tsix(tpillow_installedtimsave(tfetch_lfw_pairs(tfetch_lfw_people(tassert_array_equal(t assert_equal(tSkipTest(t assert_raises(tcheck_return_X_ytprefixtscikit_learn_lfw_test_tscikit_learn_empty_test_tlfw_hometAbdelatif_Smitht Abhati_Keplert Camara_Alvarot Chen_DuponttJohn_Leet Lin_Baumant Onur_Lopezc Csitstdƒ‚ntjjtƒs7tjtƒntjdƒ}t jj dƒ}i}xát D]Ù}tjj td|ƒ}tjj|ƒs¥tj|ƒn|j ddƒ}|||lib/python2.7/site-packages/sklearn/datasets/tests/test_lfw.pyt setup_module,sP   $!"'!!cCsHtjjtƒr"tjtƒntjjtƒrDtjtƒndS(s?Test fixture (clean up) run once after all tests of this moduleN(RRtisdirtSCIKIT_LEARN_DATAtshutiltrmtreetSCIKIT_LEARN_EMPTY_DATA(((s>lib/python2.7/site-packages/sklearn/datasets/tests/test_lfw.pytteardown_moduledscCstttdtdtƒdS(Nt data_hometdownload_if_missing(R tIOErrorRREtFalse(((s>lib/python2.7/site-packages/sklearn/datasets/tests/test_lfw.pyttest_load_empty_lfw_peoplelscCsktdtdddtƒ}t|jjdƒt|jjdƒt|jd d d d d d d d d d g ƒd d dg}t|j |ƒtdtdddddt dtƒ}t|jjdƒt|jd d d dddddd ddd d ddd d gƒt|j d d dddddgƒt tdtdddddt dtƒ}t ||ƒdS(NRGtmin_faces_per_personiRHi i>i/ib iiisAbdelatif Smiths Abhati Keplers Onur Lopeztresizetslice_tcoloriiúiiis Camara Alvaros Chen DupontsJohn Lees Lin Bauman(i i>i/(i ib (iiúiúi(RRBRJRtimagestshapetdataRttargett target_namestNonetTrueRR (t lfw_peopletexpected_classest fetch_func((s>lib/python2.7/site-packages/sklearn/datasets/tests/test_lfw.pyttest_load_fake_lfw_peopleqs,  .   :    c Cs#tttdtdddtƒdS(NRGRLidRH(R t ValueErrorRRBRJ(((s>lib/python2.7/site-packages/sklearn/datasets/tests/test_lfw.pyt)test_load_fake_lfw_people_too_restrictive˜scCstttdtdtƒdS(NRGRH(R RIRRERJ(((s>lib/python2.7/site-packages/sklearn/datasets/tests/test_lfw.pyttest_load_empty_lfw_pairss c Csîtdtdtƒ}t|jjdƒt|jddddddddddg ƒd d g}t|j|ƒtdtd dd dd t dtƒ}t|jjdƒt|jddddddddddg ƒt|j|ƒdS(NRGRHi ii>i/iisDifferent personss Same personRMRNROiúi(i ii>i/(i iiúiúi( RRBRJRtpairsRQRRSRTRURV(tlfw_pairs_trainRX((s>lib/python2.7/site-packages/sklearn/datasets/tests/test_lfw.pyttest_load_fake_lfw_pairs£s  .   .(&t__doc__RRRCttempfiletnumpyR t functoolsRtsklearn.externalsRtsklearn.externals._pilutilRRtsklearn.datasetsRRtsklearn.utils.testingRRRR t"sklearn.datasets.tests.test_commonR tmkdtempRBRERR#RR"R@RFRKRZR\R]R`(((s>lib/python2.7/site-packages/sklearn/datasets/tests/test_lfw.pyt s@       8   '