diff --git a/pystencils/stencil.py b/pystencils/stencil.py index 1d02996df2ba50a42a1833592aa2177c934b1175..aa86427fcd59113e502c91e7444a6815a5307f96 100644 --- a/pystencils/stencil.py +++ b/pystencils/stencil.py @@ -319,18 +319,14 @@ def adjacent_directions(direction): return tuple(result) binary_numbers_list = binary_numbers(len(direction)) for adjacent_direction in binary_numbers_list: - for i in range(len(direction)): - if direction[i] == 0: + for i, entry in enumerate(direction): + if entry == 0: adjacent_direction[i] = 0 - if direction[i] == -1 and adjacent_direction[i] == 1: + if entry == -1 and adjacent_direction[i] == 1: adjacent_direction[i] = -1 if not all(e == 0 for e in adjacent_direction): result.add(tuple(adjacent_direction)) - result = list(result) - result.sort() - return tuple(result) - - + return tuple(sorted(result)) # -------------------------------------- Visualization -----------------------------------------------------------------