""" from __future__ import division import math import unittest from WMCore_t.WMSpec_t.samples.MultiMergeProductionWorkload import workload as MultiMergeProductionWorkload from WMCore_t.WMSpec_t.samples.MultiTaskProductionWorkload import workload as MultiTaskProductionWorkload from WMCore_t.WorkQueue_t.WorkQueue_t import getFirstTask from WMCore.WorkQueue.Policy.Start.MonteCarlo import MonteCarlo from WMCore.WorkQueue.WorkQueueExceptions import WorkQueueNoWorkError, WorkQueueWMSpecError from WMQuality.Emulators.EmulatedUnitTestCase import EmulatedUnitTestCase from WMQuality.Emulators.WMSpecGenerator.Samples.TestMonteCarloWorkload import getMCArgs, monteCarloWorkload mcArgs = getMCArgs() class MonteCarloTestCase(EmulatedUnitTestCase): """Test case MonteCarlo Workload""" splitArgs = dict(SliceType='NumEvents', SliceSize=100, MaxJobsPerElement=1) def testBasicProductionWorkload(self): """Basic Production Workload""" # change split defaults for this test totalevents = 1000000 splitArgs = dict(SliceType='NumberOfEvents', SliceSize=100, MaxJobsPerElement=5) mcArgs["EventsPerJob"] = 100
import unittest from WMCore.WorkQueue.Policy.Start.MonteCarlo import MonteCarlo from WMQuality.Emulators.WMSpecGenerator.Samples.TestMonteCarloWorkload \ import monteCarloWorkload, getMCArgs from WMCore_t.WMSpec_t.samples.MultiMergeProductionWorkload \ import workload as MultiMergeProductionWorkload from WMCore_t.WMSpec_t.samples.MultiTaskProductionWorkload \ import workload as MultiTaskProductionWorkload from WMCore.WorkQueue.WorkQueueExceptions import * from WMCore_t.WorkQueue_t.WorkQueue_t import getFirstTask from WMQuality.Emulators.DataBlockGenerator import Globals import math mcArgs = getMCArgs() class MonteCarloTestCase(unittest.TestCase): splitArgs = dict(SliceType = 'NumEvents', SliceSize = 100, MaxJobsPerElement = 1) def testBasicProductionWorkload(self): """Basic Production Workload""" # change split defaults for this test totalevents = 1000000 splitArgs = dict(SliceType = 'NumberOfEvents', SliceSize = 100, MaxJobsPerElement = 5) BasicProductionWorkload = monteCarloWorkload('MonteCarloWorkload', mcArgs) getFirstTask(BasicProductionWorkload).setSiteWhitelist(['T2_XX_SiteA', 'T2_XX_SiteB']) getFirstTask(BasicProductionWorkload).addProduction(totalevents = totalevents) getFirstTask(BasicProductionWorkload).setSiteWhitelist(['T2_XX_SiteA', 'T2_XX_SiteB'])