jan/gridlock
2
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Improve error checking

Browse source
This commit is contained in:
Jan Petykiewicz 2017-09-25 00:43:26 -07:00
commit a76fef0449
1 changed files with 7 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

7
gridlock/grid.py
View file

@ -317,6 +317,13 @@ class Grid(object):
raise GridError('Misshapen shifts; second axis size should be 3,' raise GridError('Misshapen shifts; second axis size should be 3,'
' shape is {}'.format(self.shifts.shape)) ' shape is {}'.format(self.shifts.shape))
if (numpy.abs(self.shifts) > 1).any():
raise GridError('Only shifts in the range [-1, 1] are currently supported')
if (self.shifts < 0).any():
# TODO: Test negative shifts
warnings.warn('Negative shifts are still experimental and mostly untested, be careful!')
num_shifts = self.shifts.shape[0] num_shifts = self.shifts.shape[0]
if num_grids is None: if num_grids is None:
num_grids = num_shifts num_grids = num_shifts