B 0d@sddlmZddlmZddlZddlZddlZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZdd lmZmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZeejeekred eeeeeeeedZ!ddZ"e"eZ#e"eZ$e"eZ%e"eZ&e"eZ'e"eZ(e"eZ)ddZ*ej+ddddZ,ddZ-ddZ.dS) )environ)wrapsN)threadpool_limits) DoctestItem) _IS_32BIT)_openmp_effective_n_threads)_pilutil)PYTEST_MIN_VERSION) np_version parse_version)fetch_20newsgroups)fetch_20newsgroups_vectorized)fetch_california_housing) fetch_covtype)fetch_kddcup99)fetch_olivetti_faces) fetch_rcv1zSYour 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_fxtcs8tdddktfddtfddS)zAFetch dataset (download if missing and requested by environment).SKLEARN_SKIP_NETWORK_TESTS10c sR|d<y ||Stk rL}zt|dkr2tdWdd}~XYnXdS)Ndownload_if_missingz1Data not found and `download_if_missing` is Falsez1test is enabled when SKLEARN_SKIP_NETWORK_TESTS=0)IOErrorstrpytestskip)argskwargse)rfB/var/www/html/venv/lib/python3.7/site-packages/sklearn/conftest.pywrapped-s  z_fetch_fixture..wrappedcsS)Nr&r&)r(r&r'7z _fetch_fixture..)rgetrr fixture)r%r&)rr%r(r'_fetch_fixture)s r-cCsLtdddk}tjjdd}tt}t}xH|D]@}t|dsBq2t|j}||@}|sZq2|rh||O}q2| |q2Wtdd} | dkr|rx|D]} t| qWxn|D]f}|j d rt d krtjjd d} | | q|j d rt d krtjjdd} | | qWd} y ddl} Wntk rDd} d}YnXyBttdkr`d}d} n&trpd}d} ntj drd}d} Wntk rYnXx"|D]}t|tri|j_qW| r tjj|d}xn|D](}t|tr|j dkr| |qWn= 1.14z;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)rr+r markr!setdataset_fetchershasattrr/Z add_markernameendswithplatformpython_implementationmachineZxfail matplotlib ImportErrorr r rsys startswith isinstancerZdtestZglobsrZpillow_installed)configitemsZrun_network_testsZ skip_networkZdataset_features_setZdatasets_to_downloaditemZ item_fixturesZdataset_to_fetchZ worker_idr7markerZ skip_doctestsr<r.Z skip_markerr&r&r'pytest_collection_modifyitemsDs|                      rEfunction)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)r Z importorskipclose)pyplotr&r&r'rJs  rJcCsBtd}|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)rr+intrmaxr)rCZxdist_worker_countZopenmp_threadsZthreads_per_workerr&r&r'pytest_runtest_setups rNcCs0yddl}|dWntk r*YnXdS)NrZagg)r<Zuser=)rAr<r&r&r'pytest_configures rO)/osr functoolsrr9r>r Z threadpoolctlrZ_pytest.doctestrZ sklearn.utilsrZsklearn.utils._openmp_helpersrZsklearn.externalsrZsklearn._min_dependenciesr Zsklearn.utils.fixesr r Zsklearn.datasetsr r rrrrr __version__r=formatr5r-rrrrrrrrEr,rJrNrOr&r&r&r'sR               u