Python task_arg_generator Examples

Example #1

File: core_test.py Project: 4x/oq-engine

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

Example #2

File: core_test.py Project: 4x/oq-engine

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

Example #3

File: core_test.py Project: 4x/oq-engine

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