rtjddlZddlmZddlZddlZddlmZmZmZddlm Z ddl m Z m Z dZ dZdZd Zejd gd d Zeje d dZdZdS)N)ThreadPoolExecutor)config_context get_config set_config)_IS_WASM)Paralleldelayedcxtdddddddddddd ksJtd td dusJtd 5tdddddddddddd ksJ dddn #1swxYwYtd dusJtd 5td 5td dusJ dddn #1swxYwYtd dusJtd 5td dusJtd 5td dusJtd td dusJ dddn #1swxYwYtd dusJ dddn #1swxYwYtd dusJ dddn #1swxYwYtdddddddddddd ksJtjt 5tddddn #1swxYwYtjt 5td ddddS#1swxYwYdS) 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__W/var/www/html/CA-Chatbot/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:::;;;;;;;;;;;;;;;;;;s!BB B :G; C1% G;1C5 5G;8C5 9+G;$)G  AF G F G "F #G ? G; G G;G G;;G?G??II"I?#J//J36J3ctddusJ td5tddusJt#1swxYwYn#t$rYnwxYwtddusJdS)NrFTr)rr ValueErrorr!r"r#test_config_context_exceptionr'[s << (E 1 1 1 1 $ / / /  <<0D8888,,                << (E 1 1 1 1 1 1s.A'AAAAA A*)A*ctddusJtdtddusJtdtddusJtdtddusJtdtddusJtjt5tdddddS#1swxYwYdS)NrFrTr)rrrrrr!r"r#test_set_configr)fsO << (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`.rrN)rtimesleepr)rsleep_durations r#set_assume_finiter.vs 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皙?皙?r6r7)r/n_jobsc3XK|]%\}}tt||V&dS)N)r r.).0r sleep_durs r# z0test_config_threadsafe_joblib..sO00 $M9 #!""=)<<000000r"N)rzip)r/assume_finitessleep_durationsitemss r#test_config_threadsafe_joblibrB}s~0//N***O /HWQ / / /00(+NO(L(L000  E ... . . . . . .r"zcannot start threads)reasoncgd}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`.r4r5r8) max_workerscg|]}|Sr!r!)r;outputs r# z*test_config_threadsafe..s(       r"N)rmapr.)r?r@erAs r#test_config_threadsaferKs 0//N***O  * * * a  %% 1>?SS                  ... . . . . . .s'A  AAc|tjjjddt jtd5td5 dddn #1swxYwYdddn #1swxYwYt jtd5tdddddS#1swxYwYdS)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_scipyrVs 06 xPPP {*I J J J t 4 4 4                   {*I J J J,,d++++,,,,,,,,,,,,,,,,,,sHA=A& A=&A* *A=-A* .A==BB#CCC)r+concurrent.futuresrrrPrrrsklearn.utils.fixesrsklearn.utils.parallelrr r$r'r)r.mark parametrizerBxfailrKrVr!r"r#r]s; 111111 ::::::::::((((((44444444L;L;L;^222 + + + ---$L$L$LMM//NM/$8$:;;//<;/" , , , , ,r"