def check_backend_context_manager(backend_name): with parallel_backend(backend_name, n_jobs=3): active_backend, active_n_jobs = parallel.get_active_backend() assert active_n_jobs == 3 assert effective_n_jobs(3) == 3 p = Parallel() assert p.n_jobs == 3 if backend_name == 'multiprocessing': assert type(active_backend) == MultiprocessingBackend assert type(p._backend) == MultiprocessingBackend elif backend_name == 'threading': assert type(active_backend) == ThreadingBackend assert type(p._backend) == ThreadingBackend elif backend_name.startswith('test_'): assert type(active_backend) == FakeParallelBackend assert type(p._backend) == FakeParallelBackend
def check_backend_context_manager(backend_name): with parallel_backend(backend_name, n_jobs=3): active_backend, active_n_jobs = parallel.get_active_backend() assert_equal(active_n_jobs, 3) assert_equal(effective_n_jobs(3), 3) p = Parallel() assert_equal(p.n_jobs, 3) if backend_name == "multiprocessing": assert_equal(type(active_backend), MultiprocessingBackend) assert_equal(type(p._backend), MultiprocessingBackend) elif backend_name == "threading": assert_equal(type(active_backend), ThreadingBackend) assert_equal(type(p._backend), ThreadingBackend) elif backend_name.startswith("test_"): assert_equal(type(active_backend), FakeParallelBackend) assert_equal(type(p._backend), FakeParallelBackend)
def check_backend_context_manager(backend_name): with parallel_backend(backend_name, n_jobs=3): active_backend, active_n_jobs = parallel.get_active_backend() assert_equal(active_n_jobs, 3) assert_equal(effective_n_jobs(3), 3) p = Parallel() assert_equal(p.n_jobs, 3) if backend_name == 'multiprocessing': assert_equal(type(active_backend), MultiprocessingBackend) assert_equal(type(p._backend), MultiprocessingBackend) elif backend_name == 'threading': assert_equal(type(active_backend), ThreadingBackend) assert_equal(type(p._backend), ThreadingBackend) elif backend_name.startswith('test_'): assert_equal(type(active_backend), FakeParallelBackend) assert_equal(type(p._backend), FakeParallelBackend)
def test_effective_n_jobs(): assert effective_n_jobs() > 0