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