def test_next_turn_init(self):
game = GameBattleship()
result = game.next_turn()
expected = 'pone el barco (x, y, boat, horizontal/vertical)'
self.assertEqual(expected, result)
class test_battleship(unittest.TestCase):
def setUp(self):
self.game = GameBattleship()
input_user = [
'1, 1, 1, vertical',
'1, 2, 2, vertical',
'1, 3, 3, vertical',
'1, 4, 3, vertical',
'1, 5, 4, vertical',
'1, 6, 5, vertical',
]
board = Board()
board_table = [
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 1, 2, 31, 32, 4, 5, 0, 0, 0],
[0, 0, 2, 31, 32, 4, 5, 0, 0, 0],
[0, 0, 0, 31, 32, 4, 5, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 4, 5, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 5, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
]
board.board = board_table
for single_input in input_user:
self.game.play(single_input)
self.game.player_cpu.board_own = board
def test_shoot_water(self):
board = Board()
result = board.shoot(3, 3)
self.assertEqual(result, "water")
def test_shoot_sunked_1(self):
board = Board()
board.set_boat(1, 1, 1, "horizontal")
result = board.shoot(1, 1)
self.assertEqual(result, "sunked")
def test_shoot_sunked_2(self):
board = Board()
result = []
board.set_boat(0, 0, 2, "horizontal")
for i in range(0, 2):
result.append(board.shoot(0, i))
self.assertEqual(result[len(result) - 1], "sunked")
def test_shoot_sunked_31(self):
board = Board()
result = []
board.set_boat(1, 2, 3, "vertical")
for i in range(1, 4):
result.append(board.shoot(i, 2))
self.assertEqual(result[len(result) - 1], "sunked")
def test_shoot_sunked_32(self):
board = Board()
result = []
board.set_boat(5, 7, 3, "horizontal")
for i in range(7, 10):
result.append(board.shoot(5, i))
self.assertEqual(result[len(result) - 1], "sunked")
def test_shoot_sunked_4(self):
board = Board()
result = []
result2 = []
result3 = []
board.set_boat(9, 5, 4, "horizontal")
for i in range(5, 9):
result.append(board.shoot(9, i))
board.set_boat(0, 0, 3, "horizontal")
for i in range(0, 1):
result2.append(board.shoot(0, i))
board.set_boat(4, 5, 3, "horizontal")
for i in range(5, 8):
result3.append(board.shoot(4, i))
self.assertEqual(result[len(result) - 1], "sunked")
self.assertEqual(result2[len(result2) - 1], "hit")
self.assertEqual(result3[len(result3) - 1], "sunked")
def test_shoot_sunked_5(self):
board = Board()
result = []
board.set_boat(5, 5, 5, "vertical")
for i in range(5, 10):
result.append(board.shoot(i, 5))
self.assertEqual(result[len(result) - 1], "sunked")
def test_player_boards(self):
player = PlayerHuman()
result = player.get_boards()
self.assertEqual(len(result), 2)
def test_game_players(self):
game = GameBattleship()
result = game.get_players()
self.assertEqual(len(result), 2)
def test_own_board_is_ready_cpu(self):
p1 = PlayerCPU()
p1.fill_own_board()
result = p1.board_own.state
self.assertEqual(result, 'ready_to_war')
# def test_own_board_is_ready_human(self):
# p1 = PlayerHuman()
# p1.fill_own_board()
# result = p1.board_own.state
# self.assertEqual(result, board_states[1])
def test_game_state_init(self):
game = GameBattleship()
result = game.state
self.assertEqual(result, 'init')
def test_game_state_war(self):
game = GameBattleship()
input_user = [
'1, 1, 1, vertical',
'1, 2, 2, vertical',
'1, 3, 3, vertical',
'1, 4, 3, vertical',
'1, 5, 4, vertical',
'1, 6, 5, vertical',
]
# Ir corriendo cada uno de los inputs del user
for single_input in input_user:
result = game.state
self.assertEqual(result, 'init')
game.set_boat(single_input)
result = game.state
self.assertEqual(result, 'war')
def test_turn_init_player_wrong_param_amount_of(self):
game = GameBattleship()
result = game.set_boat('1, 1, 1, vertical, extrabadparam')
self.assertEqual(result, 'error, mas parametros de los requeridos (4)')
def test_turn_war_player(self):
game = GameBattleship()
result = game.set_boat('1, 2, 1, vertical')
self.assertTrue(result)
def test_turn_war_player_wrong_param_letter(self):
game = GameBattleship()
result = game.play('A, B, 1, vertical')
self.assertEqual(result, 'error')
def test_turn_war_player_wrong_number(self):
game = GameBattleship()
result = game.play('10, 1, 1, vertical')
self.assertFalse(result)
def test_turn_war_player_wrong_param_amount_of(self):
game = GameBattleship()
game.state = 'war'
result = game.play('1, 1, extrabadparam')
self.assertEqual(result, 'error, mas parametros de los requeridos (2)')
def test_game_war_player_can_sunk_boat_player_cpu(self):
board_cpu = Board()
board = [
[1, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 2, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 2, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
]
board_cpu.board = board
self.game.state = game_states[1]
self.game.player_cpu.board_own = board_cpu
expected = ['Congratulations! You sunk a boat.']
result = self.game.play('0, 0')
self.assertEqual(expected, result)
@unittest.mock.patch('battleship.player.PlayerCPU.pick_coordenate')
def test_game_war_player_can_water_boat_player_cpu(self, mock_pick_coordenate):
board_cpu = Board()
board = [
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 1, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
]
board_cpu.board = board
self.game.player_cpu.board_own = board_cpu
self.game.turn = possible_turn[0]
self.game.state = game_states[1]
expected = ['You only hit water! CPU turn', 'Water! Now is your turn.']
mock_pick_coordenate.return_value = [0, 0]
result = self.game.play('0, 0')
self.assertEqual(expected, result)
def test_game_war_player_can_hit_boat_player_cpu(self):
board_cpu = Board()
board = [
[1, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 2, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 2, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
]
board_cpu.board = board
self.game.player_cpu.board_own = board_cpu
expected = ['You hit a boat']
self.game.state = game_states[1]
result = self.game.play('2, 2')
self.assertEqual(expected, result)
def test_game_dont_change_state(self):
board_cpu = Board()
board = [
[1, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
]
board_cpu.board = board
self.game.player_cpu.board_own = board_cpu
expected = self.game.state
self.game.play('1, 2')
self.assertEqual(expected, self.game.state)
@unittest.mock.patch('battleship.player.PlayerCPU.pick_coordenate')
def test_game_check_cpu_sunk(self, mock_pick_coordenate):
self.game.state = game_states[1]
self.game.turn = 'cpu'
self.game.player_human.board_own.sunked.append(2)
mock_pick_coordenate.return_value = [2, 2]
result = self.game.play('')
self.assertEqual(['Your boat was sunk.'], result)
@unittest.mock.patch('battleship.player.PlayerCPU.pick_coordenate')
def test_game_check_cpu_sunk_3(self, mock_pick_coordenate):
self.game.state = game_states[1]
self.game.turn = 'cpu'
self.game.player_human.board_own.sunked.append(31)
self.game.player_human.board_own.sunked.append(31)
mock_pick_coordenate.return_value = [1, 3]
result = self.game.play('')
self.assertEqual(['Your boat was sunk.'], result)
@unittest.mock.patch('battleship.player.PlayerCPU.pick_coordenate')
def test_game_check_cpu_hit(self, mock_pick_coordenate):
self.game.state = game_states[1]
self.game.turn = 'cpu'
mock_pick_coordenate.return_value = [2, 2]
result = self.game.play('')
self.assertEqual(['Your boat was hit.'], result)
@unittest.mock.patch('battleship.player.PlayerCPU.pick_coordenate')
def test_game_check_cpu_wins(self, mock_pick_coordenate):
board_human = Board()
board_table = [
[1, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
]
board_human.board = board_table
self.game.player_human.board_own = board_human
self.game_states = game_states[1]
self.game.turn = 'cpu'
mock_pick_coordenate.return_value = [0, 0]
result = self.game.war_cpu()
self.assertEqual(['Your boat was sunk.', 'You lose.'], result)
@unittest.mock.patch('battleship.player.PlayerCPU.pick_coordenate')
def test_game_check_cpu_water(self, mock_pick_coordenate):
self.game.state = game_states[1]
self.game.turn = 'cpu'
mock_pick_coordenate.return_value = [0, 0]
result = self.game.play('')
self.assertEqual(['Water! Now is your turn.'], result)
def test_player_cpu_pick_coordenate(self):
cpu_player = PlayerCPU()
coordenate = cpu_player.pick_coordenate()
self.assertTrue(0 <= coordenate[0] < 10)
self.assertTrue(0 <= coordenate[1] < 10)
def test_doblehit(self):
board_cpu = Board()
board = [
[31, 9, 31, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
]
board_cpu.board = board
result = board_cpu.shoot(0, 1)
self.assertEqual(result, 'already shoot')
@unittest.mock.patch('battleship.player.PlayerCPU.pick_coordenate')
def test_player_cpu_return_messages(self, mock_pick_coordenate):
self.game.state = game_states[1]
self.game.turn = 'cpu'
mock_pick_coordenate.return_value = [1, 1]
self.game.play('')
mock_pick_coordenate.return_value = [1, 2]
self.game.play('')
mock_pick_coordenate.return_value = [2, 2]
self.game.play('')
mock_pick_coordenate.return_value = [0, 0]
result = self.game.play('')
expected = ['Your boat was sunk.', 'Your boat was hit.', 'Your boat was sunk.', 'Water! Now is your turn.']
self.assertEqual(expected, result)
def test_next_turn_init(self):
game = GameBattleship()
result = game.next_turn()
expected = 'pone el barco (x, y, boat, horizontal/vertical)'
self.assertEqual(expected, result)
def test_next_turn_war_start(self):
self.game.state = game_states[1]
self.game.turn = 'human'
result = self.game.next_turn()
expected = 'shoot (x, y)'
self.assertEqual(expected, result)
@unittest.mock.patch('battleship.player.PlayerCPU.pick_coordenate')
def test_play_human_water(self, mock_pick_coordenate):
board = [
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 1, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
]
self.game.player_cpu.board_own.board = board
self.game.state = game_states[1]
self.game.turn = 'human'
mock_pick_coordenate.return_value = [1, 1]
result = self.game.play('0, 0')
expected = ['You only hit water! CPU turn', 'Your boat was sunk.']
self.assertEqual(result, expected)
def test_play_human_already_shoot(self):
board = [
[9, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 4, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 4, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 4, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 4, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
]
self.game.player_cpu.board_own.board = board
self.game.state = game_states[1]
self.game.turn = 'human'
result = self.game.play('0, 0')
expected = ['You already shoot in this place. Try again']
self.assertEqual(result, expected)
def test_play_human_win(self):
board = [
[9, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 9],
[0, 0, 0, 0, 0, 9, 0, 0, 0, 9],
[0, 0, 9, 0, 0, 9, 0, 0, 0, 9],
[0, 0, 9, 0, 0, 9, 0, 0, 0, 9],
[0, 0, 9, 0, 0, 2, 2, 0, 0, 0],
[0, 0, 9, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 9, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 9, 9, 9, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
]
self.game.player_cpu.board_own.board = board
self.game.state = game_states[1]
self.game.turn = 'human'
self.game.play('5, 5')
result = self.game.play('5, 6')
expected = ['You hit a boat', 'Congratulations! You sunk a boat.', 'You Win']
self.assertEqual(result, expected)
def test_turn_war_player_wrong_number(self):
game = GameBattleship()
result = game.play('10, 1, 1, vertical')
self.assertFalse(result)
def test_turn_war_player_wrong_param_amount_of(self):
game = GameBattleship()
game.state = 'war'
result = game.play('1, 1, extrabadparam')
self.assertEqual(result, 'error, mas parametros de los requeridos (2)')
def test_turn_war_player_wrong_param_letter(self):
game = GameBattleship()
result = game.play('A, B, 1, vertical')
self.assertEqual(result, 'error')
def test_turn_war_player(self):
game = GameBattleship()
result = game.set_boat('1, 2, 1, vertical')
self.assertTrue(result)
def test_turn_init_player_wrong_param_amount_of(self):
game = GameBattleship()
result = game.set_boat('1, 1, 1, vertical, extrabadparam')
self.assertEqual(result, 'error, mas parametros de los requeridos (4)')
def test_game_state_init(self):
game = GameBattleship()
result = game.state
self.assertEqual(result, 'init')
def test_game_players(self):
game = GameBattleship()
result = game.get_players()
self.assertEqual(len(result), 2)