U 3dT@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`.FTrrr ) 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