print("AI making safe move.")
time.sleep(0.2)
# Reset game state
elif resetButton.collidepoint(mouse):
game = Minesweeper(height=HEIGHT, width=WIDTH, mines=MINES)
ai = MinesweeperAI(height=HEIGHT, width=WIDTH)
revealed = set()
flags = set()
lost = False
continue
# User-made move
elif not lost:
for i in range(HEIGHT):
for j in range(WIDTH):
if (cells[i][j].collidepoint(mouse) and (i, j) not in flags
and (i, j) not in revealed):
move = (i, j)
# Make move and update AI knowledge
if move:
if game.is_mine(move):
lost = True
else:
nearby = game.nearby_mines(move)
revealed.add(move)
ai.add_knowledge(move, nearby)
pygame.display.flip()
class MinsweeperTestCases(unittest.TestCase):
def setUp(self):
self.ai = MinesweeperAI(8, 8)
def test_moves_made(self):
tuple_example = (1, 4)
self.ai.add_knowledge(tuple_example, 0)
print(self.ai.moves_made)
self.assertTrue(tuple_example in self.ai.moves_made)
def test_mark_safe(self):
tuple_example = (1, 6)
self.ai.add_knowledge(tuple_example, 0)
self.ai.print_knowledge()
self.assertTrue(tuple_example in self.ai.safes)
def test_mark_mine(self):
sentence_1 = Sentence({(6, 4), (6, 3)}, 1)
sentence_2 = Sentence({(6, 4)}, 1)
sentence_3 = Sentence({(6, 3)}, 0)
self.ai.knowledge.append(sentence_1)
self.ai.resolve(sentence_2)
self.ai.print_knowledge()
self.assertTrue(sentence_3 in self.ai.knowledge)
self.assertTrue((6, 4) in self.ai.mines)
def test_mark_mine_2(self):
sentence_1 = Sentence({(4, 3), (4, 2), (4, 1)}, 1)
self.ai.safes = {
(4, 2), (4, 3), (1, 3), (3, 0), (2, 1), (1, 6), (3, 7), (0, 3),
(2, 5), (2, 4), (1, 2), (3, 3), (4, 4), (1, 5), (1, 0), (3, 5),
(0, 1), (2, 7), (3, 1), (2, 0), (0, 6), (1, 7), (3, 4), (0, 2)
}
self.ai.knowledge.append(sentence_1)
self.ai.update_whats_known()
self.assertTrue((4, 1) in self.ai.mines)
def test_makes_safe_move(self):
self.ai.mines = {
(7, 1), (0, 1), (0, 2), (0, 3), (0, 4), (0, 5), (0, 6), (0, 7),
(1, 0), (1, 1), (1, 2), (1, 3), (1, 4), (1, 5), (1, 6), (1, 7),
(2, 0), (2, 1), (2, 2), (2, 3), (2, 4), (2, 5), (2, 6), (2, 7),
(3, 0), (3, 1), (3, 2), (3, 3), (3, 4), (3, 5), (3, 6), (3, 7),
(4, 0), (4, 1), (4, 2), (4, 3), (4, 4), (4, 5), (4, 6), (4, 7),
(5, 0), (5, 1), (5, 2), (5, 3), (5, 4), (5, 5), (5, 6), (5, 7),
(6, 0), (6, 1), (6, 2), (6, 3), (6, 4), (6, 5), (6, 6), (6, 7),
(7, 0)
}
self.ai.safes = {(0, 0), (7, 2), (7, 3), (7, 4), (7, 5), (7, 6),
(7, 7)}
self.assertIn(self.ai.make_safe_move(),
{(0, 0), (7, 2), (7, 3), (7, 4), (7, 5), (7, 6), (7, 7)})
def test_random_move(self):
self.ai.mines = {
(7, 1), (0, 1), (0, 2), (0, 3), (0, 4), (0, 5), (0, 6), (0, 7),
(1, 0), (1, 1), (1, 2), (1, 3), (1, 4), (1, 5), (1, 6), (1, 7),
(2, 0), (2, 1), (2, 2), (2, 3), (2, 4), (2, 5), (2, 6), (2, 7),
(3, 0), (3, 1), (3, 2), (3, 3), (3, 4), (3, 5), (3, 6), (3, 7),
(4, 0), (4, 1), (4, 2), (4, 3), (4, 4), (4, 5), (4, 6), (4, 7),
(5, 0), (5, 1), (5, 2), (5, 3), (5, 4), (5, 5), (5, 6), (5, 7)
}
self.ai.moves_made = {(0, 0), (7, 2), (7, 3), (7, 4), (7, 5), (7, 6),
(7, 7)}
self.assertIn(self.ai.make_random_move(), [(6, 0), (6, 1), (6, 2),
(6, 3), (6, 4), (6, 5),
(6, 6), (6, 7), (7, 0)])
# given {a,b,c} = X in knowledge en {c} = X then add {a,b} = 0 to knowledge
def test_add_knowledge_1(self):
# given sentence_1 in knowledge
sentence_1 = Sentence({(6, 4), (6, 3), (7, 4), (6, 2), (7, 2)}, 1)
self.ai.knowledge.append(sentence_1)
# if I resolve sentence_2
sentence_2 = Sentence({(6, 4)}, 1)
self.ai.resolve(sentence_2)
# then sentence_3 should be in knowledge
sentence_3 = Sentence({(6, 3), (7, 4), (6, 2), (7, 2)}, 0)
self.assertIn(sentence_3, self.ai.knowledge)
# given {a,b,c,d} = X and {a,b} = Y add set {c,d} = X - Y
def test_add_knowledge_2(self):
# the resolved sentence (_2) is a subset of sentence in knowledge (_1)
# given sentence_1 in knowledge
sentence_1 = Sentence({(6, 4), (6, 3), (7, 4), (6, 2), (7, 2)}, 2)
self.ai.knowledge.append(sentence_1)
# resolve sentence_2
sentence_2 = Sentence({(6, 4), (6, 3)}, 1)
self.ai.resolve(sentence_2)
# then sentence_3 should be in knowledge
sentence_3 = Sentence({(7, 4), (6, 2), (7, 2)}, 1)
print(self.ai.knowledge)
self.assertIn(sentence_3, self.ai.knowledge)
# given a set {a,b} = X and {a,b,c,d} = Y add set {c,d} = Y - X
def test_add_knowledge_3(self):
# the resolved sentence (_2) is a superset of sentence in knowledge (_1)
# given sentence_1 in knowledge
sentence_1 = Sentence({(6, 4), (6, 3)}, 1)
self.ai.knowledge.append(sentence_1)
# if I resolve sentence_2
sentence_2 = Sentence({(6, 4), (6, 3), (7, 4), (6, 2), (7, 2)}, 2)
self.ai.knowledge.append(sentence_2)
self.ai.resolve(sentence_2)
# then sentence_3 should be in knowledge
sentence_3 = Sentence({(7, 4), (6, 2), (7, 2)}, 1)
self.assertIn(sentence_3, self.ai.knowledge)
# applies same logica as test_add_knowledge_4 but with safe
def test_add_knowledge_4(self):
# given sentence_1 in knowledge
sentence_1 = Sentence({(6, 4)}, 0)
self.ai.knowledge.append(sentence_1)
# if I resolve sentence_2
sentence_2 = Sentence({(6, 4), (6, 3), (7, 4), (6, 2), (7, 2)}, 2)
self.ai.resolve(sentence_2)
# then sentence_3 should be in knowledge
sentence_3 = Sentence({(6, 3), (7, 4), (6, 2), (7, 2)}, 2)
print(self.ai.knowledge)
self.assertIn(sentence_3, self.ai.knowledge)
# given a sentence {a, b, c}= 0 add {a}=0, {b}=0 and {c}=0 to knowledge
def test_add_knowledge_5(self):
# given sentence_1 in knowledge
sentence_1 = Sentence({(6, 4), (6, 3), (7, 4)}, 0)
# sentence_0
sentence_2 = Sentence({(6, 4), (5, 4), (5, 3)}, 2)
self.ai.knowledge.append(sentence_2)
# self.ai.knowledge.append(sentence)
self.ai.resolve(sentence_1)
# then ai.safes is filled with these tuples
self.assertIn((6, 4), self.ai.safes)
self.assertIn((6, 3), self.ai.safes)
self.assertIn((7, 4), self.ai.safes)
sentence_3 = Sentence({(5, 4), (5, 3)}, 2)
# then while resolving sentence_1, mark_safe will remove (6,4) from sentence_2
self.assertIn(sentence_3, self.ai.knowledge)
def test_new_functions_in_sentence(self):
# given sentence_1 in knowledge
sentence_1 = Sentence({(6, 4)}, 0)
sentence_2 = Sentence({(6, 4), (6, 3), (7, 4), (6, 2), (7, 2)}, 2)
sentence_3 = Sentence(set(), 0)
# self.assertEqual(len(sentence_1), 1)
# self.assertEqual(len(sentence_2), 5)
self.assertTrue(sentence_1.is_literal)
self.assertTrue(sentence_3.is_empty_set)
def test_known_mines(self):
sentence_1 = Sentence({(6, 4), (6, 3)}, 2)
sentence_2 = Sentence({(5, 4)}, 1)
self.ai.resolve(sentence_1)
self.ai.resolve(sentence_2)
self.assertEqual(self.ai.mines, {(5, 4), (6, 4), (6, 3)})
def test_known_safes(self):
sentence_1 = Sentence({(6, 4), (6, 3)}, 1)
sentence_2 = Sentence({(6, 4)}, 1)
self.ai.resolve(sentence_1)
print(self.ai.knowledge)
print(self.ai.safes)
# self.assertEqual(len(self.ai.safes), 0)
self.assertEqual(self.ai.safes, set())
self.ai.resolve(sentence_2)
print(self.ai.safes)
def test_add_knowledge_again(self):
# given sentence_1 in knowledge
sentence_1 = Sentence({(6, 4), (5, 0)}, 1)
sentence_2 = Sentence({(6, 4)}, 0)
# self.ai.knowledge.append(sentence_1)
self.ai.knowledge.append(sentence_2)
# if I resolve sentence_2
# sentence_2 = Sentence({(6, 4), (6, 3), (7, 4), (6, 2), (7, 2)}, 2)
self.ai.resolve(sentence_1)
# then sentence_3 should be in knowledge
sentence_3 = Sentence({(5, 0)}, 1)
# print(self.ai.knowledge)
self.assertIn(sentence_3, self.ai.knowledge)
self.assertIn((5, 0), self.ai.mines)
# Solving the corners
if initial_solve:
ai = MinesweeperAI(height=HEIGHT, width=WIDTH, kb=child)
move = (0, 0)
if move:
while mine_first:
if game.is_mine(move):
game = Minesweeper(height=HEIGHT, width=WIDTH, mines=MINES)
print("Mine in first box, creating new game.")
else:
mine_first = False
break
else:
nearby = game.nearby_mines(move)
revealed.add(move)
ai.add_knowledge(move, 'safe')
if nearby == 0:
moveStack = [
move, (move[0] + 1, move[1]),
(move[0] + 1, move[1] + 1), (move[0], move[1] + 1)
]
while len(moveStack) != 0:
move = moveStack[-1]
nearby = game.nearby_mines(move)
revealed.add(move)
ai.add_knowledge(move, 'safe')
if nearby == 0:
# If within range of board, append
if move[0] + 1 < HEIGHT:
# Down
moveStack.append((move[0] + 1, move[1]))
b) not in flags:
if game.is_mine((a, b)):
lost = True
else:
nearby = game.nearby_mines((a, b))
if nearby == 0:
cleared_squares = clear((a, b))
for square in cleared_squares:
revealed.add(square)
if game.nearby_mines(
square) == 0:
ai.moves_made.add(square)
ai.mark_safe(square)
else:
ai.add_knowledge(
square,
game.nearby_mines(
square))
else:
revealed.add((a, b))
ai.add_knowledge((a, b), nearby)
time.sleep(0.1)
elif left == 1:
mouse = pygame.mouse.get_pos()
# If AI button clicked, make an AI move
if aiButton.collidepoint(mouse) and not lost and not won:
move = ai.make_safe_move()
if move is None:
move = ai.make_random_move()
if move is None:
