o tBh/@sddlZddlmZddlmZddlZddlmZmZm Z ddl m Z ddZ dd Z d d Zd d ZejdgdddZddZdS)N)ThreadPoolExecutor)Parallel) get_config set_configconfig_context)delayedc CsPtdddddddksJtddtddusJtddtdddddddks0JWdn1s:wYtddusHJtddtddtddus]JWdn1sgwYtddusuJtddFtddusJtddtddusJtddtddusJWdn1swYtddusJWdn1swYtddusJWdn1swYtdddddddksJtt tdWdn 1swYtttdd WddS1s!wYdS) NFiTdiagram) assume_finiteworking_memoryprint_changed_onlydisplaypairwise_dist_chunk_sizeenable_cython_pairwise_distr r do_something_else)rrrpytestraises TypeError __enter__rrp/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/sklearn/tests/test_config.pytest_config_context sf          $rcCsttddus Jztdd tddusJt1swYWn ty.Ynwtddus8JdS)Nr FTr)rr ValueErrorrrrrtest_config_context_exceptionKs  rcCstddus JtddtddusJtddtddus%Jtddtddus3JtddtddusAJtttddWddS1sWwYdS)Nr FrTr)rrrrrrrrrtest_set_configVs      "rcCsBt|dt|tdWdS1swYdS)zAReturn the value of assume_finite after waiting `sleep_duration`.rr N)rtimesleepr)r sleep_durationrrrset_assume_finitefs  $r backend)lokymultiprocessing threadingcCsDddg}ddg}t|ddddt||D}|ddgks Jd S) 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. FT皙?皙?)r!n_jobscss"|] \}}tt||VqdS)N)rr ).0r sleep_durrrr xs  z0test_config_threadsafe_joblib..N)rzip)r!assume_finitessleep_durationsitemsrrrtest_config_threadsafe_joblibms r0cCsfddg}ddg}tdd}dd|t||D}Wd n1s$wY|ddgks1Jd S) zUses threads directly to test that the global config does not change between threads. Same test as `test_config_threadsafe_joblib` but with `ThreadPoolExecutor`.FTr%r&r') max_workerscSsg|]}|qSrr)r)outputrrr sz*test_config_threadsafe..N)rmapr )r-r.er/rrrtest_config_threadsafes  r6)rconcurrent.futuresrjoblibrrsklearnrrrsklearn.utils.fixesrrrrr mark parametrizer0r6rrrrs   @