i3 ddlZddlZddlZddlmZddlmZddlmZddl m Z ddl Z ddl Z ddlZddlmZddlmZddlmZdd lmZdd lmZmZmZmZmZmZmZmZm Z m!Z!dd l"m#Z#dd l$m%Z%m&Z&m'Z'm(Z( ddl)Z)d Z* ddl,m-Z-e'ej^e'ekre+dej^deddZ0eeeeeeeee e!d Z1e0e1d<ejdjgejhdddk7dZ5ejlejne jpe5e jrgdZ:dZ;e;eZ<e;eZ=e;eZ>e;eZ?e;eZ@e;eZAe;eZBe;eZCe;e ZDe;e!ZEejle0ZFdZGejld !d"ZHd#ZId$ZJd%ZKejld&ZLe-d e-_Myy#e+$rdZ*YswxYw#e.$rdZ-YzwxYw)'N)suppress)wraps)environ)SkipTest) DoctestItem)face)threadpool_limits)PYTEST_MIN_VERSION) fetch_20newsgroupsfetch_20newsgroups_vectorizedfetch_california_housing fetch_covtypefetch_kddcup99fetch_lfw_pairsfetch_lfw_peoplefetch_olivetti_faces fetch_rcv1fetch_species_distributions)get_pytest_filterwarning_lines) _IS_32BITnp_base_version parse_version sp_versionTF) dt_configz/Your version of pytest is too old. Got version z, you should have pytest >= z installed.ctjdddk(}|s td ddl}t dS#t$r tdwxYw) NSKLEARN_SKIP_NETWORK_TESTS101test is enabled when SKLEARN_SKIP_NETWORK_TESTS=0rz#test requires pooch to be installedT)gray)rgetrpooch ImportErrorr)run_network_testsr"s e/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/sklearn/conftest.pyraccoon_face_or_skipr&<sY $@#F#M JKK> T? ><==>s 8A ) fetch_20newsgroups_fxt!fetch_20newsgroups_vectorized_fxtfetch_california_housing_fxtfetch_covtype_fxtfetch_kddcup99_fxtfetch_lfw_pairs_fxtfetch_lfw_people_fxtfetch_olivetti_faces_fxtfetch_rcv1_fxtfetch_species_distributions_fxtraccoon_face_fxtSKLEARN_RUN_FLOAT32_TESTSrrz:Set SKLEARN_RUN_FLOAT32_TESTS=1 to run float32 dtype testsreason)marks)paramsc#(K|jywN)param)requests r% global_dtyper;_s --sctjdddk(tfdtjfdS)zAFetch dataset (download if missing and requested by environment).rrrc|d< |i|S#t$r.}t|dk7rtjdYd}~yd}~wwxYw)Ndownload_if_missingz1Data not found and `download_if_missing` is Falser)OSErrorstrpytestskip)argskwargser>fs r%wrappedz_fetch_fixture..wrappedhsV(;$% Md%f% % M1vLL KKK L L Ms A$AAcSr8)rGsr%z _fetch_fixture..rs')rr!rrAfixture)rFr>rGs`@@r%_fetch_fixturerMdsC!++&BCHCO 1XMM >>/ **rKctjdddk(}tjj d}t t }t }|D]}t|trAd|jvr3|jjdd}|d }t |g|z} n't|d sat |j} | |z} | s~|r|| z}|j|tjd d } | d k(r.|r,|D]'} tt5t | d d d )|D]g}|jj!dst#j$dk(s7tjj'd} |j| id} dd l}t,rd}d}n#t.j"j1drd}d}t2t5dkrd}d}t6t5dkrd}d}|D]$}t|tsi|j8_&|rYtjj }|D]4}t|ts|jdk7s$|j|6 dd l}d}|sHtjj d}|D]"}|jdvs|j|$y y #1swYxYw#t*$rd}d}YMwxYw#t*$rd}YywxYw)zCalled after collect is completed. Parameters ---------- config : pytest config items : list of collected items rrrrr3fetch_._fxt fixturenamesPYTEST_XDIST_WORKERgw0NGradientBoostingClassifieraarch64zKknow failure. See https://github.com/scikit-learn/scikit-learn/issues/17797FrTz*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 2z1.14z2Scipy sparse matrix repr has changed in scipy 1.14zsklearn._config.config_contextzpillow (or PIL) not installed!)z/sklearn.feature_extraction.image.PatchExtractorz3sklearn.feature_extraction.image.extract_patches_2d)rr!rAmarkrBsetdataset_fetchers isinstancernamesplithasattrrS add_markerrrendswithplatformmachinexfail matplotlibr#rsys startswithrrrdtestglobsPIL)configitemsr$ skip_networkdataset_features_setdatasets_to_downloaditemfetcher_function_namedataset_fetchers_keydataset_to_fetch item_fixtures worker_idr^marker skip_doctestsrfr4 skip_markerrkpillow_installeds r%pytest_collection_modifyitemsr{s $@#F#M;;##B$L /05 dK (X-B$(IIOOC$8$< !&;% N   ) 5  s++J M&))E  dK (!DJJ kk&&f&5 D$ , 99 @@OOK0! kk&&.N&O Dyy ,  M$#( > =>^ ! !s0K&K' K;K$ ' K87K8; L L function)scopec#Ktjd}|jd||jdyw)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)rA importorskipclose)pyplots r%rrs6 !4 5F LL L LLs=?cttd}tjd}dg}||}n|dk(r|}nyd|vr?|j d\}}ttt |t |dz}n t |g}t |dkst|d kDrtd |d |jvr|jd |yy) aParametrization of global_random_seed fixture based on the SKLEARN_TESTS_GLOBAL_RANDOM_SEED environment variable. The goal of this fixture is to prevent tests that use it to be sensitive to a specific seed value while still being deterministic by default. See the documentation for the SKLEARN_TESTS_GLOBAL_RANDOM_SEED variable for instructions on how to use this fixture. https://scikit-learn.org/dev/computing/parallelism.html#sklearn-tests-global-random-seed d SKLEARN_TESTS_GLOBAL_RANDOM_SEED*Nr-rczxThe value(s) of the environment variable SKLEARN_TESTS_GLOBAL_RANDOM_SEED must be in the range [0, 99] (or 'all'), got: global_random_seed) listrangerr!r_intminmax ValueErrorrS parametrize)metafuncRANDOM_SEED_RANGErandom_seed_vardefault_random_seeds random_seedsstartstops r%pytest_generate_testsrs$U3Z(kk"DEO4+ E !( / !)//4KE4c%j#d)a- @AL01L | q C $5$:$$3#46  x4441<@5rKc6ts|jddyy)Nthread_unsafe_fixturesz list of stuff)PARALLEL_RUN_AVAILABLEaddini)parser pluginmanagers r%pytest_addoptionr?s ! .@ "rKc ddl}|jdtjd}t j d}|t|t|zd}t|t j dddk7r!tD]}|jd |tsI|jd d |jd d |jd d |jd d yy#t$rYwxYw)NraggT)only_physical_coresPYTEST_XDIST_WORKER_COUNTrSKLEARN_WARNINGS_AS_ERRORSrfilterwarningsmarkerszOparallel_threads(n): run the given test function in parallel using `n` threads.z8thread_unsafe: mark the test function as single-threadedzCiterations(n): run the given test function `n` times in each thread) rfuser#joblib cpu_countrr!rrr raddinivalue_liner)rlrfallowed_parallelismxdist_worker_countlines r%pytest_configurerDs u!**tD %@A%""5=O9P"PRST)*{{/5<34D  # #$4d ;5 "  !   F    Q    Q  "%    sC11 C=<C=c^tjfd}|jtd|y)z!Pretend pandas was not installed.c:|dk(r t|g|i|S)Npandas)r#)r^rCrD import_origs r% mocked_importz,hide_available_pandas..mocked_importus( 8 - 41$1&11rK __import__N)builtinsrsetattr) monkeypatchrrs @r%hide_available_pandasrps)%%K2 , >rK)Nrrcrg contextlibr functoolsrosrunittestrrnumpynprA_pytest.doctestrscipy.datasetsr threadpoolctlr sklearn._min_dependenciesr sklearn.datasetsr r r rrrrrrrsklearn.utils._testingrsklearn.utils.fixesrrrrpytest_run_parallelrr#scipy_doctest.conftestrModuleNotFoundError __version__r&r\rZskipifr! _SKIP32_MARKrLr9float32float64r;rMr'r(r)r*r+r,r-r.r/r0r1r{rrrrr strict_checkrIrKr%rsv   '+8   B#! 0##$}5G'HH  9&:L:L9MN""4!5[ B  1)F$<&(*, 4 'B (<#${{!!GKK+S1S8 G"   RZZ|DbjjQRS+$((:;$23P$Q!-.FG"=1#N3$_5%&67)*>? +"01L"M!6>>"67y-xj!"$*AZA ) X ? ? !Ic #"# Is$4G#;G1#G.-G.1G<;G<