fix AI bug

2022-04-25 17:41:57 -07:00
parent 7f6a01e2fb
commit 335836b03e
1 changed files with 3 additions and 1 deletions
--- a/battleship/ai.py
+++ b/battleship/ai.py
@@ -60,7 +60,9 @@ class AIPlayer(Player):
    def guess(self, opponent: Player) -> Coordinate:
        for guess in self.guesses:
            if self.coord_status(guess, opponent) == CoordStatus.HIT:
-                return self.guess_near(guess, opponent)
+                candidate = self.guess_near(guess, opponent)
+                if candidate:
+                    return candidate

        while True:
            guess = random.choice(opponent.board.columns), random.choice(opponent.board.rows)