o
    ×tBh   ã                   @   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J ‚dD ]
}|| ¡ v sJ ‚q|jjdks"J ‚|jjdks*J ‚|jjdks2J ‚tt 	t 
|j¡¡t d¡ƒ |j d	¡sIJ ‚t|| ƒ d S )
NTr   )ÚshuffleÚrandom_state)ÚdataÚimagesÚtargetÚDESCR)é  i   )r   é@   r   )r   é(   z.. _olivetti_faces_dataset:)Ú
isinstancer   Úkeysr   Úshaper   r	   r   ÚnpÚuniqueÚsortÚaranger
   Ú
startswithr   )Úfetch_olivetti_faces_fxtr   Úexpected_keys© r   ú/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/sklearn/datasets/tests/test_olivetti_faces.pyÚtest_olivetti_faces   s   r   )
Ú__doc__Únumpyr   Úsklearn.utilsr   Ú"sklearn.datasets.tests.test_commonr   Úsklearn.utils._testingr   r   r   r   r   r   Ú<module>   s    