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)])