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

fix and improve error-checking for shifts parameter

Browse Source
This commit is contained in:
jan 2017-09-05 11:07:32 -07:00
parent a94a91446e
commit e930baf2b4
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
gridlock/grid.py
View File

@ -307,8 +307,11 @@ class Grid(object):
self.periodic = periodic self.periodic = periodic
self.shifts = numpy.array(shifts, dtype=float) self.shifts = numpy.array(shifts, dtype=float)
if len(self.shifts.shape) != 2:
raise GridError('Misshapen shifts: shifts must have two axes! '
' The given shifts has shape {}'.format(self.shifts.shape))
if self.shifts.shape[1] != 3: if self.shifts.shape[1] != 3:
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))
num_shifts = self.shifts.shape[0] num_shifts = self.shifts.shape[0]