Skip to content
Snippets Groups Projects

Fix: shift_slice to accept single slices as argument, and return tuples

Merged Frederik Hennig requested to merge da15siwa/pystencils:fix_slicing into master
Compare and
2 files
+ 28
2
Preferences
Compare changes
Files
2
+ 5
2
@@ -89,9 +89,12 @@ def shift_slice(slices, offset):
raise ValueError()
if hasattr(offset, '__len__'):
return [shift_slice_component(k, off) for k, off in zip(slices, offset)]
return tuple(shift_slice_component(k, off) for k, off in zip(slices, offset))
else:
return [shift_slice_component(k, offset) for k in slices]
if isinstance(slices, slice) or isinstance(slices, int) or isinstance(slices, float):
return shift_slice_component(slices, offset)
else:
return tuple(shift_slice_component(k, offset) for k in slices)
def slice_from_direction(direction_name, dim, normal_offset=0, tangential_offset=0):