Massive test runner speedup

tests/test_daq.py

@@ -5,12 +5,11 @@ import unittest
 from km3io.daq import DAQReader, get_rate, has_udp_trailer, get_udp_max_sequence_number, get_channel_flags, get_number_udp_packets
 
 SAMPLES_DIR = os.path.join(os.path.dirname(__file__), "samples")
-
+DAQ_FILE = DAQReader(os.path.join(SAMPLES_DIR, "daq_v1.0.0.root"))
 
 class TestDAQEvents(unittest.TestCase):
     def setUp(self):
-        self.events = DAQReader(os.path.join(SAMPLES_DIR,
-                                             "daq_v1.0.0.root")).events
+        self.events = DAQ_FILE.events
 
     def test_index_lookup(self):
         assert 3 == len(self.events)
@@ -24,8 +23,7 @@ class TestDAQEvents(unittest.TestCase):
 
 class TestDAQEvent(unittest.TestCase):
     def setUp(self):
-        self.event = DAQReader(os.path.join(SAMPLES_DIR,
-                                            "daq_v1.0.0.root")).events[0]
+        self.event = DAQ_FILE.events[0]
 
     def test_str(self):
         assert re.match(".*event.*96.*snapshot.*18.*triggered",
@@ -38,8 +36,7 @@ class TestDAQEvent(unittest.TestCase):
 
 class TestDAQEventsSnapshotHits(unittest.TestCase):
     def setUp(self):
-        self.events = DAQReader(os.path.join(SAMPLES_DIR,
-                                             "daq_v1.0.0.root")).events
+        self.events = DAQ_FILE.events
         self.lengths = {0: 96, 1: 124, -1: 78}
         self.total_item_count = 298
 
@@ -77,8 +74,7 @@ class TestDAQEventsSnapshotHits(unittest.TestCase):
 
 class TestDAQEventsTriggeredHits(unittest.TestCase):
     def setUp(self):
-        self.events = DAQReader(os.path.join(SAMPLES_DIR,
-                                             "daq_v1.0.0.root")).events
+        self.events = DAQ_FILE.events
         self.lengths = {0: 18, 1: 53, -1: 9}
         self.total_item_count = 80
 
@@ -118,8 +114,7 @@ class TestDAQEventsTriggeredHits(unittest.TestCase):
 
 class TestDAQTimeslices(unittest.TestCase):
     def setUp(self):
-        self.ts = DAQReader(os.path.join(SAMPLES_DIR,
-                                         "daq_v1.0.0.root")).timeslices
+        self.ts = DAQ_FILE.timeslices
 
     def test_data_lengths(self):
         assert 3 == len(self.ts._timeslices["L1"][0])
@@ -144,8 +139,7 @@ class TestDAQTimeslices(unittest.TestCase):
 
 class TestDAQTimeslice(unittest.TestCase):
     def setUp(self):
-        self.ts = DAQReader(os.path.join(SAMPLES_DIR,
-                                         "daq_v1.0.0.root")).timeslices
+        self.ts = DAQ_FILE.timeslices
         self.n_frames = {"L1": [69, 69, 69], "SN": [64, 66, 68]}
 
     def test_str(self):
@@ -158,8 +152,7 @@ class TestDAQTimeslice(unittest.TestCase):
 
 class TestSummaryslices(unittest.TestCase):
     def setUp(self):
-        self.ss = DAQReader(os.path.join(SAMPLES_DIR,
-                                         "daq_v1.0.0.root")).summaryslices
+        self.ss = DAQ_FILE.summaryslices
 
     def test_headers(self):
         assert 3 == len(self.ss.headers)

tests/test_offline.py

@@ -5,15 +5,15 @@ from pathlib import Path
 from km3io import OfflineReader
 
 SAMPLES_DIR = Path(__file__).parent / 'samples'
-OFFLINE_FILE = SAMPLES_DIR / 'aanet_v2.0.0.root'
-OFFLINE_USR = SAMPLES_DIR / 'usr-sample.root'
-OFFLINE_NUMUCC = SAMPLES_DIR / "numucc.root"  # with mc data
+OFFLINE_FILE = OfflineReader(SAMPLES_DIR / 'aanet_v2.0.0.root')
+OFFLINE_USR = OfflineReader(SAMPLES_DIR / 'usr-sample.root')
+OFFLINE_NUMUCC = OfflineReader(SAMPLES_DIR / "numucc.root")  # with mc data
 
 
 class TestOfflineReader(unittest.TestCase):
     def setUp(self):
-        self.r = OfflineReader(OFFLINE_FILE)
-        self.nu = OfflineReader(OFFLINE_NUMUCC)
+        self.r = OFFLINE_FILE
+        self.nu = OFFLINE_NUMUCC
         self.Nevents = 10
 
     def test_number_events(self):
@@ -126,7 +126,7 @@ class TestOfflineReader(unittest.TestCase):
 
     def test_reading_header(self):
         # head is the supported format
-        head = OfflineReader(OFFLINE_NUMUCC).header
+        head = OFFLINE_NUMUCC.header
 
         self.assertEqual(float(head['DAQ']), 394)
         self.assertEqual(float(head['kcut']), 2)
@@ -138,7 +138,7 @@ class TestOfflineReader(unittest.TestCase):
 
 class TestOfflineEvents(unittest.TestCase):
     def setUp(self):
-        self.events = OfflineReader(OFFLINE_FILE).events
+        self.events = OFFLINE_FILE.events
         self.n_events = 10
         self.det_id = [44] * self.n_events
         self.n_hits = [176, 125, 318, 157, 83, 60, 71, 84, 255, 105]
@@ -191,7 +191,7 @@ class TestOfflineEvents(unittest.TestCase):
 
 class TestOfflineHits(unittest.TestCase):
     def setUp(self):
-        self.hits = OfflineReader(OFFLINE_FILE).hits
+        self.hits = OFFLINE_FILE.hits
         self.n_hits = 10
         self.dom_id = {
             0: [
@@ -239,8 +239,8 @@ class TestOfflineHits(unittest.TestCase):
 class TestOfflineTracks(unittest.TestCase):
     @unittest.skip
     def setUp(self):
-        self.tracks = OfflineReader(OFFLINE_FILE).tracks
-        self.r_mc = OfflineReader(OFFLINE_NUMUCC)
+        self.tracks = OFFLINE_FILE.tracks
+        self.r_mc = OFFLINE_NUMUCC
         self.Nevents = 10
 
     @unittest.skip
@@ -307,14 +307,14 @@ class TestOfflineTracks(unittest.TestCase):
 
 class TestUsr(unittest.TestCase):
     def setUp(self):
-        self.f = OfflineReader(OFFLINE_USR)
+        self.f = OFFLINE_USR
 
     def test_str(self):
         print(self.f.usr)
 
     def test_nonexistent_usr(self):
         f = OfflineReader(SAMPLES_DIR / "daq_v1.0.0.root")
-        self.assertListEqual([], f.usr.keys())
+        assert not hasattr(self.f, "usr")
 
     def test_keys(self):
         self.assertListEqual([