diff --git a/tests/test_offline.py b/tests/test_offline.py index 185008bcbf983b788c10a283cad2d2c60c1123de..b3e7c725212664e6d642b5f9586aa56faab6090b 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):