def test_task_arg_generator_many_gmfs(self):
args = list(task_arg_generator(
number_of_ground_motion_fields=7,
num_concurrent_tasks=3))
# generate 3 tasks with 2 realizations each, plus a task with a spare
# realization
self.assertEqual(
args, [(0, 2), (1, 2), (2, 2),
(0, 1)])
def test_task_arg_generator_no_spare(self):
args = list(task_arg_generator(
number_of_ground_motion_fields=6,
num_concurrent_tasks=3))
# generate 3 tasks with 2 realizations each
self.assertEqual(args, [(0, 2), (1, 2), (2, 2)])
def test_task_arg_generator_few_gmfs(self):
args = list(task_arg_generator(
number_of_ground_motion_fields=2,
num_concurrent_tasks=3))
# generate a single task with 2 spare realization
self.assertEqual(args, [(0, 2)])
