U !`9@sddlZddlmZddlmZddlZddlmZddlmZddl m Z ddl m Z ddl m Z dd l m Z dd l mZdd l mZdd l mZe e e e eeed ZddZee Zee Zee Zee ZeeZeeZeeZddZejddddZddZdS)N)environ)wraps)threadpool_limits)_openmp_effective_n_threads)fetch_20newsgroups)fetch_20newsgroups_vectorized)fetch_california_housing) fetch_covtype)fetch_kddcup99)fetch_olivetti_faces) fetch_rcv1)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_TESTS10cs8|d<z ||WStk r2tdYnXdS)Ndownload_if_missing1test is enabled when SKLEARN_SKIP_NETWORK_TESTS=0)IOErrorpytestskip)argskwargs)rf/lib/python3.8/site-packages/sklearn/conftest.pywrapped!s  z_fetch_fixture..wrappedcsS)Nrr)r!rr (z _fetch_fixture..)rgetrrfixture)rr)rrr!r _fetch_fixturesr&c Cstdddk}tjjdd}tt}t}|D]@}t|ds@q0t|j}||@}|sXq0|rf||O}q0| |q0tdd} | dkr|r|D]} t| qd S) zCalled after collect is completed. Parameters ---------- config : pytest config items : list of collected items rrrr)reason fixturenamesZPYTEST_XDIST_WORKERZgw0N) rr$rZmarkrsetdataset_fetchershasattrr(Z add_marker) ZconfigitemsZrun_network_testsZ skip_networkZdataset_features_setZdatasets_to_downloaditemZ item_fixturesZdataset_to_fetchZ worker_idnamerrr pytest_collection_modifyitems6s(      r/Zfunction)Zscopeccs2td}|dtd}|V|ddS)aASetup and teardown fixture for matplotlib. This fixture checks if we can import matplotlib. If not, the tests will be skipped. Otherwise, we setup matplotlib backend and close the figures after running the functions. Returns ------- pyplot : module The ``matplotlib.pyplot`` module. matplotlibZaggzmatplotlib.pyplotallN)rZ importorskipZuseclose)r0pyplotrrr r3^s   r3cCsNzttjd}Wntk r(YdSXt}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)intosrKeyErrorrmaxr)r-Zxdist_worker_countZopenmp_threadsZthreads_per_workerrrr pytest_runtest_setuprs r9)r6r functoolsrrZ threadpoolctlrZsklearn.utils._openmp_helpersrZsklearn.datasetsrrrr r r r r*r&r rrrrrrr/r%r3r9rrrr sB            (