assertion checks for all kernels

This commit is contained in:
Max Zwiessele 2014-03-28 12:02:34 +00:00
parent 40ade9e157
commit 1dabf67c93
3 changed files with 2 additions and 14 deletions

View file

@ -56,6 +56,7 @@ class _Slice_wrap(object):
def _slice_K(f):
@wraps(f)
def wrap(self, X, X2 = None, *a, **kw):
assert X.shape[1] > max(np.r_[self.active_dims]), "At least {} dimensional X needed".format(max(np.r_[self.active_dims]))
with _Slice_wrap(self, X, X2) as s:
ret = f(self, s.X, s.X2, *a, **kw)
return ret
@ -64,6 +65,7 @@ def _slice_K(f):
def _slice_Kdiag(f):
@wraps(f)
def wrap(self, X, *a, **kw):
assert X.shape[1] > max(np.r_[self.active_dims]), "At least {} dimensional X needed".format(max(np.r_[self.active_dims]))
with _Slice_wrap(self, X, None) as s:
ret = f(self, s.X, *a, **kw)
return ret