From a0f31a7b8ca9b54961a8890b2b36eafccd659dd6 Mon Sep 17 00:00:00 2001
From: zineb aly <aly.zineb.az@gmail.com>
Date: Mon, 3 Feb 2020 12:05:22 +0100
Subject: [PATCH] add tests for km3net-Dataformat
---
tests/test_offline.py | 27 +++++++++++++++++++++++++++
1 file changed, 27 insertions(+)
diff --git a/tests/test_offline.py b/tests/test_offline.py
index 185008b..b3e7c72 100644
--- a/tests/test_offline.py
+++ b/tests/test_offline.py
@@ -45,6 +45,33 @@ class TestOfflineKeys(unittest.TestCase):
# there are 18 fit keys
self.assertEqual(len(self.keys.fit_keys), 18)
+ def test_trigger(self):
+ # there are 4 trigger keys in v1.1.2 of km3net-Dataformat
+ trigger = self.keys.trigger
+ keys = ['JTRIGGER3DSHOWER', 'JTRIGGERMXSHOWER',
+ 'JTRIGGER3DMUON', 'JTRIGGERNB']
+
+ self.assertListEqual(keys, [*trigger.keys()])
+ self.assertListEqual([1, 2, 4, 5], [*trigger.values()])
+
+ def test_reconstruction(self):
+ # there are 34 parameters in v1.1.2 of km3net-Dataformat
+ reco = self.keys.reconstruction
+ keys = ['JPP_RECONSTRUCTION_TYPE', 'JMUONFIT', 'JMUONBEGIN',
+ 'JMUONPREFIT', 'JMUONSIMPLEX', 'JMUONGANDALF',
+ 'JMUONENERGY', 'JMUONSTART']
+ values = [4000, 0, 0, 1, 2, 3, 4, 5]
+
+ self.assertEqual(34, len([*reco.keys()]))
+ self.assertListEqual(keys, [*reco.keys()][:8])
+ self.assertListEqual(values, [*reco.values()][:8])
+
+ def test_fitparameters(self):
+ # there are 18 parameters in v1.1.2 of km3net-Dataformat
+ fit = self.keys.fitparameters
+
+ self.assertListEqual([*fit.keys()], self.keys.fit_keys)
+ self.assertListEqual([*fit.values()], [i for i in range(18)])
class TestReader(unittest.TestCase):
def setUp(self):
--
GitLab