python.EventSelectionConfigTestSupport Namespace Reference

Functions
	_container (spec)
	_new_accumulator (dataType, containers)
	run (cuts, *, dataType=DataType.FullSim, containers=None, selectionName="SR", **options)
	run_merger (region_names, *, dataType=DataType.FullSim, noFilter=False)
	selectors (algs)
	named (algs, substr)
	prop (alg, name, default=_UNSET)

Variables
	_UNSET = object()
list	_DUMMY_WPS

Detailed Description

Shared harness for the EventSelectionConfig unit tests.

Each *_unitTest.py script builds a real ConfigSequence containing a single
EventSelectionConfig block, configures it through a real ConfigAccumulator,
and inspects the resulting algorithm sequence. No mocking: the accumulator
instantiates the real CP algorithms, so a Python/C++ property-name mismatch
is caught here rather than at runtime.

In a full job the object-config blocks register the input containers and their
selection working points upstream. In isolation we register them here:
  * input containers      -> config.setSourceName(container, container)
  * object selection WPs  -> config.addSelection(container, wp, decoration)

Function Documentation

_container()

python.EventSelectionConfigTestSupport._container ( spec )

protected

Strip a trailing '.selection' to get the bare container name.

Definition at line 32 of file EventSelectionConfigTestSupport.py.

def _container(spec):
    """Strip a trailing '.selection' to get the bare container name."""
    return spec.split(".")[0]
 
 

_new_accumulator()

python.EventSelectionConfigTestSupport._new_accumulator ( dataType, containers )

protected

Definition at line 37 of file EventSelectionConfigTestSupport.py.

def _new_accumulator(dataType, containers):
    algSeq = AlgSequence()
    config = ConfigAccumulator(dataType=dataType, algSeq=algSeq)
    for cont in {_container(c) for c in containers}:
        config.setSourceName(cont, cont)
        for wp in _DUMMY_WPS:
            # decoration string deliberately embeds the WP name so the
            # extra-selection tests can assert on it via substring.
            config.addSelection(cont, wp, f"{wp}_%SYS%")
    return algSeq, config
 
 

named()

python.EventSelectionConfigTestSupport.named	(		algs,
			substr )

Definition at line 111 of file EventSelectionConfigTestSupport.py.

def named(algs, substr):
    return [a for a in algs if substr in a.getName()]
 
 

prop()

python.EventSelectionConfigTestSupport.prop	(		alg,
			name,
			default = _UNSET )

Read a property set on the algorithm; return `default` if unset.

Definition at line 115 of file EventSelectionConfigTestSupport.py.

def prop(alg, name, default=_UNSET):
    """Read a property set on the algorithm; return `default` if unset."""
    val = getattr(alg, name, _UNSET)
    if val is _UNSET:
        if default is _UNSET:
            raise AttributeError(f"{alg.getName()} has no property {name!r}")
        return default
    return val

run()

python.EventSelectionConfigTestSupport.run	(		cuts,
		*	,
			dataType = DataType.FullSim,
			containers = None,
			selectionName = "SR",
		**	options )

Configure one EventSelectionConfig block and return its algorithms.

`containers` maps EventSelectionConfig option names to container specs,
e.g. {'electrons': 'AnaElectrons'} or {'jets': 'AnaJets.baseline'}.

Definition at line 49 of file EventSelectionConfigTestSupport.py.

        selectionName="SR", **options):
    """Configure one EventSelectionConfig block and return its algorithms.
 
    `containers` maps EventSelectionConfig option names to container specs,
    e.g. {'electrons': 'AnaElectrons'} or {'jets': 'AnaJets.baseline'}.
    """
    containers = containers or {}
    algSeq, config = _new_accumulator(dataType, containers.values())
    seq = ConfigSequence()
    block = EventSelectionConfig()
    block.setOptionValue("selectionName", selectionName)
    block.setOptionValue("selectionCuts", cuts)
    for opt, cname in containers.items():
        block.setOptionValue(opt, cname)
    for opt, val in options.items():
        block.setOptionValue(opt, val)
    seq.append(block)
 
    if isAthena:
        try:
            seq.fullConfigure(config)
            return config.CA.getEventAlgos()
        finally:
            config.CA.wasMerged()
 
    seq.fullConfigure(config)
    return list(algSeq)
 
 

run_merger()

python.EventSelectionConfigTestSupport.run_merger	(		region_names,
		*	,
			dataType = DataType.FullSim,
			noFilter = False )

Build real regions then the merger, and return the algorithm sequence.

Each region is a trivial EventSelectionConfig (a single RUN_NUMBER cut) so
that the merger's `EventSelection` dependency is satisfied and
`eventSelectionNames` is populated. The event filter is emitted implicitly
at the end of each block (no explicit SAVE needed).

Definition at line 79 of file EventSelectionConfigTestSupport.py.

def run_merger(region_names, *, dataType=DataType.FullSim, noFilter=False):
    """Build real regions then the merger, and return the algorithm sequence.
 
    Each region is a trivial EventSelectionConfig (a single RUN_NUMBER cut) so
    that the merger's `EventSelection` dependency is satisfied and
    `eventSelectionNames` is populated. The event filter is emitted implicitly
    at the end of each block (no explicit SAVE needed).
    """
    algSeq, config = _new_accumulator(dataType, [])
    seq = ConfigSequence()
    for name in region_names:
        block = EventSelectionConfig()
        block.setOptionValue("selectionName", name)
        block.setOptionValue("selectionCuts", "RUN_NUMBER >= 0")
        seq.append(block)
    merger = EventSelectionMergerConfig()
    merger._instance_number = 1            # defensive: ensure the guard runs
    merger.setOptionValue("noFilter", noFilter)
    seq.append(merger)
    seq.fullConfigure(config)
    if isAthena:
        algs = config.CA.getEventAlgos()
        config.CA.wasMerged()
        return algs
    return list(algSeq)
 
 

selectors()

python.EventSelectionConfigTestSupport.selectors

(

algs

)

Selector algorithms only (exclude the SAVE filter).

Definition at line 106 of file EventSelectionConfigTestSupport.py.

def selectors(algs):
    """Selector algorithms only (exclude the SAVE filter)."""
    return [a for a in algs if a.getType() != "CP::SaveFilterAlg"]
 
 

Variable Documentation

_DUMMY_WPS

list python.EventSelectionConfigTestSupport._DUMMY_WPS

protected

Initial value:

=  ["baseline", "mysel", "goodjet", "centraljet", "topjet",
              "elsel", "musel", "esel", "msel", "tsel"]

Definition at line 28 of file EventSelectionConfigTestSupport.py.

_UNSET

python.EventSelectionConfigTestSupport._UNSET = object()

protected

Definition at line 24 of file EventSelectionConfigTestSupport.py.