U mdT@sddlZddlmZddlZddlmZmZmZddlm Z m Z ddZ ddZ d d Z d d Zejd dddgddZddZdS)N)ThreadPoolExecutor) get_config set_configconfig_context)delayedParallelc Cstdddddddddks ttddtd dks}sz0test_config_threadsafe_joblib..N)rzipr)rassume_finitessleep_durationsitemsrrrtest_config_threadsafe_joblibrs   r)c Cs^ddddg}ddddg}tdd}dd|t||D}W5QRX|ddddgksZtd 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)rmaprr)r&r'er(rrrtest_config_threadsafes    r/)rconcurrent.futuresrrZsklearnrrrZsklearn.utils.parallelrrrrrrmarkZ parametrizer)r/rrrrs F