o ڷi@s<dZddlZddlmZddlmZddlmZddZ dS)zTest 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_equalcCs|ddd}t|ts JdD] }||vsJq|jjdks"J|jjdks*J|jjdks2Jtt t |jt d|j d sIJt||dS) NTr)shuffle random_state)dataimagestargetDESCR)i)r @r )r (z.. _olivetti_faces_dataset:) isinstancerkeysrshaperr rnpuniquesortaranger startswithr)fetch_olivetti_faces_fxtr expected_keysr`/home/ubuntu/vllm_env/lib/python3.10/site-packages/sklearn/datasets/tests/test_olivetti_faces.pytest_olivetti_faces s r) __doc__numpyr"sklearn.datasets.tests.test_commonr sklearn.utilsrsklearn.utils._testingrrrrrrs