Merge pull request #846 from janga1997/typeHint-apavlov

Add type hints to apavlov

Author: drvinceknight

axelrod/strategies/apavlov.py

@@ -1,4 +1,5 @@
-from axelrod.actions import Actions
+
+from axelrod.actions import Actions, Action
 from axelrod.player import Player
 
 C, D = Actions.C, Actions.D
@@ -26,11 +27,11 @@ class APavlov2006(Player):
         'manipulates_state': False
     }
 
-    def __init__(self):
+    def __init__(self) -> None:
         super().__init__()
-        self.opponent_class = None
+        self.opponent_class = ""
 
-    def strategy(self, opponent):
+    def strategy(self, opponent: Player) -> Action:
         # TFT for six rounds
         if len(self.history) < 6:
             return D if opponent.history[-1:] == [D] else C
@@ -93,11 +94,11 @@ class APavlov2011(Player):
         'manipulates_state': False
     }
 
-    def __init__(self):
+    def __init__(self) -> None:
         super().__init__()
-        self.opponent_class = None
+        self.opponent_class = ""
 
-    def strategy(self, opponent):
+    def strategy(self, opponent: Player) -> Action:
         # TFT for six rounds
         if len(self.history) < 6:
             return D if opponent.history[-1:] == [D] else C

type_tests.sh

@@ -11,6 +11,7 @@ mypy --ignore-missing-imports --follow-imports skip axelrod/strategies/alternato
 mypy --ignore-missing-imports --follow-imports skip axelrod/strategies/ann.py
 mypy --ignore-missing-imports --follow-imports skip axelrod/strategies/appeaser.py
 mypy --ignore-missing-imports --follow-imports skip axelrod/strategies/averagecopier.py
+mypy --ignore-missing-imports --follow-imports skip axelrod/strategies/apavlov.py
 mypy --ignore-missing-imports --follow-imports skip axelrod/strategies/axelrod_second.py
 mypy --ignore-missing-imports --follow-imports skip axelrod/strategies/backstabber.py
 mypy --ignore-missing-imports --follow-imports skip axelrod/strategies/better_and_better.py