Integration/test/testSwitchProducerTask_cfg.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 import argparse
0004 import sys
0005
0006 parser = argparse.ArgumentParser(prog=sys.argv[0], description='Test SwitchProducer in Task.')
0007 parser.add_argument("--disableTest2", help="Disable test2 SwitchProducer case", action="store_true")
0008 parser.add_argument("--input", help="Read input file from a previous step of this same configuration", action="store_true")
0009 parser.add_argument("--conditionalTask", help="Use ConditionalTask instead of Task", action="store_true")
0010
0011 argv = sys.argv[:]
0012 if '--' in argv:
0013     argv.remove("--")
0014 args, unknown = parser.parse_known_args(argv)
0015
0016 enableTest2 = not args.disableTest2
0017 class SwitchProducerTest(cms.SwitchProducer):
0018     def __init__(self, **kargs):
0019         super(SwitchProducerTest,self).__init__(
0020             dict(
0021                 test1 = lambda accelerators: (True, -10),
0022                 test2 = lambda accelerators: (enableTest2, -9)
0023             ), **kargs)
0024
0025 process = cms.Process("PROD2" if args.input else "PROD1")
0026
0027 if args.input:
0028      # Test that having SwitchProducers with same labels as products from
0029     # earlier processes works
0030     process.source = cms.Source("PoolSource",
0031         fileNames = cms.untracked.vstring("file:testSwitchProducer{}Task{}.root".format(
0032             "Conditional" if args.conditionalTask else "",
0033             1 if enableTest2 else 2,
0034         ))
0035     )
0036     process.maxEvents.input = -1
0037 else:
0038     process.source = cms.Source("EmptySource")
0039     process.maxEvents.input = 3
0040     if enableTest2:
0041         process.source.firstLuminosityBlock = cms.untracked.uint32(2)
0042
0043 process.out = cms.OutputModule("PoolOutputModule",
0044     fileName = cms.untracked.string('testSwitchProducer{}Task{}{}.root'.format(
0045         "Conditional" if args.conditionalTask else "",
0046         "Input" if args.input else "",
0047         1 if enableTest2 else 2,
0048     )),
0049     outputCommands = cms.untracked.vstring(
0050         'keep *_intProducer_*_*',
0051         'keep *_intProducerOther_*_*',
0052         'keep *_intProducerAlias_*_*',
0053         'keep *_intProducerAlias2_other_*',
0054         'keep *_intProducerDep1_*_*',
0055         'keep *_intProducerDep2_*_*',
0056         'keep *_intProducerDep3_*_*',
0057     )
0058 )
0059
0060 process.intProducer1 = cms.EDProducer("ManyIntProducer", ivalue = cms.int32(1))
0061 process.intProducer2 = cms.EDProducer("ManyIntProducer", ivalue = cms.int32(2))
0062 process.intProducer3 = cms.EDProducer("ManyIntProducer", ivalue = cms.int32(2), values = cms.VPSet(cms.PSet(instance=cms.string("foo"),value=cms.int32(2))))
0063 process.intProducer4 = cms.EDProducer("ManyIntProducer", ivalue = cms.int32(42), throw = cms.untracked.bool(True))
0064 process.intProducer5 = cms.EDProducer("ManyIntProducer", ivalue = cms.int32(3), values = cms.VPSet(cms.PSet(instance=cms.string("foo"),value=cms.int32(3))))
0065 if enableTest2:
0066     process.intProducer1.throw = cms.untracked.bool(True)
0067 else:
0068     process.intProducer2.throw = cms.untracked.bool(True)
0069     process.intProducer3.throw = cms.untracked.bool(True)
0070     process.intProducer5.throw = cms.untracked.bool(True)
0071
0072 process.intProducer = SwitchProducerTest(
0073     test1 = cms.EDProducer("AddIntsProducer", labels = cms.VInputTag("intProducer1")),
0074     test2 = cms.EDProducer("AddIntsProducer", labels = cms.VInputTag("intProducer2"))
0075 )
0076 # Test also existence of another SwitchProducer here
0077 process.intProducerOther = SwitchProducerTest(
0078     test1 = cms.EDProducer("AddIntsProducer", labels = cms.VInputTag("intProducer1")),
0079     test2 = cms.EDProducer("AddIntsProducer", labels = cms.VInputTag("intProducer2"))
0080 )
0081 # SwitchProducer with an alias
0082 process.intProducerAlias = SwitchProducerTest(
0083     test1 = cms.EDProducer("AddIntsProducer", labels = cms.VInputTag("intProducer1")),
0084     test2 = cms.EDAlias(intProducer3 = cms.VPSet(cms.PSet(type = cms.string("edmtestIntProduct"), fromProductInstance = cms.string(""), toProductInstance = cms.string("")),
0085                                                  cms.PSet(type = cms.string("edmtestIntProduct"), fromProductInstance = cms.string("foo"), toProductInstance = cms.string("other"))))
0086 )
0087
0088 process.intProducerAlias2 = SwitchProducerTest(
0089     test1 = cms.EDProducer("AddIntsProducer", labels = cms.VInputTag("intProducer1")),
0090     test2 = cms.EDAlias(intProducer4 = cms.VPSet(cms.PSet(type = cms.string("edmtestIntProduct"), fromProductInstance = cms.string(""), toProductInstance = cms.string(""))),
0091                         intProducer5 = cms.VPSet(cms.PSet(type = cms.string("edmtestIntProduct"), fromProductInstance = cms.string("foo"), toProductInstance = cms.string("other"))))
0092 )
0093
0094 # Test multiple consumers of a SwitchProducer
0095 process.intProducerDep1 = cms.EDProducer("AddIntsProducer", labels = cms.VInputTag("intProducer"))
0096 process.intProducerDep2 = cms.EDProducer("AddIntsProducer", labels = cms.VInputTag("intProducer"))
0097 process.intProducerDep3 = cms.EDProducer("AddIntsProducer", labels = cms.VInputTag("intProducer"))
0098
0099 if args.conditionalTask:
0100     process.ct = cms.ConditionalTask(process.intProducer, process.intProducerOther, process.intProducerAlias, process.intProducerAlias2,
0101                                      process.intProducer1, process.intProducer2, process.intProducer3, process.intProducer4, process.intProducer5)
0102     process.p = cms.Path(process.intProducerDep1+process.intProducerDep2+process.intProducerDep3, process.ct)
0103 else:
0104     process.t = cms.Task(process.intProducer, process.intProducerOther, process.intProducerAlias, process.intProducerAlias2,
0105                          process.intProducerDep1, process.intProducerDep2, process.intProducerDep3,
0106                          process.intProducer1, process.intProducer2, process.intProducer3, process.intProducer4, process.intProducer5)
0107     process.p = cms.Path(process.t)
0108
0109 process.e = cms.EndPath(process.out)