def testSetMaxAndSkipLumis(self):
"""
test class for setMaxAndSkipLumis in Mask.py
"""
testMask = Mask()
maxLumis = 10
skipLumis = 2
testMask.setMaxAndSkipLumis(maxLumis, skipLumis)
self.assertEqual(testMask['FirstLumi'], skipLumis)
self.assertEqual(testMask['LastLumi'], maxLumis + skipLumis)
return
def testSetMaxAndSkipLumis(self):
"""
test class for setMaxAndSkipLumis in Mask.py
"""
testMask = Mask()
maxLumis = 10
skipLumis = 2
testMask.setMaxAndSkipLumis(maxLumis, skipLumis)
self.assertEqual(testMask['FirstLumi'], skipLumis)
self.assertEqual(testMask['LastLumi'], maxLumis + skipLumis)
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])
