FIX: Coerce data types on writing GIFTI DataArrays

Includes a hack to keep the old data_tag API functional

Author: effigies

nibabel/gifti/gifti.py

@@ -270,16 +270,21 @@ def _to_xml_element(self):
     return DataTag(dataarray, encoding, datatype, ordering).to_xml()
 
 
-def _data_tag_element(dataarray, encoding, datatype, ordering):
+def _data_tag_element(dataarray, encoding, dtype, ordering):
     """ Creates data tag with given `encoding`, returns as XML element
     """
     import zlib
-    ord = array_index_order_codes.npcode[ordering]
+    order = array_index_order_codes.npcode[ordering]
     enclabel = gifti_encoding_codes.label[encoding]
     if enclabel == 'ASCII':
-        da = _arr2txt(dataarray, datatype)
+        # XXX Accommodating data_tag API
+        # On removal (nibabel 4.0) drop str case
+        da = _arr2txt(dataarray, dtype if isinstance(dtype, str) else KIND2FMT[dtype.kind])
     elif enclabel in ('B64BIN', 'B64GZ'):
-        out = dataarray.tostring(ord)
+        # XXX Accommodating data_tag API - don't try to fix dtype
+        if isinstance(dtype, str):
+            dtype = dataarray.dtype
+        out = np.asanyarray(dataarray, dtype).tostring(order)
         if enclabel == 'B64GZ':
             out = zlib.compress(out)
         da = base64.b64encode(out).decode()
@@ -462,11 +467,10 @@ def _to_xml_element(self):
         if self.coordsys is not None:
             data_array.append(self.coordsys._to_xml_element())
         # write data array depending on the encoding
-        dt_kind = data_type_codes.dtype[self.datatype].kind
         data_array.append(
             _data_tag_element(self.data,
                               gifti_encoding_codes.specs[self.encoding],
-                              KIND2FMT[dt_kind],
+                              data_type_codes.dtype[self.datatype],
                               self.ind_ord))
 
         return data_array

nibabel/gifti/tests/test_gifti.py

@@ -403,7 +403,7 @@ def test_data_array_round_trip():
     assert_array_equal(vertices, verts)
 
 
-def test_type_coercion():
+def test_darray_dtype_coercion_failures():
     dtypes = (np.uint8, np.int32, np.int64, np.float32, np.float64)
     encodings = ('ASCII', 'B64BIN', 'B64GZ')
     for data_dtype, darray_dtype, encoding in itertools.product(dtypes,