hTddlZddlmZddlZddlmZmZmZddlm Z m Z dZ dZ dZ dZejd gd d Zd ZdS) N)ThreadPoolExecutor) get_config set_configconfig_context)delayedParallelc ftdddddddddksJtdtd dusJtd5tdddddddddksJ dddn #1swxYwYtd dusJtd5td5td dusJ dddn #1swxYwYtd dusJtd5td dusJtd5td dusJtdtd dusJ dddn #1swxYwYtd dusJ dddn #1swxYwYtd dusJ dddn #1swxYwYtdddddddddksJtjt 5tddddn #1swxYwYtjt 5td ddddS#1swxYwYdS) NFiTdiagramdefault) assume_finiteworking_memoryprint_changed_onlydisplayarray_api_dispatchpairwise_dist_chunk_sizeenable_cython_pairwise_disttransform_outputr r do_something_else)rrrpytestraises TypeError __enter__U/var/www/html/Sam_Eipo/venv/lib/python3.11/site-packages/sklearn/tests/test_config.pytest_config_contextr s <<"#$''+%      &&&& << (E 1 1 1 1 d + + +   ||!""& "'(++/ )                       << (E 1 1 1 1 d + + +55 $ / / / 9 9<<0D88888 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9||O,4444 % 0 0 0 : :<<0E9999d333 = =!||O4====....!||O4<<<<<  = = = = = = = = = = = = = = =<<0E99999 : : : : : : : : : : : : : : :||O,44444'555555555555555* <<"#$''+%       y ! !t y ! !;;...88:::;;;;;;;;;;;;;;;;;;sBBB4G5C+ G5+C/ /G52C/ 3+G5)GAF GF GF G9 G5G G5 G G55G9<G96III6#J&&J*-J*ctddusJ td5tddusJt#1swxYwYn#t$rYnwxYwtddusJdS)Nr FTr)rr ValueErrorrrrtest_config_context_exceptionr"Ps << (E 1 1 1 1 $ / / /  <<0D8888,,                << (E 1 1 1 1 1 1s.A'AAAAA A*)A*ctddusJtdtddusJtdtddusJtdtddusJtdtddusJtjt5tdddddS#1swxYwYdS)Nr FrTr)rrrrrrrrtest_set_configr$[sO << (E 1 1 1 1T"""" << (E 1 1 1 1T"""" << (D 0 0 0 0T"""" << (D 0 0 0 0U#### << (E 1 1 1 1 y ! !++T****++++++++++++++++++sC00C47C4ct|5tj|tdcdddS#1swxYwYdS)zAReturn the value of assume_finite after waiting `sleep_duration`.rr N)rtimesleepr)r sleep_durations rset_assume_finiter)ks m 4 4 4-- >"""||O,------------------s(AA  A backend)lokymultiprocessing threadingcgd}gd}t|ddt||D}|gdksJdS)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皙?皙?r1r2)r*n_jobsc3XK|]%\}}tt||V&dS)N)rr)).0r sleep_durs r z0test_config_threadsafe_joblib..}sO00 $M9 #!""=)<<000000rN)rzip)r*assume_finitessleep_durationsitemss rtest_config_threadsafe_joblibr=rs~0//N***O /HWQ / / /00(+NO(L(L000  E ... . . . . . .rcgd}gd}td5}d|t||D}dddn #1swxYwY|gdksJdS)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/r0r3) max_workerscg|]}|Srr)r6outputs r z*test_config_threadsafe..s(       rN)rmapr))r:r;er<s rtest_config_threadsaferEs 0//N***O  * * * a  %% 1>?SS                  ... . . . . . .s'A  AA)r&concurrent.futuresrrsklearnrrrsklearn.utils.parallelrrrr"r$r)mark parametrizer=rErrrrKs 111111 ::::::::::44444444C;C;C;L222 + + + ---$L$L$LMM//NM/$/////r