concluido doctests

TelmoCode · TelmoCode · commit 78032e2938d2 · 2022-08-03T22:10:55.000-03:00
diff --git a/oo/carro.py b/oo/carro.py
@@ -64,13 +64,13 @@
     'Oeste'
     >>> direcao.virar_a_direita()
     >>> direcao.valor
-    'Sul'
+    'Norte'
     >>> direcao.virar_a_direita()
     >>> direcao.valor
     'Leste'
     >>> direcao.virar_a_direita()
     >>> direcao.valor
-    'Norte'
+    'Sul'
     >>> carro = Carro(direcao, motor)
     >>> carro.calcular_velocidade()
     0
@@ -83,55 +83,67 @@
     >>> carro.acelerar()
     >>> carro.calcular_velocidade()
     3
-    >>> carro.frerar()
+    >>> carro.frear()
     >>> carro.calcular_velocidade()
     1
-    >>> carro.frerar()
+    >>> carro.frear()
     >>> carro.calcular_velocidade()
     0
     >>> carro.calcular_direcao()
-    'Norte'
+    'Sul'
     >>> carro.virar_a_direita()
     >>> carro.calcular_direcao()
-    'Leste'
+    'Oeste'
     >>> carro.virar_a_esquerda()
     >>> carro.calcular_direcao()
-    'Norte'
+    'Sul'
 """
-NORTE = 'Norte'
-SUL = 'Sul'
-LESTE = 'Leste'
-OESTE = 'OESTE'
-class Direcao:
-    def __init__(self):
-        self.valor = NORTE
 
-    def virar_a_direita(self):
-        if self.valor == NORTE:
-            self.valor = LESTE
-        elif self.valor == LESTE:
-            self.valor = SUL
-        elif self.valor == SUL:
-            self.valor = OESTE
-        elif self.valor == OESTE:
-            self.valor = NORTE
+class Carro:
+    def __init__(self, direcao, motor):
+        self.motor = motor
+        self.direcao = direcao
+
+    def calcular_velocidade(self):
+        return self.motor.velocidade
+
+    def acelerar(self):
+        self.motor.acelerar()
 
+    def frear(self):
+        self.motor.frear()
+
+    def calcular_direcao(self):
+        return self.direcao.valor
 
+    def virar_a_direita(self):
+        self.direcao.virar_a_direita()
 
     def virar_a_esquerda(self):
+        self.direcao.virar_a_esquerda()
+
 
-        if self.valor == NORTE:
-            self.valor = OESTE
-        elif self.valor == OESTE:
-            self.valor = SUL
-        elif self.valor == SUL:
-            self.valor = LESTE
-        elif self.valor == LESTE:
-            self.valor = NORTE
+NORTE = 'Norte'
+SUL = 'Sul'
+LESTE = 'Leste'
+OESTE = 'Oeste'
 
+class Direcao:
+    rotacao_a_direita_dct = {
+        NORTE: LESTE, LESTE: SUL, SUL: OESTE, OESTE: NORTE
+    }
+    rotacao_a_esquerda_dct = {
+        NORTE: OESTE, OESTE: SUL, SUL: LESTE, LESTE: NORTE
+    }
 
+    def __init__(self):
+        self.valor = NORTE
 
+    def virar_a_direita(self):
+        self.valor = self.rotacao_a_direita_dct[self.valor]
 
+    def virar_a_esquerda(self):
+        self.valor = self.rotacao_a_esquerda_dct[self.valor]
 
 
 class Motor:
