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)