def setUp(self):
self.logger = Mock(spec=logging.Logger)
self.pool = WorkerPool(numWorkers=1, logger=self.logger)
def test_no_workders_in_WorkerPool(self):
with self.assertRaises(Exception):
WorkerPool(numWorkers=0)
def test_print_progress_none(self, mock1):
w = WorkerPool(numWorkers=32)
w.print_progress(0)
w.join()
self.assertTrue(mock1.called_with('0.00% of jobs completed'))
w.haltWork()
def setUp(self):
self.pool = WorkerPool()
