Merge pull request #694 from shoyer/strict-reindex

Check that dimensions are all valid when using reindex

Author: shoyer

xray/core/alignment.py

@@ -84,7 +84,12 @@ def align(*objects, **kwargs):
                         % list(kwargs))
 
     joined_indexes = _join_indexes(join, objects)
-    return tuple(obj.reindex(copy=copy, **joined_indexes) for obj in objects)
+    result = []
+    for obj in objects:
+        valid_indexers = dict((k, v) for k, v in joined_indexes.items()
+                              if k in obj.dims)
+        result.append(obj.reindex(copy=copy, **valid_indexers))
+    return tuple(result)
 
 
 def partial_align(*objects, **kwargs):

xray/core/dataarray.py

@@ -659,8 +659,10 @@ def reindex_like(self, other, method=None, tolerance=None, copy=True):
         DataArray.reindex
         align
         """
+        indexers = dict((k, v) for k, v in other.indexes.items()
+                        if k in self.dims)
         return self.reindex(method=method, tolerance=tolerance, copy=copy,
-                            **other.indexes)
+                            **indexers)
 
     def reindex(self, method=None, tolerance=None, copy=True, **indexers):
         """Conform this object onto a new set of indexes, filling in

xray/core/dataset.py

@@ -1154,8 +1154,10 @@ def reindex_like(self, other, method=None, tolerance=None, copy=True):
         Dataset.reindex
         align
         """
+        indexers = dict((k, v) for k, v in other.indexes.items()
+                        if k in self.dims)
         return self.reindex(method=method, copy=copy, tolerance=tolerance,
-                            **other.indexes)
+                            **indexers)
 
     def reindex(self, indexers=None, method=None, tolerance=None, copy=True, **kw_indexers):
         """Conform this object onto a new set of indexes, filling in
@@ -1205,6 +1207,10 @@ def reindex(self, indexers=None, method=None, tolerance=None, copy=True, **kw_in
             # shortcut
             return self.copy(deep=True) if copy else self
 
+        bad_dims = [d for d in indexers if d not in self.dims]
+        if bad_dims:
+            raise ValueError('invalid reindex dimensions: %s' % bad_dims)
+
         variables = alignment.reindex_variables(
             self.variables, self.indexes, indexers, method, tolerance, copy=copy)
         return self._replace_vars_and_dims(variables)

xray/core/merge.py

@@ -151,8 +151,11 @@ def _reindex_variables_against(variables, indexes, copy=False):
     """Reindex all DataArrays in the provided dict, leaving other values alone.
     """
     alignable = [k for k, v in variables.items() if hasattr(v, 'indexes')]
-    aligned = [variables[a].reindex(copy=copy, **indexes)
-               for a in alignable]
+    aligned = []
+    for a in alignable:
+        valid_indexes = dict((k, v) for k, v in indexes.items()
+                             if k in variables[a].dims)
+        aligned.append(variables[a].reindex(copy=copy, **valid_indexes))
     new_variables = OrderedDict(variables)
     new_variables.update(zip(alignable, aligned))
     return new_variables

xray/test/test_dataset.py

@@ -881,6 +881,10 @@ def test_reindex(self):
         with self.assertRaisesRegexp(ValueError, 'dictionary'):
             data.reindex('foo')
 
+        # invalid dimension
+        with self.assertRaisesRegexp(ValueError, 'invalid reindex dim'):
+            data.reindex(invalid=0)
+
         # out of order
         expected = data.sel(dim1=data['dim1'][:10:-1])
         actual = data.reindex(dim1=data['dim1'][:10:-1])