Update warrior.py

kyu_4/the_greatest_warrior/warrior.py

@@ -37,6 +37,7 @@ class Warrior:
     """
 
     def __init__(self):
+        """Create a new warrior instance."""
         # A warrior's experience starts from 100
         self.__experience = BASIC_EXPERIENCE
         # A warrior starts at level 1
@@ -67,24 +68,23 @@ def __set_level(self) -> int:
         by another 100, the warrior's level rises to
         the next level.
 
-        :return:
+        :return: int
         """
         new_level = self.experience // BASIC_EXPERIENCE
         return new_level if new_level <= 100 else 100
 
-    def __update_experience(self, experience: int):
+    def __update_experience(self, experience: int) -> None:
         """
-        Update expirience.
+        Update experience.
 
         A warrior's experience is cumulative, and does not
         reset with each rise of level. The only exception
         is when the warrior reaches level 100, with which
         the experience stops at 10000.
         :return:
         """
-        if self.level == 100:
-            self.__experience = MAX_EXPERIENCE
-        elif self.experience + experience > MAX_EXPERIENCE:
+        if (self.level == 100
+                or self.experience + experience > MAX_EXPERIENCE):
             self.__experience = MAX_EXPERIENCE
         else:
             self.__experience += experience
@@ -118,7 +118,7 @@ def experience(self) -> int:
         """
         Return experience value.
 
-        :return:
+        :return: int
         """
         return self.__experience
 
@@ -127,16 +127,16 @@ def achievements(self) -> list:
         """
         Return achievements as a list.
 
-        :return:
+        :return: list
         """
         return self.__achievements
 
     def battle(self, enemy_level: int) -> str:
         """
         Return message based on the result of the battle.
 
-        :param enemy_level:
-        :return:
+        :param enemy_level: int
+        :return: str
         """
         msg: str = ''
         # If an enemy level does not fall in the range of 1 to 100,
@@ -174,13 +174,15 @@ def battle(self, enemy_level: int) -> str:
 
     def training(self, params: list) -> str:
         """
+        Training method.
+
         Training will accept an array of three elements:
-            the description,
-            the experience points your warrior earns,
-            and the minimum level requirement.
+            1. the description.
+            2. the experience points your warrior earns.
+            3. the minimum level requirement.
 
-        :param params:
-        :return:
+        :param params: list
+        :return: str
         """
         # If the warrior's level meets the minimum level requirement,
         # the warrior will receive the experience points from it and