def test_criar_jogo(self):
jogo = Jogo()
jogo.create_jogo()
self.assertEquals(len(jogo.jogadores), 2)
self.assertEquals(len(jogo.jogadores[0].embarcacoes), 5)
self.assertEquals(len(jogo.jogadores[1].embarcacoes), 5)
for embarcacao in (PortaAvioes, Encouracado, Submarino, Destroyer, Patrulha):
self.assertIn(embarcacao, [e.__class__ for e in jogo.jogadores[0].embarcacoes])
self.assertIn(embarcacao, [e.__class__ for e in jogo.jogadores[1].embarcacoes])
def test_status_jogo(self):
jogo = Jogo()
jogador1 = Jogador()
jogador1.adicionar_embarcacao(Submarino(is_horizontal=True, posicao=(0,0)))
jogador2 = Jogador()
jogador2.adicionar_embarcacao(Submarino(is_horizontal=True, posicao=(0,0)))
jogo.jogadores.append(jogador1)
jogo.jogadores.append(jogador2)
self.assertFalse(jogo.is_terminado())
jogador1.atirar((0,0))
jogador1.atirar((1,0))
jogador1.atirar((2,0))
self.assertTrue(jogo.is_terminado())
def main():
jogo = Jogo()
jogo.create_jogo()
print_jogo(jogo)
while not jogo.is_terminado():
key = raw_input()
print("Jogada "+str(len(jogo.jogadores[0].jogadas)+1))
for jogador in jogo.jogadores:
atirou = False
while not atirou:
try:
jogador.atirar((random.randint(0,9), random.randint(0,9)))
atirou = True
except:
continue
print_jogo(jogo)
