U 2dk@sddlmZddlmZddlZddlZddlZddlZddl m Z ddl m Z ddl mZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZeejeekre d!eeeeeeeedZ"ej#j$e%dddkddZ&ej'ej(ej)e&dej*gdddZ+ddZ,e,eZ-e,eZ.e,eZ/e,eZ0e,eZ1e,eZ2e,eZ3dd Z4ej'd!d"d#d$Z5d%d&Z6d'd(Z7dS)))environ)wrapsN)threadpool_limits) DoctestItem) _IS_32BIT)_openmp_effective_n_threads)PYTEST_MIN_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.)fetch_20newsgroups_fxt!fetch_20newsgroups_vectorized_fxtfetch_california_housing_fxtfetch_covtype_fxtfetch_kddcup99_fxtfetch_olivetti_faces_fxtfetch_rcv1_fxtZSKLEARN_RUN_FLOAT32_TESTS01z:Set SKLEARN_RUN_FLOAT32_TESTS=1 to run float32 dtype testsreason)Zmarks)paramsccs |jVdSN)param)requestr!4/tmp/pip-unpacked-wheel-zrfo1fqw/sklearn/conftest.py global_dtype0sr#cs8tdddktfddtfddS)zAFetch dataset (download if missing and requested by environment).SKLEARN_SKIP_NETWORK_TESTSrrc sT|d<z ||WStk rN}zt|dkr4tdW5d}~XYnXdS)Ndownload_if_missingz1Data not found and `download_if_missing` is False1test is enabled when SKLEARN_SKIP_NETWORK_TESTS=0)IOErrorstrpytestskip)argskwargse)r%fr!r"wrapped9s  z_fetch_fixture..wrappedcsSrr!r!)r/r!r"Cz _fetch_fixture..)rgetrr)fixture)r.r!)r%r.r/r"_fetch_fixture5s r4cCstdddk}tjjdd}tt}t}|D]@}t|ds@q0t|j}||@}|sXq0|rf||O}q0| |q0tdd} | dkr|r|D]} t| q|D]4}|j d rt d krtjjd d} | | qd } z d dl} Wntk rd} d}YnXtrd}d} ntj dr.d}d} |D]}t|tr2i|j_q2| rtjj|d}|D](}t|trf|j dkrf| |qfzd dl}d}Wntk rd }YnX|stjjdd}|D]}|j dkr| |qdS)zCalled after collect is completed. Parameters ---------- config : pytest config items : list of collected items r$rrr&r 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)rr2r)markr*setdataset_fetchershasattrr5Z add_markernameendswithplatformmachineZxfail matplotlib ImportErrorrsys startswith isinstancerZdtestZglobsPIL)configitemsZrun_network_testsZ skip_networkZdataset_features_setZdatasets_to_downloaditemZ item_fixturesZdataset_to_fetchZ worker_idr<markerZ skip_doctestsr@rZ skip_markerrEZpillow_installedr!r!r"pytest_collection_modifyitemsPsz                  rJfunction)Zscopeccs(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)r)Z importorskipclose)pyplotr!r!r"rNs  rNcCsBtd}|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)rr2intrmaxr)rHZxdist_worker_countZopenmp_threadsZthreads_per_workerr!r!r"pytest_runtest_setups rRcCsHzddl}|dWntk r*YnX|jdsD|jtdS)NrZaggzsklearn.tests.random_seed)r@ZuserAZ pluginmanagerZ haspluginregisterr)rFr@r!r!r"pytest_configures rT)8osr functoolsrr>rBr)ZnumpynpZ threadpoolctlrZ_pytest.doctestrZ sklearn.utilsrZsklearn.utils._openmp_helpersrZsklearn._min_dependenciesrZsklearn.utils.fixesr Zsklearn.datasetsr r r r rrrZ sklearn.testsr __version__rAformatr:r8Zskipifr2Z _SKIP32_MARKr3rZfloat32Zfloat64r#r4rrrrrrrrJrNrRrTr!r!r!r"sh                  k