U md$#@sddlmZddlmZddlZddlZddlmZddlm Z ddl Z ddl Z ddl mZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!ddl"m#Z#ee j$eekr&e%d&eeedkZ'ddZ(eeeeee e!dZ)e'r\e(e)d<e j*j+e,dddkddZ-e j.e j/e j0e-d e j1gd!d"d#Z2d$d%Z3e3eZ4e3eZ5e3eZ6e3eZ7e3eZ8e3e Z9e3e!Z:e .e(Z;d&d'Zd.d/Z?dS)0)environ)wrapsN)suppress)SkipTest)threadpool_limits) DoctestItem) _IS_32BIT)_openmp_effective_n_threads)PYTEST_MIN_VERSION) sp_version) parse_version)fetch_20newsgroups)fetch_20newsgroups_vectorized)fetch_california_housing) fetch_covtype)fetch_kddcup99)fetch_olivetti_faces) fetch_rcv1) random_seedzSYour version of pytest is too old, you should have at least pytest >= {} 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 I/home/sam/Atlas/atlas_env/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)IOErrorstrpytestskip)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} |D]}t|trFi|j_qF| rtjj|d}|D](}t|trz|j dkrz| |qzzd 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.zsklearn._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 startswith isinstancerZdtestZglobsPIL)configitemsrZ skip_networkZdataset_features_setZdatasets_to_downloaditemZ item_fixturesZdataset_to_fetchZ worker_idrGmarkerZ skip_doctestsrKr,Z skip_markerrOZpillow_installedr r r!pytest_collection_modifyitemsms|                  rTfunction)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!rYs  rYcCsBtd}|dkrdSt|}t}t||d}t|dddS)zSet the number of openmp threads based on the number of workers xdist is using to prevent oversubscription. Parameters ---------- item : pytest item item to be processed ZPYTEST_XDIST_WORKER_COUNTNZopenmp)Zuser_api)rrintr maxr)rRZxdist_worker_countZopenmp_threadsZthreads_per_workerr r r!pytest_runtest_setups r]cCsHzddl}|dWntk r*YnX|jdsD|jtdS)NrZaggzsklearn.tests.random_seed)rKZuserZ pluginmanagerZ haspluginregisterr)rPrKr r r!pytest_configures r_)@osr functoolsrrIrL contextlibrZunittestrr5numpynpZ threadpoolctlrZ_pytest.doctestrZ sklearn.utilsrZsklearn.utils._openmp_helpersr Zsklearn._min_dependenciesr Zsklearn.utils.fixesr r Zsklearn.datasetsr rrrrrrZ sklearn.testsr __version__rformatrr"rErCZskipifrZ _SKIP32_MARKr>r/Zfloat32Zfloat64r1r?r#r$r%r&r'r(r)r*rTrYr]r_r r r r!sx                       l