o
    Ó\i—  ã                   @   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 CI jobs).é    N)Úcheck_return_X_y)ÚBunch)Ú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/www-root/data/www/176.119.141.140/sports-predictor/venv/lib/python3.10/site-packages/sklearn/datasets/tests/test_olivetti_faces.pyÚtest_olivetti_faces   s   r   )
Ú__doc__Únumpyr   Ú"sklearn.datasets.tests.test_commonr   Úsklearn.utilsr   Úsklearn.utils._testingr   r   r   r   r   r   Ú<module>   s    