B
    0d                 @   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 )zTest 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xdD ]}|| ks tq W |jjdksHt|jjdksXt|jjdkshttt	
t	|jt	d |jd	stt||  d S )
NTr   )shuffleZrandom_state)dataimagestargetDESCR)i  i   )i  @   r
   )i  (   z.. _olivetti_faces_dataset:)
isinstancer   AssertionErrorkeysr   shaper   r   r   npuniquesortZaranger	   
startswithr   )Zfetch_olivetti_faces_fxtr   Zexpected_keys r   \/var/www/html/venv/lib/python3.7/site-packages/sklearn/datasets/tests/test_olivetti_faces.pytest_olivetti_faces   s    
r   )
__doc__numpyr   Zsklearn.utilsr   Z"sklearn.datasets.tests.test_commonr   Zsklearn.utils._testingr   r   r   r   r   r   <module>   s
   