self.reset_pos()
self.been = []
- def move(self, level, caught):
+ def move(self, level):
if not self.on_board():
# Advance towards the map a step at a time
self.pos = (self.pos[0] + 1, self.pos[1])
# AI goes here
steps = 0
self.been.append(self.pos)
- for direction in [(1, 0), (-1, 0), (0, -1), (0, 1)] * 2:
+ for direction in [(1, 0), (-1, 0), (0, -1), (0, 1)]:
new_pos = (self.pos[0] + direction[0], self.pos[1] + direction[1])
if self.can_move(direction, level) and new_pos not in self.been:
self.pos = new_pos
self.been.append(new_pos)
steps += 1
- if steps >= 2:
- break
- if caught():
break
if steps == 0:
self.been = []