U
    9%e$"                     @   s  d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlm	Z	 d dl
Z
d dlZd dlZd dlmZ d dlmZ d dlmZ d d	lmZmZmZmZmZmZmZ d d
lmZ d dlmZ d dl m!Z!m"Z"m#Z# e"ej$e"ek re%dej$ de de#e"dkZ&dd Z'eeeeeeedZ(e&r,e'e(d< ej)j*e+dddkddZ,ej-ej.ej/e,dej0gddd Z1dd Z2e2eZ3e2eZ4e2eZ5e2eZ6e2eZ7e2eZ8e2eZ9e-e'Z:d d! Z;ej-d"d#d$d% Z<d&d' Z=dS )(    N)suppress)wraps)environ)SkipTest)DoctestItem)threadpool_limits)PYTEST_MIN_VERSION)fetch_20newsgroupsfetch_20newsgroups_vectorizedfetch_california_housingfetch_covtypefetch_kddcup99fetch_olivetti_faces
fetch_rcv1)random_seed)	_IS_32BIT)np_base_versionparse_version
sp_versionz/Your version of pytest is too old. Got version z, you should have pytest >= z installed.z1.10c                  C   sn   t rXtdddk} | s tdzdd l}W n tk
rH   tdY nX ddlm} nddlm} |dd	S )
NSKLEARN_SKIP_NETWORK_TESTS101test is enabled when SKLEARN_SKIP_NETWORK_TESTS=0r   z#test requires pooch to be installed)faceT)gray)	scipy_datasets_require_networkr   getr   poochImportErrorZscipy.datasetsr   Z
scipy.misc)run_network_testsr   r    r    O/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/sklearn/conftest.pyraccoon_face_or_skip%   s    r"   )fetch_20newsgroups_fxt!fetch_20newsgroups_vectorized_fxtfetch_california_housing_fxtfetch_covtype_fxtfetch_kddcup99_fxtfetch_olivetti_faces_fxtfetch_rcv1_fxtraccoon_face_fxtZSKLEARN_RUN_FLOAT32_TESTSr   r   z:Set SKLEARN_RUN_FLOAT32_TESTS=1 to run float32 dtype testsreason)Zmarks)paramsc                 c   s   | j V  d S N)param)requestr    r    r!   global_dtypeL   s    r1   c                    s8   t dddk t fddtfddS )zAFetch dataset (download if missing and requested by environment).r   r   r   c               
      sT    |d< z| |W S  t k
rN } zt|dkr4 td W 5 d }~X Y nX d S )Ndownload_if_missingz1Data not found and `download_if_missing` is Falser   )OSErrorstrpytestskip)argskwargse)r2   fr    r!   wrappedU   s    z_fetch_fixture.<locals>.wrappedc                      s    S r.   r    r    )r;   r    r!   <lambda>_       z _fetch_fixture.<locals>.<lambda>)r   r   r   r5   fixture)r:   r    )r2   r:   r;   r!   _fetch_fixtureQ   s    	r?   c              
   C   s   t dddk}tjjdd}tt}t }|D ]@}t|ds@q0t|j}||@ }|sXq0|rf||O }q0|	| q0t dd}	|	dkr|r|D ]"}
t
t t|
   W 5 Q R X q|D ]4}|jd	rt d
krtjjdd}|	| qd}zddl}W n tk
r   d}d}Y nX tr,d}d}ntjdrBd}d}ttdkrXd}d}|D ]}t|tr\i |j_q\|rtjj|d}|D ](}t|tr|jdkr|	| qzddl}d}W n tk
r   d}Y nX |stjjdd}|D ]}|jdkr|	| qdS )zCalled after collect is completed.

    Parameters
    ----------
    config : pytest config
    items : list of collected items
    r   r   r   r   r+   fixturenamesZPYTEST_XDIST_WORKERZgw0ZGradientBoostingClassifieraarch64zKknow failure. See https://github.com/scikit-learn/scikit-learn/issues/17797Fr   NTz*matplotlib is required to run the doctestsz;doctest are only run when the default numpy int is 64 bits.win32z\doctests are not run for Windows because numpy arrays repr is inconsistent across platforms.2z6Due to NEP 51 numpy scalar repr has changed in numpy 2zsklearn._config.config_contextzpillow (or PIL) not installed!)z/sklearn.feature_extraction.image.PatchExtractorz3sklearn.feature_extraction.image.extract_patches_2d)r   r   r5   markr6   setdataset_fetchershasattrr@   Z
add_markerr   r   nameendswithplatformmachineZxfail
matplotlibr   r   sys
startswithr   r   
isinstancer   ZdtestZglobsPIL)configitemsr   Zskip_networkZdataset_features_setZdatasets_to_downloaditemZitem_fixturesZdataset_to_fetchZ	worker_idrH   markerZskip_doctestsrL   r,   Zskip_markerrP   Zpillow_installedr    r    r!   pytest_collection_modifyitemsm   s    







rU   function)scopec                  c   s(   t d} | d | V  | d dS )a/  Setup and teardown fixture for matplotlib.

    This fixture checks if we can import matplotlib. If not, the tests will be
    skipped. Otherwise, we close the figures before and after running the
    functions.

    Returns
    -------
    pyplot : module
        The ``matplotlib.pyplot`` module.
    zmatplotlib.pyplotallN)r5   Zimportorskipclose)pyplotr    r    r!   rZ      s    

rZ   c                 C   s   zdd l }|d W n tk
r*   Y nX tjdd}td}|d k	r\t|t| d}t	| | j
ds|| j
t d S )Nr   ZaggT)Zonly_physical_coresZPYTEST_XDIST_WORKER_COUNT   zsklearn.tests.random_seed)rL   Zuser   joblib	cpu_countr   r   maxintr   ZpluginmanagerZ	haspluginregisterr   )rQ   rL   Zallowed_parallelismZxdist_worker_countr    r    r!   pytest_configure   s    
ra   )>rJ   rM   
contextlibr   	functoolsr   osr   Zunittestr   r\   numpynpr5   Z_pytest.doctestr   Zthreadpoolctlr   Zsklearn._min_dependenciesr   Zsklearn.datasetsr	   r
   r   r   r   r   r   Zsklearn.testsr   Zsklearn.utilsr   Zsklearn.utils.fixesr   r   r   __version__r   r   r"   rF   rD   Zskipifr   Z_SKIP32_MARKr>   r/   Zfloat32Zfloat64r1   r?   r#   r$   r%   r&   r'   r(   r)   r*   rU   rZ   ra   r    r    r    r!   <module>   sd   $	


o

