def testGetMax(self):
"""
test class for the getMax() routine added to Mask.py
"""
testMask = Mask()
maxRuns = 1000
skipRuns = 200
testMask.setMaxAndSkipRuns(maxRuns, skipRuns)
self.assertEqual(testMask.getMax('Lumi'), None)
self.assertEqual(testMask.getMax('junk'), None)
self.assertEqual(testMask.getMax('Run'), 1000)
def testGetMax(self):
"""
test class for the getMax() routine added to Mask.py
"""
testMask = Mask()
maxRuns = 1000
skipRuns = 200
testMask.setMaxAndSkipRuns(maxRuns, skipRuns)
self.assertEqual(testMask.getMax('Lumi'), None)
self.assertEqual(testMask.getMax('junk'), None)
self.assertEqual(testMask.getMax('Run'), 1000)
def testSetMaxAndSkipRuns(self):
"""
test class for setMaxAndSkipRuns in Mask.py
"""
testMask = Mask()
maxRuns = 1000
skipRuns = 200
testMask.setMaxAndSkipRuns(maxRuns, skipRuns)
self.assertEqual(testMask['FirstRun'], skipRuns)
self.assertEqual(testMask['LastRun'], maxRuns + skipRuns)
return
def testSetMaxAndSkipRuns(self):
"""
test class for setMaxAndSkipRuns in Mask.py
"""
testMask = Mask()
maxRuns = 1000
skipRuns = 200
testMask.setMaxAndSkipRuns(maxRuns, skipRuns)
self.assertEqual(testMask['FirstRun'], skipRuns)
self.assertEqual(testMask['LastRun'], maxRuns + skipRuns)
return
def testFilterRealCase(self):
"""
Test filtering of a set(run) object based on real cases from production
"""
mask = Mask()
mask.addRunWithLumiRanges(run=1, lumiList=[[9, 9], [8, 8], [3, 4], [7, 7]])
mask.setMaxAndSkipLumis(0, 7)
mask.setMaxAndSkipRuns(0, 1)
runs = set()
runs.add(Run(1, *[(9, 500), (10, 500)]))
runs.add(Run(1, *[(3, 500), (4, 500)]))
runs.add(Run(1, *[(7, 500), (8, 500)]))
newRuns = mask.filterRunLumisByMask(runs=runs)
self.assertEqual(len(newRuns), 1)
run = newRuns.pop()
self.assertEqual(run.run, 1)
self.assertEqual(run.lumis, [3, 4, 7, 8, 9])
def testFilterRealCase(self):
"""
Test filtering of a set(run) object based on real cases from production
"""
mask = Mask()
mask.addRunWithLumiRanges(run=1,
lumiList=[[9, 9], [8, 8], [3, 4], [7, 7]])
mask.setMaxAndSkipLumis(0, 7)
mask.setMaxAndSkipRuns(0, 1)
runs = set()
runs.add(Run(1, *[(9, 500), (10, 500)]))
runs.add(Run(1, *[(3, 500), (4, 500)]))
runs.add(Run(1, *[(7, 500), (8, 500)]))
newRuns = mask.filterRunLumisByMask(runs=runs)
self.assertEqual(len(newRuns), 1)
run = newRuns.pop()
self.assertEqual(run.run, 1)
self.assertEqual(run.lumis, [3, 4, 7, 8, 9])
