a ;Za@sddlZddlmZddlmZddlZddlZddlmZmZm Z ddl m Z ddl m Z ddZ d d Zd d Zd dZejdgdddZddZdS)N)ThreadPoolExecutor)Parallel) get_config set_configconfig_context)delayed) parse_versionc CsVtdddddksJtddtddus4Jtdd(tdddddksXJWdn1sl0YtddusJtdd$tdd"tddusJWdn1s0YtddusJtddtddusJtddBtddus$JtddtddusBJWdn1sX0YtddusvJWdn1s0YtddusJWdn1s0YtdddddksJtttdWdn1s0YtttddWdn1sH0YdS) NFiTtext) assume_finiteZworking_memoryZprint_changed_onlyZdisplayr r Zdo_something_else)rrrpytestraises TypeError __enter__rr8lib/python3.9/site-packages/sklearn/tests/test_config.pytest_config_context sL   * 0   444 ( rcCstddusJzFtdd(tddus2JtWdn1sL0YWntyjYn0tddus~JdS)Nr FTr )rr ValueErrorrrrrtest_config_context_exceptionGs ( rcCstddusJtddtddus.JtddtddusJJtddtddusfJtddtddusJtttddWdn1s0YdS)Nr Fr Tr )rrr rrrrrrtest_set_configRs     rcCsBt|d$t|tdWdS1s40YdS)zAReturn the value of assume_finite after waiting `sleep_duration`.r r N)rtimesleepr)r Zsleep_durationrrrset_assume_finitebs  rbackend)lokyZmultiprocessingZ threadingcCshttjtdkr$|dkr$tdddg}ddg}t|dd d d t||D}|ddgksdJd 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. z0.12rz+loky backend does not exist in joblib <0.12FT皙?皙?)rZn_jobscss |]\}}tt||VqdS)N)rr).0r Z sleep_durrrr xsz0test_config_threadsafe_joblib..N)rjoblib __version__r skiprzip)rassume_finitessleep_durationsitemsrrrtest_config_threadsafe_joblibis  r(cCsfddg}ddg}tdd(}dd|t||D}Wd n1sH0Y|ddgksbJd 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`.FTrrr)Z max_workerscSsg|]}|qSrr)routputrrr sz*test_config_threadsafe..N)rmapr)r%r&er'rrrtest_config_threadsafes  $r-)rZconcurrent.futuresrr!rr ZsklearnrrrZsklearn.utils.fixesrrrrrrZmarkZ parametrizer(r-rrrrs    :