python.EventSelectionConfigSplitOutput_unitTest Namespace Reference

Classes
class	TestSplitOutput

Functions
	run_split (region_names, **options)

Detailed Description

Unit tests for OutputAnalysisConfig.splitBySelection: one tree per region,
each in its own stream and filled only for events passing that region.

Function Documentation

run_split()

run_split	(		region_names,
		**	options )

Definition at line 16 of file EventSelectionConfigSplitOutput_unitTest.py.

def run_split(region_names, **options):
    algSeq, config = _new_accumulator(DataType.FullSim, [])
    config.setDefaultHistogramStream("HISTOGRAM")  # as in the runners, apart from the tree stream
    seq = ConfigSequence()
    for name in region_names:
        block = EventSelectionConfig()
        block.setOptionValue("selectionName", name)
        block.setOptionValue("selectionCuts", "RUN_NUMBER >= 0")
        seq.append(block)
    output = OutputAnalysisConfig()
    output.setOptionValue("vars", ["EventInfo.runNumber -> runNumber"])
    output.setOptionValue("splitBySelection", True)
    for opt, val in options.items():
        output.setOptionValue(opt, val)
    seq.append(output)
    seq.fullConfigure(config)
    if isAthena:
        algs = config.CA.getEventAlgos()
        config.CA.wasMerged()
        return algs, config.outputStreams()
    return list(algSeq), config.outputStreams()