def testSerial(self): os.environ['WM_WORK_MANAGER'] = 'serial' work_manager = make_work_manager() assert isinstance(work_manager, SerialWorkManager) with work_manager: future = work_manager.submit(will_succeed) future.get_result()
def testSerial(self): os.environ['WM_WORK_MANAGER'] = 'serial' work_manager = make_work_manager() assert isinstance(work_manager, SerialWorkManager) with work_manager: future = work_manager.submit(will_succeed) future.get_result()
def testZeroMQ(self): os.environ['WM_WORK_MANAGER'] = 'zmq' os.environ['WM_N_WORKERS'] = str(3) work_manager = make_work_manager() assert isinstance(work_manager, ZMQWorkManager) with work_manager: future = work_manager.submit(will_succeed) future.get_result() assert work_manager.n_workers == 3
def testProcesses(self): os.environ['WM_WORK_MANAGER'] = 'processes' os.environ['WM_N_WORKERS'] = str(3) work_manager = make_work_manager() assert isinstance(work_manager, ProcessWorkManager) assert work_manager.n_workers == 3 with work_manager: future = work_manager.submit(will_succeed) future.get_result()
def testArgs(self): parser = argparse.ArgumentParser() add_wm_args(parser) args = '--work-manager=threads --n-workers=3'.split() args = parser.parse_args(args) process_wm_args(args) work_manager = make_work_manager() assert isinstance(work_manager, ThreadsWorkManager) assert work_manager.n_workers == 3
def testZeroMQ(self): os.environ['WM_WORK_MANAGER'] = 'zmq' os.environ['WM_N_WORKERS'] = str(3) work_manager = make_work_manager() assert isinstance(work_manager, ZMQWorkManager) assert work_manager.internal_client.n_workers == 3 with work_manager: future = work_manager.submit(will_succeed) future.get_result()
def testProcesses(self): os.environ['WM_WORK_MANAGER'] = 'processes' os.environ['WM_N_WORKERS'] = str(3) work_manager = make_work_manager() assert isinstance(work_manager, ProcessWorkManager) assert work_manager.n_workers == 3 with work_manager: future = work_manager.submit(will_succeed) future.get_result()
def testArgs(self): parser = argparse.ArgumentParser() add_wm_args(parser) args='--work-manager=threads --n-workers=3'.split() args = parser.parse_args(args) process_wm_args(args) work_manager = make_work_manager() assert isinstance(work_manager, ThreadsWorkManager) assert work_manager.n_workers == 3
def testArgFallthrough(self): # this test specifies the work manager on the command line, and the worker count in the environment # this simply tests whether we look to the environment in the case of a missing command line argument os.environ['WM_N_WORKERS'] = str(3) parser = argparse.ArgumentParser() add_wm_args(parser) args = '--work-manager=threads'.split() args = parser.parse_args(args) process_wm_args(args) work_manager = make_work_manager() assert isinstance(work_manager, ThreadsWorkManager) assert work_manager.n_workers == 3
def testArgFallthrough(self): # this test specifies the work manager on the command line, and the worker count in the environment # this simply tests whether we look to the environment in the case of a missing command line argument os.environ['WM_N_WORKERS'] = str(3) parser = argparse.ArgumentParser() add_wm_args(parser) args='--work-manager=threads'.split() args = parser.parse_args(args) process_wm_args(args) work_manager = make_work_manager() assert isinstance(work_manager, ThreadsWorkManager) assert work_manager.n_workers == 3