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)