Merge pull request #141 from keflavich/header_NAXIS

Include correct NAXIS in header
radio-astro-tools · Sep 8, 2014 · d2ed5cd · d2ed5cd
2 parents 29f942a + f8b89c6
commit d2ed5cd
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/spectral_cube/spectral_cube.py b/spectral_cube/spectral_cube.py
@@ -1554,6 +1554,10 @@ def header(self):
         header = self._header.copy()
         header.update(self.wcs.to_header())
         header['BUNIT'] = self.unit.to_string(format='fits')
+        header['NAXIS'] = self._data.ndim
+        header['NAXIS1'] = self.shape[2]
+        header['NAXIS2'] = self.shape[1]
+        header['NAXIS3'] = self.shape[0]
         # TODO: incorporate other relevant metadata here
         return header
 

diff --git a/spectral_cube/tests/test_spectral_cube.py b/spectral_cube/tests/test_spectral_cube.py
@@ -516,6 +516,15 @@ def test_endians():
     assert xbig.dtype.byteorder == '>'
     assert xlil.dtype.byteorder == '='
 
+def test_header_naxis():
+
+    cube, data = cube_and_raw('advs.fits')
+
+    assert cube.header['NAXIS'] == 3 # NOT data.ndim == 4
+    assert cube.header['NAXIS1'] == data.shape[3]
+    assert cube.header['NAXIS2'] == data.shape[2]
+    assert cube.header['NAXIS3'] == data.shape[1]
+
 def test_slicing():
 
     cube, data = cube_and_raw('advs.fits')