:jddlZddlmZddlZddlZddlmZmZmZddlm Z ddl m Z m Z dZ dZdZd Zej"j%d gd d Zej"j)e d dZdZy)N)ThreadPoolExecutor)config_context get_config set_config)_IS_WASM)Paralleldelayedcbtdddddddddddd k(sJtd td dusJtd 5tdddddddddddd k(sJ dddtd dusJtd 5td 5td dusJ dddtd dusJtd 5td dusJtd 5td dusJtd td dusJ dddtd dusJ dddtd dusJ dddtdddddddddddd k(sJtjt 5tddddtjt 5td j dddy#1swY{xYw#1swYCxYw#1swYxYw#1swYxYw#1swYxYw#1swYxYw#1swYyxYw) NFiTdiagramdefaultspmatrix) assume_finiteworking_memoryprint_changed_onlydisplayarray_api_dispatchpairwise_dist_chunk_sizeenable_cython_pairwise_disttransform_outputenable_metadata_routingskip_parameter_validationsparse_interfacerrdo_something_else)rrrpytestraises TypeError __enter__n/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/sklearn/tests/test_config.pytest_config_contextr$ s <"#$''+%#(%*&    & < (E 11 1 d +  |!""& "'(++/ )',). *      < (E 11 1 d +5 $ / 9<0D8 88 9|O,444 % 0 :<0E9 99d3 =!|O4===.!|O4<<<  =<0E9 99 :|O,444'5* <"#$''+%#(%*&     y !t y !;.88:;;q   " 9 9 = = : : 55H;;sxG H G('%H  H*/G5H4H  H7H%G%(G2 -H 5G> :HH H  HH"%H.ctddusJ td5tddusJt#1swYnxYwn#t$rYnwxYwtddusJy)NrFTr)rr ValueErrorr!r"r#test_config_context_exceptionr'[s < (E 11 1 $ / <0D8 88,        < (E 11 1s A;AA AActddusJtdtddusJtdtddusJtdtddusJtdtddusJtjt5tddddy#1swYyxYw)NrFrTr)rrrrrr!r"r#test_set_configr)fs < (E 11 1T" < (E 11 1T" < (D 00 0T" < (D 00 0U# < (E 11 1 y !+T*+++s  B55B>ct|5tj|tdcdddS#1swYyxYw)zAReturn the value of assume_finite after waiting `sleep_duration`.rrN)rtimesleepr)rsleep_durations r#set_assume_finiter.vs5 m 4- >"|O,---s "9Abackend)lokymultiprocessing threadingcngd}gd}t|ddt||D}|gdk(sJy)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皙?皙?r6r7)r/n_jobsc3NK|]\}}tt||yw)N)r r.).0r sleep_durs r# z0test_config_threadsafe_joblib..s+0 $M9 #!"=)<0s#%N)rzip)r/assume_finitessleep_durationsitemss r#test_config_threadsafe_joblibrB}sF0N*O /HWQ /0(+NO(L0 E . .. .r"zcannot start threads)reasoncgd}gd}td5}|jt||Dcgc]}|}}dddgdk(sJycc}w#1swYxYw)zUses threads directly to test that the global config does not change between threads. Same test as `test_config_threadsafe_joblib` but with `ThreadPoolExecutor`.r4r5r8) max_workersN)rmapr.)r?r@eoutputrAs r#test_config_threadsaferIsp 0N*O  * a%% 1>?S     . .. .    sA A A AAc|jtjjjddt j td5td5 ddddddt j td5tddddy#1swYBxYw#1swYFxYw#1swYyxYw)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) monkeypatchs r#*test_config_array_api_dispatch_error_scipyrTs 0066 xP {*I J t 4    {*I J,d+,,  ,,s0 B-B! B- B9!B* &B--B69C)r+concurrent.futuresrrrNrrrsklearn.utils.fixesrsklearn.utils.parallelrr r$r'r)r.mark parametrizerBxfailrIrTr!r"r#r[s 1 ::(4L;^2 + -$LM/N/$8$:;/</" ,r"