def get_neighbor_wrap(self, cell, direction): neighbor = list(Utils.add_tuples(cell.coordinates, direction.to_coordinates())) if neighbor[0] < 0: neighbor[0] += self.maze.width elif neighbor[0] >= self.maze.width: neighbor[0] -= self.maze.width if neighbor[1] < 0: neighbor[1] += self.maze.height elif neighbor[1] >= self.maze.height: neighbor[1] -= self.maze.height return self.maze.get(tuple(neighbor))
def get_neighbor(self, direction): return self.grid.get(Utils.add_tuples(self.coordinates, direction.to_coordinates()))