U
    3dA                      @   s   d Z ddlZddlZddlZddlZddl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dlmZ dadadadd	d
ddddgZdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd ZdS )av  This 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).
    N)partial)fetch_lfw_pairs)fetch_lfw_people)assert_array_equal)check_return_X_yZAbdelatif_SmithZAbhati_KeplerZCamara_AlvaroZChen_DupontZJohn_LeeZ
Lin_BaumanZ
Onur_Lopezc               	   C   s`  t d} tjddatjtdatjdda	tj
tsFtt td}tjd}i }tD ]}tjtd|}tj
|st| |dd	}|||< t|D ]F}tj||d
|  }|jdddd}	| |	tj}
|
| qqdttjtddd}|d W 5 Q R X ttjtdd}|d dd | D }td	D ]@}||}|t|| d\}}|d|||f   qZtd	D ]X}|td\}}|t|| }|t|| }|d||||f   qW 5 Q R X ttjtdd}|d W 5 Q R X ttjtdd}|d W 5 Q R X dS )z<Test fixture run once and common to all tests of this modulez	PIL.ImageZscikit_learn_lfw_test_)prefixZlfw_homeZscikit_learn_empty_test_*   Zlfw_funneled      z	_%04d.jpgr      )   r      )sizez	.test.swpwbs.   Text file to be ignored by the dataset loader.zpairsDevTrain.txts   10
c                 S   s   g | ]\}}|d kr|qS )    ).0namecountr   r   C/tmp/pip-unpacked-wheel-zrfo1fqw/sklearn/datasets/tests/test_lfw.py
<listcomp>O   s      z setup_module.<locals>.<listcomp>r   z	%s	%d	%d
z%s	%d	%s	%d
zpairsDevTest.txts&   Fake place holder that won't be testedz	pairs.txtN)pytestZimportorskiptempfilemkdtempSCIKIT_LEARN_DATAospathjoinLFW_HOMESCIKIT_LEARN_EMPTY_DATAexistsmakedirsrandomRandomnpZRandomState
FAKE_NAMESrandintrangeZ	fromarrayZastypeZuint8saveopenwriteitemschoicesampleencodeZarange)ZImageZrandom_stateZnp_rngcountsr   Zfolder_nameZn_facesi	file_pathZunifaceimgfZmore_than_twofirstsecondZ
first_nameZsecond_nameZfirst_indexZsecond_indexr   r   r   setup_module(   sV    






r6   c                   C   s0   t jtrtt t jtr,tt dS )z?Test fixture (clean up) run once after all tests of this moduleN)r   r   isdirr   shutilrmtreer   r   r   r   r   teardown_moduleg   s    
r:   c                	   C   s&   t t ttdd W 5 Q R X d S NF	data_homedownload_if_missing)r   raisesIOErrorr   r   r   r   r   r   test_load_empty_lfw_peopleo   s    rA   c                  C   s   t tddd} | jjdkst| jjdks.tt| jddddddddddg
 d	d
dg}t| j| t td d ddd} | jjdkst| j	
dstt| jdddddddddddddddddg t| jd	d
dddddg tt td d ddd}t| | d S )Nr   Fr=   min_faces_per_personr>   )
   >   /   )rD   ib  r   r   r	   zAbdelatif SmithzAbhati Keplerz
Onur LopezTr=   resizeslice_colorr>   )   r   r   r   &.. _labeled_faces_in_the_wild_dataset:   r
      zCamara AlvarozChen DupontzJohn Leez
Lin Bauman)r   r   imagesshapeAssertionErrordatar   targettarget_namesDESCR
startswithr   r   )Z
lfw_peopleexpected_classesZ
fetch_funcr   r   r   test_load_fake_lfw_peoplet   sV       
 $rX   c                	   C   s(   t t ttddd W 5 Q R X d S )Nd   FrB   )r   r?   
ValueErrorr   r   r   r   r   r   )test_load_fake_lfw_people_too_restrictive   s    r[   c                	   C   s&   t t ttdd W 5 Q R X d S r;   )r   r?   r@   r   r   r   r   r   r   test_load_empty_lfw_pairs   s    r\   c                  C   s   t tdd} | jjdkstt| jddddddddddg
 ddg}t| j| t td d ddd	} | jjd
ksrtt| jddddddddddg
 t| j| | j	dstd S )NFr<   )rD   r   rE   rF   r	   r   zDifferent personszSame personTrG   )rD   r   r   r   r   rL   )
r   r   pairsrP   rQ   r   rS   rT   rU   rV   )Zlfw_pairs_trainrW   r   r   r   test_load_fake_lfw_pairs   s&       r^   c                  C   sb   t ddt ddf} ttddd| d}|jd	 j| d	 j| d	 j | d
 j| d
 j fks^tdS )zCheck that we properly crop the images.

    Non-regression test for:
    https://github.com/scikit-learn/scikit-learn/issues/24942
    F      N      r   FN)r=   rC   r>   rH   rI   r   r	   )slicer   r   rO   rP   stopstartrQ   )rI   Zlfwr   r   r   'test_fetch_lfw_people_internal_cropping   s    	
rf   )__doc__r"   r   r8   r   Znumpyr$   r   	functoolsr   Zsklearn.datasetsr   r   Zsklearn.utils._testingr   Z"sklearn.datasets.tests.test_commonr   r   r   r   r%   r6   r:   rA   rX   r[   r\   r^   rf   r   r   r   r   <module>   s<   
?:	"