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
parent e65f25cc68
commit a76fef0449
1 changed files with 7 additions and 0 deletions
Show Stats 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