SSB-44: add additional test

CAL/CalibrationCommon/test/t_holography_datatable_class.py

@@ -73,7 +73,7 @@ class TestHolographyDatasetClass(unittest.TestCase):
         with tempfile.TemporaryDirectory() as temporary_dir:
             full_path = temporary_dir + '/' + temp_file_name
             logger.info('creating temp file %s', full_path)
-            h5file = h5py.File(full_path)
+            h5file = h5py.File(full_path, 'w')
             data_table = HolographyDataTable(uri=test_uri, h5_file=h5file)
 
             for key, value in test_data.items():
@@ -85,12 +85,19 @@ class TestHolographyDatasetClass(unittest.TestCase):
             data_table = HolographyDataTable(uri=test_uri, h5_file=h5file)
 
             stored_keys = data_table.keys()
+            stations = set(key[0] for key in stored_keys)
+            frequencies = set(key[1] for key in stored_keys)
+            beams = set(key[2] for key in stored_keys)
+
+            self.assertEqual(set(data_table.frequencies), frequencies)
+            self.assertEqual(set(data_table.reference_stations), stations)
+            self.assertEqual(set(data_table.beam_numbers), beams)
 
             self.assertEqual(stored_keys.difference(test_data.keys()), set())
 
             for key, value in data_table.iter_items():
                 self.assertIn(key, test_data)
-                frequency, station_name, beam = key
+                station_name, frequency, beam = key
                 numpy.testing.assert_array_equal(data_table[station_name, frequency, beam], test_data[key])
 
             for frequency in data_table.frequencies:
@@ -141,5 +148,5 @@ class TestHolographyDatasetClass(unittest.TestCase):
 
 
 if __name__ == '__main__':
-    logging.basicConfig(format='%(name)s : %(message)s')
+    logging.basicConfig(format='%(name)s : %(message)s', level=logging.DEBUG)
     unittest.main()