Simplified Carnivore.bite method and removed test changes)

mate-academy · Oct 13, 2023 · d410032 · d410032
1 parent 9238722
commit d410032
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/app/main.py b/app/main.py
@@ -23,9 +23,10 @@ def hide(self) -> None:
 
 
 class Carnivore(Animal):
-    def bite(self, herbivore: Herbivore) -> None:
+    @staticmethod
+    def bite(herbivore: Herbivore) -> None:
         if herbivore.hidden or isinstance(herbivore, Carnivore):
             return
-        herbivore.health = max(0, herbivore.health - 50)
-        if not herbivore.health:
+        herbivore.health -= 50
+        if herbivore.health <= 0:
             Animal.alive.pop(Animal.alive.index(herbivore))
diff --git a/tests/test_main.py b/tests/test_main.py
@@ -129,8 +129,6 @@ def test_print_animal_alive():
 
     f = io.StringIO()
 
-    print(Animal.alive)
-
     with redirect_stdout(f):
         print(Animal.alive)