o \i@sddlZddlmZddlZddlZddlmZmZmZddlm Z ddl m Z m Z ddZ dd Zd d Zd d ZejdgdddZejje ddddZddZdS)N)ThreadPoolExecutor)config_context get_config set_config)_IS_WASM)Paralleldelayedc Cshtddddddddddd ksJtddtd dus Jtddtddddddddddd ks8JWdn1sBwYtd dusPJtddtddtd duseJWdn1sowYtd dus}JtddFtd dusJtddtd dusJtddtd dusJWdn1swYtd dusJWdn1swYtd dusJWdn1swYtddddddddddd ksJtt tdWdn 1swYtttdd WddS1s-wYdS) NFiTdiagramdefault) assume_finiteworking_memoryprint_changed_onlydisplayarray_api_dispatchpairwise_dist_chunk_sizeenable_cython_pairwise_disttransform_outputenable_metadata_routingskip_parameter_validationr r do_something_else)rrrpytestraises TypeError __enter__rrz/var/www/www-root/data/www/176.119.141.140/sports-predictor/venv/lib/python3.10/site-packages/sklearn/tests/test_config.pytest_config_context s~           $rcCsttddus Jztdd tddusJt1swYWn ty.Ynwtddus8JdS)Nr FTr)rr ValueErrorrrrrtest_config_context_exceptionXs  r!cCstddus JtddtddusJtddtddus%Jtddtddus3JtddtddusAJtttddWddS1sWwYdS)Nr FrTr)rrrrrrrrrtest_set_configcs      "r"cCsBt|dt|tdWdS1swYdS)zAReturn the value of assume_finite after waiting `sleep_duration`.rr N)rtimesleepr)r sleep_durationrrrset_assume_finitess  $r&backend)lokymultiprocessing threadingcCsDgd}gd}t|ddddt||D}|gdks JdS)aoTest that the global config is threadsafe with all joblib backends. Two jobs are spawned and sets assume_finite to two different values. When the job with a duration 0.1s completes, the assume_finite value should be the same as the value passed to the function. In other words, it is not influenced by the other job setting assume_finite to True. FTFT皙?皙?r-r.)r'n_jobscss"|] \}}tt||VqdS)N)rr&).0r sleep_durrrr s  z0test_config_threadsafe_joblib..N)rzip)r'assume_finitessleep_durationsitemsrrrtest_config_threadsafe_joblibzs r8zcannot start threads)reasoncCsfgd}gd}tdd}dd|t||D}Wdn1s$wY|gdks1JdS)zUses threads directly to test that the global config does not change between threads. Same test as `test_config_threadsafe_joblib` but with `ThreadPoolExecutor`.r+r,r/) max_workerscSsg|]}|qSrr)r1outputrrr sz*test_config_threadsafe..N)rmapr&)r5r6er7rrrtest_config_threadsafes  r?c Cs|tjjjddtjtddtddWdn1s"wYWdn1s1wYtjtddt ddWddS1sNwYdS)z!Check error when SciPy is too old __version__z1.13.0zSciPy must be 1.14.0 or newer)matchT)rN) setattrsklearnutils _array_apiscipyrr ImportErrorrr) monkeypatchrrr*test_config_array_api_dispatch_error_scipys  "rI)r#concurrent.futuresrrrCrrrsklearn.utils.fixesrsklearn.utils.parallelrrrr!r"r&mark parametrizer8xfailr?rIrrrrs   L