def test_max_memory(self): worker = Pool() count_worker = worker.map(self.task, self.generate_data(30, 100000))[0] mem_for_worker = worker.map(self.task, self.generate_data(30, 100000))[1] self.assertLessEqual(count_worker * mem_for_worker, 500)
def test_count_worker_max(self): worker = Pool(1, 2, 512) self.assertEqual(2, worker.map(self.task, self.generate_data(30, 100000))[0])
def test_memory(self): worker = Pool() self.assertLessEqual(30, int(worker.map(self.task, self.generate_data(30, 100000))[1]))
def test_count_worker(self): worker = Pool() self.assertEqual(10, worker.map(self.task, self.generate_data(30, 100000))[0])