def test_closed_neighbours(self):
sea1 = single_player.player.game.Sea()
grid1 = [[0 for i in range(10)] for i in range(10)]
grid2 = [[0 for i in range(10)] for i in range(10)]
single_player.computer_open_cell([3, 3], grid1, sea1, grid2)
count34 = single_player.closed_neighbours([3, 4], grid1, 2)
count88 = single_player.closed_neighbours([9, 9], grid1, 2)
self.assertEqual(count34, 5)
self.assertEqual(count88, 4)
def test_computer_open_cell(self):
sea1 = single_player.player.game.Sea()
grid1 = [[0 for i in range(10)] for i in range(10)]
grid2 = [[0 for i in range(10)] for i in range(10)]
single_player.player.game.Ship(3, sea1, [4, 4], 0)
single_player.computer_open_cell([4, 4], grid1, sea1, grid2)
single_player.computer_open_cell([3, 3], grid1, sea1, grid2)
self.assertEqual(grid1[4][4], 3)
self.assertEqual(grid1[3][3], 4)
def test_computer_make_move(self):
player1 = single_player.player.Player(10, 2)
sea1 = player1.sea
grid1 = [[0 for i in range(10)] for i in range(10)]
grid2 = [[0 for i in range(10)] for i in range(10)]
put_one_ship(player1, [8, 8], 3, 0, grid1)
single_player.computer_open_cell([0, 0], grid1, sea1, grid2)
self.assertTrue(computer_make_move(player1, sea1, grid1, grid2))
self.assertTrue(grid1[0][1] == 3 or grid1[1][0] == 4)
computer_make_move(player1, sea1, grid1, grid2)
self.assertTrue(grid1[0][1] == 3)
computer_make_move(player1, sea1, grid1, grid2)
self.assertTrue(grid1[0][2] == 3)
