def test_waitfortasks_multiple(self):
"""Waitfortasks waits for multiple tasks."""
tasks = []
for _ in range(5):
tasks.append(concurrent_function(0.01))
time.sleep(0.1)
self.assertEqual(waitfortasks(tasks), tasks)
def test_waitfortasks_restore(self):
"""Waitfortasks Task object is restored to original one."""
task = concurrent_function(0.01)
expected = sorted(dir(task))
waitfortasks([task])
self.assertEqual(sorted(dir(task)), expected)
def test_waitfortasks_timeout(self):
"""Waitfortasks returns empty list if timeout."""
task = concurrent_function(0.1)
self.assertEqual(waitfortasks([task], timeout=0.01), [])
def test_waitfortasks_single(self):
"""Waitfortasks waits for a single task."""
task = concurrent_function(0.01)
self.assertEqual(waitfortasks([task])[0], task)
