From ff786b94ad0c40df3ffd30173be34e857fe3e905 Mon Sep 17 00:00:00 2001
From: Stefan Reck <stefan.reck@fau.de>
Date: Mon, 19 Apr 2021 17:26:15 +0200
Subject: [PATCH] rename keys for extractors & add missing ones
---
orcasong/from_toml.py | 8 ++++++--
tests/test_from_toml.py | 2 +-
2 files changed, 7 insertions(+), 3 deletions(-)
diff --git a/orcasong/from_toml.py b/orcasong/from_toml.py
index 563aab5..8f795a9 100644
--- a/orcasong/from_toml.py
+++ b/orcasong/from_toml.py
@@ -5,8 +5,12 @@ import orcasong.extractors as extractors
# built-in extractors. First argument has to be the input filename,
# other parameters can be set via 'extractor_config' dict in the toml
EXTRACTORS = {
- "neutrino_mc": extractors.get_neutrino_mc_info_extr,
- "neutrino_data": extractors.get_real_data_info_extr,
+ "nu_chain_neutrino": extractors.get_neutrino_mc_info_extr,
+ "nu_chain_muon": extractors.get_muon_mc_info_extr,
+ "nu_chain_noise": extractors.get_random_noise_mc_info_extr,
+ "nu_chain_data": extractors.get_real_data_info_extr,
+ # TODO "bundle_mc": ???,
+ # TODO "bundle_data": ???,
}
MODES = {
diff --git a/tests/test_from_toml.py b/tests/test_from_toml.py
index f3985c0..bd3d3d6 100644
--- a/tests/test_from_toml.py
+++ b/tests/test_from_toml.py
@@ -12,7 +12,7 @@ def _test_extr(infile):
return infile + "_extr"
-orcasong.from_toml.EXTRACTORS["neutrino_mc"] = _test_extr
+orcasong.from_toml.EXTRACTORS["nu_chain_neutrino"] = _test_extr
class TestSetupProcessorExampleConfig(TestCase):
--
GitLab