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