U
    3‰d   ã                   @   s<   d Z ddlZddlmZ ddlmZ ddlmZ dd„ Z	dS )zˆTest Olivetti faces fetcher, if the data is available,
or if specifically requested via environment variable
(e.g. for travis cron job).é    N)ÚBunch)Úcheck_return_X_y)Úassert_array_equalc                 C   s    | ddd}t |tƒst‚dD ]}|| ¡ kst‚q|jjdksDt‚|jjdksTt‚|jjdksdt‚tt	 
t	 |j¡¡t	 d¡ƒ |j d	¡s’t‚t|| ƒ d S )
NTr   )ÚshuffleZrandom_state)ÚdataÚimagesÚtargetÚDESCR)é  i   )r
   é@   r   )r
   é(   z.. _olivetti_faces_dataset:)Ú
isinstancer   ÚAssertionErrorÚkeysr   Úshaper   r   r   ÚnpÚuniqueÚsortZaranger	   Ú
startswithr   )Zfetch_olivetti_faces_fxtr   Zexpected_keys© r   úN/tmp/pip-unpacked-wheel-zrfo1fqw/sklearn/datasets/tests/test_olivetti_faces.pyÚtest_olivetti_faces   s    r   )
Ú__doc__Znumpyr   Zsklearn.utilsr   Z"sklearn.datasets.tests.test_commonr   Zsklearn.utils._testingr   r   r   r   r   r   Ú<module>   s
   