Add km3net dataformat

This closes issue #21 (closed)

tests/test_offline.py

@@ -46,6 +46,36 @@ 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']
+        values = [1, 2, 4, 5]
+
+        for k, v in zip(keys, values):
+            self.assertEqual(v, trigger[k])
+
+    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()]))
+        for k, v in zip(keys, values):
+            self.assertEqual(v, reco[k])
+
+    def test_fitparameters(self):
+        # there are 18 parameters in v1.1.2 of km3net-Dataformat
+        fit = self.keys.fitparameters
+        values = [i for i in range(18)]
+
+        self.assertEqual(18, len([*fit.keys()]))
+        for k, v in fit.items():
+            self.assertEqual(values[v], fit[k])
 
 class TestReader(unittest.TestCase):
     def setUp(self):
@@ -399,6 +429,4 @@ class TestOfflineTrack(unittest.TestCase):
 
     def test_str(self):
         self.assertEqual(repr(self.track).split('\n\t')[0], 'offline track:')
-        self.assertEqual(
-            repr(self.track).split('\n\t')[28],
-            'JGANDALF_LAMBDA                :      4.2409761837248484e-12')
+        self.assertTrue("JGANDALF_LAMBDA" in repr(self.track))

[Tamas Gal]

Create a dict with the known values, then iterate and check for equality...