pythonprobr
diff --git a/‎atores.py‎
Lines changed: 3 additions & 6 deletions b/‎atores.py‎
Lines changed: 3 additions & 6 deletions
diff --git a/‎testes/atores_testes.py‎
Lines changed: 21 additions & 0 deletions b/‎testes/atores_testes.py‎
Lines changed: 21 additions & 0 deletions
@@ -11,17 +11,14 @@ class Ator():
     _caracter_ativo = 'A'
     _caracter_destruido = ' '
 
-    def caracter(self, tempo):
-        if self.status(tempo) == ATIVO:
-            return self._caracter_ativo
-        else:
-            return self._caracter_destruido
-
     def __init__(self, x=0, y=0):
         self.y = y
         self.x = x
         self._tempo_de_colisao = None
 
+    def caracter(self, tempo):
+        return self._caracter_ativo if self.status(tempo) == ATIVO else self._caracter_destruido
+
     def resetar(self):
         self._tempo_de_colisao = None
 
 
@@ -32,6 +32,14 @@ def teste_ator_posicao(self):
         ator = Ator(0.6, 2.1)
         self.assertTupleEqual((1, 2), ator.calcular_posicao(3.14), 'Deveria arredondar para inteiro')
 
+    def teste_resetar(self):
+        ator = Ator()
+        self.assertIsNone(ator._tempo_de_colisao)
+        ator._tempo_de_colisao = 1
+        ator.resetar()
+        self.assertIsNone(ator._tempo_de_colisao)
+
+
     def teste_status(self):
         ator = Ator()
         'Confere status de um ator'
@@ -157,6 +165,19 @@ def teste_status(self):
     def teste_velocidade_escalar(self):
         self.assertEqual(20, PassaroVermelho.velocidade_escalar)
 
+    def teste_resetar(self):
+        passaro = PassaroVermelho()
+        self.assertIsNone(passaro._tempo_de_colisao)
+        self.assertIsNone(passaro._tempo_de_lancamento)
+        self.assertIsNone(passaro._angulo_de_lancamento)
+        passaro._tempo_de_colisao = 1
+        passaro._tempo_de_lancamento = 2
+        passaro._angulo_de_lancamento = 90
+        passaro.resetar()
+        self.assertIsNone(passaro._tempo_de_colisao)
+        self.assertIsNone(passaro._tempo_de_lancamento)
+
     def teste_foi_lancado(self):
         passaro_vermelho = PassaroVermelho(1, 1)
         self.assertFalse(passaro_vermelho.foi_lancado(),