def testMultiTaskProcessingWorkload(self):
"""Multi Task Processing Workflow"""
count = 0
getFirstTask(MultiTaskProductionWorkload).setSiteWhitelist(['T2_XX_SiteA', 'T2_XX_SiteB'])
for task in MultiTaskProductionWorkload.taskIterator():
count += 1
task.setFirstEventAndLumi(1, 1)
units, _ = MonteCarlo(**self.splitArgs)(MultiTaskProductionWorkload, task)
self.assertEqual(10 * count, len(units))
for unit in units:
self.assertEqual(1, unit['Jobs'])
self.assertEqual(unit['WMSpec'], MultiTaskProductionWorkload)
self.assertEqual(unit['Task'], task)
self.assertEqual(count, 2)
def testMultiTaskProcessingWorkload(self):
"""Multi Task Processing Workflow"""
count = 0
getFirstTask(MultiTaskProductionWorkload).setSiteWhitelist(['T2_XX_SiteA', 'T2_XX_SiteB'])
for task in MultiTaskProductionWorkload.taskIterator():
count += 1
task.setFirstEventAndLumi(1, 1)
units, _, _ = MonteCarlo(**self.splitArgs)(MultiTaskProductionWorkload, task)
self.assertEqual(10 * count, len(units))
for unit in units:
self.assertEqual(1, unit['Jobs'])
self.assertEqual(unit['WMSpec'], MultiTaskProductionWorkload)
self.assertEqual(unit['Task'], task)
self.assertEqual(count, 2)