Merge pull request #424 from astrofrog/no-data-hdu-0

Nice error when no data is present in HDU
radio-astro-tools · Sep 28, 2017 · da62d16 · da62d16
2 parents c1f9a99 + c2c7b8f
commit da62d16
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -1,5 +1,9 @@
 0.4.1 (unreleased)
 ------------------
+
+ - Raise a nicer error if no data is present in the default or
+   selected HDU
+   (https://github.com/radio-astro-tools/spectral-cube/pull/424)
  - Check mask inputs to OneDSpectrum and add mask handling for
    OneDSpectrum.spectral_interpolate
    (https://github.com/radio-astro-tools/spectral-cube/pull/400)

diff --git a/spectral_cube/io/fits.py b/spectral_cube/io/fits.py
@@ -134,6 +134,10 @@ def load_fits_cube(input, hdu=0, meta=None, **kwargs):
 
     data, header, beam_table = read_data_fits(input, hdu=hdu, **kwargs)
 
+    if data is None:
+        raise Exception('No data found in HDU {0}. You can try using the hdu= '
+                        'keyword argument to read data from another HDU.'.format(hdu))
+
     if meta is None:
         meta = {}