U ^{j$"@sddlZddlZddlmZddlmZddlmZddlm Z ddl Z ddl Z ddl Z ddlmZddlmZddlmZdd lmZmZmZmZmZmZmZdd lmZdd lmZdd l m!Z!m"Z"m#Z#e"e j$e"ekre%d e j$dede#e"dkZ&ddZ'eeeeeeedZ(e&r,e'e(d<e j)j*e+dddkddZ,e j-e j.e j/e,de j0gdddZ1ddZ2e2eZ3e2eZ4e2eZ5e2eZ6e2eZ7e2eZ8e2eZ9e -e'Z:d d!Z;e j-d"d#d$d%Z= z installed.z1.10cCsntrXtdddk}|s tdz ddl}Wntk rHtdYnXddlm}n ddlm}|dd S) NSKLEARN_SKIP_NETWORK_TESTS101test is enabled when SKLEARN_SKIP_NETWORK_TESTS=0rz#test requires pooch to be installed)faceT)gray) scipy_datasets_require_networkrgetrpooch ImportErrorZscipy.datasetsrZ scipy.misc)run_network_testsrrr O/var/www/html/bwave/python/venv/lib/python3.8/site-packages/sklearn/conftest.pyraccoon_face_or_skip%s  r")fetch_20newsgroups_fxt!fetch_20newsgroups_vectorized_fxtfetch_california_housing_fxtfetch_covtype_fxtfetch_kddcup99_fxtfetch_olivetti_faces_fxtfetch_rcv1_fxtraccoon_face_fxtZSKLEARN_RUN_FLOAT32_TESTSrrz:Set SKLEARN_RUN_FLOAT32_TESTS=1 to run float32 dtype testsreason)Zmarks)paramsccs |jVdSN)param)requestr r r! global_dtypeLsr1cs8tdddktfddtfddS)zAFetch dataset (download if missing and requested by environment).rrrc sT|d<z ||WStk rN}zt|dkr4tdW5d}~XYnXdS)Ndownload_if_missingz1Data not found and `download_if_missing` is Falser)OSErrorstrpytestskip)argskwargse)r2fr r!wrappedUs  z_fetch_fixture..wrappedcsSr.r r )r;r r!_z _fetch_fixture..)rrrr5fixture)r:r )r2r:r;r!_fetch_fixtureQs r?c Cs tdddk}tjjdd}tt}t}|D]@}t|ds@q0t|j}||@}|sXq0|rf||O}q0| |q0tdd} | dkr|r|D]"} t t t| W5QRXq|D]4}|j d rtd krtjjd d} | | qd } z d dl} Wntk rd} d}YnXtr,d}d} ntjdrBd}d} ttdkrXd}d} |D]}t|tr\i|j_q\| rtjj|d}|D](}t|tr|j dkr| |qzd dl}d}Wntk rd }YnX|stjjdd}|D]}|j dkr| |qdS)zCalled after collect is completed. Parameters ---------- config : pytest config items : list of collected items rrrrr+ fixturenamesZPYTEST_XDIST_WORKERZgw0ZGradientBoostingClassifieraarch64zKknow failure. See https://github.com/scikit-learn/scikit-learn/issues/17797FrNTz*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 2zsklearn._config.config_contextzpillow (or PIL) not installed!)z/sklearn.feature_extraction.image.PatchExtractorz3sklearn.feature_extraction.image.extract_patches_2d)rrr5markr6setdataset_fetchershasattrr@Z add_markerrrnameendswithplatformmachineZxfail matplotlibrrsys startswithrr isinstancerZdtestZglobsPIL)configitemsrZ skip_networkZdataset_features_setZdatasets_to_downloaditemZ item_fixturesZdataset_to_fetchZ worker_idrHmarkerZ skip_doctestsrLr,Z skip_markerrPZpillow_installedr r r!pytest_collection_modifyitemsms                  rUfunction)scopeccs(td}|d|V|ddS)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. zmatplotlib.pyplotallN)r5Z importorskipclose)pyplotr r r!rZs  rZcCszddl}|dWntk r*YnXtjdd}td}|dk r\t|t|d}t ||j ds||j t dS)NrZaggT)Zonly_physical_coresZPYTEST_XDIST_WORKER_COUNTzsklearn.tests.random_seed)rLZuserjoblib cpu_countrrmaxintrZ pluginmanagerZ haspluginregisterr)rQrLZallowed_parallelismZxdist_worker_countr r r!pytest_configures   ra)>rJrM contextlibr functoolsrosrZunittestrr\numpynpr5Z_pytest.doctestrZ threadpoolctlrZsklearn._min_dependenciesrZsklearn.datasetsr r r r r rrZ sklearn.testsrZ sklearn.utilsrZsklearn.utils.fixesrrr __version__rrr"rFrDZskipifrZ _SKIP32_MARKr>r/Zfloat32Zfloat64r1r?r#r$r%r&r'r(r)r*rUrZrar r r r!sd       $      o