diff --git a/CAL/CalibrationCommon/test/t_holography_datatable_class.py b/CAL/CalibrationCommon/test/t_holography_datatable_class.py index d4bc32f7a29f886392f640dd75b4b77b4c0ae285..28dd6b888e4586d288d6148243c10824392e8a1c 100755 --- a/CAL/CalibrationCommon/test/t_holography_datatable_class.py +++ b/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()