diff --git a/python/bifrost/header_standard.py b/python/bifrost/header_standard.py index df438e01f..11428fe8a 100644 --- a/python/bifrost/header_standard.py +++ b/python/bifrost/header_standard.py @@ -53,13 +53,13 @@ # Format: # 'parameter name':(type, minimum) STANDARD_HEADER = { - 'nchans': ((int, np.int64), 1), - 'nifs': ((int, np.int64), 1), - 'nbits': ((int, np.int64), 1), - 'fch1': ((float, np.float64), 0), - 'foff': ((float, np.float64), None), - 'tstart': ((float, np.float64), 0), - 'tsamp': ((float, np.float64), 0)} + 'nchans': ((int, np.integer), 1), + 'nifs': ((int, np.integer), 1), + 'nbits': ((int, np.integer), 1), + 'fch1': ((float, np.floating), 0), + 'foff': ((float, np.floating), None), + 'tstart': ((float, np.floating), 0), + 'tsamp': ((float, np.floating), 0)} def enforce_header_standard(header_dict): """Raise an error if the header dictionary passed diff --git a/test/test_header_standard.py b/test/test_header_standard.py index b32598b68..ff87aae1c 100644 --- a/test/test_header_standard.py +++ b/test/test_header_standard.py @@ -48,9 +48,9 @@ def test_simple_header(self): def test_numpy_types(self): """Same values, but some are numpy types""" self.header_dict = { - 'nchans': np.int64(1), 'nifs': 1, 'nbits': 8, - 'fch1': np.float64(100.0), 'foff': np.float64(1e-5), - 'tstart': 1e5, 'tsamp': np.float64(1e-5)} + 'nchans': np.int32(1), 'nifs': 1, 'nbits': 8, + 'fch1': np.float32(100.0), 'foff': np.float64(1e-5), + 'tstart': 1e5, 'tsamp': np.float32(1e-5)} def test_extra_parameters(self): """Add some extra parameters""" self.header_dict = {