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

Use f-strings where possible

Browse Source
This commit is contained in:
Jan Petykiewicz 2021-10-24 19:05:48 -07:00
parent ff5ffb2f40
commit 7340d41f10
3 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
gridlock/draw.py
View File

@ -48,7 +48,7 @@ def draw_polygons(self,
surface = numpy.delete(range(3), surface_normal)
for i, polygon in enumerate(polygons):
malformed = 'Malformed polygon: (%i)' % i
malformed = f'Malformed polygon: ({i})'
if polygon.shape[1] not in (2, 3):
raise GridError(malformed + 'must be a Nx2 or Nx3 ndarray')
if polygon.shape[1] == 3:
@ -106,9 +106,9 @@ def draw_polygons(self,
# evaluate on the meshgrid
eps_i = eps[i](x0, y0, z0)
if not numpy.isfinite(eps_i).all():
raise GridError('Non-finite values in eps[%u]' % i)
raise GridError('Unsupported eps[{}]: {}'.format(i, type(eps[i])))
raise GridError(f'Non-finite values in eps[{i}]')
elif numpy.size(eps[i]) != 1:
raise GridError(f'Unsupported eps[{i}]: {type(eps[i])}')
else:
# eps[i] is scalar non-callable
eps_i = eps[i]
@ -232,7 +232,7 @@ def draw_slab(self,
if len(center) == 3:
center = center[surface_normal]
else:
raise GridError('Bad center: {}'.format(center))
raise GridError(f'Bad center: {center}')
# Find center of slab
center_shift = self.center
@ -331,7 +331,7 @@ def draw_extrude_rectangle(self,
if s == 0:
raise GridError('0 is not a valid polarity')
if direction not in range(3):
raise GridError('Invalid direction: {}'.format(direction))
raise GridError(f'Invalid direction: {direction}')
if rectangle[0, direction] != rectangle[1, direction]:
raise GridError('Rectangle entries along extrusion direction do not match.')

6
gridlock/grid.py
View File

@ -279,7 +279,7 @@ class Grid:
for i in range(3):
if len(self.exyz[i]) != len(pixel_edge_coordinates[i]):
warnings.warn('Dimension {} had duplicate edge coordinates'.format(i), stacklevel=2)
warnings.warn(f'Dimension {i} had duplicate edge coordinates', stacklevel=2)
if isinstance(periodic, bool):
self.periodic = [periodic] * 3
@ -288,10 +288,10 @@ class Grid:
if len(self.shifts.shape) != 2:
raise GridError('Misshapen shifts: shifts must have two axes! '
' The given shifts has shape {}'.format(self.shifts.shape))
f' The given shifts has shape {self.shifts.shape}')
if self.shifts.shape[1] != 3:
raise GridError('Misshapen shifts; second axis size should be 3,'
' shape is {}'.format(self.shifts.shape))
f' shape is {self.shifts.shape}')
if (numpy.abs(self.shifts) > 1).any():
raise GridError('Only shifts in the range [-1, 1] are currently supported')

8
gridlock/position.py
View File

@ -46,7 +46,7 @@ def ind2pos(self,
low_bound = -0.5
high_bound = -0.5
if (ind < low_bound).any() or (ind > self.shape - high_bound).any():
raise GridError('Position outside of grid: {}'.format(ind))
raise GridError(f'Position outside of grid: {ind}')
if round_ind:
rind = numpy.clip(numpy.round(ind).astype(int), 0, self.shape - 1)
@ -84,17 +84,17 @@ def pos2ind(self,
"""
r = numpy.squeeze(r)
if r.size != 3:
raise GridError('r must be 3-element vector: {}'.format(r))
raise GridError(f'r must be 3-element vector: {r}')
if (which_shifts is not None) and (which_shifts >= self.shifts.shape[0]):
raise GridError('Invalid which_shifts: {}'.format(which_shifts))
raise GridError(f'Invalid which_shifts: {which_shifts}')
sexyz = self.shifted_exyz(which_shifts)
if check_bounds:
for a in range(3):
if self.shape[a] > 1 and (r[a] < sexyz[a][0] or r[a] > sexyz[a][-1]):
raise GridError('Position[{}] outside of grid!'.format(a))
raise GridError(f'Position[{a}] outside of grid!')
grid_pos = numpy.zeros((3,))
for a in range(3):