30 def __init__(self, ID: str, run: WorkflowRun, type: WorkflowType, steps: list[str], setup: TestSetup, extra_args: str = "") -> None:
31 if "maxEvents" not in extra_args:
32 if type == WorkflowType.MCPileUpReco or run == WorkflowRun.Run4:
33 extra_args += " --maxEvents 5"
34 else:
35 extra_args += " --maxEvents 20"
36
37 if type == WorkflowType.MCPileUpReco:
38 if "inputHITSFile" not in extra_args:
39 extra_args += f" --inputHITSFile {input_HITS[run]}"
40 if "inputRDO_BKGFile" not in extra_args:
41 extra_args += " --inputRDO_BKGFile ../run_d*/myRDO.pool.root"
42
43 threads = 1
44 threads_argument = '--multithreaded'
45 if setup.custom_threads is not None:
46 threads = setup.custom_threads
47 if threads <= 0:
48 threads_argument = ''
49
50 self.command = \
51 (f"ATHENA_CORE_NUMBER={threads} Reco_tf.py {threads_argument} --AMIConfig {ID}"
52 f" --imf False {extra_args}")
53
54 self.output_checks = []
55
56
57
58 self.output_checks.append(FrozenTier0PolicyCheck(setup, "AOD", 60))
59 self.output_checks.append(FrozenTier0PolicyCheck(setup, "ESD", 20))
60 self.output_checks.append(MetadataCheck(setup, "AOD"))
61
62 self.digest_checks = []
63 if not setup.disable_output_checks:
64 self.digest_checks.append(AODContentCheck(setup))
65 self.digest_checks.append(AODDigestCheck(setup))
66
67 super().__init__(ID, run, type, steps, setup)
68
69