def test_shrink_many(self):
pool = TaskPool(10)
pool.start()
self.assertEqual(pool._pool.size, 10)
pool.shrink(2)
self.assertEqual(pool._pool.size, 8)
def test_num_processes(self):
pool = TaskPool(10)
pool.start()
pool.apply_async(lambda x: x, (2, ), {})
self.assertEqual(pool.num_processes, 1)
def test_grow_many(self):
pool = TaskPool(10)
pool.start()
self.assertEqual(pool._pool.size, 10)
pool.grow(2)
self.assertEqual(pool._pool.size, 12)
